lsp workspace stuff (#1677)

* some lsp shit Signed-off-by: Jess Frazelle <github@jessfraz.com> * more stuffs Signed-off-by: Jess Frazelle <github@jessfraz.com> * on open send close and open events Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * update the path Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * send on close Signed-off-by: Jess Frazelle <github@jessfraz.com> * on close project Signed-off-by: Jess Frazelle <github@jessfraz.com> * update on close Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * initpromise Signed-off-by: Jess Frazelle <github@jessfraz.com> * add to wasm.ts Signed-off-by: Jess Frazelle <github@jessfraz.com> * Update src/lang/wasm.ts Co-authored-by: Kurt Hutten <k.hutten@protonmail.ch> * restart lsps on failure Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * add panic hook Signed-off-by: Jess Frazelle <github@jessfraz.com> * updartes Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> --------- Signed-off-by: Jess Frazelle <github@jessfraz.com> Co-authored-by: Kurt Hutten <k.hutten@protonmail.ch>
2024-03-11 17:50:31 -07:00
parent cd158f8db0
commit db5657a298
17 changed files with 436 additions and 137 deletions
--- a/src/components/LspProvider.tsx
+++ b/src/components/LspProvider.tsx
@ -0,0 +1,191 @@
+import { LanguageServerClient } from 'editor/plugins/lsp'
+import type * as LSP from 'vscode-languageserver-protocol'
+import React, { createContext, useMemo, useContext } from 'react'
+import { FromServer, IntoServer } from 'editor/plugins/lsp/codec'
+import Server from '../editor/plugins/lsp/server'
+import Client from '../editor/plugins/lsp/client'
+import { TEST } from 'env'
+import kclLanguage from 'editor/plugins/lsp/kcl/language'
+import { copilotPlugin } from 'editor/plugins/lsp/copilot'
+import { useStore } from 'useStore'
+import { useSettingsAuthContext } from 'hooks/useSettingsAuthContext'
+import { Extension } from '@codemirror/state'
+import { LanguageSupport } from '@codemirror/language'
+import { useNavigate } from 'react-router-dom'
+import { basename } from './FileTree'
+import { paths } from 'lib/paths'
+import { FileEntry } from '@tauri-apps/api/fs'
+import { ProjectWithEntryPointMetadata } from 'lib/types'
+
+function getWorkspaceFolders(): LSP.WorkspaceFolder[] {
+  return []
+}
+
+type LspContext = {
+  lspClients: LanguageServerClient[]
+  copilotLSP: Extension | null
+  kclLSP: LanguageSupport | null
+  onProjectClose: (file: FileEntry | null, redirect: boolean) => void
+  onProjectOpen: (
+    project: ProjectWithEntryPointMetadata | null,
+    file: FileEntry | null
+  ) => void
+}
+
+export const LspStateContext = createContext({} as LspContext)
+export const LspProvider = ({ children }: { children: React.ReactNode }) => {
+  const {
+    isKclLspServerReady,
+    isCopilotLspServerReady,
+    setIsKclLspServerReady,
+    setIsCopilotLspServerReady,
+  } = useStore((s) => ({
+    isKclLspServerReady: s.isKclLspServerReady,
+    isCopilotLspServerReady: s.isCopilotLspServerReady,
+    setIsKclLspServerReady: s.setIsKclLspServerReady,
+    setIsCopilotLspServerReady: s.setIsCopilotLspServerReady,
+  }))
+
+  const { auth } = useSettingsAuthContext()
+  const navigate = useNavigate()
+
+  // So this is a bit weird, we need to initialize the lsp server and client.
+  // But the server happens async so we break this into two parts.
+  // Below is the client and server promise.
+  const { lspClient: kclLspClient } = useMemo(() => {
+    const intoServer: IntoServer = new IntoServer()
+    const fromServer: FromServer = FromServer.create()
+    const client = new Client(fromServer, intoServer)
+    if (!TEST) {
+      Server.initialize(intoServer, fromServer).then((lspServer) => {
+        lspServer.start('kcl')
+        setIsKclLspServerReady(true)
+      })
+    }
+
+    const lspClient = new LanguageServerClient({ client, name: 'kcl' })
+    return { lspClient }
+  }, [setIsKclLspServerReady])
+
+  // Here we initialize the plugin which will start the client.
+  // Now that we have multi-file support the name of the file is a dep of
+  // this use memo, as well as the directory structure, which I think is
+  // a good setup because it will restart the client but not the server :)
+  // We do not want to restart the server, its just wasteful.
+  const kclLSP = useMemo(() => {
+    let plugin = null
+    if (isKclLspServerReady && !TEST) {
+      // Set up the lsp plugin.
+      const lsp = kclLanguage({
+        documentUri: `file:///main.kcl`,
+        workspaceFolders: getWorkspaceFolders(),
+        client: kclLspClient,
+      })
+
+      plugin = lsp
+    }
+    return plugin
+  }, [kclLspClient, isKclLspServerReady])
+
+  const { lspClient: copilotLspClient } = useMemo(() => {
+    const intoServer: IntoServer = new IntoServer()
+    const fromServer: FromServer = FromServer.create()
+    const client = new Client(fromServer, intoServer)
+    if (!TEST) {
+      Server.initialize(intoServer, fromServer).then((lspServer) => {
+        const token = auth?.context?.token
+        lspServer.start('copilot', token)
+        setIsCopilotLspServerReady(true)
+      })
+    }
+
+    const lspClient = new LanguageServerClient({ client, name: 'copilot' })
+    return { lspClient }
+  }, [setIsCopilotLspServerReady])
+
+  // Here we initialize the plugin which will start the client.
+  // When we have multi-file support the name of the file will be a dep of
+  // this use memo, as well as the directory structure, which I think is
+  // a good setup because it will restart the client but not the server :)
+  // We do not want to restart the server, its just wasteful.
+  const copilotLSP = useMemo(() => {
+    let plugin = null
+    if (isCopilotLspServerReady && !TEST) {
+      // Set up the lsp plugin.
+      const lsp = copilotPlugin({
+        documentUri: `file:///main.kcl`,
+        workspaceFolders: getWorkspaceFolders(),
+        client: copilotLspClient,
+        allowHTMLContent: true,
+      })
+
+      plugin = lsp
+    }
+    return plugin
+  }, [copilotLspClient, isCopilotLspServerReady])
+
+  const lspClients = [kclLspClient, copilotLspClient]
+
+  const onProjectClose = (file: FileEntry | null, redirect: boolean) => {
+    const currentFilePath = basename(file?.name || 'main.kcl')
+    lspClients.forEach((lspClient) => {
+      lspClient.textDocumentDidClose({
+        textDocument: {
+          uri: `file:///${currentFilePath}`,
+        },
+      })
+    })
+
+    if (redirect) {
+      navigate(paths.HOME)
+    }
+  }
+
+  const onProjectOpen = (
+    project: ProjectWithEntryPointMetadata | null,
+    file: FileEntry | null
+  ) => {
+    const projectName = project?.name || 'ProjectRoot'
+    // Send that the workspace folders changed.
+    lspClients.forEach((lspClient) => {
+      lspClient.workspaceDidChangeWorkspaceFolders(
+        [{ uri: 'file://', name: projectName }],
+        []
+      )
+    })
+    if (file) {
+      // Send that the file was opened.
+      const filename = basename(file?.name || 'main.kcl')
+      lspClients.forEach((lspClient) => {
+        lspClient.textDocumentDidOpen({
+          textDocument: {
+            uri: `file:///${filename}`,
+            languageId: 'kcl',
+            version: 1,
+            text: '',
+          },
+        })
+      })
+    }
+  }
+
+  return (
+    <LspStateContext.Provider
+      value={{
+        lspClients,
+        copilotLSP,
+        kclLSP,
+        onProjectClose,
+        onProjectOpen,
+      }}
+    >
+      {children}
+    </LspStateContext.Provider>
+  )
+}
+
+export default LspProvider
+
+export const useLspContext = () => {
+  return useContext(LspStateContext)
+}