@dudousxd/nestjs-inertia-codegen 1.2.0 → 1.4.0

package/dist/cli/main.js

@@ -216,16 +216,6 @@ function validateNameSegment(seg, fullName) {
   }
 }
 __name(validateNameSegment, "validateNameSegment");
-function detectCollisions(tree, name) {
-  for (const [key, node] of tree) {
-    if (node.kind === "leaf") {
-    } else {
-      void key;
-    }
-  }
-  void name;
-}
-__name(detectCollisions, "detectCollisions");
 function insertIntoTree(tree, segments, leaf, fullName) {
   const head = segments[0];
   const rest = segments.slice(1);
@@ -255,7 +245,30 @@ function insertIntoTree(tree, segments, leaf, fullName) {
   }
 }
 __name(insertIntoTree, "insertIntoTree");
-function emitRouterTypeBlock(tree, indent) {
+function buildParamsType(params) {
+  const pathParams = params.filter((p) => p.source === "path");
+  if (pathParams.length === 0) return "never";
+  return `{ ${pathParams.map((p) => `${p.name}: string`).join("; ")} }`;
+}
+__name(buildParamsType, "buildParamsType");
+function hasPathParams(params) {
+  return params.some((p) => p.source === "path");
+}
+__name(hasPathParams, "hasPathParams");
+function buildResponseType(c, outDir) {
+  if (c.controllerRef) {
+    let relPath = relative3(outDir, c.controllerRef.filePath).replace(/\.ts$/, "");
+    if (!relPath.startsWith(".")) relPath = `./${relPath}`;
+    return `Awaited<ReturnType<import('${relPath}').${c.controllerRef.className}['${c.controllerRef.methodName}']>>`;
+  }
+  const respRef = c.contractSource.responseRef;
+  if (respRef) {
+    return respRef.isArray ? `Array<${respRef.name}>` : respRef.name;
+  }
+  return c.contractSource.response;
+}
+__name(buildResponseType, "buildResponseType");
+function emitRouterTypeBlock(tree, indent, outDir) {
   const pad = " ".repeat(indent);
   const lines = [];
   for (const [key, node] of tree) {
@@ -267,14 +280,14 @@ function emitRouterTypeBlock(tree, indent) {
       const query = queryRef ? queryRef.isArray ? `Array<${queryRef.name}>` : queryRef.name : c.contractSource.query ?? "never";
       const bodyRef = c.contractSource.bodyRef;
       const body = method === "GET" ? "never" : bodyRef ? bodyRef.isArray ? `Array<${bodyRef.name}>` : bodyRef.name : c.contractSource.body ?? "never";
-      const respRef = c.contractSource.responseRef;
-      const response = respRef ? respRef.isArray ? `Array<${respRef.name}>` : respRef.name : c.contractSource.response;
+      const response = buildResponseType(c, outDir);
+      const params = buildParamsType(c.params);
       const safeMethod = JSON.stringify(method);
       const safeUrl = JSON.stringify(c.path);
-      lines.push(`${pad}${objKey}: { method: ${safeMethod}; url: ${safeUrl}; query: ${query}; body: ${body}; response: ${response} };`);
+      lines.push(`${pad}${objKey}: { method: ${safeMethod}; url: ${safeUrl}; params: ${params}; query: ${query}; body: ${body}; response: ${response} };`);
     } else {
       lines.push(`${pad}${objKey}: {`);
-      lines.push(...emitRouterTypeBlock(node.children, indent + 2));
+      lines.push(...emitRouterTypeBlock(node.children, indent + 2, outDir));
       lines.push(`${pad}};`);
     }
   }
@@ -294,20 +307,61 @@ function emitApiObjectBlock(tree, indent) {
       const fetcherMethod = method.toLowerCase();
       if (method === "GET") {
         const typeAccess = buildRouterTypeAccess(c.name);
+        const withParams = hasPathParams(c.params);
         lines.push(`${pad}${objKey}: {`);
-        lines.push(`${pad}  queryKey: (query?: ${typeAccess}['query']) => query !== undefined ? [${flatName}, query] as const : [${flatName}] as const,`);
-        lines.push(`${pad}  queryOptions: (query?: ${typeAccess}['query']) => ({`);
-        lines.push(`${pad}    queryKey: query !== undefined ? [${flatName}, query] as const : [${flatName}] as const,`);
-        lines.push(`${pad}    queryFn: () => fetcher.get<${typeAccess}['response']>(route(${flatName} as never) || ${safePath}, { query }),`);
-        lines.push(`${pad}  }),`);
+        if (withParams) {
+          lines.push(`${pad}  queryKey: (params: ${typeAccess}['params'], query?: ${typeAccess}['query']) => query !== undefined ? [${flatName}, params, query] as const : [${flatName}, params] as const,`);
+          lines.push(`${pad}  queryOptions: (params: ${typeAccess}['params'], query?: ${typeAccess}['query']) =>`);
+          lines.push(`${pad}    _queryOptions({`);
+          lines.push(`${pad}      queryKey: query !== undefined ? [${flatName}, params, query] as const : [${flatName}, params] as const,`);
+          lines.push(`${pad}      queryFn: () => fetcher.get<${typeAccess}['response']>(route(${flatName} as never, params as never) || ${safePath}, { query }),`);
+          lines.push(`${pad}    }),`);
+          lines.push(`${pad}  infiniteQueryOptions: (params: ${typeAccess}['params'], query?: ${typeAccess}['query']) => ({`);
+          lines.push(`${pad}    queryKey: query !== undefined ? [${flatName}, params, query] as const : [${flatName}, params] as const,`);
+          lines.push(`${pad}    queryFn: ({ pageParam }: { pageParam: number }) => fetcher.get<${typeAccess}['response']>(route(${flatName} as never, params as never) || ${safePath}, { query: { ...query, page: pageParam } }),`);
+          lines.push(`${pad}    initialPageParam: 1,`);
+          lines.push(`${pad}    getNextPageParam: (lastPage: ${typeAccess}['response']) => {`);
+          lines.push(`${pad}      const meta = (lastPage as any)?.meta;`);
+          lines.push(`${pad}      if (meta?.page != null && meta?.lastPage != null) {`);
+          lines.push(`${pad}        return meta.page < meta.lastPage ? meta.page + 1 : undefined;`);
+          lines.push(`${pad}      }`);
+          lines.push(`${pad}      return undefined;`);
+          lines.push(`${pad}    },`);
+          lines.push(`${pad}  }),`);
+        } else {
+          lines.push(`${pad}  queryKey: (query?: ${typeAccess}['query']) => query !== undefined ? [${flatName}, query] as const : [${flatName}] as const,`);
+          lines.push(`${pad}  queryOptions: (query?: ${typeAccess}['query']) =>`);
+          lines.push(`${pad}    _queryOptions({`);
+          lines.push(`${pad}      queryKey: query !== undefined ? [${flatName}, query] as const : [${flatName}] as const,`);
+          lines.push(`${pad}      queryFn: () => fetcher.get<${typeAccess}['response']>(route(${flatName} as never) || ${safePath}, { query }),`);
+          lines.push(`${pad}    }),`);
+          lines.push(`${pad}  infiniteQueryOptions: (query?: ${typeAccess}['query']) => ({`);
+          lines.push(`${pad}    queryKey: query !== undefined ? [${flatName}, query] as const : [${flatName}] as const,`);
+          lines.push(`${pad}    queryFn: ({ pageParam }: { pageParam: number }) => fetcher.get<${typeAccess}['response']>(route(${flatName} as never) || ${safePath}, { query: { ...query, page: pageParam } }),`);
+          lines.push(`${pad}    initialPageParam: 1,`);
+          lines.push(`${pad}    getNextPageParam: (lastPage: ${typeAccess}['response']) => {`);
+          lines.push(`${pad}      const meta = (lastPage as any)?.meta;`);
+          lines.push(`${pad}      if (meta?.page != null && meta?.lastPage != null) {`);
+          lines.push(`${pad}        return meta.page < meta.lastPage ? meta.page + 1 : undefined;`);
+          lines.push(`${pad}      }`);
+          lines.push(`${pad}      return undefined;`);
+          lines.push(`${pad}    },`);
+          lines.push(`${pad}  }),`);
+        }
         lines.push(`${pad}},`);
       } else {
         const typeAccess = buildRouterTypeAccess(c.name);
+        const withParams = hasPathParams(c.params);
         lines.push(`${pad}${objKey}: {`);
         lines.push(`${pad}  queryKey: () => [${flatName}] as const,`);
-        lines.push(`${pad}  mutationOptions: () => ({`);
-        lines.push(`${pad}    mutationFn: (body: ${typeAccess}['body']) => fetcher.${fetcherMethod}<${typeAccess}['response']>(route(${flatName} as never) || ${safePath}, { body }),`);
-        lines.push(`${pad}  }),`);
+        lines.push(`${pad}  mutationOptions: () =>`);
+        lines.push(`${pad}    _mutationOptions({`);
+        if (withParams) {
+          lines.push(`${pad}      mutationFn: (input: { params: ${typeAccess}['params']; body: ${typeAccess}['body'] }) => fetcher.${fetcherMethod}<${typeAccess}['response']>(route(${flatName} as never, input.params as never) || ${safePath}, { body: input.body }),`);
+        } else {
+          lines.push(`${pad}      mutationFn: (body: ${typeAccess}['body']) => fetcher.${fetcherMethod}<${typeAccess}['response']>(route(${flatName} as never) || ${safePath}, { body }),`);
+        }
+        lines.push(`${pad}    }),`);
         lines.push(`${pad}},`);
       }
     } else {
@@ -330,11 +384,15 @@ function buildApiFile(routes, outDir) {
   for (const r of contracted) {
     const cs = r.contract?.contractSource;
     if (!cs) continue;
-    for (const ref of [
+    const refs = r.controllerRef ? [
+      cs.queryRef,
+      cs.bodyRef
+    ] : [
       cs.queryRef,
       cs.bodyRef,
       cs.responseRef
-    ]) {
+    ];
+    for (const ref of refs) {
       if (!ref) continue;
       let names = importsByFile.get(ref.filePath);
       if (!names) {
@@ -344,10 +402,18 @@ function buildApiFile(routes, outDir) {
       names.add(ref.name);
     }
   }
+  const hasGetRoutes = contracted.some((r) => r.method === "GET");
+  const hasMutationRoutes = contracted.some((r) => r.method !== "GET");
   const lines = [
     "// Generated by @dudousxd/nestjs-inertia-codegen. Do not edit.",
     ""
   ];
+  const tqImports = [];
+  if (hasGetRoutes) tqImports.push("queryOptions as _queryOptions");
+  if (hasMutationRoutes) tqImports.push("mutationOptions as _mutationOptions");
+  if (tqImports.length > 0) {
+    lines.push(`import { ${tqImports.join(", ")} } from '@tanstack/react-query';`);
+  }
   lines.push("import { route } from './routes.js';");
   lines.push("import { createFetcher } from '@dudousxd/nestjs-inertia-client';");
   if (importsByFile.size > 0 && outDir) {
@@ -401,13 +467,14 @@ function buildApiFile(routes, outDir) {
       method: r.method,
       name,
       path: r.path,
+      params: r.params,
+      controllerRef: r.controllerRef,
       contractSource: c.contractSource
     };
     insertIntoTree(tree, segments, leaf, name);
   }
-  void detectCollisions;
   lines.push("export type ApiRouter = {");
-  lines.push(...emitRouterTypeBlock(tree, 2));
+  lines.push(...emitRouterTypeBlock(tree, 2, outDir ?? ""));
   lines.push("};");
   lines.push("");
   lines.push("export const api = {");
@@ -503,8 +570,9 @@ __name(emitIndex, "emitIndex");
 
 // src/emit/emit-pages.ts
 import { mkdir as mkdir4, writeFile as writeFile4 } from "fs/promises";
-import { join as join5 } from "path";
-async function emitPages(pages, outDir) {
+import { join as join5, relative as relative4 } from "path";
+async function emitPages(pages, outDir, options = {}) {
+  const propsExport = options.propsExport ?? "ComponentProps";
   await mkdir4(outDir, {
     recursive: true
   });
@@ -513,14 +581,40 @@ async function emitPages(pages, outDir) {
     const key = needsQuotes(p.name) ? JSON.stringify(p.name) : p.name;
     return `  ${key}: ${propType};`;
   }).join("\n");
+  const pageNameUnion = pages.length > 0 ? pages.map((p) => JSON.stringify(p.name)).join(" | ") : "never";
+  const augBody = pages.map((p) => {
+    const key = needsQuotes(p.name) ? JSON.stringify(p.name) : p.name;
+    const valueType = buildAugmentationType(p, outDir, propsExport);
+    return `    ${key}: ${valueType};`;
+  }).join("\n");
+  const propsHelper = "\nexport type InertiaProps<K extends InertiaPageName> = import('@dudousxd/nestjs-inertia').InertiaPages[K];\n";
   const content = `// Generated by @dudousxd/nestjs-inertia-codegen. Do not edit.
 export interface InertiaPages {
 ${body}
 }
+
+export type InertiaPageName = ${pageNameUnion};
+` + propsHelper + `
+declare module '@dudousxd/nestjs-inertia' {
+  interface InertiaPages {
+${augBody}
+  }
+}
 `;
   await writeFile4(join5(outDir, "pages.d.ts"), content, "utf8");
 }
 __name(emitPages, "emitPages");
+function buildAugmentationType(page, outDir, propsExport) {
+  if (!page.propsSource) {
+    return "Record<string, unknown>";
+  }
+  let importPath = relative4(outDir, page.absolutePath).replace(/\.(tsx?|vue|svelte)$/, "");
+  if (!importPath.startsWith(".")) {
+    importPath = `./${importPath}`;
+  }
+  return `import('${importPath}').${propsExport}`;
+}
+__name(buildAugmentationType, "buildAugmentationType");
 function needsQuotes(name) {
   return !/^[A-Za-z_$][A-Za-z0-9_$]*$/.test(name);
 }
@@ -657,7 +751,9 @@ async function generate(config, routes = []) {
     propsExport: config.pages.propsExport,
     componentNameStrategy: config.pages.componentNameStrategy
   });
-  await emitPages(pages, config.codegen.outDir);
+  await emitPages(pages, config.codegen.outDir, {
+    propsExport: config.pages.propsExport
+  });
   await emitCache(pages, config.codegen.outDir);
   const hasRoutes = routes.length > 0;
   const hasContracts = routes.some((r) => r.contract);
@@ -677,9 +773,38 @@ import { join as join9 } from "path";
 import chokidar from "chokidar";
 
 // src/discovery/contracts-fast.ts
+import { readFileSync } from "fs";
 import { dirname, join as join7, resolve as resolve2 } from "path";
 import fg2 from "fast-glob";
 import { Node, Project, SyntaxKind } from "ts-morph";
+var _ctx = {
+  projectRoot: "",
+  tsconfigPaths: null
+};
+function _projectRoot() {
+  return _ctx.projectRoot;
+}
+__name(_projectRoot, "_projectRoot");
+function _tsconfigPaths() {
+  return _ctx.tsconfigPaths;
+}
+__name(_tsconfigPaths, "_tsconfigPaths");
+var _debug = process.env.NESTJS_INERTIA_DEBUG === "1";
+function dbg(...args) {
+  if (_debug) console.log("[codegen:debug]", ...args);
+}
+__name(dbg, "dbg");
+function loadTsconfigPaths(tsconfigPath) {
+  try {
+    const raw = readFileSync(tsconfigPath, "utf8");
+    const stripped = raw.replace(/\/\/.*$/gm, "");
+    const parsed = JSON.parse(stripped);
+    return parsed.compilerOptions?.paths ?? null;
+  } catch {
+    return null;
+  }
+}
+__name(loadTsconfigPaths, "loadTsconfigPaths");
 async function discoverContractsFast(opts) {
   const { cwd, glob, tsconfig } = opts;
   const tsconfigPath = tsconfig ? resolve2(tsconfig) : join7(cwd, "tsconfig.json");
@@ -712,8 +837,17 @@ async function discoverContractsFast(opts) {
     project.addSourceFileAtPath(f);
   }
   const routes = [];
-  for (const sourceFile of project.getSourceFiles()) {
-    routes.push(...extractFromSourceFile(sourceFile, project));
+  const prevCtx = _ctx;
+  _ctx = {
+    projectRoot: cwd,
+    tsconfigPaths: loadTsconfigPaths(tsconfigPath)
+  };
+  try {
+    for (const sourceFile of project.getSourceFiles()) {
+      routes.push(...extractFromSourceFile(sourceFile, project));
+    }
+  } finally {
+    _ctx = prevCtx;
   }
   return routes;
 }
@@ -913,17 +1047,42 @@ function findTypeInFile(name, file) {
   return null;
 }
 __name(findTypeInFile, "findTypeInFile");
+function resolveModuleSpecifier(moduleSpecifier, sourceFile, project) {
+  if (moduleSpecifier.startsWith(".")) {
+    const dir = dirname(sourceFile.getFilePath());
+    return [
+      resolve2(dir, `${moduleSpecifier}.ts`),
+      resolve2(dir, moduleSpecifier, "index.ts")
+    ];
+  }
+  const baseUrl = _projectRoot();
+  const tsconfigPaths = _tsconfigPaths();
+  dbg("resolveModuleSpecifier", moduleSpecifier, "paths:", JSON.stringify(tsconfigPaths), "baseUrl:", baseUrl);
+  if (tsconfigPaths) {
+    for (const [pattern, mappings] of Object.entries(tsconfigPaths)) {
+      const prefix = pattern.replace("*", "");
+      if (moduleSpecifier.startsWith(prefix)) {
+        const rest = moduleSpecifier.slice(prefix.length);
+        const candidates = [];
+        for (const mapping of mappings) {
+          const resolved = resolve2(baseUrl, mapping.replace("*", rest));
+          candidates.push(`${resolved}.ts`, resolve2(resolved, "index.ts"));
+        }
+        dbg("  resolved candidates:", candidates);
+        return candidates;
+      }
+    }
+  }
+  return [];
+}
+__name(resolveModuleSpecifier, "resolveModuleSpecifier");
 function resolveImportedType(name, sourceFile, project) {
   for (const importDecl of sourceFile.getImportDeclarations()) {
     const namedImport = importDecl.getNamedImports().find((n) => n.getName() === name);
     if (!namedImport) continue;
     const moduleSpecifier = importDecl.getModuleSpecifierValue();
-    if (!moduleSpecifier.startsWith(".")) return null;
-    const dir = dirname(sourceFile.getFilePath());
-    const candidates = [
-      resolve2(dir, `${moduleSpecifier}.ts`),
-      resolve2(dir, moduleSpecifier, "index.ts")
-    ];
+    const candidates = resolveModuleSpecifier(moduleSpecifier, sourceFile, project);
+    if (candidates.length === 0) continue;
     for (const candidate of candidates) {
       let importedFile = project.getSourceFile(candidate);
       if (!importedFile) {
@@ -967,6 +1126,18 @@ function resolveTypeNodeToString(typeNode, sourceFile, project, depth) {
       }
       return "Array<unknown>";
     }
+    if ([
+      "Record",
+      "Omit",
+      "Pick",
+      "Partial",
+      "Required",
+      "Readonly",
+      "Map",
+      "Set"
+    ].includes(name)) {
+      return typeNode.getText();
+    }
     if (name === "Promise") {
       const typeArgs = typeNode.getTypeArguments();
       const firstTypeArg = typeArgs[0];
@@ -979,7 +1150,8 @@ function resolveTypeNodeToString(typeNode, sourceFile, project, depth) {
     if (resolved) {
       return expandTypeDecl(resolved, project, depth - 1);
     }
-    return name;
+    dbg("unresolvable type:", name, "in", sourceFile.getFilePath());
+    return "unknown";
   }
   const kind = typeNode.getKind();
   if (kind === SyntaxKind.StringKeyword) return "string";
@@ -1141,7 +1313,7 @@ function tryResolveTypeRef(typeNode, sourceFile, project) {
       return null;
     }
     const localDecl = sourceFile.getInterface(name) || sourceFile.getClass(name) || sourceFile.getTypeAlias(name);
-    if (localDecl && localDecl.isExported()) {
+    if (localDecl?.isExported()) {
       return {
         name,
         filePath: sourceFile.getFilePath()
@@ -1203,7 +1375,7 @@ function extractDtoContract(method, sourceFile, project) {
             if (val && Node.isIdentifier(val)) {
               const name = val.getText();
               const localDecl = sourceFile.getInterface(name) || sourceFile.getClass(name) || sourceFile.getTypeAlias(name);
-              if (localDecl && localDecl.isExported()) {
+              if (localDecl?.isExported()) {
                 responseRef = {
                   name,
                   filePath: sourceFile.getFilePath()
@@ -1329,6 +1501,11 @@ function extractFromSourceFile(sourceFile, project) {
           path: combined,
           name: routeName,
           params,
+          controllerRef: {
+            className,
+            methodName,
+            filePath: sourceFile.getFilePath()
+          },
           contract: {
             contractSource: {
               query: contractDef.query,
@@ -1363,19 +1540,21 @@ function extractFromSourceFile(sourceFile, project) {
           path: combined,
           name: routeName,
           params,
-          // Attach contract if DTO extraction produced useful type info
-          ...dtoContract ? {
-            contract: {
-              contractSource: {
-                query: dtoContract.query,
-                body: dtoContract.body,
-                response: dtoContract.response,
-                queryRef: dtoContract.queryRef,
-                bodyRef: dtoContract.bodyRef,
-                responseRef: dtoContract.responseRef
-              }
+          controllerRef: {
+            className,
+            methodName,
+            filePath: sourceFile.getFilePath()
+          },
+          contract: {
+            contractSource: {
+              query: dtoContract?.query ?? null,
+              body: dtoContract?.body ?? null,
+              response: dtoContract?.response ?? "unknown",
+              queryRef: dtoContract?.queryRef,
+              bodyRef: dtoContract?.bodyRef,
+              responseRef: dtoContract?.responseRef
             }
-          } : {}
+          }
         });
       }
     }
@@ -1385,7 +1564,8 @@ function extractFromSourceFile(sourceFile, project) {
 __name(extractFromSourceFile, "extractFromSourceFile");
 
 // src/watch/lock-file.ts
-import { mkdir as mkdir6, readFile as readFile2, unlink, writeFile as writeFile6 } from "fs/promises";
+import { open } from "fs/promises";
+import { mkdir as mkdir6, readFile as readFile2, unlink } from "fs/promises";
 import { join as join8 } from "path";
 var LOCK_FILE = ".watcher.lock";
 function isProcessAlive(pid) {
@@ -1402,20 +1582,29 @@ async function acquireLock(outDir) {
     recursive: true
   });
   const lockPath = join8(outDir, LOCK_FILE);
-  try {
-    const raw = await readFile2(lockPath, "utf8");
-    const existing = JSON.parse(raw);
-    if (isProcessAlive(existing.pid)) {
-      return null;
-    }
-  } catch {
-  }
   const lockData = {
     pid: process.pid,
     startedAt: (/* @__PURE__ */ new Date()).toISOString()
   };
-  await writeFile6(lockPath, `${JSON.stringify(lockData, null, 2)}
+  try {
+    const fd = await open(lockPath, "wx");
+    await fd.writeFile(`${JSON.stringify(lockData, null, 2)}
 `, "utf8");
+    await fd.close();
+  } catch (err) {
+    if (err.code === "EEXIST") {
+      try {
+        const raw = await readFile2(lockPath, "utf8");
+        const existing = JSON.parse(raw);
+        if (isProcessAlive(existing.pid)) return null;
+        await unlink(lockPath);
+        return acquireLock(outDir);
+      } catch {
+        return null;
+      }
+    }
+    return null;
+  }
   return {
     release: /* @__PURE__ */ __name(async () => {
       try {
@@ -1479,7 +1668,8 @@ async function watch(config, onChange) {
       pagesDebounceTimer = void 0;
       try {
         await generate(config);
-      } catch {
+      } catch (err) {
+        console.error("[nestjs-inertia-codegen] Pages generation failed:", err instanceof Error ? err.message : err);
       }
       onChange?.();
     }, PAGES_DEBOUNCE_MS);
@@ -1517,7 +1707,8 @@ async function watch(config, onChange) {
         if (hasContracts) {
           await emitApi(routes, config.codegen.outDir);
         }
-      } catch {
+      } catch (err) {
+        console.error("[nestjs-inertia-codegen] Contracts generation failed:", err instanceof Error ? err.message : err);
       }
       onChange?.();
     }, config.contracts.debounceMs);
@@ -1545,7 +1736,7 @@ async function watch(config, onChange) {
 __name(watch, "watch");
 
 // src/index.ts
-var VERSION = "1.2.0";
+var VERSION = "1.4.0";
 
 // src/cli/codegen.ts
 async function runCodegen(opts = {}) {
@@ -1575,11 +1766,163 @@ async function runCodegen(opts = {}) {
 }
 __name(runCodegen, "runCodegen");
 
-// src/cli/init.ts
-import { execSync } from "child_process";
-import { readFileSync, writeFileSync } from "fs";
-import { access as access2, mkdir as mkdir7, readFile as readFile4, writeFile as writeFile7 } from "fs/promises";
+// src/cli/doctor.ts
+import { existsSync, readFileSync as readFileSync2 } from "fs";
 import { join as join10 } from "path";
+function checkFileExists(cwd, file) {
+  return existsSync(join10(cwd, file));
+}
+__name(checkFileExists, "checkFileExists");
+function readJson(path) {
+  try {
+    const raw = readFileSync2(path, "utf8").replace(/\/\/.*$/gm, "");
+    return JSON.parse(raw);
+  } catch {
+    return null;
+  }
+}
+__name(readJson, "readJson");
+function getPackageVersion(cwd, pkg) {
+  try {
+    const pkgJson = readJson(join10(cwd, "node_modules", pkg, "package.json"));
+    return pkgJson?.version ?? null;
+  } catch {
+    return null;
+  }
+}
+__name(getPackageVersion, "getPackageVersion");
+async function runDoctor(opts) {
+  const { cwd } = opts;
+  const checks = [];
+  checks.push({
+    name: "nestjs-inertia.config.ts exists",
+    pass: checkFileExists(cwd, "nestjs-inertia.config.ts"),
+    fix: "Run: pnpm exec nestjs-inertia init"
+  });
+  const hasApi = checkFileExists(cwd, ".nestjs-inertia/api.ts");
+  const hasRoutes = checkFileExists(cwd, ".nestjs-inertia/routes.ts");
+  const hasPages = checkFileExists(cwd, ".nestjs-inertia/pages.d.ts");
+  checks.push({
+    name: ".nestjs-inertia/ codegen output exists",
+    pass: hasApi && hasRoutes && hasPages,
+    fix: "Run: pnpm exec nestjs-inertia codegen"
+  });
+  const tsconfig = readJson(join10(cwd, "tsconfig.json"));
+  const paths = tsconfig?.compilerOptions?.paths;
+  checks.push({
+    name: "tsconfig.json has @/* path alias",
+    pass: !!paths?.["@/*"],
+    fix: 'Add to tsconfig.json compilerOptions.paths: { "@/*": ["./src/*"] }'
+  });
+  const inertiaTsconfig = readJson(join10(cwd, "tsconfig.inertia.json"));
+  if (inertiaTsconfig) {
+    const inertiaPaths = inertiaTsconfig.compilerOptions?.paths;
+    checks.push({
+      name: "tsconfig.inertia.json has ~/* and ~codegen/* aliases",
+      pass: !!inertiaPaths?.["~/*"] && !!inertiaPaths?.["~codegen/*"],
+      fix: 'Add paths: { "~/*": ["inertia/*"], "~codegen/*": [".nestjs-inertia/*"] }'
+    });
+  }
+  if (checkFileExists(cwd, "vite.config.ts")) {
+    const viteContent = readFileSync2(join10(cwd, "vite.config.ts"), "utf8");
+    checks.push({
+      name: "vite.config.ts has resolve.alias",
+      pass: viteContent.includes("resolve") && viteContent.includes("alias"),
+      fix: "Add resolve.alias with @\u2192src, ~\u2192inertia, ~codegen\u2192.nestjs-inertia"
+    });
+    checks.push({
+      name: "vite.config.ts references nestjs-inertia",
+      pass: viteContent.includes("nestInertia") || viteContent.includes("nestjs-inertia") || viteContent.includes("setupInertiaVite"),
+      fix: "Add: import nestInertia from '@dudousxd/nestjs-inertia-vite/plugin'"
+    });
+  }
+  const libPackages = [
+    "@dudousxd/nestjs-inertia",
+    "@dudousxd/nestjs-inertia-codegen",
+    "@dudousxd/nestjs-inertia-client",
+    "@dudousxd/nestjs-inertia-vite",
+    "@dudousxd/nestjs-inertia-testing"
+  ];
+  const versions = libPackages.map((pkg) => ({
+    pkg,
+    version: getPackageVersion(cwd, pkg)
+  }));
+  const installed = versions.filter((v) => v.version !== null);
+  const uniqueVersions = new Set(installed.map((v) => v.version));
+  const requiredPkgs = [
+    "@dudousxd/nestjs-inertia",
+    "@dudousxd/nestjs-inertia-codegen",
+    "@dudousxd/nestjs-inertia-client"
+  ];
+  const missingRequired = requiredPkgs.filter((pkg) => !getPackageVersion(cwd, pkg));
+  checks.push({
+    name: "Core packages installed (core + codegen + client)",
+    pass: missingRequired.length === 0,
+    fix: missingRequired.length > 0 ? `Missing: ${missingRequired.join(", ")}` : void 0
+  });
+  if (installed.length > 1) {
+    checks.push({
+      name: "All packages on same version",
+      pass: uniqueVersions.size === 1,
+      fix: `Versions: ${installed.map((v) => `${v.pkg.replace("@dudousxd/", "")}@${v.version}`).join(", ")}`
+    });
+  }
+  const inertiaReact = getPackageVersion(cwd, "@inertiajs/react");
+  const inertiaVue = getPackageVersion(cwd, "@inertiajs/vue3");
+  const inertiaSvelte = getPackageVersion(cwd, "@inertiajs/svelte");
+  const inertiaVersion = inertiaReact ?? inertiaVue ?? inertiaSvelte;
+  const inertiaFramework = inertiaReact ? "react" : inertiaVue ? "vue" : inertiaSvelte ? "svelte" : null;
+  if (inertiaVersion) {
+    const majorVersion = Number.parseInt(inertiaVersion.split(".")[0] ?? "0", 10);
+    checks.push({
+      name: `@inertiajs/${inertiaFramework} is v3+`,
+      pass: majorVersion >= 3,
+      fix: `Current: v${inertiaVersion}. Run: pnpm add @inertiajs/${inertiaFramework}@^3.0.0`
+    });
+  }
+  if (checkFileExists(cwd, ".gitignore")) {
+    const gitignore = readFileSync2(join10(cwd, ".gitignore"), "utf8");
+    checks.push({
+      name: ".gitignore includes .nestjs-inertia/",
+      pass: gitignore.includes(".nestjs-inertia"),
+      fix: "Add .nestjs-inertia/ to .gitignore"
+    });
+  }
+  const pkgJson = readJson(join10(cwd, "package.json"));
+  const scripts = pkgJson?.scripts ?? {};
+  checks.push({
+    name: "package.json has build:client script",
+    pass: !!scripts["build:client"],
+    fix: 'Add: "build:client": "vite build"'
+  });
+  console.log("");
+  console.log("\x1B[1mnestjs-inertia doctor\x1B[0m");
+  console.log("");
+  let hasFailures = false;
+  for (const check of checks) {
+    const icon = check.pass ? "\x1B[32m\u2713\x1B[0m" : "\x1B[31m\u2717\x1B[0m";
+    console.log(`  ${icon} ${check.name}`);
+    if (!check.pass && check.fix) {
+      console.log(`    \x1B[2m${check.fix}\x1B[0m`);
+      hasFailures = true;
+    }
+  }
+  console.log("");
+  if (hasFailures) {
+    console.log(`\x1B[33m${checks.filter((c) => !c.pass).length} issue(s) found\x1B[0m`);
+  } else {
+    console.log("\x1B[32mAll checks passed!\x1B[0m");
+  }
+  console.log("");
+  return hasFailures ? 1 : 0;
+}
+__name(runDoctor, "runDoctor");
+
+// src/cli/init.ts
+import { execFileSync } from "child_process";
+import { readFileSync as readFileSync3, writeFileSync } from "fs";
+import { access as access2, mkdir as mkdir7, readFile as readFile4, writeFile as writeFile6 } from "fs/promises";
+import { join as join11 } from "path";
 import { createInterface } from "readline";
 var GITIGNORE_ENTRY = ".nestjs-inertia/";
 var green = /* @__PURE__ */ __name((s) => `\x1B[32m${s}\x1B[0m`, "green");
@@ -1610,7 +1953,7 @@ ${bold(title)}`);
 __name(logSection, "logSection");
 async function readPackageJson(cwd) {
   try {
-    const raw = await readFile4(join10(cwd, "package.json"), "utf8");
+    const raw = await readFile4(join11(cwd, "package.json"), "utf8");
     return JSON.parse(raw);
   } catch {
     return {};
@@ -1648,7 +1991,7 @@ __name(detectTemplateEngine, "detectTemplateEngine");
 async function detectPackageManager(cwd) {
   async function exists(file) {
     try {
-      await access2(join10(cwd, file));
+      await access2(join11(cwd, file));
       return true;
     } catch {
       return false;
@@ -1697,12 +2040,12 @@ async function writeIfNotExists(filePath, content, label) {
       recursive: true
     });
   }
-  await writeFile7(filePath, content, "utf8");
+  await writeFile6(filePath, content, "utf8");
   logCreated(label);
 }
 __name(writeIfNotExists, "writeIfNotExists");
 async function handleViteConfig(cwd, framework) {
-  const filePath = join10(cwd, "vite.config.ts");
+  const filePath = join11(cwd, "vite.config.ts");
   if (await fileExists2(filePath)) {
     const existing = await readFile4(filePath, "utf8");
     const hasPlugin = existing.includes("nestInertia") || existing.includes("nestjs-inertia-vite/plugin");
@@ -1722,7 +2065,7 @@ async function handleViteConfig(cwd, framework) {
       recursive: true
     });
   }
-  await writeFile7(filePath, viteConfigTemplate(framework), "utf8");
+  await writeFile6(filePath, viteConfigTemplate(framework), "utf8");
   logCreated("vite.config.ts");
 }
 __name(handleViteConfig, "handleViteConfig");
@@ -1739,27 +2082,33 @@ async function patchGitignore(gitignorePath) {
 ` : `${existing}
 ${GITIGNORE_ENTRY}
 `;
-  await writeFile7(gitignorePath, newContent, "utf8");
+  await writeFile6(gitignorePath, newContent, "utf8");
   logPatched(".gitignore", "added .nestjs-inertia/");
 }
 __name(patchGitignore, "patchGitignore");
 function installDeps(pkgManager, deps, dev) {
   if (deps.length === 0) return;
-  const flag = dev ? pkgManager === "npm" ? "--save-dev" : "-D" : "";
-  const cmd = pkgManager === "npm" ? `npm install ${flag} ${deps.join(" ")}` : pkgManager === "yarn" ? `yarn add ${flag} ${deps.join(" ")}` : `pnpm add ${flag} ${deps.join(" ")}`;
+  const args = [];
+  if (pkgManager === "npm") {
+    args.push("install");
+    if (dev) args.push("--save-dev");
+  } else {
+    args.push("add");
+    if (dev) args.push("-D");
+  }
+  args.push(...deps);
   logPatched(deps.join(", "), "installed");
   try {
-    execSync(cmd, {
+    execFileSync(pkgManager, args, {
       stdio: "inherit"
     });
   } catch {
-    logWarning(`Failed to install deps. Run manually:
-    ${cmd}`);
+    logWarning(`Failed to install: ${deps.join(", ")}`);
   }
 }
 __name(installDeps, "installDeps");
 async function patchPackageJsonScripts(cwd, scripts) {
-  const pkgPath = join10(cwd, "package.json");
+  const pkgPath = join11(cwd, "package.json");
   let pkg = {};
   try {
     pkg = JSON.parse(await readFile4(pkgPath, "utf8"));
@@ -1781,7 +2130,7 @@ async function patchPackageJsonScripts(cwd, scripts) {
     return;
   }
   pkg.scripts = existing;
-  await writeFile7(pkgPath, `${JSON.stringify(pkg, null, 2)}
+  await writeFile6(pkgPath, `${JSON.stringify(pkg, null, 2)}
 `, "utf8");
 }
 __name(patchPackageJsonScripts, "patchPackageJsonScripts");
@@ -1801,7 +2150,7 @@ __name(findAfterLastImport, "findAfterLastImport");
 function patchAppModule(filePath, rootView) {
   let content;
   try {
-    content = readFileSync(filePath, "utf8");
+    content = readFileSync3(filePath, "utf8");
   } catch {
     return "skipped";
   }
@@ -1846,7 +2195,7 @@ __name(patchAppModule, "patchAppModule");
 function patchMainTs(filePath) {
   let content;
   try {
-    content = readFileSync(filePath, "utf8");
+    content = readFileSync3(filePath, "utf8");
   } catch {
     return "skipped";
   }
@@ -1919,11 +2268,19 @@ function htmlShellTemplate(framework, _engine) {
 __name(htmlShellTemplate, "htmlShellTemplate");
 function viteConfigTemplate(framework) {
   const pluginOption = `{ ${framework}: true }`;
-  return `import { defineConfig } from 'vite';
+  return `import { resolve } from 'node:path';
+import { defineConfig } from 'vite';
 import nestInertia from '@dudousxd/nestjs-inertia-vite/plugin';
 
 export default defineConfig({
   plugins: [nestInertia(${pluginOption})],
+  resolve: {
+    alias: {
+      '@': resolve(__dirname, 'src'),
+      '~': resolve(__dirname, 'inertia'),
+      '~codegen': resolve(__dirname, '.nestjs-inertia'),
+    },
+  },
 });
 `;
 }
@@ -2043,16 +2400,16 @@ ${bold("nestjs-inertia init")}`);
   const entryExt = framework === "react" ? "tsx" : "ts";
   const pageExt = framework === "react" ? "tsx" : framework === "vue" ? "vue" : "svelte";
   logSection("Scaffold files");
-  await writeIfNotExists(join10(cwd, "nestjs-inertia.config.ts"), configTemplate(framework), "nestjs-inertia.config.ts");
-  await writeIfNotExists(join10(cwd, "nestjs-inertia.d.ts"), DTS_TEMPLATE, "nestjs-inertia.d.ts");
-  await writeIfNotExists(join10(cwd, "inertia", shellFileName), htmlShellTemplate(framework, engine), `inertia/${shellFileName}`);
+  await writeIfNotExists(join11(cwd, "nestjs-inertia.config.ts"), configTemplate(framework), "nestjs-inertia.config.ts");
+  await writeIfNotExists(join11(cwd, "nestjs-inertia.d.ts"), DTS_TEMPLATE, "nestjs-inertia.d.ts");
+  await writeIfNotExists(join11(cwd, "inertia", shellFileName), htmlShellTemplate(framework, engine), `inertia/${shellFileName}`);
   await handleViteConfig(cwd, framework);
-  await writeIfNotExists(join10(cwd, "inertia", `app.${entryExt}`), entryPointTemplate(framework), `inertia/app.${entryExt}`);
-  await writeIfNotExists(join10(cwd, "inertia", "pages", `Home.${pageExt}`), samplePageTemplate(framework), `inertia/pages/Home.${pageExt}`);
-  await writeIfNotExists(join10(cwd, "src", "home.controller.ts"), SAMPLE_CONTROLLER, "src/home.controller.ts");
+  await writeIfNotExists(join11(cwd, "inertia", `app.${entryExt}`), entryPointTemplate(framework), `inertia/app.${entryExt}`);
+  await writeIfNotExists(join11(cwd, "inertia", "pages", `Home.${pageExt}`), samplePageTemplate(framework), `inertia/pages/Home.${pageExt}`);
+  await writeIfNotExists(join11(cwd, "src", "home.controller.ts"), SAMPLE_CONTROLLER, "src/home.controller.ts");
   logSection("Patch existing files");
   const rootView = engine === "html" ? "inertia/index.html" : `inertia/index.${engine === "handlebars" ? "hbs" : engine}`;
-  const appModulePath = join10(cwd, "src", "app.module.ts");
+  const appModulePath = join11(cwd, "src", "app.module.ts");
   const appModuleResult = patchAppModule(appModulePath, rootView);
   if (appModuleResult === "patched") {
     logPatched("src/app.module.ts", "added InertiaModule.forRoot");
@@ -2062,7 +2419,7 @@ ${bold("nestjs-inertia init")}`);
   } else {
     logWarning("src/app.module.ts not found \u2014 add InertiaModule.forRoot() manually");
   }
-  const mainTsPath = join10(cwd, "src", "main.ts");
+  const mainTsPath = join11(cwd, "src", "main.ts");
   const mainTsResult = patchMainTs(mainTsPath);
   if (mainTsResult === "patched") {
     logPatched("src/main.ts", "added setupInertiaVite after NestFactory.create");
@@ -2071,7 +2428,7 @@ ${bold("nestjs-inertia init")}`);
   } else {
     logWarning("src/main.ts not found \u2014 add setupInertiaVite() manually");
   }
-  await patchGitignore(join10(cwd, ".gitignore"));
+  await patchGitignore(join11(cwd, ".gitignore"));
   await patchPackageJsonScripts(cwd, {
     "build:client": "vite build",
     "build:ssr": "VITE_SSR=1 vite build --ssr"
@@ -2148,6 +2505,12 @@ async function run(argv) {
       cwd: process.cwd()
     });
   });
+  cli.command("doctor", "Diagnose your nestjs-inertia setup").action(async () => {
+    const code = await runDoctor({
+      cwd: process.cwd()
+    });
+    process.exitCode = code;
+  });
   cli.help();
   cli.version(VERSION);
   try {