feat(compiler-core/internal): add onContextCreated option to generate (#1672)

2020-07-22 00:46:20 +05:30
parent 4951d43526
commit 615dccd00e
1 changed files with 4 additions and 1 deletions
--- 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,