zodvex 0.7.7-beta.2 → 0.7.7

package/dist/cli/index.js

@@ -2405,19 +2405,29 @@ ${dtsDeclarations.join("\n")}
 async function generate(convexDir2, options) {
   const resolved = resolveConvexDir(convexDir2);
   const zodvexDir = path4.join(resolved, "_zodvex");
-  writeStubApi(zodvexDir);
-  const result = await discoverModules(resolved);
-  const schemaContent = generateSchemaFile(result.models);
-  const apiContent = generateApiFile(
-    result.functions,
-    result.models,
-    result.codecs,
-    result.modelCodecs,
-    result.functionCodecs,
-    { mini: options?.mini }
-  );
-  const clientContent = generateClientFile({ mini: options?.mini });
-  const serverContent = generateServerFile();
+  const restoreStubbedApi = writeStubApi(zodvexDir);
+  let result;
+  let schemaContent;
+  let apiContent;
+  let clientContent;
+  let serverContent;
+  try {
+    result = await discoverModules(resolved);
+    schemaContent = generateSchemaFile(result.models);
+    apiContent = generateApiFile(
+      result.functions,
+      result.models,
+      result.codecs,
+      result.modelCodecs,
+      result.functionCodecs,
+      { mini: options?.mini }
+    );
+    clientContent = generateClientFile({ mini: options?.mini });
+    serverContent = generateServerFile();
+  } catch (err) {
+    restoreStubbedApi();
+    throw err;
+  }
   fs3.mkdirSync(zodvexDir, { recursive: true });
   writeIfChanged(path4.join(zodvexDir, "schema.js"), schemaContent.js);
   writeIfChanged(path4.join(zodvexDir, "schema.d.ts"), schemaContent.dts);
@@ -2471,6 +2481,16 @@ function regenerate(resolved, options) {
   });
 }
 function writeStubApi(zodvexDir) {
+  const stubTargets = ["api.js", "api.d.ts"].map((name) => {
+    const filePath = path4.join(zodvexDir, name);
+    let original;
+    try {
+      original = fs3.readFileSync(filePath, "utf-8");
+    } catch {
+      original = null;
+    }
+    return { filePath, original };
+  });
   fs3.mkdirSync(zodvexDir, { recursive: true });
   fs3.writeFileSync(
     path4.join(zodvexDir, "api.js"),
@@ -2480,6 +2500,18 @@ function writeStubApi(zodvexDir) {
     path4.join(zodvexDir, "api.d.ts"),
     "// AUTO-GENERATED by zodvex \u2014 do not edit\n// Stub created for codegen bootstrap\n\nexport declare const zodvexRegistry: Record<string, any>\n"
   );
+  return () => {
+    for (const { filePath, original } of stubTargets) {
+      try {
+        if (original !== null) {
+          fs3.writeFileSync(filePath, original);
+        } else {
+          fs3.unlinkSync(filePath);
+        }
+      } catch {
+      }
+    }
+  };
 }
 function writeIfChanged(filePath, content) {
   try {