feat: generateTemplate fun support context

MrLiangLong · LeoYuan · commit e875e33dc543 · 2023-07-25T10:09:07.000+08:00
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 {

Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,7 @@ export interface ISchemaParser {`
`127`	`127`
`128`	`128`	`export interface IProjectTemplate {`
`129`	`129`	`slots: Record<string, IProjectSlot>;`
`130`		`- generateTemplate: () => ResultDir \| Promise<ResultDir>;`
	`130`	`+ generateTemplate: (data: IParseResult) => ResultDir \| Promise<ResultDir>;`
`131`	`131`	`}`
`132`	`132`
`133`	`133`	`export interface IProjectSlot {`