From 096c88c954698ad1c2d45c0baa3e29dcc3b23b63 Mon Sep 17 00:00:00 2001
From: Siegfried Pammer <siegfriedpammer@gmail.com>
Date: Tue, 21 Jan 2014 21:03:02 +0100
Subject: [PATCH] fix #282: Pressing Ctrl+Z in InsertWithCursor mode causes
 InvalidOperationException

---
 .../Project/Src/Refactoring/EditorScript.cs   | 87 +++++++++++++------
 .../Document/UndoStack.cs                     |  2 +-
 .../Refactoring/DocumentScript.cs             |  2 +-
 3 files changed, 63 insertions(+), 28 deletions(-)

diff --git a/src/AddIns/BackendBindings/CSharpBinding/Project/Src/Refactoring/EditorScript.cs b/src/AddIns/BackendBindings/CSharpBinding/Project/Src/Refactoring/EditorScript.cs
index 1e22a1af6b..3948f72f10 100644
--- a/src/AddIns/BackendBindings/CSharpBinding/Project/Src/Refactoring/EditorScript.cs
+++ b/src/AddIns/BackendBindings/CSharpBinding/Project/Src/Refactoring/EditorScript.cs
@@ -114,35 +114,44 @@ namespace CSharpBinding.Refactoring
 					}
 					break;
 			}
-			operationsRunning++;
 			InsertWithCursorOnLayer(this, layer, tcs, nodes, editor.Document);
 			return tcs.Task;
 		}
 		
 		void InsertWithCursorOnLayer(EditorScript currentScript, InsertionCursorLayer layer, TaskCompletionSource<Script> tcs, IList<AstNode> nodes, IDocument target)
 		{
+			var doc = target as TextDocument;
+			var op = new UndoOperation(layer, tcs);
+			if (doc != null) {
+				doc.UndoStack.Push(op);
+			}
 			layer.Exited += delegate(object s, InsertionCursorEventArgs args) {
-				if (args.Success) {
-					if (args.InsertionPoint.LineAfter == NewLineInsertion.None &&
-					    args.InsertionPoint.LineBefore == NewLineInsertion.None && nodes.Count > 1) {
-						args.InsertionPoint.LineAfter = NewLineInsertion.BlankLine;
-					}
-					foreach (var node in nodes.Reverse ()) {
-						int indentLevel = currentScript.GetIndentLevelAt(target.GetOffset(args.InsertionPoint.Location));
-						var output = currentScript.OutputNode(indentLevel, node);
-						var offset = target.GetOffset(args.InsertionPoint.Location);
-						var delta = args.InsertionPoint.Insert(target, output.Text);
-						output.RegisterTrackedSegments(currentScript, delta + offset);
+				doc.UndoStack.StartContinuedUndoGroup();
+				try {
+					if (args.Success) {
+						if (args.InsertionPoint.LineAfter == NewLineInsertion.None &&
+						    args.InsertionPoint.LineBefore == NewLineInsertion.None && nodes.Count > 1) {
+							args.InsertionPoint.LineAfter = NewLineInsertion.BlankLine;
+						}
+						foreach (var node in nodes.Reverse ()) {
+							int indentLevel = currentScript.GetIndentLevelAt(target.GetOffset(args.InsertionPoint.Location));
+							var output = currentScript.OutputNode(indentLevel, node);
+							var offset = target.GetOffset(args.InsertionPoint.Location);
+							var delta = args.InsertionPoint.Insert(target, output.Text);
+							output.RegisterTrackedSegments(currentScript, delta + offset);
+						}
+						tcs.SetResult(currentScript);
 					}
-					tcs.SetResult(currentScript);
+					layer.Dispose();
+					DisposeOnClose();
+				} finally {
+					doc.UndoStack.EndUndoGroup();
 				}
-				layer.Dispose();
-				DisposeOnClose();
+				op.Reset();
 			};
 		}
 		
 		readonly List<Script> startedScripts = new List<Script>();
-		int operationsRunning;
 		
 		public override Task<Script> InsertWithCursor(string operation, ITypeDefinition parentType, Func<Script, RefactoringContext, IList<AstNode>> nodeCallback)
 		{
@@ -195,7 +204,6 @@ namespace CSharpBinding.Refactoring
 				tcs.SetResult(script);
 				return tcs.Task;
 			}
-			operationsRunning++;
 			InsertWithCursorOnLayer(script, layer, tcs, nodes, area.Document);
 			return tcs.Task;
 		}
@@ -205,14 +213,10 @@ namespace CSharpBinding.Refactoring
 		{
 			if (isDisposed)
 				return;
-			if (force)
-				operationsRunning = 0;
-			if (operationsRunning-- == 0) {
-				isDisposed = true;
-				base.Dispose ();
-				// refresh parse information so that the issue can disappear immediately
-				SD.ParserService.ParseAsync(editor.FileName, editor.Document).FireAndForget();
-			}
+			isDisposed = true;
+			base.Dispose();
+			// refresh parse information so that the issue can disappear immediately
+			SD.ParserService.ParseAsync(editor.FileName, editor.Document).FireAndForget();
 			foreach (var script in startedScripts)
 				script.Dispose();
 		}
@@ -221,7 +225,38 @@ namespace CSharpBinding.Refactoring
 		{
 			DisposeOnClose();
 		}
-
+		
+		class UndoOperation : IUndoableOperation
+		{
+			InsertionCursorLayer layer;
+			TaskCompletionSource<Script> tcs;
+			
+			public UndoOperation(InsertionCursorLayer layer, TaskCompletionSource<Script> tcs)
+			{
+				this.layer = layer;
+				this.tcs = tcs;
+			}
+			
+			public void Reset()
+			{
+				layer = null;
+				tcs = null;
+			}
+			
+			public void Undo()
+			{
+				if (layer != null)
+					layer.Dispose();
+				layer = null;
+				if (tcs != null)
+					tcs.SetCanceled();
+				tcs = null;
+			}
+			
+			public void Redo()
+			{
+			}
+		}
 	}
 	
 	class InsertionCursorLayer : UIElement, IDisposable
