initial commit

src/module.ts

@@ -0,0 +1,27 @@
+import { dirname, resolve } from 'path'
+import { defineNuxtModule } from '@nuxt/kit'
+import fs from 'fs-extra'
+
+export interface ModuleOptions {}
+
+export default defineNuxtModule<ModuleOptions>({
+  meta: {
+    name: 'trpc-nuxt',
+    configKey: 'trpc',
+  },
+  defaults: {},
+  async setup(_options, nuxt) {
+    const root = nuxt.options.rootDir
+    const apiPath = resolve(root, 'server/api/trpc/[query].ts')
+
+    await fs.ensureDir(dirname(apiPath))
+    await fs.writeFile(apiPath, `
+      // generated by trpc-nuxt
+      import { createTRPCHandler } from 'trpc-nuxt/handler'
+      import * as functions from '../../fn'
+      export default createTRPCHandler({
+        router: functions.router
+      })
+    `.trimStart())
+  },
+})

src/runtime/handler.ts

@@ -0,0 +1,68 @@
+import { resolveHTTPResponse } from '@trpc/server'
+import type {
+  AnyRouter,
+  ProcedureType,
+  ResponseMeta,
+  TRPCError,
+  inferRouterContext,
+  inferRouterError,
+} from '@trpc/server'
+import { createURL } from 'ufo'
+import type { IncomingMessage } from 'h3'
+import type { TRPCResponse } from '@trpc/server/dist/declarations/src/rpc'
+import { isMethod, useBody } from 'h3'
+
+type MaybePromise<T> = T | Promise<T>
+
+type CreateContextFn<TRouter extends AnyRouter> = (req: IncomingMessage) => MaybePromise<inferRouterContext<TRouter>>
+
+type ResponseMetaFn<TRouter extends AnyRouter> = (opts: {
+  data: TRPCResponse<unknown, inferRouterError<TRouter>>[]
+  ctx?: inferRouterContext<TRouter>
+  paths?: string[]
+  type: ProcedureType | 'unknown'
+  errors: TRPCError[]
+}) => ResponseMeta
+
+export function createTRPCHandler<Router extends AnyRouter>({
+  router,
+  createContext,
+  responseMeta,
+}: {
+  router: Router
+  createContext?: CreateContextFn<Router>
+  responseMeta?: ResponseMetaFn<Router>
+}) {
+  return async (event) => {
+    const {
+      req,
+      res,
+      context,
+    } = event
+
+    const $url = createURL(req.url)
+
+    const httpResponse = await resolveHTTPResponse({
+      router,
+      req: {
+        method: req.method,
+        headers: req.headers,
+        body: isMethod(event, 'GET') ? null : await useBody(event),
+        query: $url.searchParams,
+      },
+      path: context.params.query,
+      createContext: async () => createContext?.(req),
+      responseMeta,
+    })
+
+    const { status, headers, body } = httpResponse
+
+    res.statusCode = status
+
+    Object.keys(headers).forEach((key) => {
+      res.setHeader(key, headers[key])
+    })
+
+    return body
+  }
+}