tmp

zkoss · Oct 10, 2024 · 3cf56ec · 3cf56ec
1 parent 3b788a9
commit 3cf56ec
Showing 1 changed file with 22 additions and 16 deletions.
diff --git a/babel-plugin-expose-private-functions-and-variables.js b/babel-plugin-expose-private-functions-and-variables.js
@@ -179,23 +179,29 @@ module.exports = function ({types: t}) {
 					}
 
 					function callExp(callPath) {
-						const callee = callPath.node.callee;
-						if (t.isIdentifier(callee)) {
-							// case: FUNC() -> window.PACKAGE._.FUNC()
-							// this case includes `FUNC() in ?: conditional`
-							if (privateFuncs.has(callee.name)) {
-								funcCallCount.set(callee.name, funcCallCount.get(callee.name) + 1);
-								callPath.node.callee = createNestedMemberExpression([callee.name, ...dir]);
-							}
-							const args = callPath.get('arguments');
-							args.forEach(arg => {
-								// case: xxx(FUNC) -> xxx(window.PACKAGE._.FUNC)
-								if (t.isIdentifier(arg.node) && privateFuncs.has(arg.node.name)) {
-									funcCallCount.set(arg.node.name, funcCallCount.get(arg.node.name) + 1);
-									arg.replaceWith(createNestedMemberExpression([arg.node.name, ...dir]));
-								}
-							});
+						// CallExpression (callee, arguments)
+						// x(y...) -> callee: x, arguments: y...
+
+
+						dfs(callPath.get('callee'));
+
+						const { callee, arguments: args } = callPath.node;
+
+						// case: FUNC() -> window.PACKAGE._.FUNC()
+						if (t.isIdentifier(callee) && privateFuncs.has(callee.name)) {
+							funcCallCount.set(callee.name, funcCallCount.get(callee.name) + 1);
+							callPath.node.callee = createNestedMemberExpression([callee.name, ...dir]);
 						}
+
+						// const args = callPath.get('arguments');
+						args.forEach(arg => {
+							// dfs(arg);
+							// case: xxx(FUNC...) -> xxx(window.PACKAGE._.FUNC...)
+							if (t.isIdentifier(arg.node) && privateFuncs.has(arg.node.name)) {
+								funcCallCount.set(arg.node.name, funcCallCount.get(arg.node.name) + 1);
+								arg.replaceWith(createNestedMemberExpression([arg.node.name, ...dir]));
+							}
+						});
 					}
 
 					function objExp(objectPath) {