hide it under a flag

package.json

@@ -11,7 +11,7 @@
     "test": "jest",
     "special": "ts-node src/index.ts '../../personal/Tangent-Mobile/api/entities/*.d.ts' -i -t -o ./schema/Tangent",
     "special2": "ts-node src/index.ts 'tests/comparisons/**/*.d.ts' -i -t -o ./schema/Tests/",
-    "zod": "ts-node src/index.ts 'tests/comparisons/single/zodStuff.ts' -i -t -o ./schema/Tests/ZodStuff.kt",
+    "zod": "ts-node src/index.ts 'tests/comparisons/single/zodStuff.ts' -i -t -z -o ./schema/Tests/ZodStuff.kt",
     "lint": "eslint . --ext .ts"
   },
   "author": "aaron@sarazan.net",

src/index.ts

@@ -49,6 +49,13 @@ const args = yargs
     default: false,
     describe: "Just throw in an import kotlinx.serialization.*",
   })
+  .option("experimentalZodSupport", {
+    alias: "z",
+    type: "boolean",
+    default: false,
+    describe:
+      "WIP support for Zod transpilations. Need to manually infer and expand",
+  })
   .showHelpOnFail(true)
   .help()
   .strict()
@@ -64,6 +71,7 @@ export type TranspileSingleArgs = {
   snakeToCamelCase: boolean;
   annotationNewLines: boolean;
   importStar: boolean;
+  experimentalZodSupport: boolean;
 };
 
 async function transpile(args: TranspileSingleArgs) {
@@ -74,6 +82,7 @@ async function transpile(args: TranspileSingleArgs) {
     snakeToCamelCase,
     annotationNewLines,
     importStar,
+    experimentalZodSupport,
   } = args;
   const config: MartokConfig = {
     files: resolve.files,
@@ -84,6 +93,7 @@ async function transpile(args: TranspileSingleArgs) {
       snakeToCamelCase,
       annotationNewLines,
       importStar,
+      experimentalZodSupport,
     },
   };
   const martok = new Martok(config);

src/martok/Martok.ts

@@ -38,6 +38,8 @@ export class Martok {
 
   public readonly imports;
 
+  public readonly zodProcessor: ZodProcessor;
+
   public get checker(): TypeChecker {
     return this.program.getTypeChecker();
   }
@@ -73,6 +75,7 @@ export class Martok {
     this.program = this.compiler.compileFiles(fsMap);
     this.program = new TypeExpander(this).expand();
     this.imports = new ImportGenerator(this);
+    this.zodProcessor = new ZodProcessor(this);
 
     this.declarations = new DeclarationGenerator(this);
     this.storage = new AsyncLocalStorage<MartokState>();
@@ -196,7 +199,7 @@ export class Martok {
     let relativePath = path.resolve(path.dirname(file.fileName));
     if (relativePath.startsWith(this.config.sourceRoot)) {
       relativePath = relativePath.slice(this.config.sourceRoot.length);
-    } else if (!ZodProcessor.allowImportThrough(file)) {
+    } else if (!this.zodProcessor.allowImportThrough(file)) {
       throw new Error(
         `${file.fileName} is not within the given source root, it can't be included in this project.`
       );

src/martok/MartokOptions.ts

@@ -3,4 +3,5 @@ export type MartokOptions = {
   snakeToCamelCase?: boolean;
   annotationNewLines?: boolean;
   importStar?: boolean;
+  experimentalZodSupport?: boolean;
 };

src/martok/processing/ZodProcessor.ts

@@ -1,7 +1,10 @@
 import ts from "typescript";
+import { Martok } from "../Martok";
 
 export class ZodProcessor {
-  public static allowImportThrough(file: ts.SourceFile): boolean {
+  public constructor(private readonly martok: Martok) {}
+  public allowImportThrough(file: ts.SourceFile): boolean {
+    if (!this.martok.config.options?.experimentalZodSupport) return false;
     return file.fileName.includes("/martok/node_modules/zod/lib/");
   }
 }