feat(compiler-core/internal): add onContextCreated option to `gener…

…ate` (#1672)
vuejs · Jul 21, 2020 · 615dccd · 615dccd
1 parent 4951d43
commit 615dccd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/compiler-core/src/codegen.ts b/packages/compiler-core/src/codegen.ts
@@ -180,9 +180,12 @@ function createCodegenContext(
 
 export function generate(
   ast: RootNode,
-  options: CodegenOptions = {}
+  options: CodegenOptions & {
+    onContextCreated?: (context: CodegenContext) => void
+  } = {}
 ): CodegenResult {
   const context = createCodegenContext(ast, options)
+  if (options.onContextCreated) options.onContextCreated(context)
   const {
     mode,
     push,