#15904

Committed 11 May 2026 06:41PM UTC coverage: 86.89% (-0.006%) from 86.896%

Build Type

push

Committed by

Commit Message

Merge 70dfd6181 into ce68f5cb7

Coverage Stats

15596 of 18958 branches covered (82.27%)

Branch coverage included in aggregate %.

9 of 9 new or added lines in 3 files covered. (100.0%)

70 existing lines in 2 files now uncovered.

16350 of 17808 relevant lines covered (91.81%)

27324.07 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

69.24

/src/Scope.ts

/* eslint-disable @typescript-eslint/dot-notation */
import * as path from 'path';
import chalk from 'chalk';
import type { Range } from 'vscode-languageserver-protocol';
import type { CallableContainer, FileReference, FileLink, Callable, ScopeValidationOptions } from './interfaces';
import type { Program } from './Program';
import type { NamespaceStatement, FunctionStatement, ClassStatement, EnumStatement, InterfaceStatement, EnumMemberStatement, ConstStatement, TypeStatement } from './parser/Statement';
import { ParseMode } from './parser/Parser';
import { util } from './util';
import { Cache } from './Cache';
import type { BrsFile } from './files/BrsFile';
import type { Identifier } from './lexer/Token';
import type { DependencyGraph, DependencyChangedEvent } from './DependencyGraph';
import { isBrsFile, isXmlFile, isEnumMemberStatement, isNamespaceStatement, isTypeStatement, isXmlScope } from './astUtils/reflection';
import { WalkMode } from './astUtils/visitors';
import { SymbolTable } from './SymbolTable';
import { SymbolTypeFlag } from './SymbolTypeFlag';
import type { BscFile } from './files/BscFile';
import { referenceTypeFactory } from './types/ReferenceType';
import { unionTypeFactory } from './types/UnionType';
import { AssociativeArrayType } from './types/AssociativeArrayType';
import type { Statement } from './parser/AstNode';
import { performance } from 'perf_hooks';
import { LogLevel } from './logging';
import { uninitializedTypeFactory } from './types/UninitializedType';
import { ScopeNamespaceLookup } from './ScopeNamespaceLookup';

/**
 * Assign some few factories to the SymbolTable to prevent cyclical imports. This file seems like the most intuitive place to do the linking
 * since Scope will be used by pretty much everything
 */
SymbolTable.referenceTypeFactory = referenceTypeFactory;
SymbolTable.unionTypeFactory = unionTypeFactory;
SymbolTable.uninitializedTypeFactory = uninitializedTypeFactory;
/**
 * A class to keep track of all declarations within a given scope (like source scope, component scope)
 */
export class Scope {
    constructor(
        public name: string,
        public program: Program,
        private _dependencyGraphKey?: string
    ) {
        this.isValidated = false;
        //used for improved logging performance
        this._debugLogComponentName = `Scope '${chalk.redBright(this.name)}'`;
    }

    /**
     * Indicates whether this scope needs to be validated.
     * Will be true when first constructed, or anytime one of its dependencies changes
     */
    public readonly isValidated: boolean;

    protected cache = new Cache();

    public get dependencyGraphKey() {
        return this._dependencyGraphKey;
    }

    /**
     * A dictionary of namespaces, indexed by the lower case full name of each namespace.
     * If a namespace is declared as "NameA.NameB.NameC", there will be 3 entries in this dictionary,
     * "namea", "namea.nameb", "namea.nameb.namec"
     */
    public get namespaceLookup() {
        let allFilesValidated = true;
        for (const file of this.getAllFiles()) {
            if (isBrsFile(file) && !file.hasTypedef) {
                allFilesValidated = allFilesValidated && file.isValidated;
                if (!allFilesValidated) {
                    break;
                }
            }
        }
        if (!allFilesValidated) {
            // This is not fit to cache
            // Since the files have not been validated, all namespace info might not have been available
            return this.buildNamespaceLookup();
        }
        return this.cache.getOrAdd('namespaceLookup', () => this.buildNamespaceLookup());
    }

    /**
     * A dictionary of namespaces, indexed by the lower case full name of each namespace.
     * If a namespace is declared as "NameA.NameB.NameC", there will be 3 entries in this dictionary,
     * "namea", "namea.nameb", "namea.nameb.namec"
     */
    public get namespaceNameSet() {
        return this.cache.getOrAdd('namespaceNameSet', () => {
            const lowerNamespaceNames = new Set<string>();

            this.enumerateBrsFiles((file) => {
                const fileNamespaceLookup = file.getNamespaceLookupObject();
                for (const [lowerNamespaceName, _] of fileNamespaceLookup) {
                    lowerNamespaceNames.add(lowerNamespaceName);
                }
            });
            return lowerNamespaceNames;
        });
    }


    /**
     * Get a NamespaceContainer by its name, looking for a fully qualified version first, then global version next if not found
     */
    public getNamespace(name: string, containingNamespace?: string) {
        const nameLower = name?.toLowerCase();
        const lookup = this.namespaceLookup;

        let ns: NamespaceContainer;
        if (containingNamespace) {
            ns = lookup.get(`${containingNamespace?.toLowerCase()}.${nameLower}`);
        }
        //if we couldn't find the namespace by its full namespaced name, look for a global version
        if (!ns) {
            ns = lookup.get(nameLower);
        }
        return ns;
    }

