← Back to branch summary

~ubuntu-branches/ubuntu/trusty/monodevelop/trusty-proposed

« back to all changes in this revision

Viewing changes to external/nrefactory/ICSharpCode.NRefactory.CSharp/Refactoring/CodeActions/SimplifyIfFlowAction.cs

  • Committer: Package Import Robot
  • Author(s): Jo Shields
  • Date: 2013-05-12 09:46:03 UTC
  • mto: This revision was merged to the branch mainline in revision 29.
  • Revision ID: package-import@ubuntu.com-20130512094603-mad323bzcxvmcam0
Tags: upstream-4.0.5+dfsg
Import upstream version 4.0.5+dfsg

Show diffs side-by-side

added added

removed removed

Lines of Context:
external/nrefactory/ICSharpCode.NRefactory.CSharp/Refactoring/CodeActions/SimplifyIfFlowAction.cs
 
1
//
 
2
// SimplifyIfFlowAction.cs
 
3
//
 
4
// Author:
 
5
//      Ciprian Khlud <ciprian.mustiata@yahoo.com>
 
6
//
 
7
// Copyright (c) 2013 Ciprian Khlud <ciprian.mustiata@yahoo.com>
 
8
//
 
9
// Permission is hereby granted, free of charge, to any person obtaining a copy
 
10
// of this software and associated documentation files (the "Software"), to deal
 
11
// in the Software without restriction, including without limitation the rights
 
12
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 
13
// copies of the Software, and to permit persons to whom the Software is
 
14
// furnished to do so, subject to the following conditions:
 
15
//
 
16
// The above copyright notice and this permission notice shall be included in
 
17
// all copies or substantial portions of the Software.
 
18
//
 
19
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 
20
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 
21
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 
22
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 
23
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 
24
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 
25
// THE SOFTWARE.
 
26
using System.Collections.Generic;
 
27
 
 
28
namespace ICSharpCode.NRefactory.CSharp.Refactoring
 
29
{
 
30
        [ContextAction("Simplify if flow", Description = "Inverts if and reduces branching ")]
 
31
        public class SimplifyIfFlowAction : ICodeActionProvider
 
32
        {
 
33
                readonly InsertParenthesesVisitor _insertParenthesesVisitor = new InsertParenthesesVisitor();
 
34
 
 
35
                public IEnumerable<CodeAction> GetActions(RefactoringContext context)
 
36
                {
 
37
                        // TODO: Invert if without else
 
38
                        // ex. if (cond) DoSomething () == if (!cond) return; DoSomething ()
 
39
                        // beware of loop contexts return should be continue then.
 
40
                        var ifStatement = GetIfElseStatement(context);
 
41
                        if (ifStatement == null)
 
42
                                yield break;
 
43
                        yield return new CodeAction(context.TranslateString("Invert if"), script =>
 
44
                        {
 
45
                                GenerateNewScript(script, ifStatement);
 
46
                        }, ifStatement);
 
47
                }
 
48
 
 
49
                internal static void InsertBody(Script script, IfElseStatement ifStatement)
 
50
                {
 
51
                        var ifBody = ifStatement.TrueStatement.Clone();
 
52
 
 
53
                        if (ifBody is BlockStatement) {
 
54
                                AstNode last = ifStatement;
 
55
                                foreach (var stmt in ((BlockStatement)ifBody).Children) {
 
56
                                        if (stmt.Role == Roles.LBrace || stmt.Role == Roles.RBrace || stmt.Role == Roles.NewLine)
 
57
                                                continue;
 
58
                                        script.InsertAfter(last, stmt);
 
59
                                        last = stmt;
 
60
                                }
 
61
                        } else {
 
62
                                script.InsertAfter(ifStatement, ifBody);
 
63
                        }
 
64
                        script.FormatText(ifStatement.Parent);
 
65
                }
 
66
                
 
67
                void GenerateNewScript(Script script, IfElseStatement ifStatement)
 
68
                {
 
69
                        var mergedIfStatement = new IfElseStatement {
 
70
                                Condition = CSharpUtil.InvertCondition(ifStatement.Condition),
 
71
                                TrueStatement = new ReturnStatement()
 
72
                        };
 
73
                        mergedIfStatement.Condition.AcceptVisitor(_insertParenthesesVisitor);
 
74
                        
 
75
                        script.Replace(ifStatement, mergedIfStatement);
 
76
 
 
77
                        InsertBody(script, ifStatement);
 
78
                }
 
79
                
 
80
                static IfElseStatement GetIfElseStatement(RefactoringContext context)
 
81
                {
 
82
                        var result = context.GetNode<IfElseStatement>();
 
83
                        if (result == null)
 
84
                                return null;
 
85
                        if (!(result.IfToken.Contains(context.Location)
 
86
                                && !result.TrueStatement.IsNull
 
87
                                && result.FalseStatement.IsNull))
 
88
                                return null;
 
89
                        if (!(result.Parent is BlockStatement) || !(result.Parent.Parent is MethodDeclaration))
 
90
                                return null;
 
91
                        var parentMethod = (MethodDeclaration)result.Parent.Parent;
 
92
                        if (parentMethod.ReturnType.ToString() != "void")
 
93
                                return null;
 
94
                        var nextSibling = result.GetNextSibling(n => n is Statement);
 
95
                        if (nextSibling == null)
 
96
                                return result;
 
97
                        nextSibling = nextSibling.GetNextSibling (n => n is Statement);
 
98
                        if (nextSibling != null)
 
99
                                return null;
 
100
                        return result;
 
101
                }
 
102
        }
 
103
}
 
 
b'\\ No newline at end of file'