add try-catch，能多hook一点是一点

hhxlrm · Jan 18, 2021 · b59240d · b59240d
1 parent bfce7cd
commit b59240d
Showing 1 changed file with 44 additions and 27 deletions.
diff --git a/src/components/global-assign-hook-component/core/inject-hook.js b/src/components/global-assign-hook-component/core/inject-hook.js
@@ -27,12 +27,16 @@ function injectHook(jsCode) {
                     varName = generator.default(variableDeclarator.id).code;
                 }
 
-                const hookFunctionArguments = [
-                    types.stringLiteral(varName),
-                    variableDeclarator.init,
-                    types.stringLiteral("var-init")
-                ];
-                variableDeclarator.init = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments)
+                try {
+                    const hookFunctionArguments = [
+                        types.stringLiteral(varName),
+                        variableDeclarator.init,
+                        types.stringLiteral("var-init")
+                    ];
+                    variableDeclarator.init = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments)
+                } catch (e) {
+                    console.error(e);
+                }
             }
         },
 
@@ -47,12 +51,16 @@ function injectHook(jsCode) {
                 varName = generator.default(node.left).code;
             }
 
-            const hookFunctionArguments = [
-                types.stringLiteral(varName),
-                node.right,
-                types.stringLiteral("assign")
-            ];
-            node.right = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments)
+            try {
+                const hookFunctionArguments = [
+                    types.stringLiteral(varName),
+                    node.right,
+                    types.stringLiteral("assign")
+                ];
+                node.right = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments)
+            } catch (e) {
+                console.error(e);
+            }
         },
 
         // 对象表达式
@@ -79,13 +87,18 @@ function injectHook(jsCode) {
                     objectKey = types.stringLiteral(objectKey.name);
                 }
 
-                const hookFunctionArguments = [
-                    objectKey,
-                    propertyValue,
-                    types.stringLiteral("object-key-init")
-                ];
-                objectProperty.value = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments);
+                try {
+                    const hookFunctionArguments = [
+                        objectKey,
+                        propertyValue,
+                        types.stringLiteral("object-key-init")
+                    ];
+                    objectProperty.value = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments);
+                } catch (e) {
+                    console.error(e);
+                }
             }
+
         },
 
         // 函数的形参
@@ -97,15 +110,19 @@ function injectHook(jsCode) {
             const params = node.params;
             if (types.isBlockStatement(node.body)) {
                 // 函数体是个代码块的，则在代码块最前面插入Hook，检查参数的值
-                for(let i=params.length-1; i>=0; i--) {
-                    const paramName = params[i];
-                    const hookFunctionArguments = [
-                        types.stringLiteral(generator.default(paramName).code),
-                        paramName,
-                        types.stringLiteral("function-parameter")
-                    ];
-                    const hookFunction = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments);
-                    node.body.body.unshift(types.expressionStatement(hookFunction));
+                for (let i = params.length - 1; i >= 0; i--) {
+                    try {
+                        const paramName = params[i];
+                        const hookFunctionArguments = [
+                            types.stringLiteral(generator.default(paramName).code),
+                            paramName,
+                            types.stringLiteral("function-parameter")
+                        ];
+                        const hookFunction = types.callExpression(types.identifier(hookFunctionName), hookFunctionArguments);
+                        node.body.body.unshift(types.expressionStatement(hookFunction));
+                    } catch (e) {
+                        console.error(e);
+                    }
                 }
             }
         }