@spfn/core 0.1.0-alpha.20 → 0.1.0-alpha.22

package/dist/codegen/index.d.ts

@@ -212,11 +212,13 @@ declare function createContractGenerator(config?: ContractGeneratorConfig): Gene
  */
 
 interface CodegenConfig {
-    generators?: {
-        contract?: ContractGeneratorConfig & {
-            enabled?: boolean;
-        };
-    };
+    generators?: Array<{
+        path: string;
+    } | ({
+        name: 'contract';
+    } & ContractGeneratorConfig & {
+        enabled?: boolean;
+    })>;
 }
 /**
  * Load codegen configuration from .spfnrc.json or package.json
@@ -225,6 +227,6 @@ declare function loadCodegenConfig(cwd: string): CodegenConfig;
 /**
  * Create generator instances from configuration
  */
-declare function createGeneratorsFromConfig(config: CodegenConfig): Generator[];
+declare function createGeneratorsFromConfig(config: CodegenConfig, cwd: string): Promise<Generator[]>;
 
 export { type ClientGenerationOptions, type CodegenConfig, CodegenOrchestrator, type ContractGeneratorConfig, type GenerationStats, type Generator, type GeneratorOptions, HttpMethod, type OrchestratorOptions, type ResourceRoutes, type RouteContractMapping, createContractGenerator, createGeneratorsFromConfig, generateClient, groupByResource, loadCodegenConfig, scanContracts, watchAndGenerate };

package/dist/codegen/index.js

@@ -5,6 +5,7 @@ import { existsSync, mkdirSync, createWriteStream, readFileSync } from 'fs';
 import { watch } from 'chokidar';
 import pino from 'pino';
 import mm from 'micromatch';
+import { createJiti } from 'jiti';
 
 // src/codegen/contract-scanner.ts
 async function scanContracts(routesDir) {
@@ -1201,19 +1202,51 @@ function loadCodegenConfig(cwd) {
   }
   configLogger.info("Using default config");
   return {
-    generators: {
-      contract: { enabled: true }
-    }
+    generators: [
+      { name: "contract", enabled: true }
+    ]
   };
 }
-function createGeneratorsFromConfig(config) {
+async function createGeneratorsFromConfig(config, cwd) {
   const generators = [];
-  if (config.generators?.contract?.enabled !== false) {
-    const contractConfig = {
-      routesDir: config.generators?.contract?.routesDir,
-      outputPath: config.generators?.contract?.outputPath};
-    generators.push(createContractGenerator(contractConfig));
-    configLogger.info("Contract generator enabled");
+  if (!config.generators || config.generators.length === 0) {
+    return generators;
+  }
+  for (const generatorConfig of config.generators) {
+    try {
+      if ("path" in generatorConfig) {
+        const generatorPath = generatorConfig.path.startsWith(".") ? join(cwd, generatorConfig.path) : generatorConfig.path;
+        configLogger.info(`Loading custom generator: ${generatorPath}`);
+        let module;
+        if (generatorPath.endsWith(".ts")) {
+          const jiti = createJiti(cwd, {
+            interopDefault: true
+          });
+          module = jiti(generatorPath);
+        } else {
+          module = await import(generatorPath);
+        }
+        const createGenerator = module.default || module.createGenerator || module;
+        if (typeof createGenerator === "function") {
+          const generator = createGenerator();
+          generators.push(generator);
+          configLogger.info(`Custom generator loaded: ${generator.name}`);
+        } else {
+          configLogger.warn(`Invalid generator at ${generatorPath}: expected function`);
+        }
+      } else if ("name" in generatorConfig && generatorConfig.name === "contract") {
+        if (generatorConfig.enabled !== false) {
+          const contractConfig = {
+            routesDir: generatorConfig.routesDir,
+            outputPath: generatorConfig.outputPath};
+          generators.push(createContractGenerator(contractConfig));
+          configLogger.info("Contract generator enabled");
+        }
+      }
+    } catch (error) {
+      const err = error instanceof Error ? error : new Error(String(error));
+      configLogger.error("Failed to load generator", err);
+    }
   }
   return generators;
 }