(GH-70) Initial implementation of Language Server

- This still needs some work, but it is at least working now
- Single rule for templated values has been implemented
- Question remains over how to get server into file system
- Question remains over sending entire file to server

.gitignore

@@ -17,3 +17,6 @@ yarn.lock
 # License file copied to chocolatey directory
 chocolatey/tools/LICENSE.txt
 config.wyam.hash
+
+[Bb]in/
+[Oo]bj/

package.json

@@ -27,10 +27,11 @@
 		"onCommand:chocolatey.pack",
 		"onCommand:chocolatey.delete",
 		"onCommand:chocolatey.push",
-		"onCommand:chocolatey.installTemplates"
+		"onCommand:chocolatey.installTemplates",
+		"workspaceContains:**/*.nuspec"
 	],
 	"engines": {
-		"vscode": "^1.24.0"
+		"vscode": "^1.30.0"
 	},
 	"categories": [
 		"Other"
@@ -127,7 +128,9 @@
 		"tslint": "^5.12.1",
 		"typemoq": "^2.1.0",
 		"typescript": "^3.3.3",
-		"vscode": "^1.1.29"
+		"vscode": "^1.1.29",
+		"vscode-languageclient": "^5.2.1",
+		"vscode-jsonrpc": "^4.0.0"
 	},
 	"dependencies": {
 		"xml2js": "^0.4.19"

src/Chocolatey.Language.Server/BufferManager.cs

@@ -0,0 +1,40 @@
+using System;
+using System.Collections.Concurrent;
+using OmniSharp.Extensions.LanguageServer.Protocol.Server;
+using Buffer = Microsoft.Language.Xml.Buffer;
+
+namespace Chocolatey.Language.Server
+{
+    public class BufferManager
+    {
+        public EventHandler<DocumentUpdatedEventArgs> BufferUpdated;
+
+        public BufferManager()
+        {
+        }
+
+        private ConcurrentDictionary<Uri, Buffer> _buffers = new ConcurrentDictionary<Uri, Buffer>();
+
+        public void UpdateBuffer(Uri uri, Buffer buffer)
+        {
+            _buffers.AddOrUpdate(uri, buffer, (k, v) => buffer);
+
+            BufferUpdated?.Invoke(this, new DocumentUpdatedEventArgs(uri));
+        }
+
+        public Buffer GetBuffer(Uri uri)
+        {
+            return _buffers.TryGetValue(uri, out var buffer) ? buffer : null;
+        }
+    }
+
+    public class DocumentUpdatedEventArgs : EventArgs
+    {
+        public Uri Uri { get; }
+
+        public DocumentUpdatedEventArgs(Uri uri)
+        {
+            Uri = uri;
+        }
+    }
+}

src/Chocolatey.Language.Server/Chocolatey.Language.Server.csproj

@@ -0,0 +1,14 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+  <PropertyGroup>
+    <OutputType>Exe</OutputType>
+    <TargetFramework>netcoreapp2.1</TargetFramework>
+    <LangVersion>latest</LangVersion>
+  </PropertyGroup>
+
+  <ItemGroup>
+    <PackageReference Include="GuiLabs.Language.Xml" Version="1.2.35" />
+    <PackageReference Include="OmniSharp.Extensions.LanguageServer" Version="0.11.3" />
+  </ItemGroup>
+
+</Project>

src/Chocolatey.Language.Server/DiagnosticsHandler.cs

@@ -0,0 +1,68 @@
+using System;
+using System.Collections.Generic;
+using System.Collections.Immutable;
+using System.IO;
+using System.Linq;
+using System.Xml;
+using System.Xml.Schema;
+using Microsoft.Language.Xml;
+using OmniSharp.Extensions.LanguageServer.Protocol.Models;
+using OmniSharp.Extensions.LanguageServer.Protocol.Server;
+using Buffer = Microsoft.Language.Xml.Buffer;
+using DiagnosticSeverity = OmniSharp.Extensions.LanguageServer.Protocol.Models.DiagnosticSeverity;
+
+namespace Chocolatey.Language.Server
+{
+    public class DiagnosticsHandler
+    {
+        private readonly ILanguageServer _router;
+        private readonly BufferManager _bufferManager;
+        private static readonly IReadOnlyCollection<string> TemplatedValues = new []
+        {
+            "__replace",
+            "space_separated",
+            "tag1"
+        };
+
+        public DiagnosticsHandler(ILanguageServer router, BufferManager bufferManager)
+        {
+            _router = router;
+            _bufferManager = bufferManager;
+        }
+
+        public void PublishDiagnostics(Uri uri, Buffer buffer)
+        {
+            var text = buffer.GetText(0, buffer.Length);
+            var syntaxTree = Parser.Parse(buffer);
+            var textPositions = new TextPositions(text);
+            var diagnostics = new List<Diagnostic>();
+
+            diagnostics.AddRange(NuspecDoesNotContainTemplatedValuesRequirement(syntaxTree, textPositions));
+
+            _router.Document.PublishDiagnostics(new PublishDiagnosticsParams
+            {
+                Uri = uri,
+                Diagnostics = diagnostics
+            });
+        }
+
+        private IEnumerable<Diagnostic> NuspecDoesNotContainTemplatedValuesRequirement(XmlDocumentSyntax syntaxTree, TextPositions textPositions)
+        {
+            foreach (var node in syntaxTree.DescendantNodesAndSelf().OfType<XmlTextSyntax>())
+            {
+                if (!TemplatedValues.Any(x => node.Value.Contains(x, StringComparison.OrdinalIgnoreCase)))
+                {
+                    continue;
+                }
+
+                var range = textPositions.GetRange(node.Start, node.End);
+
+                yield return new Diagnostic {
+                    Message = "Templated value which should be removed",
+                    Severity = DiagnosticSeverity.Error,
+                    Range = range
+                };
+            }
+        }
+    }
+}

src/Chocolatey.Language.Server/Program.cs

@@ -0,0 +1,43 @@
+using System;
+using System.Threading.Tasks;
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Logging;
+using OmniSharp.Extensions.LanguageServer.Server;
+
+namespace Chocolatey.Language.Server
+{
+    class Program
+    {
+        static async Task Main(string[] args)
+        {
+            var options = new LanguageServerOptions()
+                .WithInput(Console.OpenStandardInput())
+                .WithOutput(Console.OpenStandardOutput())
+                .WithLoggerFactory(new LoggerFactory())
+                .AddDefaultLoggingProvider()
+                .WithMinimumLogLevel(LogLevel.Trace)
+                .WithServices(ConfigureServices)
+                .WithHandler<TextDocumentSyncHandler>()
+                .OnInitialize((s, _) => {
+                    var serviceProvider = (s as LanguageServer).Services;
+                    var bufferManager = serviceProvider.GetService<BufferManager>();
+                    var diagnosticsHandler = serviceProvider.GetService<DiagnosticsHandler>();
+
+                    // Hook up diagnostics
+                    bufferManager.BufferUpdated += (__, x) => diagnosticsHandler.PublishDiagnostics(x.Uri, bufferManager.GetBuffer(x.Uri));
+
+                    return Task.CompletedTask;
+                });
+
+            var server = await LanguageServer.From(options);
+
+            await server.WaitForExit;
+        }
+
+        static void ConfigureServices(IServiceCollection services)
+        {
+            services.AddSingleton<BufferManager>();
+            services.AddSingleton<DiagnosticsHandler>();
+        }
+    }
+}

src/Chocolatey.Language.Server/TextDocumentSyncHandler.cs

@@ -0,0 +1,103 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using System.Xml;
+using System.Xml.Schema;
+using Microsoft.Language.Xml;
+using OmniSharp.Extensions.Embedded.MediatR;
+using OmniSharp.Extensions.LanguageServer.Protocol;
+using OmniSharp.Extensions.LanguageServer.Protocol.Client.Capabilities;
+using OmniSharp.Extensions.LanguageServer.Protocol.Models;
+using OmniSharp.Extensions.LanguageServer.Protocol.Server;
+using OmniSharp.Extensions.LanguageServer.Protocol.Server.Capabilities;
+using DiagnosticSeverity = OmniSharp.Extensions.LanguageServer.Protocol.Models.DiagnosticSeverity;
+
+namespace Chocolatey.Language.Server
+{
+    public class TextDocumentSyncHandler : ITextDocumentSyncHandler
+    {
+        private readonly ILanguageServer _router;
+        private readonly BufferManager _bufferManager;
+
+        private readonly DocumentSelector _documentSelector = new DocumentSelector(
+            new DocumentFilter()
+            {
+                Pattern = "**/*.nuspec"
+            }
+        );
+
+        private SynchronizationCapability _capability;
+
+        public TextDocumentSyncHandler(ILanguageServer router, BufferManager bufferManager)
+        {
+            _router = router;
+            _bufferManager = bufferManager;
+        }
+
+        public TextDocumentSyncKind Change { get; } = TextDocumentSyncKind.Full;
+
+        public TextDocumentChangeRegistrationOptions GetRegistrationOptions()
+        {
+            return new TextDocumentChangeRegistrationOptions()
+            {
+                DocumentSelector = _documentSelector,
+                SyncKind = Change
+            };
+        }
+
+        public TextDocumentAttributes GetTextDocumentAttributes(Uri uri)
+        {
+            return new TextDocumentAttributes(uri, "xml");
+        }
+
+        public Task<Unit> Handle(DidChangeTextDocumentParams request, CancellationToken cancellationToken)
+        {
+            var uri = request.TextDocument.Uri;
+            var text = request.ContentChanges.FirstOrDefault()?.Text;
+
+            _bufferManager.UpdateBuffer(uri, new StringBuffer(text));
+            return Unit.Task;
+        }
+
+        public Task<Unit> Handle(DidOpenTextDocumentParams request, CancellationToken cancellationToken)
+        {
+            _bufferManager.UpdateBuffer(request.TextDocument.Uri, new StringBuffer(request.TextDocument.Text));
+            return Unit.Task;
+        }
+
+        public Task<Unit> Handle(DidCloseTextDocumentParams request, CancellationToken cancellationToken)
+        {
+            return Unit.Task;
+        }
+
+        public Task<Unit> Handle(DidSaveTextDocumentParams request, CancellationToken cancellationToken)
+        {
+            return Unit.Task;
+        }
+
+        public void SetCapability(SynchronizationCapability capability)
+        {
+            _capability = capability;
+        }
+
+        TextDocumentRegistrationOptions IRegistration<TextDocumentRegistrationOptions>.GetRegistrationOptions()
+        {
+            return new TextDocumentRegistrationOptions()
+            {
+                DocumentSelector = _documentSelector
+            };
+        }
+
+        TextDocumentSaveRegistrationOptions IRegistration<TextDocumentSaveRegistrationOptions>.GetRegistrationOptions()
+        {
+            return new TextDocumentSaveRegistrationOptions()
+            {
+                DocumentSelector = _documentSelector,
+                IncludeText = default
+            };
+        }
+    }
+}