feat: get{Script,Flow}ArgsSchema helpers

invakid404 · Nov 28, 2024 · dead3c9 · dead3c9
1 parent dead4c3
commit dead3c9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/generator/flows.ts b/src/generator/flows.ts
@@ -24,6 +24,12 @@ const preamble = dedent`
 
     return wmill.runFlowAsync(flowPath, schema.parse(args));
   };
+
+  export const getFlowArgsSchema = <Path extends keyof typeof ${mapName}>(
+    flowPath: Path
+  ) => {
+    return ${mapName}[flowPath];
+  }
 `;
 
 export const generateFlows = async () => {

diff --git a/src/generator/scripts.ts b/src/generator/scripts.ts
@@ -24,6 +24,12 @@ const preamble = dedent`
 
     return wmill.runScriptAsync(scriptPath, null, schema.parse(args));
   };
+
+  export const getScriptArgsSchema = <Path extends keyof typeof ${mapName}>(
+    scriptPath: Path
+  ) => {
+    return ${mapName}[scriptPath];
+  }
 `;
 
 export const generateScripts = async () => {