7 changed files with 245 additions and 11 deletions
@ -0,0 +1,99 @@
@@ -0,0 +1,99 @@
|
||||
//
|
||||
// RemoveRegion.cs
|
||||
//
|
||||
// Author:
|
||||
// Mike Krüger <mkrueger@xamarin.com>
|
||||
//
|
||||
// Copyright (c) 2012 Xamarin Inc. (http://xamarin.com)
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
using System; |
||||
|
||||
namespace ICSharpCode.NRefactory.CSharp.Refactoring |
||||
{ |
||||
public class RemoveRegion : IContextAction |
||||
{ |
||||
public bool IsValid (RefactoringContext context) |
||||
{ |
||||
return GetDirective (context) != null; |
||||
} |
||||
|
||||
public void Run (RefactoringContext context) |
||||
{ |
||||
var directive = GetDirective (context); |
||||
var visitor = new DirectiveVisitor (directive); |
||||
context.Unit.AcceptVisitor (visitor); |
||||
Console.WriteLine ("directive:" + directive + "/" + visitor.Endregion); |
||||
if (visitor.Endregion == null) |
||||
return; |
||||
using (var script = context.StartScript ()) { |
||||
script.Remove (directive); |
||||
script.Remove (visitor.Endregion); |
||||
} |
||||
} |
||||
|
||||
class DirectiveVisitor : DepthFirstAstVisitor |
||||
{ |
||||
readonly PreProcessorDirective startDirective; |
||||
bool searchDirectives = false; |
||||
int depth; |
||||
|
||||
public PreProcessorDirective Endregion { |
||||
get; |
||||
set; |
||||
} |
||||
|
||||
public DirectiveVisitor (PreProcessorDirective startDirective) |
||||
{ |
||||
this.startDirective = startDirective; |
||||
} |
||||
|
||||
public override void VisitPreProcessorDirective (PreProcessorDirective preProcessorDirective) |
||||
{ |
||||
if (searchDirectives) { |
||||
if (preProcessorDirective.Type == PreProcessorDirectiveType.Region) |
||||
depth++; |
||||
if (preProcessorDirective.Type == PreProcessorDirectiveType.Endregion) { |
||||
depth--; |
||||
if (depth == 0) { |
||||
Endregion = preProcessorDirective; |
||||
searchDirectives = false; |
||||
} |
||||
} |
||||
} |
||||
|
||||
if (preProcessorDirective == startDirective) { |
||||
searchDirectives = true; |
||||
depth = 1; |
||||
} |
||||
|
||||
base.VisitPreProcessorDirective (preProcessorDirective); |
||||
} |
||||
} |
||||
|
||||
static PreProcessorDirective GetDirective (RefactoringContext context) |
||||
{ |
||||
var directive = context.GetNode<PreProcessorDirective> (); |
||||
if (directive == null || directive.Type != PreProcessorDirectiveType.Region) |
||||
return null; |
||||
return directive; |
||||
} |
||||
} |
||||
} |
||||
|
||||
@ -0,0 +1,91 @@
@@ -0,0 +1,91 @@
|
||||
//
|
||||
// RemoveRegionTests.cs
|
||||
//
|
||||
// Author:
|
||||
// Mike Krüger <mkrueger@xamarin.com>
|
||||
//
|
||||
// Copyright (c) 2012 Xamarin Inc.
|
||||
//
|
||||
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
// of this software and associated documentation files (the "Software"), to deal
|
||||
// in the Software without restriction, including without limitation the rights
|
||||
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
// copies of the Software, and to permit persons to whom the Software is
|
||||
// furnished to do so, subject to the following conditions:
|
||||
//
|
||||
// The above copyright notice and this permission notice shall be included in
|
||||
// all copies or substantial portions of the Software.
|
||||
//
|
||||
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||
// THE SOFTWARE.
|
||||
|
||||
using System; |
||||
using NUnit.Framework; |
||||
using ICSharpCode.NRefactory.CSharp.Refactoring; |
||||
|
||||
namespace ICSharpCode.NRefactory.CSharp.ContextActions |
||||
{ |
||||
[TestFixture] |
||||
public class RemoveRegionTests : ContextActionTestBase |
||||
{ |
||||
[Test()] |
||||
public void TestSimpleRegion () |
||||
{ |
||||
string result = RunContextAction ( |
||||
new RemoveRegion (), |
||||
"class TestClass" + Environment.NewLine + |
||||
"{" + Environment.NewLine + |
||||
" #region$ Foo" + Environment.NewLine + |
||||
" void Test ()" + Environment.NewLine + |
||||
" {" + Environment.NewLine + |
||||
" }" + Environment.NewLine + |
||||
" #endregion" + Environment.NewLine + |
||||
"}" |
||||
); |
||||
|
||||
Assert.AreEqual ( |
||||
"class TestClass" + Environment.NewLine + |
||||
"{" + Environment.NewLine + |
||||
" void Test ()" + Environment.NewLine + |
||||
" {" + Environment.NewLine + |
||||
" }" + Environment.NewLine + |
||||
"}", result); |
||||
} |
||||
|
||||
[Test()] |
||||
public void TestNestedRegion () |
||||
{ |
||||
string result = RunContextAction ( |
||||
new RemoveRegion (), |
||||
"class TestClass" + Environment.NewLine + |
||||
"{" + Environment.NewLine + |
||||
" #region$ Foo" + Environment.NewLine + |
||||
" void Test ()" + Environment.NewLine + |
||||
" {" + Environment.NewLine + |
||||
" #region Nested" + Environment.NewLine + |
||||
" Foo ();" + Environment.NewLine + |
||||
" #endregion" + Environment.NewLine + |
||||
" }" + Environment.NewLine + |
||||
" #endregion" + Environment.NewLine + |
||||
"}" |
||||
); |
||||
|
||||
Assert.AreEqual ( |
||||
"class TestClass" + Environment.NewLine + |
||||
"{" + Environment.NewLine + |
||||
" void Test ()" + Environment.NewLine + |
||||
" {" + Environment.NewLine + |
||||
" #region Nested" + Environment.NewLine + |
||||
" Foo ();" + Environment.NewLine + |
||||
" #endregion" + Environment.NewLine + |
||||
" }" + Environment.NewLine + |
||||
"}", result); |
||||
} |
||||
} |
||||
} |
||||
|
||||
Loading…
Reference in new issue