英文:
Fetch function names from a file using tree-sitter-javascript
问题
Using the tree-sitter-javascript库，我正在尝试从JavaScript代码（以字符串格式）中提取所有函数名称，并将其存储在functionNames数组中。
const Parser = require('tree-sitter');const JavaScript = require('tree-sitter-javascript');const parser = new Parser();parser.setLanguage(JavaScript);const sourceCode = `function foo() {console.log('hello world');}function bar() {console.log('bye world');}`;const tree = parser.parse(sourceCode);const functionNames = [];tree.walk((node) => {console.log("Node: ", node);if (node.type === 'function_declaration') {functionNames.push(node.child(1).text);}})console.log(functionNames);// 期望输出 ['foo', 'bar']// 实际输出 []
然而，tree.walk()的回调根本不执行。
日志中什么都不打印，只是一个空数组[]。
这段代码有什么问题？
我在官方文档（https://github.com/tree-sitter/tree-sitter-javascript）中找不到太多关于此的文档。
是否有其他方法可以实现这个目标？提前谢谢。
英文:
Using tree-sitter-javascript library, I'm trying to fetch all function names from a Javascript code (in string format) and storing it in a functionNames array.
const Parser = require('tree-sitter');const JavaScript = require('tree-sitter-javascript');const parser = new Parser();parser.setLanguage(JavaScript);const sourceCode = `function foo() {console.log('hello world');}function bar() {console.log('bye world');}`;const tree = parser.parse(sourceCode);const functionNames = [];tree.walk(()=>{console.log("Node: ", node);if (node.type === 'function_declaration') {functionNames.push(node.child(1).text);}})console.log(functionNames);// Expected output ['foo', 'bar']// Actual output []
However, the callback to tree.walk() is not executing at all.
Nothing gets printed in the log. just an empty array []
What is wrong with this code ?
I couldn't find much documentation on this in the official docs (https://github.com/tree-sitter/tree-sitter-javascript)
Is there an alternate way to do this ? Thanks in advance
答案1
得分: 2
parser.parse已经给你解析好的结果。该对象上没有.walk()函数。要获取所有函数名称，你可以递归遍历rootNode：
const tree = parser.parse(sourceCode);const rootNode = tree.rootNode;const functionNames = [];function findFunctionNames(node) {if (node.type === 'function_declaration') {for (let child of node.namedChildren) {if (child.type === 'identifier') {functionNames.push(child.text);}}}for (let child of node.namedChildren) {findFunctionNames(child);}}findFunctionNames(rootNode);console.log(functionNames);// -> [ 'foo', 'bar' ]
英文:
That's because parser.parse already gives you the parsed result. There is no .walk() function on that object. To get all the function names you can traverse the rootNode recursively:
const tree = parser.parse(sourceCode);const rootNode = tree.rootNode;const functionNames = [];function findFunctionNames(node) {if (node.type === 'function_declaration') {for (let child of node.namedChildren) {if (child.type === 'identifier') {functionNames.push(child.text);}}}for (let child of node.namedChildren) {findFunctionNames(child);}}findFunctionNames(rootNode);console.log(functionNames);// -> [ 'foo', 'bar' ]
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论