    /**
     * Get a NamespaceContainer by its name, looking for a fully qualified version first, then global version next if not found
     */
    public getNamespacesWithRoot(rootName: string, containingNamespace?: string) {
        const nameLower = rootName?.toLowerCase();
        const lookup = this.namespaceLookup;
        const lookupKeys = [...lookup.keys()];
        let lookupName = nameLower;
        if (containingNamespace) {
            lookupName = `${containingNamespace?.toLowerCase()}.${nameLower}`;
        }
        const nsList = lookupKeys.filter(key => key === lookupName).map(key => lookup.get(key));
        return nsList;
    }


    /**
     * Get a NamespaceContainer by its name, looking for a fully qualified version first, then global version next if not found
     */
    public getFirstNamespaceWithRoot(rootName: string, containingNamespace?: string) {
        const nameLower = rootName?.toLowerCase();

        let lookupName = nameLower;
        if (containingNamespace) {
            lookupName = `${containingNamespace?.toLowerCase()}.${nameLower}`;
        }
        return this.namespaceLookup.get(lookupName);
    }

    /**
     * Get the class with the specified name.
     * @param className - The class name, including the namespace of the class if possible
     * @param containingNamespace - The namespace used to resolve relative class names. (i.e. the namespace around the current statement trying to find a class)
     */
    public getClass(className: string, containingNamespace?: string): ClassStatement {
        return this.getClassFileLink(className, containingNamespace)?.item;
    }

    /**
     * Get the interface with the specified name.
     * @param ifaceName - The interface name, including the namespace of the interface if possible
     * @param containingNamespace - The namespace used to resolve relative interface names. (i.e. the namespace around the current statement trying to find a interface)
     */
    public getInterface(ifaceName: string, containingNamespace?: string): InterfaceStatement {
        return this.getInterfaceFileLink(ifaceName, containingNamespace)?.item;
    }

    /**
     * Get the enum with the specified name.
     * @param enumName - The enum name, including the namespace if possible
     * @param containingNamespace - The namespace used to resolve relative enum names. (i.e. the namespace around the current statement trying to find an enum)
     */
    public getEnum(enumName: string, containingNamespace?: string): EnumStatement {
        return this.getEnumFileLink(enumName, containingNamespace)?.item;
    }

    private useFileCachesForFileLinkLookups = false;

    private getFileLinkFromFileMap<T>(cachedMapName: string, itemName: string, containingNamespace?: string): FileLink<T> {
        let result: FileLink<T>;
        const fullNameLower = util.getFullyQualifiedClassName(itemName, containingNamespace)?.toLowerCase();
        const itemNameLower = itemName?.toLowerCase();
        if (fullNameLower) {
            this.enumerateBrsFilesWithBreak((file) => {
                let stmt = file['_cachedLookups'][cachedMapName].get(fullNameLower);
                if (stmt) {
                    result = { item: stmt, file: file };
                }
                return !!stmt;
            });
        }
        if (!result && itemNameLower && fullNameLower !== itemNameLower) {
            this.enumerateBrsFilesWithBreak((file) => {
                let stmt = file['_cachedLookups'][cachedMapName].get(itemNameLower);
                if (stmt) {
                    result = { item: stmt, file: file };
                }
                return !!stmt;
            });
        }
        return result;
    }

    public getTypeStatement(typeName: string, containingNamespace?: string): Statement {
        return this.getTypeStatementFileLink(typeName, containingNamespace)?.item;
    }

    /**
     * Get a class and its containing file by the class name
     * @param className - The class name, including the namespace of the class if possible
     * @param containingNamespace - The namespace used to resolve relative class names. (i.e. the namespace around the current statement trying to find a class)
     */
    public getClassFileLink(className: string, containingNamespace?: string): FileLink<ClassStatement> {
        if (this.useFileCachesForFileLinkLookups) {
            return this.getFileLinkFromFileMap('classStatementMap', className, containingNamespace);
        }
        const lowerName = className?.toLowerCase();
        const fullNameLower = util.getFullyQualifiedClassName(lowerName, containingNamespace)?.toLowerCase();
        const classMap = this.getClassMap();

        let cls = classMap.get(fullNameLower);
        //if we couldn't find the class by its full namespaced name, look for a global class with that name
        if (!cls && lowerName && lowerName !== fullNameLower) {
            cls = classMap.get(lowerName);
        }
        return cls;
    }

    /**
     * Get an interface and its containing file by the interface name
     * @param ifaceName - The interface name, including the namespace of the interface if possible
     * @param containingNamespace - The namespace used to resolve relative interface names. (i.e. the namespace around the current statement trying to find a interface)
     */
    public getInterfaceFileLink(ifaceName: string, containingNamespace?: string): FileLink<InterfaceStatement> {
        if (this.useFileCachesForFileLinkLookups) {
            return this.getFileLinkFromFileMap('interfaceStatementMap', ifaceName, containingNamespace);
        }
        const lowerName = ifaceName?.toLowerCase();
        const fullNameLower = util.getFullyQualifiedClassName(lowerName, containingNamespace)?.toLowerCase();
        const ifaceMap = this.getInterfaceMap();

        let iface = ifaceMap.get(fullNameLower);
        //if we couldn't find the iface by its full namespaced name, look for a global class with that name
        if (!iface && lowerName && lowerName !== fullNameLower) {
            iface = ifaceMap.get(lowerName);
        }
        return iface;
    }

