diff --git a/test/fixtures/apilinks/reverse.js b/test/fixtures/apilinks/reverse.js new file mode 100644 index 00000000000000..5a61e50dc27581 --- /dev/null +++ b/test/fixtures/apilinks/reverse.js @@ -0,0 +1,13 @@ +'use strict'; + +// Parallel assignment to the exported variable and module.exports. + +function ok() { +} + +const asserts = module.exports = ok; + +asserts.ok = ok; + +asserts.strictEqual = function() { +} diff --git a/test/fixtures/apilinks/reverse.json b/test/fixtures/apilinks/reverse.json new file mode 100644 index 00000000000000..aa32e4a90a745c --- /dev/null +++ b/test/fixtures/apilinks/reverse.json @@ -0,0 +1,5 @@ +{ + "asserts": "reverse.js#L8", + "asserts.ok": "reverse.js#L5", + "asserts.strictEqual": "reverse.js#L12" +} diff --git a/tools/doc/apilinks.js b/tools/doc/apilinks.js index 98dd7827d67ac0..35183912d31870 100644 --- a/tools/doc/apilinks.js +++ b/tools/doc/apilinks.js @@ -55,35 +55,50 @@ process.argv.slice(2).forEach((file) => { const ast = acorn.parse(source, { ecmaVersion: 10, locations: true }); const program = ast.body; + // Build link + const link = `https://github.com/${repo}/blob/${tag}/` + + path.relative('.', file).replace(/\\/g, '/'); + // Scan for exports. const exported = { constructors: [], identifiers: [] }; program.forEach((statement) => { - if (statement.type !== 'ExpressionStatement') return; - const expr = statement.expression; - if (expr.type !== 'AssignmentExpression') return; - - let lhs = expr.left; - if (expr.left.object.type === 'MemberExpression') lhs = lhs.object; - if (lhs.type !== 'MemberExpression') return; - if (lhs.object.name !== 'module') return; - if (lhs.property.name !== 'exports') return; - - let rhs = expr.right; - while (rhs.type === 'AssignmentExpression') rhs = rhs.right; - - if (rhs.type === 'NewExpression') { - exported.constructors.push(rhs.callee.name); - } else if (rhs.type === 'ObjectExpression') { - rhs.properties.forEach((property) => { - if (property.value.type === 'Identifier') { - exported.identifiers.push(property.value.name); - if (/^[A-Z]/.test(property.value.name[0])) { - exported.constructors.push(property.value.name); + if (statement.type === 'ExpressionStatement') { + const expr = statement.expression; + if (expr.type !== 'AssignmentExpression') return; + + let lhs = expr.left; + if (expr.left.object.type === 'MemberExpression') lhs = lhs.object; + if (lhs.type !== 'MemberExpression') return; + if (lhs.object.name !== 'module') return; + if (lhs.property.name !== 'exports') return; + + let rhs = expr.right; + while (rhs.type === 'AssignmentExpression') rhs = rhs.right; + + if (rhs.type === 'NewExpression') { + exported.constructors.push(rhs.callee.name); + } else if (rhs.type === 'ObjectExpression') { + rhs.properties.forEach((property) => { + if (property.value.type === 'Identifier') { + exported.identifiers.push(property.value.name); + if (/^[A-Z]/.test(property.value.name[0])) { + exported.constructors.push(property.value.name); + } } - } - }); - } else if (rhs.type === 'Identifier') { - exported.identifiers.push(rhs.name); + }); + } else if (rhs.type === 'Identifier') { + exported.identifiers.push(rhs.name); + } + } else if (statement.type === 'VariableDeclaration') { + for (const decl of statement.declarations) { + let init = decl.init; + while (init && init.type === 'AssignmentExpression') init = init.left; + if (!init || init.type !== 'MemberExpression') continue; + if (init.object.name !== 'module') continue; + if (init.property.name !== 'exports') continue; + exported.constructors.push(decl.id.name); + definition[decl.id.name] = `${link}#L${statement.loc.start.line}`; + } } }); @@ -93,8 +108,7 @@ process.argv.slice(2).forEach((file) => { // ClassName.prototype.foo = ...; // function Identifier(...) {...}; // - const link = `https://github.com/${repo}/blob/${tag}/` + - path.relative('.', file).replace(/\\/g, '/'); + const indirect = {}; program.forEach((statement) => { if (statement.type === 'ExpressionStatement') { @@ -128,6 +142,11 @@ process.argv.slice(2).forEach((file) => { } definition[name] = `${link}#L${statement.loc.start.line}`; + + if (expr.left.property.name === expr.right.name) { + indirect[expr.right.name] = name; + } + } else if (statement.type === 'FunctionDeclaration') { const name = statement.id.name; if (!exported.identifiers.includes(name)) return; @@ -136,6 +155,18 @@ process.argv.slice(2).forEach((file) => { `${link}#L${statement.loc.start.line}`; } }); + + // Search for indirect references of the form ClassName.foo = foo; + if (Object.keys(indirect).length > 0) { + program.forEach((statement) => { + if (statement.type === 'FunctionDeclaration') { + const name = statement.id.name; + if (indirect[name]) { + definition[indirect[name]] = `${link}#L${statement.loc.start.line}`; + } + } + }); + } }); console.log(JSON.stringify(definition, null, 2));