cha0s/flecks
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: visitor and tests

Browse Source
This commit is contained in:
cha0s 2024-02-14 08:35:57 -06:00
parent 477bb740e9
commit 696c0c885a
2 changed files with 26 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
packages/dox/build/parser.js
View File

@ -30,7 +30,6 @@ exports.parseNormalSource = async (path, source, root, request, options) => {
const ast = await exports.parseCode(source, options);
const buildFiles = [];
const configs = [];
const hookBases = [];
const hookImplementations = [];
const hookInvocations = [];
const todos = [];
@ -99,7 +98,6 @@ exports.parseNormalSource = async (path, source, root, request, options) => {
return {
buildFiles,
config: configs,
hookBases,
hookImplementations,
hookInvocations,
todos,

49
packages/dox/build/visitors.js
View File

@ -24,6 +24,29 @@ function visitProperties(properties, fn) {
});
}
// Test Flecks.hooks(require.context(...))
function testRight(right, fn) {
if (isCallExpression(right)) {
if (isMemberExpression(right.callee)) {
if (
isIdentifier(right.callee.object) && 'Flecks' === right.callee.object.name
&& isIdentifier(right.callee.property) && 'hooks' === right.callee.property.name
) {
if (isCallExpression(right.arguments[0])) {
if (
isIdentifier(right.arguments[0].callee.object)
&& 'require' === right.arguments[0].callee.object.name
&& isIdentifier(right.arguments[0].callee.property)
&& 'context' === right.arguments[0].callee.property.name
) {
fn(right.arguments[0].arguments);
}
}
}
}
}
}
exports.hookBaseVisitor = (fn) => ({
// exports.hooks = Flecks.hooks(require.context(...))
AssignmentExpression(path) {
@ -31,39 +54,19 @@ exports.hookBaseVisitor = (fn) => ({
if (isMemberExpression(left)) {
if (isIdentifier(left.object) && 'exports' === left.object.name) {
if (isIdentifier(left.property) && 'hooks' === left.property.name) {
if (isCallExpression(right)) {
if (isMemberExpression(right.callee)) {
if (
isIdentifier(right.callee.object) && 'Flecks' === right.callee.object.name
&& isIdentifier(right.callee.property) && 'hooks' === right.callee.property.name
) {
if (isCallExpression(right.arguments[0])) {
if (
isIdentifier(right.arguments[0].callee.object)
&& 'require' === right.arguments[0].callee.object.name
&& isIdentifier(right.arguments[0].callee.property)
&& 'context' === right.arguments[0].callee.property.name
) {
fn(right.arguments[0].arguments);
}
}
}
}
}
testRight(right, fn);
}
}
}
},
// export const hooks = Flecks.hooks(...)
// export const hooks = Flecks.hooks(require.context(...))
ExportNamedDeclaration(path) {
const {declaration} = path.node;
if (isVariableDeclaration(declaration)) {
const {declarations} = declaration;
declarations.forEach((declarator) => {
if ('hooks' === declarator.id.name) {
if (isObjectExpression(declarator.init)) {
visitProperties(declarator.init.properties, fn);
}
testRight(declarator.init, fn);
}
});
}