feat: generateTemplate fun support context

alibaba · Jul 25, 2023 · e875e33 · e875e33
1 parent 02a3361
commit e875e33
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/modules/code-generator/src/generator/ProjectBuilder.ts b/modules/code-generator/src/generator/ProjectBuilder.ts
@@ -111,8 +111,6 @@ export class ProjectBuilder implements IProjectBuilder {
     // Init
     const { schemaParser } = this;
 
-    const projectRoot = await this.template.generateTemplate();
-
     let schema: IPublicTypeProjectSchema =
       typeof originalSchema === 'string' ? JSON.parse(originalSchema) : originalSchema;
 
@@ -131,6 +129,9 @@ export class ProjectBuilder implements IProjectBuilder {
     // Collect Deps
     // Parse JSExpression
     const parseResult: IParseResult = schemaParser.parse(schema);
+
+    const projectRoot = await this.template.generateTemplate(parseResult);
+
     let buildResult: IModuleInfo[] = [];
 
     const builders = this.createModuleBuilders({

diff --git a/modules/code-generator/src/types/core.ts b/modules/code-generator/src/types/core.ts
@@ -127,7 +127,7 @@ export interface ISchemaParser {
 
 export interface IProjectTemplate {
   slots: Record<string, IProjectSlot>;
-  generateTemplate: () => ResultDir | Promise<ResultDir>;
+  generateTemplate: (data: IParseResult) => ResultDir | Promise<ResultDir>;
 }
 
 export interface IProjectSlot {