← 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/GenerateSwitchLabelsAction.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/GenerateSwitchLabelsAction.cs
 
1
// 
 
2
// GenerateSwitchLabels.cs
 
3
//  
 
4
// Author:
 
5
//       Mike Krüger <mkrueger@novell.com>
 
6
// 
 
7
// Copyright (c) 2011 Mike Krüger <mkrueger@novell.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;
 
27
using ICSharpCode.NRefactory.TypeSystem;
 
28
using System.Threading;
 
29
using System.Collections.Generic;
 
30
 
 
31
namespace ICSharpCode.NRefactory.CSharp.Refactoring
 
32
{
 
33
        [ContextAction("Generate switch labels", Description = "Creates switch lables for enumerations.")]
 
34
        public class GenerateSwitchLabelsAction : ICodeActionProvider
 
35
        {
 
36
                public IEnumerable<CodeAction> GetActions(RefactoringContext context)
 
37
                {
 
38
                        var switchStatement = GetSwitchStatement(context);
 
39
                        if (switchStatement == null) {
 
40
                                yield break;
 
41
                        }
 
42
                        var result = context.Resolve(switchStatement.Expression);
 
43
                        if (result.Type.Kind != TypeKind.Enum) {
 
44
                                yield break;
 
45
                        }
 
46
                        yield return new CodeAction (context.TranslateString("Create switch labels"), script => {
 
47
                                var type = result.Type;
 
48
                                var newSwitch = (SwitchStatement)switchStatement.Clone();
 
49
                                
 
50
                                var target = new TypeReferenceExpression (context.CreateShortType(result.Type));
 
51
                                foreach (var field in type.GetFields ()) {
 
52
                                        if (field.IsSynthetic || !field.IsConst) {
 
53
                                                continue;
 
54
                                        }
 
55
                                        newSwitch.SwitchSections.Add(new SwitchSection () {
 
56
                                                CaseLabels = {
 
57
                                                        new CaseLabel (new MemberReferenceExpression (target.Clone(), field.Name))
 
58
                                                },
 
59
                                                Statements = {
 
60
                                                        new BreakStatement ()
 
61
                                                }
 
62
                                        });
 
63
                                }
 
64
                                
 
65
                                newSwitch.SwitchSections.Add(new SwitchSection () {
 
66
                                        CaseLabels = {
 
67
                                                new CaseLabel ()
 
68
                                        },
 
69
                                        Statements = {
 
70
                                                new ThrowStatement (new ObjectCreateExpression (context.CreateShortType("System", "ArgumentOutOfRangeException")))
 
71
                                        }
 
72
                                });
 
73
                                
 
74
                                script.Replace(switchStatement, newSwitch);
 
75
                        }, switchStatement);
 
76
                }
 
77
                
 
78
                static SwitchStatement GetSwitchStatement (RefactoringContext context)
 
79
                {
 
80
                        var switchStatment = context.GetNode<SwitchStatement> ();
 
81
                        if (switchStatment != null && switchStatment.SwitchSections.Count == 0)
 
82
                                return switchStatment;
 
83
                        return null;
 
84
                }
 
85
        }
 
86
}