    /**
     * Get an Enum and its containing file by the Enum name
     * @param enumName - The Enum name, including the namespace of the enum if possible
     * @param containingNamespace - The namespace used to resolve relative enum names. (i.e. the namespace around the current statement trying to find a enum)
     */
    public getEnumFileLink(enumName: string, containingNamespace?: string): FileLink<EnumStatement> {
        if (this.useFileCachesForFileLinkLookups) {
            return this.getFileLinkFromFileMap('enumStatementMap', enumName, containingNamespace);
        }
        const lowerName = enumName?.toLowerCase();
        const fullNameLower = util.getFullyQualifiedClassName(lowerName, containingNamespace)?.toLowerCase();
        const enumMap = this.getEnumMap();

        let enumeration = enumMap.get(fullNameLower);
        //if we couldn't find the enum by its full namespaced name, look for a global enum with that name
        if (!enumeration && lowerName && lowerName !== fullNameLower) {
            enumeration = enumMap.get(lowerName);
        }
        return enumeration;
    }

    /**
     * Get an Enum and its containing file by the Enum name
     * @param enumMemberName - The Enum name, including the namespace of the enum if possible
     * @param containingNamespace - The namespace used to resolve relative enum names. (i.e. the namespace around the current statement trying to find a enum)
     */
    public getEnumMemberFileLink(enumMemberName: string, containingNamespace?: string): FileLink<EnumMemberStatement> {
        let lowerNameParts = enumMemberName?.toLowerCase()?.split('.');
        let memberName = lowerNameParts?.splice(lowerNameParts.length - 1, 1)?.[0];
        let lowerName = lowerNameParts?.join('.').toLowerCase();
        const enumMap = this.getEnumMap();

        let enumeration = enumMap.get(
            util.getFullyQualifiedClassName(lowerName, containingNamespace?.toLowerCase())
        );
        //if we couldn't find the enum by its full namespaced name, look for a global enum with that name
        if (!enumeration) {
            enumeration = enumMap.get(lowerName);
        }
        if (enumeration) {
            let member = enumeration.item.findChild<EnumMemberStatement>((child) => isEnumMemberStatement(child) && child.name?.toLowerCase() === memberName);
            return member ? { item: member, file: enumeration.file } : undefined;
        }
    }

    /**
     * Get a constant and its containing file by the constant name
     * @param constName - The constant name, including the namespace of the constant if possible
     * @param containingNamespace - The namespace used to resolve relative constant names. (i.e. the namespace around the current statement trying to find a constant)
     */
    public getConstFileLink(constName: string, containingNamespace?: string): FileLink<ConstStatement> {
        if (this.useFileCachesForFileLinkLookups) {
            return this.getFileLinkFromFileMap('constStatementMap', constName, containingNamespace);
        }
        const lowerName = constName?.toLowerCase();
        const fullNameLower = util.getFullyQualifiedClassName(lowerName, containingNamespace)?.toLowerCase();

        const constMap = this.getConstMap();

        let result = constMap.get(fullNameLower);
        //if we couldn't find the constant by its full namespaced name, look for a global constant with that name
        if (!result && lowerName !== fullNameLower) {
            result = constMap.get(lowerName);
        }
        return result;
    }

    public getAllFileLinks(name: string, containingNamespace?: string, includeNamespaces = false, includeNameShadowsOutsideNamespace = false): FileLink<Statement>[] {
        let links: FileLink<Statement>[] = [];

        links.push(this.getClassFileLink(name, containingNamespace),
            this.getInterfaceFileLink(name, containingNamespace),
            this.getConstFileLink(name, containingNamespace),
            this.getEnumFileLink(name, containingNamespace));
        if (includeNameShadowsOutsideNamespace && containingNamespace) {
            links.push(this.getClassFileLink(name),
                this.getInterfaceFileLink(name),
                this.getConstFileLink(name),
                this.getEnumFileLink(name));
        }
        if (includeNamespaces) {
            const nameSpaceContainer = this.getFirstNamespaceWithRoot(name, containingNamespace);
            if (nameSpaceContainer) {
                links.push({ item: nameSpaceContainer.namespaceStatements?.[0], file: nameSpaceContainer?.file as BrsFile });
            }
        }
        const fullNameLower = (containingNamespace ? `${containingNamespace}.${name}` : name).toLowerCase();
        const callable = this.getCallableByName(name);
        if (callable) {
            if ((!callable.hasNamespace && includeNameShadowsOutsideNamespace) || callable.getName(ParseMode.BrighterScript).toLowerCase() === fullNameLower) {
                // this callable has no namespace, or has same namespace
                links.push({ item: callable.functionStatement, file: callable.file as BrsFile });
            }
        }
        // remove empty links
        return links.filter(link => link);
    }

