feat: initial setup for nuxt devtools

src/server-rpc/index.ts

@@ -0,0 +1,100 @@
+import type { TinyWSRequest } from 'tinyws'
+import type { NodeIncomingMessage, NodeServerResponse } from 'h3'
+import type { WebSocket } from 'ws'
+import { createBirpcGroup } from 'birpc'
+import type { ChannelOptions } from 'birpc'
+
+import { parse, stringify } from 'flatted'
+import type { Nuxt } from 'nuxt/schema'
+import type { ClientFunctions, ModuleOptions, NuxtDevtoolsServerContext, ServerFunctions } from '../types'
+
+export function setupRPC(nuxt: Nuxt, options: ModuleOptions): any {
+  const serverFunctions = {} as ServerFunctions
+  const extendedRpcMap = new Map<string, any>()
+  const rpc = createBirpcGroup<ClientFunctions, ServerFunctions>(
+    serverFunctions,
+    [],
+    {
+      resolver: (name, fn) => {
+        if (fn)
+          return fn
+
+        if (!name.includes(':'))
+          return
+
+        const [namespace, fnName] = name.split(':')
+        return extendedRpcMap.get(namespace)?.[fnName]
+      },
+      onError(error, name) {
+        console.error(`[nuxt-devtools] RPC error on executing "${name}":`, error)
+      },
+    },
+  )
+
+  function refresh(event: keyof ServerFunctions) {
+    rpc.broadcast.refresh.asEvent(event)
+  }
+
+  function extendServerRpc(namespace: string, functions: any): any {
+    extendedRpcMap.set(namespace, functions)
+
+    return {
+      broadcast: new Proxy({}, {
+        get: (_, key) => {
+          if (typeof key !== 'string')
+            return
+          return (rpc.broadcast as any)[`${namespace}:${key}`]
+        },
+      }),
+    }
+  }
+
+  const ctx: NuxtDevtoolsServerContext = {
+    nuxt,
+    options,
+    rpc,
+    refresh,
+    extendServerRpc,
+  }
+
+  // @ts-expect-error untyped
+  nuxt.devtools = ctx
+
+  Object.assign(serverFunctions, {
+    // TODO: add rpc
+  } satisfies Partial<ServerFunctions>)
+
+  const wsClients = new Set<WebSocket>()
+  const middleware = async (req: NodeIncomingMessage & TinyWSRequest, _res: NodeServerResponse, next: Function) => {
+    // Handle WebSocket
+    if (req.ws) {
+      const ws = await req.ws()
+      wsClients.add(ws)
+      const channel: ChannelOptions = {
+        post: d => ws.send(d),
+        on: fn => ws.on('message', fn),
+        serialize: stringify,
+        deserialize: parse,
+      }
+      rpc.updateChannels((c) => {
+        c.push(channel)
+      })
+      ws.on('close', () => {
+        wsClients.delete(ws)
+        rpc.updateChannels((c) => {
+          const index = c.indexOf(channel)
+          if (index >= 0)
+            c.splice(index, 1)
+        })
+      })
+    }
+    else {
+      next()
+    }
+  }
+
+  return {
+    middleware,
+    ...ctx,
+  }
+}