From e3b8664259d39ce0a3fed7a523d58f484c9f19cd Mon Sep 17 00:00:00 2001
From: meixg <meixg@foxmail.com>
Date: Wed, 20 Apr 2022 20:23:09 +0800
Subject: [PATCH] feat: support custom output data

---
 src/ast/renderer-ast-dfn.ts     | 12 ++++++++++--
 src/ast/renderer-ast-walker.ts  |  1 +
 src/compilers/anode-compiler.ts | 20 ++++++++++++++++++--
 src/target-js/js-emitter.ts     |  5 +++++
 4 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/src/ast/renderer-ast-dfn.ts b/src/ast/renderer-ast-dfn.ts
index e660e708..d9fb2bb6 100644
--- a/src/ast/renderer-ast-dfn.ts
+++ b/src/ast/renderer-ast-dfn.ts
@@ -66,14 +66,15 @@ export enum SyntaxKind {
     TryStatement = 36,
     CatchClause = 37,
     ComponentClassReference = 38,
-    CreateComponentPrototype = 39
+    CreateComponentPrototype = 39,
+    Typeof = 40
 }
 
 export type Expression = Identifier | FunctionDefinition | Literal | BinaryExpression | UnaryExpression |
     CreateComponentInstance | NewExpression | MapLiteral | ComponentRendererReference | FunctionCall | Null |
     Undefined | MapAssign | ArrayIncludes | ConditionalExpression | FilterCall | HelperCall | EncodeURIComponent |
     ArrayLiteral | RegexpReplace | JSONStringify | ComputedCall | GetRootCtxCall | ComponentReferenceLiteral |
-    SlotRendererDefinition | SlotRenderCall | ComponentClassReference | CreateComponentPrototype
+    SlotRendererDefinition | SlotRenderCall | ComponentClassReference | CreateComponentPrototype | Typeof
 
 export type Statement = ReturnStatement | ImportHelper | VariableDefinition | AssignmentStatement | If | ElseIf | Else |
     Foreach | ExpressionStatement | TryStatement
@@ -425,3 +426,10 @@ export class ReturnStatement implements SyntaxNode {
         public value: Expression
     ) {}
 }
+
+export class Typeof implements SyntaxNode {
+    public readonly kind = SyntaxKind.Typeof
+    constructor (
+        public value: Expression
+    ) {}
+}
diff --git a/src/ast/renderer-ast-walker.ts b/src/ast/renderer-ast-walker.ts
index bdeaffda..cbe7fb47 100644
--- a/src/ast/renderer-ast-walker.ts
+++ b/src/ast/renderer-ast-walker.ts
@@ -17,6 +17,7 @@ export function * walk (node: Expression | Statement): Iterable<Expression | Sta
     case SyntaxKind.EncodeURIComponent:
     case SyntaxKind.ReturnStatement:
     case SyntaxKind.ExpressionStatement:
+    case SyntaxKind.Typeof:
         yield * walk(node.value)
         break
     case SyntaxKind.Literal:
diff --git a/src/compilers/anode-compiler.ts b/src/compilers/anode-compiler.ts
index 5f1bd04b..36ba4707 100644
--- a/src/compilers/anode-compiler.ts
+++ b/src/compilers/anode-compiler.ts
@@ -15,7 +15,11 @@ import { IDGenerator } from '../utils/id-generator'
 import {
     JSONStringify, RegexpReplace, Statement, SlotRendererDefinition, ElseIf, Else, MapAssign, Foreach, If, MapLiteral,
     ComponentRendererReference, FunctionCall, SlotRenderCall, Expression, GetRootCtxCall, ComponentReferenceLiteral,
-    ComponentClassReference
+    ComponentClassReference,
+    VariableDefinition,
+    ConditionalExpression,
+    Typeof,
+    AssignmentStatement
 } from '../ast/renderer-ast-dfn'
 import {
     CTX_DATA, createHTMLExpressionAppend, createHTMLLiteralAppend, L, I, ASSIGN, STATEMENT, UNARY, DEF, BINARY, RETURN
@@ -211,9 +215,21 @@ export class ANodeCompiler {
 
     private createDataComment () {
         const dataExpr = BINARY(new GetRootCtxCall([I('ctx')]), '.', I('data'))
+        const outputDataExpr = BINARY(I('info'), '.', I('outputData'))
         return [
+            new VariableDefinition('data', dataExpr),
+            new If(outputDataExpr, [
+                new AssignmentStatement(
+                    I('data'),
+                    new ConditionalExpression(
+                        BINARY(new Typeof(outputDataExpr), '===', L('function')),
+                        new FunctionCall(outputDataExpr, [dataExpr]),
+                        outputDataExpr
+                    )
+                )
+            ]),
             createHTMLLiteralAppend('<!--s-data:'),
-            createHTMLExpressionAppend(new RegexpReplace(new JSONStringify(dataExpr), '(?<=-)-', L('\\-'))),
+            createHTMLExpressionAppend(new RegexpReplace(new JSONStringify(I('data')), '(?<=-)-', L('\\-'))),
             createHTMLLiteralAppend('-->')
         ]
     }
diff --git a/src/target-js/js-emitter.ts b/src/target-js/js-emitter.ts
index 324c8b20..8d3a76a0 100644
--- a/src/target-js/js-emitter.ts
+++ b/src/target-js/js-emitter.ts
@@ -209,6 +209,11 @@ export class JSEmitter extends Emitter {
             this.write(') ')
             this.writeBlockStatements(node.handler.body)
             break
+        case SyntaxKind.Typeof:
+            this.write('typeof (')
+            this.writeSyntaxNode(node.value)
+            this.write(')')
+            break
         default: assertNever(node)
         }
     }