diff --git a/src/Libraries/AvalonEdit/ICSharpCode.AvalonEdit/Document/UndoStack.cs b/src/Libraries/AvalonEdit/ICSharpCode.AvalonEdit/Document/UndoStack.cs
index 918277fd4a..d7883edb5a 100644
--- a/src/Libraries/AvalonEdit/ICSharpCode.AvalonEdit/Document/UndoStack.cs
+++ b/src/Libraries/AvalonEdit/ICSharpCode.AvalonEdit/Document/UndoStack.cs
@@ -193,7 +193,7 @@ namespace ICSharpCode.AvalonEdit.Document
 		/// </summary>
 		/// <param name="groupDescriptor">An object that is stored with the undo group.
 		/// If this is not a top-level undo group, the parameter is ignored.</param>
-		public void StartContinuedUndoGroup(object groupDescriptor)
+		public void StartContinuedUndoGroup(object groupDescriptor = null)
 		{
 			if (undoGroupDepth == 0) {
 				actionCountInUndoGroup = (allowContinue && undostack.Count > 0) ? 1 : 0;
diff --git a/src/Libraries/NRefactory/ICSharpCode.NRefactory.CSharp/Refactoring/DocumentScript.cs b/src/Libraries/NRefactory/ICSharpCode.NRefactory.CSharp/Refactoring/DocumentScript.cs
index d4c11cf2d4..189a80804a 100644
--- a/src/Libraries/NRefactory/ICSharpCode.NRefactory.CSharp/Refactoring/DocumentScript.cs
+++ b/src/Libraries/NRefactory/ICSharpCode.NRefactory.CSharp/Refactoring/DocumentScript.cs
@@ -58,7 +58,7 @@ namespace ICSharpCode.NRefactory.CSharp.Refactoring
 				undoGroup.Dispose();
 		}
 		
-		public override void Remove(AstNode node, bool removeEmptyLine)
+		public override void Remove(AstNode node, bool removeEmptyLine = true)
 		{
 			var segment = GetSegment (node);
 			int startOffset = segment.Offset;