getReferenceDocument.ts

//===----------------------------------------------------------------------===//
//
// This source file is part of the VS Code Swift open source project
//
// Copyright (c) 2024 the VS Code Swift project authors
// Licensed under Apache License v2.0
//
// See LICENSE.txt for license information
// See CONTRIBUTORS.txt for the list of VS Code Swift project authors
//
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//

import * as vscode from "vscode";
import * as langclient from "vscode-languageclient/node";
import {
    LegacyGetReferenceDocumentParams,
    LegacyGetReferenceDocumentRequest,
} from "./lspExtensions";

export function activateLegacyGetReferenceDocument(
    client: langclient.LanguageClient
): vscode.Disposable {
    const getReferenceDocument = vscode.workspace.registerTextDocumentContentProvider(
        "sourcekit-lsp",
        {
            provideTextDocumentContent: async (uri, token) => {
                const params: LegacyGetReferenceDocumentParams = {
                    uri: client.code2ProtocolConverter.asUri(uri),
                };

                const result = await client.sendRequest(
                    LegacyGetReferenceDocumentRequest,
                    params,
                    token
                );

                if (result) {
                    return result.content;
                } else {
                    return "Unable to retrieve reference document";
                }
            },
        }
    );

    return getReferenceDocument;
}