116 lines
3.7 KiB
TypeScript
116 lines
3.7 KiB
TypeScript
import { hover } from "@github/actions-languageservice/hover";
|
|
import { registerLogger, setLogLevel } from "@github/actions-languageservice/log";
|
|
import { validate } from "@github/actions-languageservice/validate";
|
|
import {
|
|
CompletionItem,
|
|
Connection,
|
|
Hover,
|
|
HoverParams,
|
|
InitializeParams,
|
|
InitializeResult,
|
|
TextDocumentPositionParams,
|
|
TextDocuments,
|
|
TextDocumentSyncKind
|
|
} from "vscode-languageserver";
|
|
import { TextDocument } from "vscode-languageserver-textdocument";
|
|
import { contextProviders } from "./context-providers";
|
|
import { InitializationOptions, RepositoryContext } from "./initializationOptions";
|
|
import { onCompletion } from "./on-completion";
|
|
import { TTLCache } from "./utils/cache";
|
|
import { valueProviders } from "./value-providers";
|
|
|
|
export function initConnection(connection: Connection) {
|
|
const documents: TextDocuments<TextDocument> = new TextDocuments(TextDocument);
|
|
|
|
let sessionToken: string | undefined;
|
|
let repos: RepositoryContext[] = [];
|
|
const cache = new TTLCache();
|
|
|
|
let hasWorkspaceFolderCapability = false;
|
|
let hasDiagnosticRelatedInformationCapability = false;
|
|
|
|
// Register remote console logger with language service
|
|
registerLogger(connection.console);
|
|
|
|
connection.onInitialize((params: InitializeParams) => {
|
|
const capabilities = params.capabilities;
|
|
|
|
hasWorkspaceFolderCapability = !!(capabilities.workspace && !!capabilities.workspace.workspaceFolders);
|
|
hasDiagnosticRelatedInformationCapability = !!(
|
|
capabilities.textDocument &&
|
|
capabilities.textDocument.publishDiagnostics &&
|
|
capabilities.textDocument.publishDiagnostics.relatedInformation
|
|
);
|
|
|
|
const options: InitializationOptions = params.initializationOptions;
|
|
sessionToken = options.sessionToken;
|
|
if (options.repos) {
|
|
repos = options.repos;
|
|
}
|
|
|
|
if (options.logLevel !== undefined) {
|
|
setLogLevel(options.logLevel);
|
|
}
|
|
|
|
const result: InitializeResult = {
|
|
capabilities: {
|
|
textDocumentSync: TextDocumentSyncKind.Full,
|
|
completionProvider: {
|
|
resolveProvider: false,
|
|
triggerCharacters: [":", "."]
|
|
},
|
|
hoverProvider: true
|
|
}
|
|
};
|
|
|
|
if (hasWorkspaceFolderCapability) {
|
|
result.capabilities.workspace = {
|
|
workspaceFolders: {
|
|
supported: true
|
|
}
|
|
};
|
|
}
|
|
|
|
return result;
|
|
});
|
|
|
|
// The content of a text document has changed. This event is emitted
|
|
// when the text document first opened or when its content has changed.
|
|
documents.onDidChangeContent(change => {
|
|
validateTextDocument(change.document);
|
|
});
|
|
|
|
async function validateTextDocument(textDocument: TextDocument): Promise<void> {
|
|
const repoContext = repos.find(repo => textDocument.uri.startsWith(repo.workspaceUri));
|
|
const result = await validate(
|
|
textDocument,
|
|
valueProviders(sessionToken, repoContext, cache),
|
|
contextProviders(sessionToken, repoContext, cache)
|
|
);
|
|
|
|
connection.sendDiagnostics({uri: textDocument.uri, diagnostics: result});
|
|
}
|
|
|
|
// This handler provides the initial list of the completion items.
|
|
connection.onCompletion(async ({position, textDocument}: TextDocumentPositionParams): Promise<CompletionItem[]> => {
|
|
return await onCompletion(
|
|
position,
|
|
documents.get(textDocument.uri)!,
|
|
sessionToken,
|
|
repos.find(repo => textDocument.uri.startsWith(repo.workspaceUri)),
|
|
cache
|
|
);
|
|
});
|
|
|
|
connection.onHover(async ({position, textDocument}: HoverParams): Promise<Hover | null> => {
|
|
return hover(documents.get(textDocument.uri)!, position);
|
|
});
|
|
|
|
// Make the text document manager listen on the connection
|
|
// for open, change and close text document events
|
|
documents.listen(connection);
|
|
|
|
// Listen on the connection
|
|
connection.listen();
|
|
}
|