    /**
     * Get an TypeStatement and its containing file by the type name
     * @param typeName - The TypeStatement name, including the namespace of the type if possible
     * @param containingNamespace - The namespace used to resolve relative type names. (i.e. the namespace around the current statement trying to find a type)
     */
    public getTypeStatementFileLink(typeName: string, containingNamespace?: string): FileLink<TypeStatement> {
        const lowerName = typeName?.toLowerCase();
        const typeStatementMap = this.getTypeStatementMap();

        let typeStatement = typeStatementMap.get(
            util.getFullyQualifiedClassName(lowerName, containingNamespace?.toLowerCase())
        );
        //if we couldn't find the enum by its full namespaced name, look for a global enum with that name
        if (!typeStatement) {
            typeStatement = typeStatementMap.get(lowerName);
        }
        return typeStatement;
    }

    /**
     * Get a map of all enums by their member name.
     * The keys are lower-case fully-qualified paths to the enum and its member. For example:
     * namespace.enum.value
     */
    public getEnumMemberMap() {
        return this.cache.getOrAdd('enumMemberMap', () => {
            const result = new Map<string, EnumMemberStatement>();
            for (const [key, eenum] of this.getEnumMap()) {
                for (const member of eenum.item.getMembers()) {
                    result.set(`${key}.${member.name.toLowerCase()}`, member);
                }
            }
            return result;
        });
    }


    /**
     * Tests if a class exists with the specified name
     * @param className - the all-lower-case namespace-included class name
     * @param namespaceName - The namespace used to resolve relative class names. (i.e. the namespace around the current statement trying to find a class)
     */
    public hasClass(className: string, namespaceName?: string): boolean {
        return !!this.getClass(className, namespaceName);
    }

    /**
     * Tests if an interface exists with the specified name
     * @param ifaceName - the all-lower-case namespace-included interface name
     * @param namespaceName - the current namespace name
     */
    public hasInterface(ifaceName: string, namespaceName?: string): boolean {
        return !!this.getInterface(ifaceName, namespaceName);
    }

    /**
     * Tests if an enum exists with the specified name
     * @param enumName - the all-lower-case namespace-included enum name
     * @param namespaceName - the current namespace name
     */
    public hasEnum(enumName: string, namespaceName?: string): boolean {
        return !!this.getEnum(enumName, namespaceName);
    }

    public hasTypeStatementType(typeName: string, namespaceName?: string): boolean {
        return !!this.getTypeStatement(typeName, namespaceName);
    }

    /**
     * A dictionary of all classes in this scope. This includes namespaced classes always with their full name.
     * The key is stored in lower case
     */
    public getClassMap(): Map<string, FileLink<ClassStatement>> {
        return this.cache.getOrAdd('classMap', () => {
            const map = new Map<string, FileLink<ClassStatement>>();
            this.enumerateBrsFiles((file) => {
                if (isBrsFile(file)) {
                    for (let cls of file['_cachedLookups'].classStatements) {
                        const className = cls.getName(ParseMode.BrighterScript);
                        //only track classes with a defined name (i.e. exclude nameless malformed classes)
                        if (className) {
                            map.set(className.toLowerCase(), { item: cls, file: file });
                        }
                    }
                }
            });
            return map;
        });
    }

    /**
     * A dictionary of all Interfaces in this scope. This includes namespaced Interfaces always with their full name.
     * The key is stored in lower case
     */
    public getInterfaceMap(): Map<string, FileLink<InterfaceStatement>> {
        return this.cache.getOrAdd('interfaceMap', () => {
            const map = new Map<string, FileLink<InterfaceStatement>>();
            this.enumerateBrsFiles((file) => {
                if (isBrsFile(file)) {
                    for (let iface of file['_cachedLookups'].interfaceStatements) {
                        const ifaceName = iface.getName(ParseMode.BrighterScript);
                        //only track classes with a defined name (i.e. exclude nameless malformed classes)
                        if (ifaceName) {
                            map.set(ifaceName.toLowerCase(), { item: iface, file: file });
                        }
                    }
                }
            });
            return map;
        });
    }

    /**
     * A dictionary of all enums in this scope. This includes namespaced enums always with their full name.
     * The key is stored in lower case
     */
    public getEnumMap(): Map<string, FileLink<EnumStatement>> {
        return this.cache.getOrAdd('enumMap', () => {
            const map = new Map<string, FileLink<EnumStatement>>();
            this.enumerateBrsFiles((file) => {
                for (let enumStmt of file['_cachedLookups'].enumStatements) {
                    //only track enums with a defined name (i.e. exclude nameless malformed enums)
                    if (enumStmt.fullName) {
                        map.set(enumStmt.fullName.toLowerCase(), { item: enumStmt, file: file });
                    }
                }
            });
            return map;
        });
    }

    /**
     * A dictionary of all constants in this scope. This includes namespaced constants always with their full name.
     * The key is stored in lower case
     */
    public getConstMap(): Map<string, FileLink<ConstStatement>> {
        return this.cache.getOrAdd('constMap', () => {
            const map = new Map<string, FileLink<ConstStatement>>();
            this.enumerateBrsFiles((file) => {
                for (let stmt of file['_cachedLookups'].constStatements) {
                    //only track enums with a defined name (i.e. exclude nameless malformed enums)
                    if (stmt.fullName) {
                        map.set(stmt.fullName.toLowerCase(), { item: stmt, file: file });
                    }
                }
            });
            return map;
        });
    }

