feat(plugin-compiler): runtime 方法替换兼容 return 语句

packages/plugin-compiler/src/parsers/scriptParserPlugin.ts

@@ -494,14 +494,16 @@ export class ScriptParserPlugin implements Plugin {
 
     function visitor(node: ts.Node, ctx: ts.TransformationContext) {
       const factory = ctx.factory
+      // 判断  statement 的同时，要判断一下  node.expression 是否存在，如果业务上返回的不是一个 expression，调用 ts.isCallExpression 会报错
+      const isReturnStatement = ts.isReturnStatement(node) && node.expression
 
       // FIXME: 这里的条件无法匹配
       // 箭头函数（ArrowFunction）
       // 二元表达式（BinaryExpression）
       // 三元表达式（ConditionalExpression）
       // 需要提供逻辑进一步支持以上几种情况
       if (
-        ts.isExpressionStatement(node) &&
+        (ts.isExpressionStatement(node) || isReturnStatement) &&
         ts.isCallExpression(node.expression) &&
         ts.isIdentifier(node.expression.expression)
       ) {
@@ -529,14 +531,18 @@ export class ScriptParserPlugin implements Plugin {
             )
           }
 
-          return factory.updateExpressionStatement(
-            node,
-            factory.createCallExpression(
-              factory.createIdentifier(MOR_IDENTIFIERS[identfierName]),
-              node.expression.typeArguments,
-              args
-            )
+          const updateExpression = factory.createCallExpression(
+            factory.createIdentifier(MOR_IDENTIFIERS[identfierName]),
+            node.expression.typeArguments,
+            args
           )
+
+          return isReturnStatement
+            ? factory.updateReturnStatement(node, updateExpression)
+            : factory.updateExpressionStatement(
+                node as ts.ExpressionStatement,
+                updateExpression
+              )
         }
       }