Skip to content

Commit

Permalink
all good
Browse files Browse the repository at this point in the history
  • Loading branch information
@JamsonChan
JamsonChan committed Oct 10, 2024
1 parent ef000da commit ee089c0
Showing 1 changed file with 50 additions and 64 deletions.
114 changes: 50 additions & 64 deletions babel-plugin-expose-private-functions-and-variables.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -129,89 +129,75 @@ module.exports = function ({types: t}) {
funcCallCount.set(right.name, funcCallCount.get(right.name) + 1);
}
},
CallExpression(callPath) {
const callee = callPath.node.callee;

// 替換直接的函式呼叫
if (t.isIdentifier(callee) && privateFuncs.has(callee.name)) {
callPath.node.callee = createNestedMemberExpression([callee.name, ...dir]);
funcCallCount.set(callee.name, funcCallCount.get(callee.name) + 1);
ConditionalExpression(condPath) {
const { test, consequent, alternate } = condPath.node;
// TODO: test ?
// case: x ? FUNC : x -> x ? window.PACKAGE._.FUNC : x
if (t.isIdentifier(consequent) && privateFuncs.has(consequent.name)) {
funcCallCount.set(consequent.name, funcCallCount.get(consequent.name) + 1);
condPath.get('consequent').replaceWith(createNestedMemberExpression([consequent.name, ...dir]));
}

// 檢查是否是 setTimeout 的函式參數
if (t.isIdentifier(callee) && callee.name === 'setTimeout') {
// case: x ? x : FUNC -> x ? x : window.PACKAGE._.FUNC
if (t.isIdentifier(alternate) && privateFuncs.has(alternate.name)) {
funcCallCount.set(alternate.name, funcCallCount.get(alternate.name) + 1);
condPath.get('alternate').replaceWith(createNestedMemberExpression([alternate.name, ...dir]));
}
},
CallExpression(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 => {
// 檢查是否為 FunctionExpression
if (t.isFunctionExpression(arg.node)) {
arg.traverse({
CallExpression(innerCallPath) {
const innerCallee = innerCallPath.node.callee;

// 如果內部的呼叫是私有函式
if (t.isIdentifier(innerCallee) && privateFuncs.has(innerCallee.name)) {
innerCallPath.node.callee = createNestedMemberExpression([innerCallee.name, ...dir]);
funcCallCount.set(innerCallee.name, funcCallCount.get(innerCallee.name) + 1);
}
}
});
} else if (t.isConditionalExpression(arg.node)) {
// 處理條件運算子
const { test, consequent, alternate } = arg.node;

// 檢查 consequent 和 alternate 是否是 Identifier
if (t.isIdentifier(consequent) && privateFuncs.has(consequent.name)) {
arg.get('consequent').replaceWith(createNestedMemberExpression([consequent.name, ...dir]));
funcCallCount.set(consequent.name, funcCallCount.get(consequent.name) + 1);
}
if (t.isIdentifier(alternate) && privateFuncs.has(alternate.name)) {
arg.get('alternate').replaceWith(createNestedMemberExpression([alternate.name, ...dir]));
funcCallCount.set(alternate.name, funcCallCount.get(alternate.name) + 1);
}
} else if (t.isIdentifier(arg.node) && privateFuncs.has(arg.node.name)) {
// 替換 setTimeout 中的函式參數
// 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]));
// funcCallCount.set(arg.node.name, funcCallCount.get(arg.node.name) + 1);
}
});
}
},
FunctionExpression(funcPath) {
// 遍歷函式內部的語句，檢查是否使用私有函式
funcPath.traverse({
CallExpression(innerCallPath) {
const callee = innerCallPath.node.callee;

if (t.isIdentifier(callee) && privateFuncs.has(callee.name)) {
innerCallPath.node.callee = createNestedMemberExpression([callee.name, ...dir]);
funcCallCount.set(callee.name, funcCallCount.get(callee.name) + 1);
}
}
});
},
ObjectExpression(objectPath) {
objectPath.node.properties.forEach((property) => {
// 確保屬性值是 Identifier，且在私有函式中
if (t.isIdentifier(property.value) && privateFuncs.has(property.value.name)) {
// 替換屬性值
property.value = createNestedMemberExpression([property.value.name, ...dir]);
// funcCallCount.set(property.value.name, funcCallCount.get(property.value.name) + 1);
}
});
}
// FIXME: 這塊反而讓結果錯誤 window.zk.mount._.doAfterMount(window.zk.mount._.mtBL1); -> window.zk.mount._.doAfterMount(mtBL1);
// FunctionExpression(funcPath) {
// // 遍歷函式內部的語句，檢查是否使用私有函式
// funcPath.traverse({
// CallExpression(innerCallPath) {
// const callee = innerCallPath.node.callee;
//
// if (t.isIdentifier(callee) && privateFuncs.has(callee.name)) {
// innerCallPath.node.callee = createNestedMemberExpression([callee.name, ...dir]);
// funcCallCount.set(callee.name, funcCallCount.get(callee.name) + 1);
// }
// }
// });
// },
// ObjectExpression(objectPath) {
// objectPath.node.properties.forEach((property) => {
// // 確保屬性值是 Identifier，且在私有函式中
// if (t.isIdentifier(property.value) && privateFuncs.has(property.value.name)) {
// // 替換屬性值
// property.value = createNestedMemberExpression([property.value.name, ...dir]);
// // funcCallCount.set(property.value.name, funcCallCount.get(property.value.name) + 1);
// }
// });
// }
});

// 4. 構建一個變數來輸出呼叫次數結果
const callCountArray = Array.from(funcCallCount.entries()).map(([name, count]) =>
t.objectProperty(t.identifier(name), t.numericLiteral(count))
);
const callCountVariable = t.variableDeclaration('var', [
t.variableDeclarator(
t.identifier('dlwlrmaXXX'),
t.identifier('dlwlrma'),
t.objectExpression(callCountArray)
)
]);
// 5. 在程式結尾插入計算結果變數
path.pushContainer('body', callCountVariable);
}
}
Expand Down

0 comments on commit ee089c0

Please sign in to comment.