PesterCodeLensProvider.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.PowerShell.EditorServices.Commands;
using Microsoft.PowerShell.EditorServices.Symbols;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;

namespace Microsoft.PowerShell.EditorServices.CodeLenses
{
    internal class PesterCodeLensProvider : FeatureProviderBase, ICodeLensProvider
    {
        private static char[] QuoteChars = new char[] { '\'', '"'};

        private EditorSession editorSession;
        private IDocumentSymbolProvider symbolProvider;

        public PesterCodeLensProvider(EditorSession editorSession)
        {
            this.editorSession = editorSession;
            this.symbolProvider = new PesterDocumentSymbolProvider();
        }

        private IEnumerable<CodeLens> GetPesterLens(
            PesterSymbolReference pesterSymbol,
            ScriptFile scriptFile)
        {
            var clientCommands = new ClientCommand[]
            {
                new ClientCommand(
                    "PowerShell.RunPesterTests",
                    "Run tests",
                    new object[]
                    {
                        scriptFile.ClientFilePath,
                        false, // Don't debug
                        pesterSymbol.TestName,
                    }),

                new ClientCommand(
                    "PowerShell.RunPesterTests",
                    "Debug tests",
                    new object[]
                    {
                        scriptFile.ClientFilePath,
                        true, // Run in debugger
                        pesterSymbol.TestName,
                    }),
            };

            return
                clientCommands.Select(
                    command =>
                        new CodeLens(
                            this,
                            scriptFile,
                            pesterSymbol.ScriptRegion,
                            command));
        }

        public CodeLens[] ProvideCodeLenses(ScriptFile scriptFile)
        {
            var symbols =
                this.symbolProvider
                    .ProvideDocumentSymbols(scriptFile);

            var lenses =
                symbols
                    .OfType<PesterSymbolReference>()
                    .Where(s => s.Command == PesterCommandType.Describe)
                    .SelectMany(s => this.GetPesterLens(s, scriptFile))
                    .Where(codeLens => codeLens != null)
                    .ToArray();

            return lenses;
        }

        public Task<CodeLens> ResolveCodeLensAsync(
            CodeLens codeLens,
            CancellationToken cancellationToken)
        {
            // This provider has no specific behavior for
            // resolving CodeLenses.
            return Task.FromResult(codeLens);
        }
    }
}