public sealed override async Task ComputeRefactoringsAsync(CodeRefactoringContext context)
{
CurrentCompilationOptions.Nullability = context.Document.Project.CompilationOptions?.NullableContextOptions.Equals(NullableContextOptions.Enable) ?? true;
var root = await context.Document.GetSyntaxRootAsync(context.CancellationToken).ConfigureAwait(false);
var node = root.FindToken(context.Span.Start);
context.Register("Expand enum to union type", UnionTypeCodeProvider.TryGetEnumDeclaration,
async(enumDeclaration, c) =>
{
var updatedDoc = await UnionTypeCodeProvider.EnumToClass(context.Document, enumDeclaration, c);
return(updatedDoc.Project.Solution);
}, node);
context.Register("Generate 'With' extension", ImmutableHelpersCodeProvider.TryGetClassDeclaration,
async(classDeclaration, c) =>
{
var updatedDoc = await ImmutableHelpersCodeProvider.GenerateWithExtension(context.Document, classDeclaration, c);
return(updatedDoc.Project.Solution);
}, node);
context.Register(dotFilePath => $"Generate state machine from {Path.GetFileName(dotFilePath)}",
_ => StateMachineCodeProvider.TryGetDotFilename(context.Document),
async(dotFilePath, c) =>
{
var updatedDoc = await StateMachineCodeProvider.GenerateStateMachine(context.Document, dotFilePath, c);
return(updatedDoc.Project.Solution);
}, node);
}
protected override async Task Refactor(AdhocWorkspace workspace, Document document, SyntaxNode root)
{
var updatedDocument = await ImmutableHelpersCodeProvider.GenerateWithExtension(
document,
root.DescendantNodes().OfType <ClassDeclarationSyntax>().Last(), CancellationToken.None)
.ConfigureAwait(false);
workspace.TryApplyChanges(updatedDocument.Project.Solution);
}
