modeling-app/src/editor/plugins/lsp/kcl/language.ts

// Code mirror language implementation for kcl.

import {
  Language,
  defineLanguageFacet,
  LanguageSupport,
} from '@codemirror/language'
import { LanguageServerClient } from 'editor/plugins/lsp'
import { kclPlugin } from '.'
import type * as LSP from 'vscode-languageserver-protocol'
import KclParser from './parser'
import { semanticTokenField } from '../plugin'

const data = defineLanguageFacet({
  // https://codemirror.net/docs/ref/#commands.CommentTokens
  commentTokens: {
    line: '//',
    block: {
      open: '/*',
      close: '*/',
    },
  },
})

export interface LanguageOptions {
  workspaceFolders: LSP.WorkspaceFolder[]
  documentUri: string
  client: LanguageServerClient
}

class KclLanguage extends Language {
  constructor(options: LanguageOptions) {
    const plugin = kclPlugin({
      documentUri: options.documentUri,
      workspaceFolders: options.workspaceFolders,
      allowHTMLContent: true,
      client: options.client,
    })

    const parser = new KclParser()

    super(
      data,
      // For now let's use the javascript parser.
      // It works really well and has good syntax highlighting.
      // We can use our lsp for the rest.
      parser,
      [plugin],
      'kcl'
    )
  }
}

export default class KclLanguageSupport extends LanguageSupport {
  constructor(options: LanguageOptions) {
    const lang = new KclLanguage(options)

    super(lang, [semanticTokenField])
  }
}