    /**
     * A dictionary of all TypeStatements in this scope. This includes namespaced types always with their full name.
     * The key is stored in lower case
     */
    public getTypeStatementMap(): Map<string, FileLink<TypeStatement>> {
        return this.cache.getOrAdd('typeStatementMap', () => {
            const map = new Map<string, FileLink<TypeStatement>>();
            this.enumerateBrsFiles((file) => {
                file.ast.walk((node) => {
                    if (isTypeStatement(node)) {
                        const namespaceName = node.findAncestor<NamespaceStatement>(isNamespaceStatement)?.getName(ParseMode.BrighterScript);
                        const typeName = node.tokens.name?.text;
                        if (typeName) {
                            const fullName = (namespaceName ? `${namespaceName}.${typeName}` : typeName).toLowerCase();
                            map.set(fullName, { item: node, file: file });
                        }
                    }
                }, { walkMode: WalkMode.visitAllRecursive });
            });
            return map;
        });
    }

    protected onDependenciesChanged(event: DependencyChangedEvent) {
        this.logDebug('invalidated because dependency graph said [', event.sourceKey, '] changed');
        this.invalidate();
    }

    /**
     * Clean up all event handles
     */
    public dispose() {
        this.unlinkSymbolTable();
        this.cache.clear();
        this.unsubscribeFromDependencyGraph?.();
    }

    /**
     * Does this scope know about the given namespace name?
     * @param namespaceName - the name of the namespace (i.e. "NameA", or "NameA.NameB", etc...)
     */
    public isKnownNamespace(namespaceName: string) {
        let namespaceNameLower = namespaceName.toLowerCase();
        this.enumerateBrsFiles((file) => {
            for (let namespace of file['_cachedLookups'].namespaceStatements) {
                let loopNamespaceNameLower = namespace.name.toLowerCase();
                if (loopNamespaceNameLower === namespaceNameLower || loopNamespaceNameLower.startsWith(namespaceNameLower + '.')) {
                    return true;
                }
            }
        });
        return false;
    }

    /**
     * Get the parent scope for this scope (for source scope this will always be the globalScope).
     * XmlScope overrides this to return the parent xml scope if available.
     * For globalScope this will return null.
     */
    public getParentScope(): Scope | null {
        let scope: Scope | undefined;
        //use the global scope if we didn't find a sope and this is not the global scope
        if (this.program.globalScope !== this) {
            scope = this.program.globalScope;
        }
        if (scope) {
            return scope;
        } else {
            //passing null to the cache allows it to skip the factory function in the future
            return null;
        }
    }

    private dependencyGraph: DependencyGraph;
    /**
     * An unsubscribe function for the dependencyGraph subscription
     */
    private unsubscribeFromDependencyGraph: () => void;

    public attachDependencyGraph(dependencyGraph: DependencyGraph) {
        this.dependencyGraph = dependencyGraph;
        if (this.unsubscribeFromDependencyGraph) {
            this.unsubscribeFromDependencyGraph();
        }

        //anytime a dependency for this scope changes, we need to be revalidated
        this.unsubscribeFromDependencyGraph = this.dependencyGraph.onchange(this.dependencyGraphKey, this.onDependenciesChanged.bind(this));

        //invalidate immediately since this is a new scope
        this.invalidate();
    }

    /**
     * Get the file from this scope with the given path.
     * @param filePath can be a srcPath or destPath
     * @param normalizePath should this function repair and standardize the path? Passing false should have a performance boost if you can guarantee your path is already sanitized
     */
    public getFile<TFile extends BscFile>(filePath: string, normalizePath = true) {
        if (typeof filePath !== 'string') {
            return undefined;
        }

        const key: keyof Pick<BscFile, 'srcPath' | 'destPath'> = path.isAbsolute(filePath) ? 'srcPath' : 'destPath';
        let map = this.cache.getOrAdd('fileMaps-srcPath', () => {
            const result = new Map<string, BscFile>();
            for (const file of this.getAllFiles()) {
                result.set(file[key].toLowerCase(), file);
            }
            return result;
        });
        return map.get(
            (normalizePath ? util.standardizePath(filePath) : filePath).toLowerCase()
        ) as TFile;
    }

    /**
     * Get the list of files referenced by this scope that are actually loaded in the program.
     * Excludes files from ancestor scopes
     */
    public getOwnFiles() {
        //source scope only inherits files from global, so just return all files. This function mostly exists to assist XmlScope
        return this.getAllFiles();
    }

    /**
     * Get the list of files referenced by this scope that are actually loaded in the program.
     * Includes files from this scope and all ancestor scopes
     */
    public getAllFiles(): BscFile[] {
        return this.cache.getOrAdd('getAllFiles', () => {
            let result = [] as BscFile[];
            let dependencies = this.dependencyGraph.getAllDependencies(this.dependencyGraphKey);
            for (let dependency of dependencies) {
                //load components by their name
                if (dependency.startsWith('component:')) {
                    let comp = this.program.getComponent(dependency.replace(/^component:/, ''));
                    if (comp) {
                        result.push(comp.file);
                    }
                } else {
                    let file = this.program.getFile(dependency);
                    if (file) {
                        result.push(file);
                    }
                }
            }
            return result;
        });
    }

