Added FoldingRanges support to TextDocumentClient.

Author: ezolotko

src/Client/Clients/TextDocumentClient.DocumentHighlight.cs renamed to src/Client/Clients/TextDocumentClient.DocumentHighlights.cs

@@ -30,14 +30,14 @@ public partial class TextDocumentClient
         /// <returns>
         ///     A <see cref="Task{TResult}"/> that resolves to the completions or <c>null</c> if no document highlights are available at the specified position.
         /// </returns>
-        public Task<DocumentHighlightContainer> DocumentHighlight(string filePath, int line, int column, CancellationToken cancellationToken = default(CancellationToken))
+        public Task<DocumentHighlightContainer> DocumentHighlights(string filePath, int line, int column, CancellationToken cancellationToken = default(CancellationToken))
         {
             if (string.IsNullOrWhiteSpace(filePath))
                 throw new ArgumentException($"Argument cannot be null, empty, or entirely composed of whitespace: {nameof(filePath)}.", nameof(filePath));
 
             var documentUri = DocumentUri.FromFileSystemPath(filePath);
 
-            return DocumentHighlight(documentUri, line, column, cancellationToken);
+            return DocumentHighlights(documentUri, line, column, cancellationToken);
         }
 
         /// <summary>
@@ -58,7 +58,7 @@ public partial class TextDocumentClient
         /// <returns>
         ///     A <see cref="Task{TResult}"/> that resolves to the completions or <c>null</c> if no document highlights are available at the specified position.
         /// </returns>
-        public Task<DocumentHighlightContainer> DocumentHighlight(Uri documentUri, int line, int column, CancellationToken cancellationToken = default(CancellationToken))
+        public Task<DocumentHighlightContainer> DocumentHighlights(Uri documentUri, int line, int column, CancellationToken cancellationToken = default(CancellationToken))
         {
             return PositionalRequest<DocumentHighlightContainer>(DocumentNames.DocumentHighlight, documentUri, line, column, cancellationToken);
         }

src/Client/Clients/TextDocumentClient.FoldingRanges.cs

@@ -0,0 +1,66 @@
+using System;
+using System.Threading;
+using System.Threading.Tasks;
+using OmniSharp.Extensions.LanguageServer.Client.Utilities;
+using OmniSharp.Extensions.LanguageServer.Protocol;
+using OmniSharp.Extensions.LanguageServer.Protocol.Models;
+
+namespace OmniSharp.Extensions.LanguageServer.Client.Clients
+{
+    /// <summary>
+    ///     Client for the LSP Text Document API.
+    /// </summary>
+    public partial class TextDocumentClient
+    {
+        /// <summary>
+        ///     Request document folding ranges.
+        /// </summary>
+        /// <param name="filePath">
+        ///     The full file-system path of the text document.
+        /// </param>
+        /// <param name="cancellationToken">
+        ///     An optional <see cref="CancellationToken"/> that can be used to cancel the request.
+        /// </param>
+        /// <returns>
+        ///     A <see cref="Task{TResult}"/> that resolves to the completions or <c>null</c> if no document highlights are available at the specified position.
+        /// </returns>
+        public Task<Container<FoldingRange>> FoldingRanges(string filePath, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            if (string.IsNullOrWhiteSpace(filePath))
+                throw new ArgumentException($"Argument cannot be null, empty, or entirely composed of whitespace: {nameof(filePath)}.", nameof(filePath));
+
+            var documentUri = DocumentUri.FromFileSystemPath(filePath);
+
+            return FoldingRanges(documentUri, cancellationToken);
+        }
+
+        /// <summary>
+        ///     Request document highlights at the specified document position.
+        /// </summary>
+        /// <param name="documentUri">
+        ///     The document URI.
+        /// </param>
+        /// <param name="line">
+        ///     The target line (0-based).
+        /// </param>
+        /// <param name="column">
+        ///     The target column (0-based).
+        /// </param>
+        /// <param name="cancellationToken">
+        ///     An optional <see cref="CancellationToken"/> that can be used to cancel the request.
+        /// </param>
+        /// <returns>
+        ///     A <see cref="Task{TResult}"/> that resolves to the completions or <c>null</c> if no document highlights are available at the specified position.
+        /// </returns>
+        public async Task<Container<FoldingRange>> FoldingRanges(Uri documentUri, CancellationToken cancellationToken = default(CancellationToken))
+        {
+            var request = new FoldingRangeRequestParam {
+                TextDocument = new TextDocumentItem {
+                    Uri = documentUri
+                }
+            };
+
+            return await Client.SendRequest<Container<FoldingRange>>(DocumentNames.FoldingRange, request, cancellationToken).ConfigureAwait(false);
+        }
+    }
+}

