1 files changed, 157 insertions, 0 deletions
diff --git a/main/src/addins/MonoDevelop.Refactoring/MonoDevelop.CodeActions/CodeRefactoringService.cs b/main/src/addins/MonoDevelop.Refactoring/MonoDevelop.CodeActions/CodeRefactoringService.cs
new file mode 100644
index 0000000000..38b246031f
--- /dev/null
+++ b/main/src/addins/MonoDevelop.Refactoring/MonoDevelop.CodeActions/CodeRefactoringService.cs
@@ -0,0 +1,157 @@
+//
+// CodeActionService.cs
+//
+// Author:
+//       Mike Krüger <mkrueger@xamarin.com>
+//
+// Copyright (c) 2014 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;
+using Microsoft.CodeAnalysis.CodeRefactorings;
+using System.Collections.Generic;
+using System.Threading.Tasks;
+using System.Threading;
+using Microsoft.CodeAnalysis.Text;
+using Microsoft.CodeAnalysis;
+using Microsoft.CodeAnalysis.CodeActions;
+using MonoDevelop.Core;
+using MonoDevelop.Ide.Editor;
+using MonoDevelop.CodeIssues;
+using Mono.Addins;
+using ICSharpCode.NRefactory6.CSharp.Refactoring;
+using MonoDevelop.Core.Text;
+using System.Linq;
+
+namespace MonoDevelop.CodeActions
+{
+	static class CodeRefactoringService
+	{
+		readonly static List<CodeDiagnosticProvider> providers = new List<CodeDiagnosticProvider> ();
+
+		static CodeRefactoringService ()
+		{	
+			providers.Add (new BuiltInCodeDiagnosticProvider ());
+
+			AddinManager.AddExtensionNodeHandler ("/MonoDevelop/Refactoring/CodeDiagnosticProvider", delegate(object sender, ExtensionNodeEventArgs args) {
+				var node = (TypeExtensionNode)args.ExtensionNode;
+				switch (args.Change) {
+				case ExtensionChange.Add:
+					providers.Add ((CodeDiagnosticProvider)node.CreateInstance ());
+					break;
+				}
+			});
+		}
+
+		public async static Task<IEnumerable<CodeDiagnosticDescriptor>> GetCodeDiagnosticsAsync (DocumentContext documentContext, string language, CancellationToken cancellationToken = default (CancellationToken))
+		{
+			var result = new List<CodeDiagnosticDescriptor> ();
+
+			foreach (var provider in providers) {
+				if (cancellationToken.IsCancellationRequested)
+					return Enumerable.Empty<CodeDiagnosticDescriptor> ();
+				result.AddRange (await provider.GetCodeDiagnosticDescriptorsAsync (documentContext, language, cancellationToken).ConfigureAwait (false));
+			}
+			return result;
+		}
+
+		public async static Task<IEnumerable<CodeDiagnosticFixDescriptor>> GetCodeFixesAsync (DocumentContext documentContext, string language, CancellationToken cancellationToken = default (CancellationToken))
+		{
+			var result = new List<CodeDiagnosticFixDescriptor> ();
+			foreach (var provider in providers) {
+				result.AddRange (await provider.GetCodeFixDescriptorsAsync (documentContext, language, cancellationToken).ConfigureAwait (false));
+			}
+			return result;
+		}
+
+		public async static Task<IEnumerable<CodeRefactoringDescriptor>> GetCodeRefactoringsAsync (DocumentContext documentContext, string language, CancellationToken cancellationToken = default (CancellationToken))
+		{
+			var result = new List<CodeRefactoringDescriptor> ();
+			foreach (var provider in providers) {
+				result.AddRange (await provider.GetCodeRefactoringDescriptorsAsync (documentContext, language, cancellationToken).ConfigureAwait (false));
+			}
+			return result;
+		}
+
+		public static async Task<IEnumerable<ValidCodeAction>> GetValidActionsAsync (TextEditor editor, DocumentContext doc, TextSpan span, CancellationToken cancellationToken = default (CancellationToken))
+		{
+			if (editor == null)
+				throw new ArgumentNullException ("editor");
+			if (doc == null)
+				throw new ArgumentNullException ("doc");
+			var parsedDocument = doc.ParsedDocument;
+			var actions = new List<ValidCodeAction> ();
+			if (parsedDocument == null)
+				return actions;
+			var model = parsedDocument.GetAst<SemanticModel> ();
+			if (model == null)
+				return actions;
+			var root = await model.SyntaxTree.GetRootAsync(cancellationToken).ConfigureAwait (false);
+			if (span.End > root.Span.End)
+				return actions;
+			TextSpan tokenSegment = span;
+			var token = root.FindToken (span.Start);
+			if (!token.IsMissing)
+				tokenSegment = token.Span;
+			try {
+				foreach (var descriptor in await GetCodeRefactoringsAsync (doc, MimeTypeToLanguage(editor.MimeType), cancellationToken).ConfigureAwait (false)) {
+					if (cancellationToken.IsCancellationRequested)
+						return Enumerable.Empty<ValidCodeAction> ();
+					try {
+						await descriptor.GetProvider ().ComputeRefactoringsAsync (
+							new CodeRefactoringContext (doc.AnalysisDocument, span, delegate (CodeAction ca) {
+								var nrca = ca as NRefactoryCodeAction;
+								var validSegment = tokenSegment;
+								if (nrca != null)
+									validSegment = nrca.TextSpan;
+								actions.Add (new ValidCodeAction (ca, validSegment));
+							}, cancellationToken)
+						).ConfigureAwait (false);
+					} catch (OperationCanceledException) {
+						if (cancellationToken.IsCancellationRequested)
+							return Enumerable.Empty<ValidCodeAction> ();
+					} catch (AggregateException) {
+						if (cancellationToken.IsCancellationRequested)
+							return Enumerable.Empty<ValidCodeAction> ();
+					} catch (Exception e) {
+						LoggingService.LogError ("Error while getting refactorings from " + descriptor.IdString, e); 
+						continue;
+					}
+				}
+			} catch (OperationCanceledException) {
+				if (cancellationToken.IsCancellationRequested)
+					return Enumerable.Empty<ValidCodeAction> ();
+			} catch (AggregateException) {
+				if (cancellationToken.IsCancellationRequested)
+					return Enumerable.Empty<ValidCodeAction> ();
+			}
+			return actions;
+		}
+
+		public static string MimeTypeToLanguage (string mimeType)
+		{
+			switch (mimeType) {
+			case "text/x-csharp":
+				return LanguageNames.CSharp;
+			}
+			return null;
+		}
+	}
+}