DocumentHighlightHandler.cs

//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE file in the project root for full license information.
//

using Microsoft.Extensions.Logging;
using Microsoft.PowerShell.EditorServices.Symbols;
using OmniSharp.Extensions.LanguageServer.Protocol.Client.Capabilities;
using OmniSharp.Extensions.LanguageServer.Protocol.Models;
using OmniSharp.Extensions.LanguageServer.Protocol.Server;
using PowerShellEditorServices.Engine.Utility;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;

namespace Microsoft.PowerShell.EditorServices.TextDocument
{
    public class DocumentHighlightHandler : IDocumentHighlightHandler
    {
        private static readonly DocumentHighlightContainer s_emptyHighlightContainer = new DocumentHighlightContainer();

        private readonly ILogger _logger;

        private readonly WorkspaceService _workspaceService;

        private readonly SymbolsService _symbolsService;

        private readonly TextDocumentRegistrationOptions _registrationOptions;

        private DocumentHighlightCapability _capability;

        public DocumentHighlightHandler(
            ILoggerFactory loggerFactory,
            WorkspaceService workspaceService,
            SymbolsService symbolService)
        {
            _logger = loggerFactory.CreateLogger<OmniSharp.Extensions.LanguageServer.Protocol.Server.DocumentHighlightHandler>();
            _workspaceService = workspaceService;
            _symbolsService = symbolService;
            _registrationOptions = new TextDocumentRegistrationOptions()
            {
                DocumentSelector = new DocumentSelector(new DocumentFilter() { Pattern = "**/*.ps*1" } )
            };
            _logger.LogInformation("highlight handler loaded");
        }

        public TextDocumentRegistrationOptions GetRegistrationOptions()
        {
            return _registrationOptions;
        }

        public Task<DocumentHighlightContainer> Handle(
            DocumentHighlightParams request,
            CancellationToken cancellationToken)
        {
            ScriptFile scriptFile = _workspaceService.GetFile(PathUtils.FromUri(request.TextDocument.Uri));

            IReadOnlyList<SymbolReference> symbolOccurrences = _symbolsService.FindOccurrencesInFile(
                scriptFile,
                (int)request.Position.Line,
                (int)request.Position.Character);

            if (symbolOccurrences == null)
            {
                return Task.FromResult(s_emptyHighlightContainer);
            }

            var highlights = new DocumentHighlight[symbolOccurrences.Count];
            for (int i = 0; i < symbolOccurrences.Count; i++)
            {
                highlights[i] = new DocumentHighlight
                {
                    Kind = DocumentHighlightKind.Write, // TODO: Which symbol types are writable?
                    Range = symbolOccurrences[i].ScriptRegion.ToRange()
                };
            }

            return Task.FromResult(new DocumentHighlightContainer(highlights));
        }

        public void SetCapability(DocumentHighlightCapability capability)
        {
            _capability = capability;
        }
    }
}