Function queryNodes

masfactory-visualizer/src/parser/astUtils.ts:21–36

(node: TSNode, type: string)

 * Query all nodes of a specific type in the AST
 */
export function queryNodes(node: TSNode, type: string): TSNode[] {
    const results: TSNode[] = [];
    
    function traverse(n: TSNode) {
        if (n.type === type) {
            results.push(n);
        }
        for (const child of n.children) {
            if (!child) {continue;}
            traverse(child);
        }
    }
    
    traverse(node);
    return results;
}

/**
 * Get base classes from a class definition node