// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.
 
using Microsoft.CodeAnalysis.LanguageServer.HostWorkspace;
using Roslyn.LanguageServer.Protocol;
 
namespace Microsoft.CodeAnalysis.LanguageServer.UnitTests;
 
internal static class LspClientExtensions
{
    public static async Task Initialized(this ILspClient lspClient)
    {
        await lspClient.ExecuteRequestAsync<InitializedParams, object>(Methods.InitializedName, new InitializedParams(), CancellationToken.None);
    }
 
    public static async Task<InitializeResult?> Initialize(this ILspClient lspClient, ClientCapabilities clientCapabilities)
    {
        return await lspClient.ExecuteRequestAsync<InitializeParams, InitializeResult>(Methods.InitializeName, new InitializeParams { Capabilities = clientCapabilities }, CancellationToken.None);
    }
 
    public static async Task OpenProjectsAsync(this ILspClient lspClient, Uri[] projects)
    {
        await lspClient.ExecuteNotificationAsync<OpenProjectHandler.NotificationParams>(OpenProjectHandler.OpenProjectName, new() { Projects = projects });
    }
 
    public static async Task<VSInternalCodeAction[]> RunGetCodeActionsAsync(
        this ILspClient lspClient,
        CodeActionParams codeActionParams)
    {
        var result = await lspClient.ExecuteRequestAsync<CodeActionParams, CodeAction[]>(Methods.TextDocumentCodeActionName, codeActionParams, CancellationToken.None);
        Assert.NotNull(result);
        return [.. result.Cast<VSInternalCodeAction>()];
    }
 
    public static async Task<VSInternalCodeAction> RunGetCodeActionResolveAsync(
        this ILspClient lspClient,
        VSInternalCodeAction unresolvedCodeAction)
    {
        var result = (VSInternalCodeAction?)await lspClient.ExecuteRequestAsync<CodeAction, CodeAction>(Methods.CodeActionResolveName, unresolvedCodeAction, CancellationToken.None);
        Assert.NotNull(result);
        return result;
    }
}