public async override Task RegisterCodeFixesAsync(CodeFixContext context)
        {
            var document          = context.Document;
            var cancellationToken = context.CancellationToken;
            var span        = context.Span;
            var diagnostics = context.Diagnostics;
            var root        = await document.GetSyntaxRootAsync(cancellationToken);

            var semanticModel = await document.GetSemanticModelAsync(cancellationToken);

            var diagnostic = diagnostics.First();
            var node       = root.FindNode(context.Span) as IfStatementSyntax;

            var switchExpr = ConvertIfStatementToSwitchStatementCodeRefactoringProvider.GetSwitchExpression(semanticModel, node.Condition);

            if (switchExpr == null)
            {
                return;
            }
            var switchSections = new List <SwitchSectionSyntax>();

            if (!ConvertIfStatementToSwitchStatementCodeRefactoringProvider.CollectSwitchSections(switchSections, semanticModel, node, switchExpr))
            {
                return;
            }
            if (switchSections.Count(s => !s.Labels.OfType <DefaultSwitchLabelSyntax>().Any()) <= 2)
            {
                return;
            }

            var switchStatement = SyntaxFactory.SwitchStatement(switchExpr, new SyntaxList <SwitchSectionSyntax>().AddRange(switchSections));
            var newRoot         = root.ReplaceNode((SyntaxNode)node, switchStatement.WithAdditionalAnnotations(Formatter.Annotation));

            context.RegisterCodeFix(CodeActionFactory.Create(node.Span, diagnostic.Severity, "Convert to 'switch' statement", document.WithSyntaxRoot(newRoot)), diagnostic);
        }
Exemplo n.º 2
0
        static bool TryGetDiagnostic(SyntaxNodeAnalysisContext nodeContext, out Diagnostic diagnostic)
        {
            var node              = nodeContext.Node as IfStatementSyntax;
            var semanticModel     = nodeContext.SemanticModel;
            var cancellationToken = nodeContext.CancellationToken;

            diagnostic = default(Diagnostic);
            if (nodeContext.IsFromGeneratedCode())
            {
                return(false);
            }
            if (node.Parent is IfStatementSyntax || node.Parent is ElseClauseSyntax)
            {
                return(false);
            }

            var switchExpr = ConvertIfStatementToSwitchStatementCodeRefactoringProvider.GetSwitchExpression(semanticModel, node.Condition);

            if (switchExpr == null)
            {
                return(false);
            }
            var switchSections = new List <SwitchSectionSyntax>();

            if (!ConvertIfStatementToSwitchStatementCodeRefactoringProvider.CollectSwitchSections(switchSections, semanticModel, node, switchExpr))
            {
                return(false);
            }
            if (switchSections.Count(s => !s.Labels.OfType <DefaultSwitchLabelSyntax>().Any()) <= 2)
            {
                return(false);
            }

            diagnostic = Diagnostic.Create(
                descriptor,
                node.IfKeyword.GetLocation()
                );
            return(true);
        }