update types

src/module.ts

@@ -0,0 +1,84 @@
+import { dirname, join, resolve } from 'pathe'
+import { addServerHandler, 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 clientPath = join(nuxt.options.buildDir, 'trpc-client.ts')
+    const handlerPath = join(nuxt.options.buildDir, 'trpc-handler.ts')
+
+    nuxt.hook('config', (options) => {
+      options?.build?.transpile?.push('trpc-nuxt/client')
+    })
+
+    addServerHandler({
+      route: '/trpc/*',
+      handler: handlerPath,
+    })
+
+    nuxt.hook('autoImports:extend', (imports) => {
+      imports.push(
+        { name: 'useTrpcQuery', from: clientPath },
+        { name: 'useLazyTrpcQuery', from: clientPath },
+        { name: 'useTrpcMutation', from: clientPath },
+      )
+    })
+
+    const srcDir = nuxt.options.srcDir
+    const optionsPath = resolve(srcDir, 'server/fn/index.ts')
+
+    await fs.ensureDir(dirname(clientPath))
+
+    await fs.writeFile(clientPath, `
+      import * as trpc from '@trpc/client'
+      import { createTRPCComposables } from 'trpc-nuxt/client'
+      import { router } from '~/server/fn'
+
+      const client = trpc.createTRPCClient<typeof router>({
+        url: process.browser ? '/trpc' : 'http://localhost:3000/trpc',
+      })
+
+      const {
+        useTrpcQuery,
+        useLazyTrpcQuery,
+        useTrpcMutation
+      } = createTRPCComposables<typeof router>(client)
+
+      export {
+        useTrpcQuery,
+        useLazyTrpcQuery,
+        useTrpcMutation
+      }
+    `)
+
+    await fs.writeFile(handlerPath, `
+      import { createTRPCHandler } from 'trpc-nuxt/api'
+      import * as functions from '~/server/fn'
+      export default createTRPCHandler({
+        router: functions.router
+      })
+    `)
+
+    if (!fs.existsSync(optionsPath)) {
+      await fs.writeFile(optionsPath, `
+      // Generated by trpc-nuxt
+      import * as trpc from '@trpc/server'
+      
+      export const router = trpc
+        .router()
+        .query('hello', {
+          resolve: () => 'world',
+        });
+      
+      export type Router = typeof router
+    `.trimStart())
+    }
+  },
+})