fix t.isFunctionExpression改t.isCallExpression

This commit is contained in:
Wang Kejun 2023-04-14 11:56:36 +08:00
parent 292895b328
commit 01190a44bb
1 changed files with 23 additions and 15 deletions

View File

@ -236,24 +236,32 @@ module.exports = (babel) => {
}
}
// exports.default = OakPage({})、exports.default = OakComponent({})
else if (t.isExpressionStatement(node2) && t.isAssignmentExpression(node2.expression) &&
t.isFunctionExpression(node2.expression.right) &&
t.isIdentifier(node2.expression.right.callee) && node2.expression.right.callee.name === 'OakComponent'
else if (
t.isExpressionStatement(node2) &&
t.isAssignmentExpression(node2.expression) &&
t.isCallExpression(node2.expression.right) &&
t.isIdentifier(node2.expression.right.callee) &&
node2.expression.right.callee.name ===
'OakComponent'
) {
const statements = getStatements();
node2.expression.right.arguments.forEach((node3) => {
if (t.isObjectExpression(node3)) {
const propertyRender = t.objectProperty(
t.identifier('getRender'),
t.functionExpression(
null,
[],
t.blockStatement(statements)
)
);
node3.properties.unshift(propertyRender);
node2.expression.right.arguments.forEach(
(node3) => {
if (t.isObjectExpression(node3)) {
const propertyRender = t.objectProperty(
t.identifier('getRender'),
t.functionExpression(
null,
[],
t.blockStatement(statements)
)
);
node3.properties.unshift(
propertyRender
);
}
}
});
);
}
});
}