@kezh/my-custom-js-converter 0.0.18 → 0.0.20

package/core/encode.js

@@ -108,6 +108,7 @@ function getImportMethods(modPathByName, methodNamesByMod) {
             ...new Set([...useMethodNames, ...inlineMethodNames])
         ];
         modifyModMethods(name, mergeMethodNames, modAst);
+        console.log((0, generator_1.default)(modAst).code);
         mergeMethodNames = mergeMethodNames.map((m) => `${m}__${name}`);
         const mergeMethods = mergeMethodNames
             .map((name) => {
@@ -244,13 +245,27 @@ function getSrcMethodNames(entry, srcAst) {
 function getMethods(entry, ast) {
     const methods = [];
     (0, traverse_1.default)(ast, {
-        Identifier(path) {
-            if (path.node.name === entry) {
-                const declarator = path.findParent((path) => path.isVariableDeclarator());
-                const methodBody = declarator === null || declarator === void 0 ? void 0 : declarator.node.init;
-                const inlineMethods = methodBody === null || methodBody === void 0 ? void 0 : methodBody.properties;
-                (inlineMethods === null || inlineMethods === void 0 ? void 0 : inlineMethods.length) && methods.push(...inlineMethods);
-            }
+        Program(path) {
+            path.node.body.forEach((statement) => {
+                if (core_1.types.isVariableDeclaration(statement)) {
+                    const variableDeclaration = statement;
+                    variableDeclaration.declarations.filter((declaration) => {
+                        if (core_1.types.isIdentifier(declaration.id)) {
+                            const id = declaration.id;
+                            if (id.name === entry) {
+                                return true;
+                            }
+                        }
+                        return false;
+                    }).forEach((declaration) => {
+                        if (core_1.types.isObjectExpression(declaration.init)) {
+                            const init = declaration.init;
+                            const objectMethods = init.properties.filter(property => core_1.types.isObjectMethod(property));
+                            methods.push(...objectMethods);
+                        }
+                    });
+                }
+            });
         }
     });
     return methods;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kezh/my-custom-js-converter",
-  "version": "0.0.18",
+  "version": "0.0.20",
   "description": "自定义代码转换器",
   "main": "index.js",
   "bin": {