// 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 System;
using Roslyn.LanguageServer.Protocol;
 
namespace Microsoft.CommonLanguageServerProtocol.Framework.Example;
 
internal class CapabilitiesManager : IInitializeManager<InitializeParams, InitializeResult>
{
    private InitializeParams? _initializeParams;
 
    public void SetInitializeParams(InitializeParams request)
    {
        _initializeParams = request;
    }
 
    public InitializeResult GetInitializeResult()
    {
        var serverCapabilities = new ServerCapabilities()
        {
            SemanticTokensOptions = new SemanticTokensOptions
            {
                Range = true,
            },
        };
 
        var initializeResult = new InitializeResult
        {
            Capabilities = serverCapabilities,
        };
 
        return initializeResult;
    }
 
    public InitializeParams GetInitializeParams()
    {
        if (_initializeParams is null)
        {
            throw new ArgumentNullException(nameof(_initializeParams));
        }
 
        return _initializeParams;
    }
}