    /**
     * Gets a list of all files in this scope, but not imported files, and not from ancestor scopes
     */
    public getImmediateFiles(): BscFile[] {
        return this.cache.getOrAdd('getImmediateFiles', () => {
            let result = [] as BscFile[];
            if (isXmlScope(this)) {
                result.push(this.xmlFile);
            }
            let dependencies = this.dependencyGraph.getImmediateDependencies(this.dependencyGraphKey);
            for (let dependency of dependencies) {
                //load components by their name
                if (dependency.startsWith('component:')) {
                    let comp = this.program.getComponent(dependency.replace(/^component:/, ''));
                    if (comp) {
                        result.push(...comp.scope.getImmediateFiles());
                        result.push(comp.file);
                    }
                } else {
                    let file = this.program.getFile(dependency);
                    if (file) {
                        result.push(file);
                    }
                }
            }
            this.logDebug('getImmediateFiles', () => result.map(x => x.destPath));
            return result;
        });
    }

    /**
     * Get the list of callables available in this scope (either declared in this scope or in a parent scope)
     */
    public getAllCallables(): CallableContainer[] {
        //get callables from parent scopes
        let parentScope = this.getParentScope();
        if (parentScope) {
            return [...this.getOwnCallables(), ...parentScope.getAllCallables()];
        } else {
            return [...this.getOwnCallables()];
        }
    }

    /**
     * Get the callable with the specified name.
     * If there are overridden callables with the same name, the closest callable to this scope is returned
     */
    public getCallableByName(name: string) {
        return this.getCallableMap().get(
            name.toLowerCase()
        );
    }

    public getCallableMap() {
        return this.cache.getOrAdd('callableMap', () => {
            const result = new Map<string, Callable>();
            for (let callable of this.getAllCallables()) {
                const callableName = callable.callable.getName(ParseMode.BrighterScript)?.toLowerCase();
                result.set(callableName, callable.callable);
                result.set(
                    // Split by `.` and check the last term to consider namespaces.
                    callableName.split('.').pop()?.toLowerCase(),
                    callable.callable
                );
            }
            return result;
        });
    }

    public getCallableContainerMap() {
        return this.cache.getOrAdd('callableContainerMap', () => {
            let callables = this.getAllCallables();

            //get a list of all callables, indexed by their lower case names
            return util.getCallableContainersByLowerName(callables);
        });
    }

    /**
     * Iterate over Brs files not shadowed by typedefs
     */
    public enumerateBrsFiles(callback: (file: BrsFile) => void) {
        const files = this.getAllFiles();
        for (const file of files) {
            //only brs files without a typedef
            if (isBrsFile(file) && !file.hasTypedef) {
                callback(file);
            }
        }
    }

    /**
     * Iterate over Brs files not shadowed by typedefs
     */
    public enumerateBrsFilesWithBreak(callback: (file: BrsFile) => boolean) {
        const files = this.getAllFiles();
        for (const file of files) {
            //only brs files without a typedef
            if (isBrsFile(file) && !file.hasTypedef) {
                if (callback(file)) {
                    break;
                }
            }
        }
    }

    /**
     * Call a function for each file directly included in this scope (excluding files found only in parent scopes).
     */
    public enumerateOwnFiles(callback: (file: BscFile) => void) {
        const files = this.getOwnFiles();
        for (const file of files) {
            //either XML components or files without a typedef
            if (isXmlFile(file) || (isBrsFile(file) && !file.hasTypedef)) {
                callback(file);
            }
        }
    }

    /**
     * Get the list of callables explicitly defined in files in this scope.
     * This excludes ancestor callables
     */
    public getOwnCallables(): CallableContainer[] {
        let result = [] as CallableContainer[];
        //get callables from own files
        this.enumerateOwnFiles((file) => {
            if (isBrsFile(file)) {
                for (let callable of file?.callables ?? []) {
                    result.push({
                        callable: callable,
                        scope: this
                    });
                }
            }
        });
        return result;
    }

    /**
     * Build the namespace lookup for this scope.
     *
     * The lookup is now backed by `ScopeNamespaceLookup`, which queries the
     * Program-level `getNamespaceContributors` map lazily on each
     * `.get(name)` call. Per-file contributions are pre-built by
     * `BrsFile.getNamespaceContributions` and shared across every scope that
     * pulls in the file, so single-contribution containers reuse the file's
     * pre-built statement collections and symbolTable instead of allocating
     * per-scope copies.
     *
     * The return type is `Map<string, NamespaceContainer>` for backward
     * compatibility with plugins that consume the public API.
     */
    public buildNamespaceLookup(): Map<string, NamespaceContainer> {
        return new ScopeNamespaceLookup(this);
    }

    public getAllNamespaceStatements() {
        let result = [] as NamespaceStatement[];
        this.enumerateBrsFiles((file) => {
            result.push(...file['_cachedLookups'].namespaceStatements);
        });
        return result;
    }

