opencode/packages/opencode/src/util/rpc.ts at dev · argszero/opencode · GitHub

History

66 lines (62 loc) · 1.92 KB

Raw

type Definition = {

[method: string]: (input: any) => any

}

export function listen(rpc: Definition) {

onmessage = async (evt) => {

const parsed = JSON.parse(evt.data)

if (parsed.type === "rpc.request") {

const result = await rpc[parsed.method](parsed.input)

postMessage(JSON.stringify({ type: "rpc.result", result, id: parsed.id }))

}

export function emit(event: string, data: unknown) {

postMessage(JSON.stringify({ type: "rpc.event", event, data }))

}

export function client<T extends Definition>(target: {

postMessage: (data: string) => void | null

onmessage: ((this: Worker, ev: MessageEvent<any>) => any) | null

}) {

const pending = new Map<number, (result: any) => void>()

const listeners = new Map<string, Set<(data: any) => void>>()

let id = 0

target.onmessage = async (evt) => {

const parsed = JSON.parse(evt.data)

if (parsed.type === "rpc.result") {

const resolve = pending.get(parsed.id)

if (resolve) {

resolve(parsed.result)

pending.delete(parsed.id)

}

if (parsed.type === "rpc.event") {

const handlers = listeners.get(parsed.event)

if (handlers) {

for (const handler of handlers) {

handler(parsed.data)

}

return {

call<Method extends keyof T>(method: Method, input: Parameters<T[Method]>[0]): Promise<ReturnType<T[Method]>> {

const requestId = id++

return new Promise((resolve) => {

pending.set(requestId, resolve)

target.postMessage(JSON.stringify({ type: "rpc.request", method, input, id: requestId }))

})

on<Data>(event: string, handler: (data: Data) => void) {

let handlers = listeners.get(event)

if (!handlers) {

handlers = new Set()

listeners.set(event, handlers)

}

handlers.add(handler)

return () => {

handlers!.delete(handler)

}

export * as Rpc from "./rpc"

opencode/packages/opencode/src/util/rpc.ts at dev · argszero/opencode · GitHub

Expand file tree

Breadcrumbs

rpc.ts

Latest commit

History

Breadcrumbs

rpc.ts

File metadata and controls