Function getRecursiveBases

packages/language/src/utils.ts:203–241

(
    decl: DataModel | TypeDef,
    includeDelegate = true,
    documents?: LangiumDocuments,
    seen = new Set<DataModel | TypeDef>(),
)

 * Gets all base models and mixins of a data model or type def, recursively.
 */
export function getRecursiveBases(
    decl: DataModel | TypeDef,
    includeDelegate = true,
    documents?: LangiumDocuments,
    seen = new Set<DataModel | TypeDef>(),
): (TypeDef | DataModel)[] {
    const result: (TypeDef | DataModel)[] = [];
    if (seen.has(decl)) {
        return result;
    }
    seen.add(decl);
    const bases = [...decl.mixins, ...(isDataModel(decl) && decl.baseModel ? [decl.baseModel] : [])];
    bases.forEach((base) => {
        let baseDecl: TypeDef | DataModel | undefined;

        if (base.ref && (isTypeDef(base.ref) || isDataModel(base.ref))) {
            // base is already resolved
            baseDecl = base.ref;
        } else {
            // otherwise, search by name, in all imported documents if provided
            const declarations = documents
                ? getAllDeclarationsIncludingImports(documents, decl.$container)
                : decl.$container.declarations;

            baseDecl = declarations.find(
                (d): d is TypeDef | DataModel => (isTypeDef(d) || isDataModel(d)) && d.name === base.$refText,
            );
        }

        if (baseDecl) {
            if (!includeDelegate && isDelegateModel(baseDecl)) {
                return;
            }
            result.push(baseDecl);
            result.push(...getRecursiveBases(baseDecl, includeDelegate, documents, seen));
        }
    });
    return result;
}

/**
 * Gets `@@id` fields declared at the data model level (including search in base models)