test/Client.Tests/ClientTests.cs

@@ -320,7 +320,7 @@ public async Task Definition_Success()
         ///     Ensure that the language client can successfully request DocumentHighlight.
         /// </summary>
         [Fact(DisplayName = "Language client can successfully request document highlights", Skip = "Periodic failures")]
-        public async Task DocumentHighlight_Success()
+        public async Task DocumentHighlights_Success()
         {
             await Connect();
 
@@ -355,7 +355,7 @@ public async Task DocumentHighlight_Success()
                 return Task.FromResult(expectedHighlights);
             });
 
-            var definitions = await LanguageClient.TextDocument.DocumentHighlight(AbsoluteDocumentPath, line, column);
+            var definitions = await LanguageClient.TextDocument.DocumentHighlights(AbsoluteDocumentPath, line, column);
 
             var actualDefinitions = definitions.ToArray();
             Assert.Collection(actualDefinitions, actualHighlight => {
@@ -373,6 +373,47 @@ public async Task DocumentHighlight_Success()
             });
         }
 
+        /// <summary>
+        ///     Ensure that the language client can successfully request FoldingRanges.
+        /// </summary>
+        [Fact(DisplayName = "Language client can successfully request document folding ranges", Skip = "Periodic failures")]
+        public async Task FoldingRanges_Success()
+        {
+            await Connect();
+
+            var expectedDocumentPath = AbsoluteDocumentPath;
+            var expectedDocumentUri = DocumentUri.FromFileSystemPath(expectedDocumentPath);
+
+            var expectedFoldingRanges = new Container<FoldingRange>(
+                new FoldingRange {
+                    Kind = FoldingRangeKind.Region,
+                    StartLine = 5,
+                    StartCharacter = 1,
+                    EndLine = 7,
+                    EndCharacter = 2,
+                });
+
+            ServerDispatcher.HandleRequest<FoldingRangeRequestParam, Container<FoldingRange>>(DocumentNames.FoldingRange, (request, cancellationToken) => {
+                Assert.NotNull(request.TextDocument);
+                Assert.Equal(expectedDocumentUri, request.TextDocument.Uri);
+                return Task.FromResult(expectedFoldingRanges);
+            });
+
+            var foldingRanges = await LanguageClient.TextDocument.FoldingRanges(AbsoluteDocumentPath);
+
+            var actualFoldingRanges = foldingRanges.ToArray();
+            Assert.Collection(actualFoldingRanges, actualFoldingRange => {
+                var expectedFoldingRange = expectedFoldingRanges.Single();
+
+                Assert.Equal(FoldingRangeKind.Region, expectedFoldingRange.Kind);
+
+                Assert.Equal(expectedFoldingRange.StartLine, actualFoldingRange.StartLine);
+                Assert.Equal(expectedFoldingRange.StartCharacter, actualFoldingRange.StartCharacter);
+                Assert.Equal(expectedFoldingRange.EndLine, actualFoldingRange.EndLine);
+                Assert.Equal(expectedFoldingRange.EndCharacter, actualFoldingRange.EndCharacter);
+            });
+        }
+
         /// <summary>
         ///     Ensure that the language client can successfully receive Diagnostics from the server.
         /// </summary>