    protected logDebug(...args: any[]) {
        this.program.logger.debug(this._debugLogComponentName, ...args);
    }
    private _debugLogComponentName: string;


    public validationMetrics = {
        linkTime: 0,
        validationTime: 0
    };

    public shouldValidate(validationOptions: ScopeValidationOptions = { force: false }) {
        //if this scope is already validated, no need to revalidate
        if (this.isValidated === true && !validationOptions.force) {
            this.logDebug('validate(): already validated');
            return false;
        }

        if (!validationOptions.initialValidation && validationOptions.filesToBeValidatedInScopeContext?.size === 0) {
            // There was no need to validate this scope.
            (this as any).isValidated = true;
            return false;
        }
        return true;
    }


    public validate(validationOptions: ScopeValidationOptions = { force: false }) {
        this.validationMetrics = {
            linkTime: 0,
            validationTime: 0
        };

        //if this scope is already validated, no need to revalidate
        if (!this.shouldValidate(validationOptions)) {
            this.logDebug('validate(): already validated');
            // There was no need to validate this scope.
            (this as any).isValidated = true;
            return false;
        }

        this.useFileCachesForFileLinkLookups = !validationOptions.initialValidation;

        this.program.logger.time(LogLevel.debug, [this._debugLogComponentName, 'validate()'], () => {

            let parentScope = this.getParentScope();

            //validate our parent before we validate ourself
            if (parentScope && parentScope.isValidated === false) {
                this.logDebug('validate(): validating parent first');
                parentScope.validate(validationOptions);
            }

            //Since statements from files are shared across multiple scopes, we need to link those statements to the current scope

            let t0 = performance.now();
            this.linkSymbolTable();
            this.validationMetrics.linkTime = performance.now() - t0;
            const scopeValidateEvent = {
                program: this.program,
                scope: this,
                changedFiles: validationOptions?.changedFiles ?? [],
                changedSymbols: validationOptions?.changedSymbols
            };
            t0 = performance.now();
            this.program.plugins.emit('validateScope', scopeValidateEvent);
            this.validationMetrics.validationTime = performance.now() - t0;
            //unlink all symbol tables from this scope (so they don't accidentally stick around)
            this.unlinkSymbolTable();
            (this as any).isValidated = true;
        });
        for (let file of this.getAllFiles()) {
            validationOptions.filesToBeValidatedInScopeContext?.delete(file);
        }
        return true;
    }

    /**
     * Mark this scope as invalid, which means its `validate()` function needs to be called again before use.
     */
    public invalidate() {
        (this as any).isValidated = false;
        //clear out various lookups (they'll get regenerated on demand the next time they're requested)
        this.cache.clear();
    }

    public get symbolTable(): SymbolTable {
        return this.cache.getOrAdd('symbolTable', () => {
            const result = new SymbolTable(`Scope: '${this.name}'`, () => this.getParentScope()?.symbolTable);
            result.addSymbol('m', undefined, new AssociativeArrayType(), SymbolTypeFlag.runtime);
            for (let file of this.getOwnFiles()) {
                if (isBrsFile(file)) {
                    result.mergeSymbolTable(file.parser?.symbolTable);
                }
            }
            return result;
        });
    }

    /**
     * A list of functions that will be called whenever `unlinkSymbolTable` is called
     */
    private linkSymbolTableDisposables = [];

    private symbolsAddedDuringLinking: { symbolTable: SymbolTable; name: string; flags: number }[] = [];

    public get allNamespaceTypeTable() {
        return this._allNamespaceTypeTable;
    }

    private _allNamespaceTypeTable: SymbolTable;

    /**
     * Builds the current symbol table for the scope, by merging the tables for all the files in this scope.
     * Also links all file symbols tables to this new table
     * This will only rebuilt if the symbol table has not been built before
     *
     *  Tree of symbol tables:
     *  ```
     *  Global Scope Symbol Table
     *      -  Source Scope Symbol Table :: Aggregate Namespaces Symbol Table (Siblings)
     *          - File 1 Symbol Table
     *          - File 2 Symbol Table
     *      -  Component A Scope Symbol Table :: Aggregate Namespaces Symbol Table (Siblings)
     *          - File 1 Symbol Table
     *          - File 2 Symbol Table
     *      -  Component B Scope Symbol Table :: Aggregate Namespaces Symbol Table (Siblings)
     *          - File 1 Symbol Table
     *          - File 2 Symbol Table
     * ```
     */
    public linkSymbolTable() {
        SymbolTable.cacheVerifier.generateToken();
        this._allNamespaceTypeTable = new SymbolTable(`Scope NamespaceTypes ${this.name}`);
        for (const file of this.getAllFiles()) {
            if (isBrsFile(file)) {
                this.linkSymbolTableDisposables.push(
                    file.parser.symbolTable.pushParentProvider(() => this.symbolTable)
                );

                //link each NamespaceStatement's SymbolTable with the aggregate NamespaceLookup SymbolTable.
                //Leaf containers always have symbolTable populated, so the lookup below is safe; the null
                //guard tolerates edge cases like a namespace that failed to register.
                for (const namespace of file['_cachedLookups'].namespaceStatements) {
                    const namespaceNameLower = namespace.getName(ParseMode.BrighterScript).toLowerCase();
                    const aggregate = this.namespaceLookup.get(namespaceNameLower)?.symbolTable;
                    if (aggregate) {
                        this.linkSymbolTableDisposables.push(
                            namespace.getSymbolTable().addSibling(aggregate)
                        );
                    }
                }
            }
        }
        this.enumerateBrsFiles((file) => {
            const namespaceTypes = file.getNamespaceSymbolTable();

            this.linkSymbolTableDisposables.push(
                ...this._allNamespaceTypeTable.mergeNamespaceSymbolTables(namespaceTypes)
            );
        });
        this.linkSymbolTableDisposables.push(
            this.symbolTable.addSibling(this._allNamespaceTypeTable)
        );
    }

