index.ts

import * as shvl from "shvl";
import { InitializeParams, TextDocumentSyncKind } from "vscode-languageserver";

import { projectRootPatterns } from "./common/constant";
import { IConfig, IDiagnostic, IIndexes, ISuggest } from "./common/types";
import { completionProvider } from "./handles/completion";
import { completionResolveProvider } from "./handles/completionResolve";
import { definitionProvider } from "./handles/definition";
import { documentHighlightProvider } from "./handles/documentHighlight";
import {foldingRangeProvider} from "./handles/foldingRange";
import { hoverProvider } from "./handles/hover";
import { referencesProvider } from "./handles/references";
import { prepareProvider, renameProvider } from "./handles/rename";
import { signatureHelpProvider } from "./handles/signatureHelp";
import { builtinDocs } from "./server/builtin";
import config from "./server/config";
import { connection } from "./server/connection";
import { documents } from "./server/documents";
import { next, unsubscribe } from "./server/parser";
import {selectionRangeProvider} from "./handles/selectionRange";
import {documentSymbolProvider} from "./handles/documentSymbol";
import logger from "./common/logger";

// lsp initialize
connection.onInitialize((param: InitializeParams) => {
  const renamePrepareSupport = param.capabilities.textDocument && param.capabilities.textDocument.rename && param.capabilities.textDocument.rename.prepareSupport === true;
  const { initializationOptions = {} } = param;
  const {
    isNeovim,
    iskeyword,
    runtimepath,
    vimruntime,
    diagnostic,
    suggest,
    indexes,
  }: {
    isNeovim: boolean
    iskeyword: string
    runtimepath: string
    vimruntime: string
    diagnostic: IDiagnostic
    suggest: ISuggest
    indexes: IIndexes,
  } = initializationOptions;

  const runtimepaths = runtimepath ? runtimepath.split(",") : [];

  // config by user's initializationOptions
  const conf: IConfig = {
    isNeovim: isNeovim || false,
    iskeyword: iskeyword || "",
    runtimepath: runtimepaths,
    vimruntime: (vimruntime || "").trim(),
    diagnostic: {
      enable: true,
      ...(diagnostic || {}),
    },
    snippetSupport: shvl.get(param, "capabilities.textDocument.completion.completionItem.snippetSupport"),
    suggest: {
      fromRuntimepath: false,
      fromVimruntime: true,
      ...(suggest || {}),
    },
    indexes: {
      runtimepath: true,
      gap: 100,
      count: 1,
      projectRootPatterns,
      ...(indexes || {}),
    },
    capabilities: param.capabilities
  };

  // init config
  config.init(conf);

  // init builtin docs
  builtinDocs.init();

  return {
    capabilities: {
      textDocumentSync: TextDocumentSyncKind.Incremental,
      documentHighlightProvider: true,
      foldingRangeProvider: true,
      selectionRangeProvider: true,
      documentSymbolProvider: true,
      hoverProvider: true,
      completionProvider: {
        triggerCharacters: [".", ":", "#", "[", "&", "$", "<", '"', "'"],
        resolveProvider: true,
      },
      signatureHelpProvider: {
        triggerCharacters: ["(", ","],
      },
      definitionProvider: true,
      referencesProvider: true,
      renameProvider: renamePrepareSupport ? {
        prepareProvider: true,
      } : true,
    },
  };
});

// document change or open
documents.onDidChangeContent(( change ) => {
  next(change.document);
});

documents.onDidClose((evt) => {
  unsubscribe(evt.document);
});

// listen for document's open/close/change
documents.listen(connection);

// handle completion
connection.onCompletion(completionProvider);

// handle completion resolve
connection.onCompletionResolve(completionResolveProvider);

// handle signature help
connection.onSignatureHelp(signatureHelpProvider);

// handle hover
connection.onHover(hoverProvider);

// handle definition request
connection.onDefinition(definitionProvider);

// handle references
connection.onReferences(referencesProvider);

// handle rename
connection.onPrepareRename(prepareProvider);
connection.onRenameRequest(renameProvider);

// document highlight
connection.onDocumentHighlight(documentHighlightProvider);

// folding range
connection.onFoldingRanges(foldingRangeProvider);

// select range
connection.onSelectionRanges(selectionRangeProvider);

// document symbols
connection.onDocumentSymbol(documentSymbolProvider);

connection.onNotification('$/change/iskeyword', (iskeyword: string) => {
  config.changeByKey('iskeyword', iskeyword)
})

// lsp start
connection.listen();