Function getBaseClasses

masfactory-visualizer/src/parser/astUtils.ts:41–56

(classNode: TSNode)

 * Get base classes from a class definition node
 */
export function getBaseClasses(classNode: TSNode): string[] {
    const bases: string[] = [];
    const argList = classNode.childForFieldName('superclasses');
    if (argList) {
        for (const child of argList.namedChildren) {
            if (!child) {continue;}
            if (child.type === 'identifier') {
                bases.push(child.text);
            } else if (child.type === 'attribute') {
                // Handle module.ClassName
                bases.push(child.text);
            }
        }
    }
    return bases;
}

/**
 * Collect local class -> base class mappings from a module.