    public unlinkSymbolTable() {
        for (const symbolToRemove of this.symbolsAddedDuringLinking) {
            this.symbolTable.removeSymbol(symbolToRemove.name);
        }
        this.symbolsAddedDuringLinking = [];
        for (const dispose of this.linkSymbolTableDisposables) {
            dispose();
        }
        this.linkSymbolTableDisposables = [];

        this.cache.delete('namespaceLookup');
    }

    /**
     * Get the list of all script imports for this scope
     */
    public getOwnScriptImports() {
        let result = [] as FileReference[];
        this.enumerateOwnFiles((file) => {
            if (isBrsFile(file)) {
                result.push(...file.ownScriptImports);
            } else if (isXmlFile(file)) {
                result.push(...file.scriptTagImports);
            }
        });
        return result;
    }


    /**
     * Find the file with the specified relative path
     */
    public getFileByRelativePath(relativePath: string) {
        if (!relativePath) {
            return;
        }
        let files = this.getAllFiles();
        for (let file of files) {
            if (file.destPath.toLowerCase() === relativePath.toLowerCase()) {
                return file;
            }
        }
    }

    /**
     * Determine if this file is included in this scope (excluding parent scopes)
     */
    public hasFile(file: BscFile) {
        let files = this.getOwnFiles();
        let hasFile = files.includes(file);
        return hasFile;
    }

    /**
     * @param className - The name of the class (including namespace if possible)
     * @param callsiteNamespace - the name of the namespace where the call site resides (this is NOT the known namespace of the class).
     *                            This is used to help resolve non-namespaced class names that reside in the same namespac as the call site.
     */
    public getClassHierarchy(className: string, callsiteNamespace?: string) {
        let items = [] as FileLink<ClassStatement>[];
        let link = this.getClassFileLink(className, callsiteNamespace);
        while (link) {
            items.push(link);
            link = this.getClassFileLink(link.item.parentClassName?.getName()?.toLowerCase(), callsiteNamespace);
        }
        return items;
    }
}

/**
 * A single file's contribution to a namespace name. Cached on `BrsFile` and shared
 * across every scope that pulls in this file. The fields here are intrinsic to the
 * file's parsed AST, so they never need to be rebuilt per scope.
 *
 * Pure-intermediate contributions (a namespace name part that this file only references
 * as a dotted prefix, e.g. `A` from `namespace A.B`) carry only the structural fields
 * (file, fullName, lastPartName, nameRange). Leaf contributions populate the relevant
 * statement collections and the per-file `symbolTable`.
 *
 * The `symbolTable` here has no parent provider; sibling resolution in `linkSymbolTable`
 * does not walk into a sibling's parent, so the parent-provider plumbing was dead code.
 */
export interface NamespaceFileContribution {
    file: BrsFile;
    fullName: string;
    lastPartName: string;
    nameRange: Range;
    namespaceStatements?: NamespaceStatement[];
    statements?: Statement[];
    classStatements?: Record<string, ClassStatement>;
    functionStatements?: Record<string, FunctionStatement>;
    enumStatements?: Map<string, EnumStatement>;
    constStatements?: Map<string, ConstStatement>;
    symbolTable?: SymbolTable;
}

/**
 * A node in the per-scope namespace tree.
 *
 * `namespaces` is always allocated so parent-child wiring works for every container.
 * The remaining collections are lazily allocated by `buildNamespaceLookup` only when
 * a corresponding declaration is encountered, so pure-intermediate containers and
 * sparsely-populated leaves do not pay the cost of empty Maps/Records they will never use.
 *
 * Consumers must handle these fields being undefined.
 */
export interface NamespaceContainer {
    file: BscFile;
    fullName: string;
    nameRange: Range;
    lastPartName: string;
    namespaces?: Map<string, NamespaceContainer>;
    namespaceStatements?: NamespaceStatement[];
    statements?: Statement[];
    classStatements?: Record<string, ClassStatement>;
    functionStatements?: Record<string, FunctionStatement>;
    enumStatements?: Map<string, EnumStatement>;
    constStatements?: Map<string, ConstStatement>;
    symbolTable?: SymbolTable;
    //fields used by per-file namespace lookups; optional so scope-level callers don't have to populate them
    fullNameLower?: string;
    parentNameLower?: string;
    nameParts?: Identifier[];
    lastPartNameLower?: string;
    isTopLevel?: boolean;
}


rokucommunity / brighterscript / #15904

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous