cloudaxe-opencode/packages/opencode/src/lsp/client.ts

import { Readable, Writable } from "stream"
import path from "path"
import {
  createMessageConnection,
  StreamMessageReader,
  StreamMessageWriter,
} from "vscode-jsonrpc/node"
import type { Diagnostic as VSCodeDiagnostic } from "vscode-languageserver-types"
import { App } from "../app/app"
import { Log } from "../util/log"
import { LANGUAGE_EXTENSIONS } from "./language"
import { Bus } from "../bus"
import z from "zod"

export namespace LSPClient {
  const log = Log.create({ service: "lsp.client" })

  export type Info = Awaited<ReturnType<typeof create>>

  export type Diagnostic = VSCodeDiagnostic

  export const Event = {
    Diagnostics: Bus.event(
      "lsp.client.diagnostics",
      z.object({
        serverID: z.string(),
        path: z.string(),
      }),
    ),
  }

  export async function create(input: {
    cmd: string[]
    serverID: string
    initialization?: any
  }) {
    const app = App.info()
    log.info("starting client", {
      ...input,
      cwd: app.path.cwd,
    })

    const server = Bun.spawn({
      cmd: input.cmd,
      stdin: "pipe",
      stdout: "pipe",
      stderr: "pipe",
      cwd: app.path.cwd,
    })

    const stdout = new Readable({
      read() {},
      construct(callback) {
        const reader = server.stdout.getReader()
        const pump = async () => {
          try {
            while (true) {
              const { done, value } = await reader.read()
              if (done) {
                this.push(null)
                break
              }
              this.push(Buffer.from(value))
            }
          } catch (error) {
            this.destroy(
              error instanceof Error ? error : new Error(String(error)),
            )
          }
        }
        pump()
        callback()
      },
    })

    const stdin = new Writable({
      write(chunk, _encoding, callback) {
        server.stdin.write(chunk)
        callback()
      },
    })

    const connection = createMessageConnection(
      new StreamMessageReader(stdout),
      new StreamMessageWriter(stdin),
    )

    const diagnostics = new Map<string, Diagnostic[]>()
    connection.onNotification("textDocument/publishDiagnostics", (params) => {
      const path = new URL(params.uri).pathname
      log.info("textDocument/publishDiagnostics", {
        path,
      })
      diagnostics.set(path, params.diagnostics)
      Bus.publish(Event.Diagnostics, { path, serverID: input.serverID })
    })
    connection.onRequest("workspace/configuration", async () => {
      return [{}]
    })
    connection.listen()

    const response = await connection.sendRequest("initialize", {
      processId: server.pid,
      workspaceFolders: [
        {
          name: "workspace",
          uri: "file://" + app.path.cwd,
        },
      ],
      initializationOptions: {
        ...input.initialization,
      },
      capabilities: {
        workspace: {
          configuration: true,
        },
        textDocument: {
          synchronization: {
            didOpen: true,
          },
          publishDiagnostics: {
            versionSupport: true,
          },
        },
      },
    })
    await connection.sendNotification("initialized", {})
    log.info("initialized")

    const files = new Set<string>()

    const result = {
      get clientID() {
        return input.serverID
      },
      get connection() {
        return connection
      },
      notify: {
        async open(input: { path: string }) {
          input.path = path.isAbsolute(input.path)
            ? input.path
            : path.resolve(app.path.cwd, input.path)
          const file = Bun.file(input.path)
          const text = await file.text()
          const opened = files.has(input.path)
          if (!opened) {
            log.info("textDocument/didOpen", input)
            diagnostics.delete(input.path)
            const extension = path.extname(input.path)
            const languageId = LANGUAGE_EXTENSIONS[extension] ?? "plaintext"
            await connection.sendNotification("textDocument/didOpen", {
              textDocument: {
                uri: `file://` + input.path,
                languageId,
                version: Date.now(),
                text,
              },
            })
            files.add(input.path)
            return
          }

          log.info("textDocument/didChange", input)
          diagnostics.delete(input.path)
          await connection.sendNotification("textDocument/didChange", {
            textDocument: {
              uri: `file://` + input.path,
              version: Date.now(),
            },
            contentChanges: [
              {
                text,
              },
            ],
          })
        },
      },
      get diagnostics() {
        return diagnostics
      },
      async waitForDiagnostics(input: { path: string }) {
        input.path = path.isAbsolute(input.path)
          ? input.path
          : path.resolve(app.path.cwd, input.path)
        log.info("waiting for diagnostics", input)
        let unsub: () => void
        let timeout: NodeJS.Timeout
        return await Promise.race([
          new Promise<void>(async (resolve) => {
            unsub = Bus.subscribe(Event.Diagnostics, (event) => {
              if (
                event.properties.path === input.path &&
                event.properties.serverID === result.clientID
              ) {
                log.info("got diagnostics", input)
                clearTimeout(timeout)
                unsub?.()
                resolve()
              }
            })
          }),
          new Promise<void>((resolve) => {
            timeout = setTimeout(() => {
              log.info("timed out refreshing diagnostics", input)
              unsub?.()
              resolve()
            }, 5000)
          }),
        ])
      },
      async shutdown() {
        log.info("shutting down")
        connection.end()
        connection.dispose()
      },
    }

    return result
  }
}