@itwin/tree-widget-react 3.4.1 → 4.0.0-alpha.1

package/lib/cjs/tree-widget-react/components/trees/models-tree/internal/ModelsTreeIdsCache.js

@@ -1,385 +0,0 @@
-"use strict";
-/*---------------------------------------------------------------------------------------------
- * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
- * See LICENSE.md in the project root for license terms and full copyright notice.
- *--------------------------------------------------------------------------------------------*/
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ModelsTreeIdsCache = void 0;
-const rxjs_1 = require("rxjs");
-const core_bentley_1 = require("@itwin/core-bentley");
-const Utils_js_1 = require("../../common/Utils.js");
-/** @internal */
-class ModelsTreeIdsCache {
-    constructor(_queryExecutor, _hierarchyConfig) {
-        this._queryExecutor = _queryExecutor;
-        this._hierarchyConfig = _hierarchyConfig;
-        this._categoryElementCounts = new ModelCategoryElementsCountCache(async (input) => this.queryCategoryElementCounts(input));
-        this._modelKeyPaths = new Map();
-        this._subjectKeyPaths = new Map();
-        this._categoryKeyPaths = new Map();
-    }
-    [Symbol.dispose]() {
-        this._categoryElementCounts[Symbol.dispose]();
-    }
-    async *querySubjects() {
-        const subjectsQuery = `
-      SELECT
-        s.ECInstanceId id,
-        s.Parent.Id parentId,
-        (
-          SELECT m.ECInstanceId
-          FROM bis.GeometricModel3d m
-          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))
-            AND NOT m.IsPrivate
-            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)
-        ) targetPartitionId,
-        CASE
-          WHEN (
-            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL
-            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'
-          ) THEN 1
-          ELSE 0
-        END hideInHierarchy
-      FROM bis.Subject s
-    `;
-        for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
-            yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };
-        }
-    }
-    async *queryModels() {
-        const modelsQuery = `
-      SELECT p.ECInstanceId id, p.Parent.Id parentId
-      FROM bis.InformationPartitionElement p
-      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId
-      WHERE
-        NOT m.IsPrivate
-        ${this._hierarchyConfig.showEmptyModels ? "" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}
-    `;
-        for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
-            yield { id: row.id, parentId: row.parentId };
-        }
-    }
-    async getSubjectInfos() {
-        this._subjectInfos ??= (async () => {
-            const [subjectInfos, targetPartitionSubjects] = await Promise.all([
-                (async () => {
-                    const result = new Map();
-                    for await (const subject of this.querySubjects()) {
-                        const subjectInfo = {
-                            parentSubject: subject.parentId,
-                            hideInHierarchy: subject.hideInHierarchy,
-                            childSubjects: new Set(),
-                            childModels: new Set(),
-                        };
-                        if (subject.targetPartitionId) {
-                            subjectInfo.childModels.add(subject.targetPartitionId);
-                        }
-                        result.set(subject.id, subjectInfo);
-                    }
-                    return result;
-                })(),
-                (async () => {
-                    const result = new Map();
-                    for await (const model of this.queryModels()) {
-                        (0, Utils_js_1.pushToMap)(result, model.id, model.parentId);
-                    }
-                    return result;
-                })(),
-            ]);
-            for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {
-                if (parentSubjectId) {
-                    const parentSubjectInfo = subjectInfos.get(parentSubjectId);
-                    (0, core_bentley_1.assert)(!!parentSubjectInfo);
-                    parentSubjectInfo.childSubjects.add(subjectId);
-                }
-            }
-            for (const [partitionId, subjectIds] of targetPartitionSubjects) {
-                subjectIds.forEach((subjectId) => {
-                    const subjectInfo = subjectInfos.get(subjectId);
-                    (0, core_bentley_1.assert)(!!subjectInfo);
-                    subjectInfo.childModels.add(partitionId);
-                });
-            }
-            return subjectInfos;
-        })();
-        return this._subjectInfos;
-    }
-    /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */
-    async getParentSubjectIds() {
-        this._parentSubjectIds ??= (async () => {
-            const subjectInfos = await this.getSubjectInfos();
-            const parentSubjectIds = new Set();
-            subjectInfos.forEach((subjectInfo, subjectId) => {
-                if (subjectInfo.childModels.size > 0) {
-                    parentSubjectIds.add(subjectId);
-                    let currParentId = subjectInfo.parentSubject;
-                    while (currParentId) {
-                        parentSubjectIds.add(currParentId);
-                        currParentId = subjectInfos.get(currParentId)?.parentSubject;
-                    }
-                }
-            });
-            return [...parentSubjectIds];
-        })();
-        return this._parentSubjectIds;
-    }
-    /**
-     * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into
-     * account `hideInHierarchy` flag.
-     */
-    async getChildSubjectIds(parentSubjectIds) {
-        const childSubjectIds = new Array();
-        const subjectInfos = await this.getSubjectInfos();
-        parentSubjectIds.forEach((subjectId) => {
-            forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {
-                if (!childSubjectInfo.hideInHierarchy) {
-                    childSubjectIds.push(childSubjectId);
-                    return "break";
-                }
-                return "continue";
-            });
-        });
-        return childSubjectIds;
-    }
-    /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */
-    async getSubjectModelIds(subjectIds) {
-        const subjectInfos = await this.getSubjectInfos();
-        const subjectStack = [...subjectIds];
-        const result = new Array();
-        while (true) {
-            const subjectId = subjectStack.pop();
-            if (subjectId === undefined) {
-                break;
-            }
-            const subjectInfo = subjectInfos.get(subjectId);
-            if (!subjectInfo) {
-                continue;
-            }
-            result.push(...subjectInfo.childModels);
-            subjectStack.push(...subjectInfo.childSubjects);
-        }
-        return result;
-    }
-    /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */
-    async getChildSubjectModelIds(parentSubjectIds) {
-        const subjectInfos = await this.getSubjectInfos();
-        const hiddenSubjectIds = new Array();
-        parentSubjectIds.forEach((subjectId) => {
-            forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {
-                if (childSubjectInfo.hideInHierarchy) {
-                    hiddenSubjectIds.push(childSubjectId);
-                    return "continue";
-                }
-                return "break";
-            });
-        });
-        const modelIds = new Array();
-        [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {
-            const subjectInfo = subjectInfos.get(subjectId);
-            subjectInfo && modelIds.push(...subjectInfo.childModels);
-        });
-        return modelIds;
-    }
-    async createSubjectInstanceKeysPath(targetSubjectId) {
-        let entry = this._subjectKeyPaths.get(targetSubjectId);
-        if (!entry) {
-            entry = (async () => {
-                const subjectInfos = await this.getSubjectInfos();
-                const result = new Array();
-                let currParentId = targetSubjectId;
-                while (currParentId) {
-                    const parentInfo = subjectInfos.get(currParentId);
-                    if (!parentInfo?.hideInHierarchy) {
-                        result.push({ className: "BisCore.Subject", id: currParentId });
-                    }
-                    currParentId = parentInfo?.parentSubject;
-                }
-                return result.reverse();
-            })();
-            this._subjectKeyPaths.set(targetSubjectId, entry);
-        }
-        return entry;
-    }
-    async *queryModelElementCounts() {
-        const query = /* sql */ `
-      SELECT Model.Id modelId, COUNT(*) elementCount
-      FROM ${this._hierarchyConfig.elementClassSpecification}
-      GROUP BY Model.Id
-    `;
-        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
-            yield { modelId: row.modelId, elementCount: row.elementCount };
-        }
-    }
-    async *queryModelCategories() {
-        const query = /* sql */ `
-      SELECT Model.Id modelId, Category.Id categoryId
-      FROM ${this._hierarchyConfig.elementClassSpecification}
-      WHERE Parent.Id IS NULL
-      GROUP BY modelId, categoryId
-    `;
-        for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
-            yield { modelId: row.modelId, categoryId: row.categoryId };
-        }
-    }
-    async getModelInfos() {
-        this._modelInfos ??= (async () => {
-            const modelInfos = new Map();
-            await Promise.all([
-                (async () => {
-                    for await (const { modelId, categoryId } of this.queryModelCategories()) {
-                        const entry = modelInfos.get(modelId);
-                        if (entry) {
-                            entry.categories.add(categoryId);
-                        }
-                        else {
-                            modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });
-                        }
-                    }
-                })(),
-                (async () => {
-                    for await (const { modelId, elementCount } of this.queryModelElementCounts()) {
-                        const entry = modelInfos.get(modelId);
-                        if (entry) {
-                            entry.elementCount = elementCount;
-                        }
-                        else {
-                            modelInfos.set(modelId, { categories: new Set(), elementCount });
-                        }
-                    }
-                })(),
-            ]);
-            return modelInfos;
-        })();
-        return this._modelInfos;
-    }
-    async getModelCategories(modelId) {
-        const modelInfos = await this.getModelInfos();
-        const categories = modelInfos.get(modelId)?.categories;
-        return categories ? [...categories] : [];
-    }
-    async getModelElementCount(modelId) {
-        const modelInfos = await this.getModelInfos();
-        return modelInfos.get(modelId)?.elementCount ?? 0;
-    }
-    async createModelInstanceKeyPaths(modelId) {
-        let entry = this._modelKeyPaths.get(modelId);
-        if (!entry) {
-            entry = (async () => {
-                const result = new Array();
-                const subjectInfos = (await this.getSubjectInfos()).entries();
-                for (const [modelSubjectId, subjectInfo] of subjectInfos) {
-                    if (subjectInfo.childModels.has(modelId)) {
-                        const subjectPath = await this.createSubjectInstanceKeysPath(modelSubjectId);
-                        result.push([...subjectPath, { className: "BisCore.GeometricModel3d", id: modelId }]);
-                    }
-                }
-                return result;
-            })();
-            this._modelKeyPaths.set(modelId, entry);
-        }
-        return entry;
-    }
-    async queryCategoryElementCounts(input) {
-        const reader = this._queryExecutor.createQueryReader({
-            ctes: [
-                /* sql */ `
-            CategoryElements(id, modelId, categoryId) AS (
-              SELECT ECInstanceId, Model.Id, Category.Id
-              FROM ${this._hierarchyConfig.elementClassSpecification}
-              WHERE
-                Parent.Id IS NULL
-                AND (
-                  ${input.map(({ modelId, categoryId }) => `Model.Id = ${modelId} AND Category.Id = ${categoryId}`).join(" OR ")}
-                )
-
-              UNION ALL
-
-              SELECT c.ECInstanceId, p.modelId, p.categoryId
-              FROM ${this._hierarchyConfig.elementClassSpecification} c
-              JOIN CategoryElements p ON c.Parent.Id = p.id
-            )
-          `,
-            ],
-            ecsql: `
-          SELECT modelId, categoryId, COUNT(id) elementsCount
-          FROM CategoryElements
-          GROUP BY modelId, categoryId
-        `,
-        }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" });
-        const result = new Array();
-        for await (const row of reader) {
-            result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });
-        }
-        return result;
-    }
-    async getCategoryElementsCount(modelId, categoryId) {
-        return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);
-    }
-    async createCategoryInstanceKeyPaths(categoryId) {
-        let entry = this._categoryKeyPaths.get(categoryId);
-        if (!entry) {
-            entry = (async () => {
-                const result = new Set();
-                const modelInfos = await this.getModelInfos();
-                modelInfos?.forEach((modelInfo, modelId) => {
-                    if (modelInfo.categories.has(categoryId)) {
-                        result.add(modelId);
-                    }
-                });
-                const categoryPaths = new Array();
-                for (const categoryModelId of [...result]) {
-                    const modelPaths = await this.createModelInstanceKeyPaths(categoryModelId);
-                    for (const modelPath of modelPaths) {
-                        categoryPaths.push([...modelPath, { className: "BisCore.SpatialCategory", id: categoryId }]);
-                    }
-                }
-                return categoryPaths;
-            })();
-            this._categoryKeyPaths.set(categoryId, entry);
-        }
-        return entry;
-    }
-}
-exports.ModelsTreeIdsCache = ModelsTreeIdsCache;
-function forEachChildSubject(subjectInfos, parentSubject, cb) {
-    const parentSubjectInfo = typeof parentSubject === "string" ? subjectInfos.get(parentSubject) : parentSubject;
-    parentSubjectInfo &&
-        parentSubjectInfo.childSubjects.forEach((childSubjectId) => {
-            const childSubjectInfo = subjectInfos.get(childSubjectId);
-            if (cb(childSubjectId, childSubjectInfo) === "break") {
-                return;
-            }
-            forEachChildSubject(subjectInfos, childSubjectInfo, cb);
-        });
-}
-class ModelCategoryElementsCountCache {
-    constructor(_loader) {
-        this._loader = _loader;
-        this._cache = new Map();
-        this._requestsStream = new rxjs_1.Subject();
-        this._subscription = this._requestsStream
-            .pipe((0, rxjs_1.bufferTime)(20), (0, rxjs_1.filter)((requests) => requests.length > 0), (0, rxjs_1.mergeMap)(async (requests) => this._loader(requests)), (0, rxjs_1.mergeAll)())
-            .subscribe({
-            next: ({ modelId, categoryId, elementsCount }) => {
-                const subject = this._cache.get(`${modelId}${categoryId}`);
-                (0, core_bentley_1.assert)(!!subject);
-                subject.next(elementsCount);
-            },
-        });
-    }
-    [Symbol.dispose]() {
-        this._subscription.unsubscribe();
-    }
-    async getCategoryElementsCount(modelId, categoryId) {
-        const cacheKey = `${modelId}${categoryId}`;
-        let result = this._cache.get(cacheKey);
-        if (result !== undefined) {
-            return (0, rxjs_1.firstValueFrom)(result);
-        }
-        result = new rxjs_1.ReplaySubject(1);
-        this._cache.set(cacheKey, result);
-        this._requestsStream.next({ modelId, categoryId });
-        return (0, rxjs_1.firstValueFrom)(result);
-    }
-}
-//# sourceMappingURL=ModelsTreeIdsCache.js.map

package/lib/cjs/tree-widget-react/components/trees/models-tree/internal/ModelsTreeIdsCache.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ModelsTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/ModelsTreeIdsCache.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAGhG,+BAAsG;AACtG,sDAA6C;AAC7C,oDAAkD;AAqBlD,gBAAgB;AAChB,MAAa,kBAAkB;IAS7B,YACU,cAA0C,EAC1C,gBAAkD;QADlD,mBAAc,GAAd,cAAc,CAA4B;QAC1C,qBAAgB,GAAhB,gBAAgB,CAAkC;QAE1D,IAAI,CAAC,sBAAsB,GAAG,IAAI,+BAA+B,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,KAAK,CAAC,CAAC,CAAC;QAC3H,IAAI,CAAC,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,IAAI,GAAG,EAAE,CAAC;QAClC,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,EAAE,CAAC;IACrC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,CAAC,aAAa;QAC1B,MAAM,aAAa,GAAG;;;;;;;;;wCASc,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;;;;;;;KAUlF,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAChJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,iBAAiB,EAAE,GAAG,CAAC,iBAAiB,EAAE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;SAChI;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,WAAW;QACxB,MAAM,WAAW,GAAG;;;;;;UAMd,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,6BAA6B,IAAI,CAAC,gBAAgB,CAAC,yBAAyB,mCAAmC;KACjK,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAC9I,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;SAC9C;IACH,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,aAAa,KAAK,CAAC,KAAK,IAAI,EAAE;YACjC,MAAM,CAAC,YAAY,EAAE,uBAAuB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChE,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;oBAClD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,aAAa,EAAE,EAAE;wBAChD,MAAM,WAAW,GAAgB;4BAC/B,aAAa,EAAE,OAAO,CAAC,QAAQ;4BAC/B,eAAe,EAAE,OAAO,CAAC,eAAe;4BACxC,aAAa,EAAE,IAAI,GAAG,EAAE;4BACxB,WAAW,EAAE,IAAI,GAAG,EAAE;yBACvB,CAAC;wBACF,IAAI,OAAO,CAAC,iBAAiB,EAAE;4BAC7B,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;yBACxD;wBACD,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,WAAW,CAAC,CAAC;qBACrC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,MAAM,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;oBACtD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,WAAW,EAAE,EAAE;wBAC5C,IAAA,oBAAS,EAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;qBAC7C;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YAEH,KAAK,MAAM,CAAC,SAAS,EAAE,EAAE,aAAa,EAAE,eAAe,EAAE,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,EAAE;gBACpF,IAAI,eAAe,EAAE;oBACnB,MAAM,iBAAiB,GAAG,YAAY,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;oBAC5D,IAAA,qBAAM,EAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC;oBAC5B,iBAAiB,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;iBAChD;aACF;YAED,KAAK,MAAM,CAAC,WAAW,EAAE,UAAU,CAAC,IAAI,uBAAuB,EAAE;gBAC/D,UAAU,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;oBAC/B,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChD,IAAA,qBAAM,EAAC,CAAC,CAAC,WAAW,CAAC,CAAC;oBACtB,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;gBAC3C,CAAC,CAAC,CAAC;aACJ;YAED,OAAO,YAAY,CAAC;QACtB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,aAAa,CAAC;IAC5B,CAAC;IAED,kHAAkH;IAC3G,KAAK,CAAC,mBAAmB;QAC9B,IAAI,CAAC,iBAAiB,KAAK,CAAC,KAAK,IAAI,EAAE;YACrC,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;YAClD,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;YAC/C,YAAY,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,SAAS,EAAE,EAAE;gBAC9C,IAAI,WAAW,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE;oBACpC,gBAAgB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAChC,IAAI,YAAY,GAAG,WAAW,CAAC,aAAa,CAAC;oBAC7C,OAAO,YAAY,EAAE;wBACnB,gBAAgB,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;wBACnC,YAAY,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE,aAAa,CAAC;qBAC9D;iBACF;YACH,CAAC,CAAC,CAAC;YACH,OAAO,CAAC,GAAG,gBAAgB,CAAC,CAAC;QAC/B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAChC,CAAC;IAED;;;OAGG;IACI,KAAK,CAAC,kBAAkB,CAAC,gBAA8B;QAC5D,MAAM,eAAe,GAAG,IAAI,KAAK,EAAc,CAAC;QAChD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,CAAC,gBAAgB,CAAC,eAAe,EAAE;oBACrC,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACrC,OAAO,OAAO,CAAC;iBAChB;gBACD,OAAO,UAAU,CAAC;YACpB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IACzB,CAAC;IAED,+GAA+G;IACxG,KAAK,CAAC,kBAAkB,CAAC,UAAqB;QACnD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAClD,MAAM,YAAY,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,OAAO,IAAI,EAAE;YACX,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,EAAE,CAAC;YACrC,IAAI,SAAS,KAAK,SAAS,EAAE;gBAC3B,MAAM;aACP;YACD,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,IAAI,CAAC,WAAW,EAAE;gBAChB,SAAS;aACV;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;YACxC,YAAY,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,aAAa,CAAC,CAAC;SACjD;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6GAA6G;IACtG,KAAK,CAAC,uBAAuB,CAAC,gBAA8B;QACjE,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;QAElD,MAAM,gBAAgB,GAAG,IAAI,KAAK,EAAc,CAAC;QACjD,gBAAgB,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YACrC,mBAAmB,CAAC,YAAY,EAAE,SAAS,EAAE,CAAC,cAAc,EAAE,gBAAgB,EAAE,EAAE;gBAChF,IAAI,gBAAgB,CAAC,eAAe,EAAE;oBACpC,gBAAgB,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBACtC,OAAO,UAAU,CAAC;iBACnB;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,KAAK,EAAc,CAAC;QACzC,CAAC,GAAG,gBAAgB,EAAE,GAAG,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAChD,WAAW,IAAI,QAAQ,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEM,KAAK,CAAC,6BAA6B,CAAC,eAA2B;QACpE,IAAI,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;QACvD,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC;gBAClD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAe,CAAC;gBACxC,IAAI,YAAY,GAA2B,eAAe,CAAC;gBAC3D,OAAO,YAAY,EAAE;oBACnB,MAAM,UAAU,GAAG,YAAY,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;oBAClD,IAAI,CAAC,UAAU,EAAE,eAAe,EAAE;wBAChC,MAAM,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;qBACjE;oBACD,YAAY,GAAG,UAAU,EAAE,aAAa,CAAC;iBAC1C;gBACD,OAAO,MAAM,CAAC,OAAO,EAAE,CAAC;YAC1B,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,eAAe,EAAE,KAAK,CAAC,CAAC;SACnD;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,uBAAuB;QACpC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;KAEvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,YAAY,EAAE,GAAG,CAAC,YAAY,EAAE,CAAC;SAChE;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,oBAAoB;QACjC,MAAM,KAAK,GAAG,SAAS,CAAC;;aAEf,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;KAGvD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACxI,MAAM,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC5D;IACH,CAAC;IAEO,KAAK,CAAC,aAAa;QACzB,IAAI,CAAC,WAAW,KAAK,CAAC,KAAK,IAAI,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAI,GAAG,EAA6D,CAAC;YACxF,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,IAAI,CAAC,oBAAoB,EAAE,EAAE;wBACvE,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;yBAClC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC;yBACjF;qBACF;gBACH,CAAC,CAAC,EAAE;gBACJ,CAAC,KAAK,IAAI,EAAE;oBACV,IAAI,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,YAAY,EAAE,IAAI,IAAI,CAAC,uBAAuB,EAAE,EAAE;wBAC5E,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;wBACtC,IAAI,KAAK,EAAE;4BACT,KAAK,CAAC,YAAY,GAAG,YAAY,CAAC;yBACnC;6BAAM;4BACL,UAAU,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,GAAG,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;yBAClE;qBACF;gBACH,CAAC,CAAC,EAAE;aACL,CAAC,CAAC;YACH,OAAO,UAAU,CAAC;QACpB,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAEM,KAAK,CAAC,kBAAkB,CAAC,OAAmB;QACjD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,MAAM,UAAU,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,UAAU,CAAC;QACvD,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3C,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,OAAmB;QACnD,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC9C,OAAO,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,YAAY,IAAI,CAAC,CAAC;IACpD,CAAC;IAEM,KAAK,CAAC,2BAA2B,CAAC,OAAmB;QAC1D,IAAI,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC7C,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,KAAK,EAAgC,CAAC;gBACzD,MAAM,YAAY,GAAG,CAAC,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9D,KAAK,MAAM,CAAC,cAAc,EAAE,WAAW,CAAC,IAAI,YAAY,EAAE;oBACxD,IAAI,WAAW,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;wBACxC,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,6BAA6B,CAAC,cAAc,CAAC,CAAC;wBAC7E,MAAM,CAAC,IAAI,CAAC,CAAC,GAAG,WAAW,EAAE,EAAE,SAAS,EAAE,0BAA0B,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;qBACvF;iBACF;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC,CAAC,EAAE,CAAC;YAEL,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;SACzC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,0BAA0B,CACtC,KAA6D;QAE7D,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAClD;YACE,IAAI,EAAE;gBACJ,SAAS,CAAC;;;qBAGC,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;;oBAIhD,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,cAAc,OAAO,sBAAsB,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;;;;;;qBAM3G,IAAI,CAAC,gBAAgB,CAAC,yBAAyB;;;WAGzD;aACF;YACD,KAAK,EAAE;;;;SAIN;SACF,EACD,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CACxD,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,KAAK,EAAkE,CAAC;QAC3F,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE;YAC9B,MAAM,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,aAAa,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;SACrG;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,OAAO,IAAI,CAAC,sBAAsB,CAAC,wBAAwB,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,KAAK,CAAC,8BAA8B,CAAC,UAAsB;QAChE,IAAI,KAAK,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,CAAC,KAAK,IAAI,EAAE;gBAClB,MAAM,MAAM,GAAG,IAAI,GAAG,EAAc,CAAC;gBACrC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC9C,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,EAAE,OAAO,EAAE,EAAE;oBACzC,IAAI,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;wBACxC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;qBACrB;gBACH,CAAC,CAAC,CAAC;gBAEH,MAAM,aAAa,GAAG,IAAI,KAAK,EAAgC,CAAC;gBAChE,KAAK,MAAM,eAAe,IAAI,CAAC,GAAG,MAAM,CAAC,EAAE;oBACzC,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,eAAe,CAAC,CAAC;oBAC3E,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE;wBAClC,aAAa,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,EAAE,EAAE,SAAS,EAAE,yBAAyB,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;qBAC9F;iBACF;gBACD,OAAO,aAAa,CAAC;YACvB,CAAC,CAAC,EAAE,CAAC;YACL,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;SAC/C;QACD,OAAO,KAAK,CAAC;IACf,CAAC;CACF;AAhXD,gDAgXC;AAED,SAAS,mBAAmB,CAC1B,YAA0C,EAC1C,aAAuC,EACvC,EAAuF;IAEvF,MAAM,iBAAiB,GAAG,OAAO,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC;IAC9G,iBAAiB;QACf,iBAAiB,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACzD,MAAM,gBAAgB,GAAG,YAAY,CAAC,GAAG,CAAC,cAAc,CAAE,CAAC;YAC3D,IAAI,EAAE,CAAC,cAAc,EAAE,gBAAgB,CAAC,KAAK,OAAO,EAAE;gBACpD,OAAO;aACR;YACD,mBAAmB,CAAC,YAAY,EAAE,gBAAgB,EAAE,EAAE,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;AACP,CAAC;AAED,MAAM,+BAA+B;IAKnC,YACU,OAE2E;QAF3E,YAAO,GAAP,OAAO,CAEoE;QAP7E,WAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;QAC5C,oBAAe,GAAG,IAAI,cAAO,EAAmD,CAAC;QAQvF,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,eAAe;aACtC,IAAI,CACH,IAAA,iBAAU,EAAC,EAAE,CAAC,EACd,IAAA,aAAM,EAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,EACzC,IAAA,eAAQ,EAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,EACpD,IAAA,eAAQ,GAAE,CACX;aACA,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC,CAAC;gBAC3D,IAAA,qBAAM,EAAC,CAAC,CAAC,OAAO,CAAC,CAAC;gBAClB,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC;SACF,CAAC,CAAC;IACP,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,aAAa,CAAC,WAAW,EAAE,CAAC;IACnC,CAAC;IAEM,KAAK,CAAC,wBAAwB,CAAC,OAAmB,EAAE,UAAsB;QAC/E,MAAM,QAAQ,GAAG,GAAG,OAAO,GAAG,UAAU,EAAE,CAAC;QAC3C,IAAI,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,OAAO,IAAA,qBAAc,EAAC,MAAM,CAAC,CAAC;SAC/B;QAED,MAAM,GAAG,IAAI,oBAAa,CAAC,CAAC,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAClC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;QACnD,OAAO,IAAA,qBAAc,EAAC,MAAM,CAAC,CAAC;IAChC,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport type { Subscription } from \"rxjs\";\nimport { bufferTime, filter, firstValueFrom, mergeAll, mergeMap, ReplaySubject, Subject } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { pushToMap } from \"../../common/Utils.js\";\n\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\nimport type { ModelsTreeDefinition } from \"../ModelsTreeDefinition.js\";\nimport type { Id64Array, Id64Set, Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\n\ninterface SubjectInfo {\n  parentSubject: Id64String | undefined;\n  hideInHierarchy: boolean;\n  childSubjects: Id64Set;\n  childModels: Id64Set;\n}\n\ninterface ModelInfo {\n  categories: Id64Set;\n  elementCount: number;\n}\n\ntype ModelsTreeHierarchyConfiguration = ConstructorParameters<typeof ModelsTreeDefinition>[0][\"hierarchyConfig\"];\n\n/** @internal */\nexport class ModelsTreeIdsCache {\n  private readonly _categoryElementCounts: ModelCategoryElementsCountCache;\n  private _subjectInfos: Promise<Map<Id64String, SubjectInfo>> | undefined;\n  private _parentSubjectIds: Promise<Id64Array> | undefined; // the list should contain a subject id if its node should be shown as having children\n  private _modelInfos: Promise<Map<Id64String, ModelInfo>> | undefined;\n  private _modelKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n  private _subjectKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath>>;\n  private _categoryKeyPaths: Map<Id64String, Promise<HierarchyNodeIdentifiersPath[]>>;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    private _hierarchyConfig: ModelsTreeHierarchyConfiguration,\n  ) {\n    this._categoryElementCounts = new ModelCategoryElementsCountCache(async (input) => this.queryCategoryElementCounts(input));\n    this._modelKeyPaths = new Map();\n    this._subjectKeyPaths = new Map();\n    this._categoryKeyPaths = new Map();\n  }\n\n  public [Symbol.dispose]() {\n    this._categoryElementCounts[Symbol.dispose]();\n  }\n\n  private async *querySubjects(): AsyncIterableIterator<{ id: Id64String; parentId?: Id64String; targetPartitionId?: Id64String; hideInHierarchy: boolean }> {\n    const subjectsQuery = `\n      SELECT\n        s.ECInstanceId id,\n        s.Parent.Id parentId,\n        (\n          SELECT m.ECInstanceId\n          FROM bis.GeometricModel3d m\n          WHERE m.ECInstanceId = HexToId(json_extract(s.JsonProperties, '$.Subject.Model.TargetPartition'))\n            AND NOT m.IsPrivate\n            AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)\n        ) targetPartitionId,\n        CASE\n          WHEN (\n            json_extract(s.JsonProperties, '$.Subject.Job.Bridge') IS NOT NULL\n            OR json_extract(s.JsonProperties, '$.Subject.Model.Type') = 'Hierarchy'\n          ) THEN 1\n          ELSE 0\n        END hideInHierarchy\n      FROM bis.Subject s\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: subjectsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId, targetPartitionId: row.targetPartitionId, hideInHierarchy: !!row.hideInHierarchy };\n    }\n  }\n\n  private async *queryModels(): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const modelsQuery = `\n      SELECT p.ECInstanceId id, p.Parent.Id parentId\n      FROM bis.InformationPartitionElement p\n      INNER JOIN bis.GeometricModel3d m ON m.ModeledElement.Id = p.ECInstanceId\n      WHERE\n        NOT m.IsPrivate\n        ${this._hierarchyConfig.showEmptyModels ? \"\" : `AND EXISTS (SELECT 1 FROM ${this._hierarchyConfig.elementClassSpecification} WHERE Model.Id = m.ECInstanceId)`}\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: modelsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.parentId };\n    }\n  }\n\n  private async getSubjectInfos() {\n    this._subjectInfos ??= (async () => {\n      const [subjectInfos, targetPartitionSubjects] = await Promise.all([\n        (async () => {\n          const result = new Map<Id64String, SubjectInfo>();\n          for await (const subject of this.querySubjects()) {\n            const subjectInfo: SubjectInfo = {\n              parentSubject: subject.parentId,\n              hideInHierarchy: subject.hideInHierarchy,\n              childSubjects: new Set(),\n              childModels: new Set(),\n            };\n            if (subject.targetPartitionId) {\n              subjectInfo.childModels.add(subject.targetPartitionId);\n            }\n            result.set(subject.id, subjectInfo);\n          }\n          return result;\n        })(),\n        (async () => {\n          const result = new Map<Id64String, Set<Id64String>>();\n          for await (const model of this.queryModels()) {\n            pushToMap(result, model.id, model.parentId);\n          }\n          return result;\n        })(),\n      ]);\n\n      for (const [subjectId, { parentSubject: parentSubjectId }] of subjectInfos.entries()) {\n        if (parentSubjectId) {\n          const parentSubjectInfo = subjectInfos.get(parentSubjectId);\n          assert(!!parentSubjectInfo);\n          parentSubjectInfo.childSubjects.add(subjectId);\n        }\n      }\n\n      for (const [partitionId, subjectIds] of targetPartitionSubjects) {\n        subjectIds.forEach((subjectId) => {\n          const subjectInfo = subjectInfos.get(subjectId);\n          assert(!!subjectInfo);\n          subjectInfo.childModels.add(partitionId);\n        });\n      }\n\n      return subjectInfos;\n    })();\n    return this._subjectInfos;\n  }\n\n  /** Returns ECInstanceIDs of Subjects that either have direct Model or at least one child Subject with a Model. */\n  public async getParentSubjectIds(): Promise<Id64String[]> {\n    this._parentSubjectIds ??= (async () => {\n      const subjectInfos = await this.getSubjectInfos();\n      const parentSubjectIds = new Set<Id64String>();\n      subjectInfos.forEach((subjectInfo, subjectId) => {\n        if (subjectInfo.childModels.size > 0) {\n          parentSubjectIds.add(subjectId);\n          let currParentId = subjectInfo.parentSubject;\n          while (currParentId) {\n            parentSubjectIds.add(currParentId);\n            currParentId = subjectInfos.get(currParentId)?.parentSubject;\n          }\n        }\n      });\n      return [...parentSubjectIds];\n    })();\n    return this._parentSubjectIds;\n  }\n\n  /**\n   * Returns child subjects of the specified parent subjects as they're displayed in the hierarchy - taking into\n   * account `hideInHierarchy` flag.\n   */\n  public async getChildSubjectIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const childSubjectIds = new Array<Id64String>();\n    const subjectInfos = await this.getSubjectInfos();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (!childSubjectInfo.hideInHierarchy) {\n          childSubjectIds.push(childSubjectId);\n          return \"break\";\n        }\n        return \"continue\";\n      });\n    });\n    return childSubjectIds;\n  }\n\n  /** Returns ECInstanceIDs of all Models under specific parent Subjects, including their child Subjects, etc. */\n  public async getSubjectModelIds(subjectIds: Id64Array): Promise<Id64Array> {\n    const subjectInfos = await this.getSubjectInfos();\n    const subjectStack = [...subjectIds];\n    const result = new Array<Id64String>();\n    while (true) {\n      const subjectId = subjectStack.pop();\n      if (subjectId === undefined) {\n        break;\n      }\n      const subjectInfo = subjectInfos.get(subjectId);\n      if (!subjectInfo) {\n        continue;\n      }\n      result.push(...subjectInfo.childModels);\n      subjectStack.push(...subjectInfo.childSubjects);\n    }\n    return result;\n  }\n\n  /** Returns ECInstanceIDs of Models under specific parent Subjects as they are displayed in the hierarchy. */\n  public async getChildSubjectModelIds(parentSubjectIds: Id64String[]): Promise<Id64String[]> {\n    const subjectInfos = await this.getSubjectInfos();\n\n    const hiddenSubjectIds = new Array<Id64String>();\n    parentSubjectIds.forEach((subjectId) => {\n      forEachChildSubject(subjectInfos, subjectId, (childSubjectId, childSubjectInfo) => {\n        if (childSubjectInfo.hideInHierarchy) {\n          hiddenSubjectIds.push(childSubjectId);\n          return \"continue\";\n        }\n        return \"break\";\n      });\n    });\n\n    const modelIds = new Array<Id64String>();\n    [...parentSubjectIds, ...hiddenSubjectIds].forEach((subjectId) => {\n      const subjectInfo = subjectInfos.get(subjectId);\n      subjectInfo && modelIds.push(...subjectInfo.childModels);\n    });\n    return modelIds;\n  }\n\n  public async createSubjectInstanceKeysPath(targetSubjectId: Id64String): Promise<HierarchyNodeIdentifiersPath> {\n    let entry = this._subjectKeyPaths.get(targetSubjectId);\n    if (!entry) {\n      entry = (async () => {\n        const subjectInfos = await this.getSubjectInfos();\n        const result = new Array<InstanceKey>();\n        let currParentId: Id64String | undefined = targetSubjectId;\n        while (currParentId) {\n          const parentInfo = subjectInfos.get(currParentId);\n          if (!parentInfo?.hideInHierarchy) {\n            result.push({ className: \"BisCore.Subject\", id: currParentId });\n          }\n          currParentId = parentInfo?.parentSubject;\n        }\n        return result.reverse();\n      })();\n      this._subjectKeyPaths.set(targetSubjectId, entry);\n    }\n    return entry;\n  }\n\n  private async *queryModelElementCounts() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, COUNT(*) elementCount\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      GROUP BY Model.Id\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, elementCount: row.elementCount };\n    }\n  }\n\n  private async *queryModelCategories() {\n    const query = /* sql */ `\n      SELECT Model.Id modelId, Category.Id categoryId\n      FROM ${this._hierarchyConfig.elementClassSpecification}\n      WHERE Parent.Id IS NULL\n      GROUP BY modelId, categoryId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: query }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { modelId: row.modelId, categoryId: row.categoryId };\n    }\n  }\n\n  private async getModelInfos() {\n    this._modelInfos ??= (async () => {\n      const modelInfos = new Map<Id64String, { categories: Id64Set; elementCount: number }>();\n      await Promise.all([\n        (async () => {\n          for await (const { modelId, categoryId } of this.queryModelCategories()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.categories.add(categoryId);\n            } else {\n              modelInfos.set(modelId, { categories: new Set([categoryId]), elementCount: 0 });\n            }\n          }\n        })(),\n        (async () => {\n          for await (const { modelId, elementCount } of this.queryModelElementCounts()) {\n            const entry = modelInfos.get(modelId);\n            if (entry) {\n              entry.elementCount = elementCount;\n            } else {\n              modelInfos.set(modelId, { categories: new Set(), elementCount });\n            }\n          }\n        })(),\n      ]);\n      return modelInfos;\n    })();\n    return this._modelInfos;\n  }\n\n  public async getModelCategories(modelId: Id64String): Promise<Id64Array> {\n    const modelInfos = await this.getModelInfos();\n    const categories = modelInfos.get(modelId)?.categories;\n    return categories ? [...categories] : [];\n  }\n\n  public async getModelElementCount(modelId: Id64String): Promise<number> {\n    const modelInfos = await this.getModelInfos();\n    return modelInfos.get(modelId)?.elementCount ?? 0;\n  }\n\n  public async createModelInstanceKeyPaths(modelId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._modelKeyPaths.get(modelId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Array<HierarchyNodeIdentifiersPath>();\n        const subjectInfos = (await this.getSubjectInfos()).entries();\n        for (const [modelSubjectId, subjectInfo] of subjectInfos) {\n          if (subjectInfo.childModels.has(modelId)) {\n            const subjectPath = await this.createSubjectInstanceKeysPath(modelSubjectId);\n            result.push([...subjectPath, { className: \"BisCore.GeometricModel3d\", id: modelId }]);\n          }\n        }\n        return result;\n      })();\n\n      this._modelKeyPaths.set(modelId, entry);\n    }\n    return entry;\n  }\n\n  private async queryCategoryElementCounts(\n    input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n  ): Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>> {\n    const reader = this._queryExecutor.createQueryReader(\n      {\n        ctes: [\n          /* sql */ `\n            CategoryElements(id, modelId, categoryId) AS (\n              SELECT ECInstanceId, Model.Id, Category.Id\n              FROM ${this._hierarchyConfig.elementClassSpecification}\n              WHERE\n                Parent.Id IS NULL\n                AND (\n                  ${input.map(({ modelId, categoryId }) => `Model.Id = ${modelId} AND Category.Id = ${categoryId}`).join(\" OR \")}\n                )\n\n              UNION ALL\n\n              SELECT c.ECInstanceId, p.modelId, p.categoryId\n              FROM ${this._hierarchyConfig.elementClassSpecification} c\n              JOIN CategoryElements p ON c.Parent.Id = p.id\n            )\n          `,\n        ],\n        ecsql: `\n          SELECT modelId, categoryId, COUNT(id) elementsCount\n          FROM CategoryElements\n          GROUP BY modelId, categoryId\n        `,\n      },\n      { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" },\n    );\n\n    const result = new Array<{ modelId: number; categoryId: number; elementsCount: number }>();\n    for await (const row of reader) {\n      result.push({ modelId: row.modelId, categoryId: row.categoryId, elementsCount: row.elementsCount });\n    }\n    return result;\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    return this._categoryElementCounts.getCategoryElementsCount(modelId, categoryId);\n  }\n\n  public async createCategoryInstanceKeyPaths(categoryId: Id64String): Promise<HierarchyNodeIdentifiersPath[]> {\n    let entry = this._categoryKeyPaths.get(categoryId);\n    if (!entry) {\n      entry = (async () => {\n        const result = new Set<Id64String>();\n        const modelInfos = await this.getModelInfos();\n        modelInfos?.forEach((modelInfo, modelId) => {\n          if (modelInfo.categories.has(categoryId)) {\n            result.add(modelId);\n          }\n        });\n\n        const categoryPaths = new Array<HierarchyNodeIdentifiersPath>();\n        for (const categoryModelId of [...result]) {\n          const modelPaths = await this.createModelInstanceKeyPaths(categoryModelId);\n          for (const modelPath of modelPaths) {\n            categoryPaths.push([...modelPath, { className: \"BisCore.SpatialCategory\", id: categoryId }]);\n          }\n        }\n        return categoryPaths;\n      })();\n      this._categoryKeyPaths.set(categoryId, entry);\n    }\n    return entry;\n  }\n}\n\nfunction forEachChildSubject(\n  subjectInfos: Map<Id64String, SubjectInfo>,\n  parentSubject: Id64String | SubjectInfo,\n  cb: (childSubjectId: Id64String, childSubjectInfo: SubjectInfo) => \"break\" | \"continue\",\n) {\n  const parentSubjectInfo = typeof parentSubject === \"string\" ? subjectInfos.get(parentSubject) : parentSubject;\n  parentSubjectInfo &&\n    parentSubjectInfo.childSubjects.forEach((childSubjectId) => {\n      const childSubjectInfo = subjectInfos.get(childSubjectId)!;\n      if (cb(childSubjectId, childSubjectInfo) === \"break\") {\n        return;\n      }\n      forEachChildSubject(subjectInfos, childSubjectInfo, cb);\n    });\n}\n\nclass ModelCategoryElementsCountCache {\n  private _cache = new Map<string, Subject<number>>();\n  private _requestsStream = new Subject<{ modelId: Id64String; categoryId: Id64String }>();\n  private _subscription: Subscription;\n\n  public constructor(\n    private _loader: (\n      input: Array<{ modelId: Id64String; categoryId: Id64String }>,\n    ) => Promise<Array<{ modelId: number; categoryId: number; elementsCount: number }>>,\n  ) {\n    this._subscription = this._requestsStream\n      .pipe(\n        bufferTime(20),\n        filter((requests) => requests.length > 0),\n        mergeMap(async (requests) => this._loader(requests)),\n        mergeAll(),\n      )\n      .subscribe({\n        next: ({ modelId, categoryId, elementsCount }) => {\n          const subject = this._cache.get(`${modelId}${categoryId}`);\n          assert(!!subject);\n          subject.next(elementsCount);\n        },\n      });\n  }\n\n  public [Symbol.dispose]() {\n    this._subscription.unsubscribe();\n  }\n\n  public async getCategoryElementsCount(modelId: Id64String, categoryId: Id64String): Promise<number> {\n    const cacheKey = `${modelId}${categoryId}`;\n    let result = this._cache.get(cacheKey);\n    if (result !== undefined) {\n      return firstValueFrom(result);\n    }\n\n    result = new ReplaySubject(1);\n    this._cache.set(cacheKey, result);\n    this._requestsStream.next({ modelId, categoryId });\n    return firstValueFrom(result);\n  }\n}\n"]}

package/lib/cjs/tree-widget-react/components/trees/models-tree/internal/ModelsTreeNode.d.ts

@@ -1,37 +0,0 @@
-import type { Id64String } from "@itwin/core-bentley";
-import { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
-interface ModelsTreeNode {
-    key: HierarchyNodeKey;
-    extendedData?: {
-        [id: string]: any;
-    };
-}
-/**
- * @internal
- */
-export declare namespace ModelsTreeNode {
-    /**
-     * Determines if a node represents a subject.
-     */
-    const isSubjectNode: (node: Pick<ModelsTreeNode, "extendedData">) => boolean;
-    /**
-     * Determines if a node represents a model.
-     */
-    const isModelNode: (node: Pick<ModelsTreeNode, "extendedData">) => boolean;
-    /**
-     * Determines if a node represents a category.
-     */
-    const isCategoryNode: (node: Pick<ModelsTreeNode, "extendedData">) => boolean;
-    /** Returns type of the node. */
-    const getType: (node: ModelsTreeNode) => "subject" | "model" | "category" | "element" | "elements-class-group";
-    /**
-     * Retrieves model ID from node's extended data.
-     */
-    const getModelId: (node: Pick<ModelsTreeNode, "extendedData">) => Id64String | undefined;
-    /**
-     * Retrieves category ID from node's extended data.
-     */
-    const getCategoryId: (node: Pick<ModelsTreeNode, "extendedData">) => Id64String | undefined;
-}
-export {};
-//# sourceMappingURL=ModelsTreeNode.d.ts.map

package/lib/cjs/tree-widget-react/components/trees/models-tree/internal/ModelsTreeNode.js

@@ -1,57 +0,0 @@
-"use strict";
-/*---------------------------------------------------------------------------------------------
- * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
- * See LICENSE.md in the project root for license terms and full copyright notice.
- *--------------------------------------------------------------------------------------------*/
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ModelsTreeNode = void 0;
-const presentation_hierarchies_1 = require("@itwin/presentation-hierarchies");
-/**
- * @internal
- */
-var ModelsTreeNode;
-(function (ModelsTreeNode) {
-    /**
-     * Determines if a node represents a subject.
-     */
-    ModelsTreeNode.isSubjectNode = (node) => !!node.extendedData?.isSubject;
-    /**
-     * Determines if a node represents a model.
-     */
-    ModelsTreeNode.isModelNode = (node) => !!node.extendedData?.isModel;
-    /**
-     * Determines if a node represents a category.
-     */
-    ModelsTreeNode.isCategoryNode = (node) => !!node.extendedData?.isCategory;
-    /** Returns type of the node. */
-    ModelsTreeNode.getType = (node) => {
-        if (presentation_hierarchies_1.HierarchyNodeKey.isClassGrouping(node.key)) {
-            return "elements-class-group";
-        }
-        if (ModelsTreeNode.isSubjectNode(node)) {
-            return "subject";
-        }
-        if (ModelsTreeNode.isModelNode(node)) {
-            return "model";
-        }
-        if (ModelsTreeNode.isCategoryNode(node)) {
-            return "category";
-        }
-        return "element";
-    };
-    /**
-     * Retrieves model ID from node's extended data.
-     */
-    ModelsTreeNode.getModelId = (node) => {
-        if (node.extendedData?.modelId) {
-            return node.extendedData?.modelId;
-        }
-        const modelIds = node.extendedData?.modelIds?.[0];
-        return modelIds && (Array.isArray(modelIds) ? modelIds[0] : modelIds);
-    };
-    /**
-     * Retrieves category ID from node's extended data.
-     */
-    ModelsTreeNode.getCategoryId = (node) => node.extendedData?.categoryId;
-})(ModelsTreeNode || (exports.ModelsTreeNode = ModelsTreeNode = {}));
-//# sourceMappingURL=ModelsTreeNode.js.map

package/lib/cjs/tree-widget-react/components/trees/models-tree/internal/ModelsTreeNode.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ModelsTreeNode.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/ModelsTreeNode.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAGhG,8EAAmE;AAOnE;;GAEG;AACH,IAAiB,cAAc,CAiD9B;AAjDD,WAAiB,cAAc;IAC7B;;OAEG;IACU,4BAAa,GAAG,CAAC,IAA0C,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,SAAS,CAAC;IAE5G;;OAEG;IACU,0BAAW,GAAG,CAAC,IAA0C,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC;IAExG;;OAEG;IACU,6BAAc,GAAG,CAAC,IAA0C,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;IAE9G,gCAAgC;IACnB,sBAAO,GAAG,CAAC,IAAoB,EAAyE,EAAE;QACrH,IAAI,2CAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;YAC9C,OAAO,sBAAsB,CAAC;SAC/B;QACD,IAAI,eAAA,aAAa,CAAC,IAAI,CAAC,EAAE;YACvB,OAAO,SAAS,CAAC;SAClB;QACD,IAAI,eAAA,WAAW,CAAC,IAAI,CAAC,EAAE;YACrB,OAAO,OAAO,CAAC;SAChB;QACD,IAAI,eAAA,cAAc,CAAC,IAAI,CAAC,EAAE;YACxB,OAAO,UAAU,CAAC;SACnB;QACD,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;IAEF;;OAEG;IACU,yBAAU,GAAG,CAAC,IAA0C,EAA0B,EAAE;QAC/F,IAAI,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE;YAC9B,OAAO,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC;SACnC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC;QAClD,OAAO,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC,CAAC;IAEF;;OAEG;IACU,4BAAa,GAAG,CAAC,IAA0C,EAA0B,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;AACrI,CAAC,EAjDgB,cAAc,8BAAd,cAAc,QAiD9B","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport { HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\n\ninterface ModelsTreeNode {\n  key: HierarchyNodeKey;\n  extendedData?: { [id: string]: any };\n}\n\n/**\n * @internal\n */\nexport namespace ModelsTreeNode {\n  /**\n   * Determines if a node represents a subject.\n   */\n  export const isSubjectNode = (node: Pick<ModelsTreeNode, \"extendedData\">) => !!node.extendedData?.isSubject;\n\n  /**\n   * Determines if a node represents a model.\n   */\n  export const isModelNode = (node: Pick<ModelsTreeNode, \"extendedData\">) => !!node.extendedData?.isModel;\n\n  /**\n   * Determines if a node represents a category.\n   */\n  export const isCategoryNode = (node: Pick<ModelsTreeNode, \"extendedData\">) => !!node.extendedData?.isCategory;\n\n  /** Returns type of the node. */\n  export const getType = (node: ModelsTreeNode): \"subject\" | \"model\" | \"category\" | \"element\" | \"elements-class-group\" => {\n    if (HierarchyNodeKey.isClassGrouping(node.key)) {\n      return \"elements-class-group\";\n    }\n    if (isSubjectNode(node)) {\n      return \"subject\";\n    }\n    if (isModelNode(node)) {\n      return \"model\";\n    }\n    if (isCategoryNode(node)) {\n      return \"category\";\n    }\n    return \"element\";\n  };\n\n  /**\n   * Retrieves model ID from node's extended data.\n   */\n  export const getModelId = (node: Pick<ModelsTreeNode, \"extendedData\">): Id64String | undefined => {\n    if (node.extendedData?.modelId) {\n      return node.extendedData?.modelId;\n    }\n\n    const modelIds = node.extendedData?.modelIds?.[0];\n    return modelIds && (Array.isArray(modelIds) ? modelIds[0] : modelIds);\n  };\n\n  /**\n   * Retrieves category ID from node's extended data.\n   */\n  export const getCategoryId = (node: Pick<ModelsTreeNode, \"extendedData\">): Id64String | undefined => node.extendedData?.categoryId;\n}\n"]}

package/lib/cjs/tree-widget-react/components/trees/models-tree/internal/ModelsTreeVisibilityHandler.d.ts

@@ -1,106 +0,0 @@
-import type { GroupingHierarchyNode, HierarchyFilteringPath } from "@itwin/presentation-hierarchies";
-import type { HierarchyVisibilityHandler, HierarchyVisibilityHandlerOverridableMethod, VisibilityStatus } from "../../common/UseHierarchyVisibility.js";
-import type { ModelsTreeIdsCache } from "./ModelsTreeIdsCache.js";
-import type { Id64Arg, Id64Array, Id64Set, Id64String } from "@itwin/core-bentley";
-import type { Viewport } from "@itwin/core-frontend";
-import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
-/** @beta */
-interface GetCategoryVisibilityStatusProps {
-    categoryId: Id64String;
-    modelId: Id64String;
-}
-/** @beta */
-interface ChangeCategoryVisibilityStateProps extends GetCategoryVisibilityStatusProps {
-    on: boolean;
-}
-/** @beta */
-interface GetGeometricElementVisibilityStatusProps {
-    elementId: Id64String;
-    modelId: Id64String;
-    categoryId: Id64String;
-}
-/** @beta */
-interface ChangeGeometricElementsDisplayStateProps {
-    elementIds: Id64Set;
-    modelId: Id64String;
-    categoryId: Id64String;
-    on: boolean;
-}
-/** @beta */
-interface ChangeModelVisibilityStateProps {
-    ids: Id64Arg;
-    on: boolean;
-}
-/**
- * Functionality of Models tree visibility handler that can be overridden.
- * Each callback is provided original implementation and reference to a `HierarchyVisibilityHandler`.
- * @beta
- */
-export interface ModelsTreeVisibilityHandlerOverrides {
-    getSubjectNodeVisibility?: HierarchyVisibilityHandlerOverridableMethod<(props: {
-        ids: Id64Array;
-    }) => Promise<VisibilityStatus>>;
-    getModelDisplayStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: {
-        id: Id64String;
-    }) => Promise<VisibilityStatus>>;
-    getCategoryDisplayStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: GetCategoryVisibilityStatusProps) => Promise<VisibilityStatus>>;
-    getElementGroupingNodeDisplayStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: {
-        node: GroupingHierarchyNode;
-    }) => Promise<VisibilityStatus>>;
-    getElementDisplayStatus?: HierarchyVisibilityHandlerOverridableMethod<(props: GetGeometricElementVisibilityStatusProps) => Promise<VisibilityStatus>>;
-    changeSubjectNodeState?: HierarchyVisibilityHandlerOverridableMethod<(props: {
-        ids: Id64Array;
-        on: boolean;
-    }) => Promise<void>>;
-    changeModelState?: HierarchyVisibilityHandlerOverridableMethod<(props: ChangeModelVisibilityStateProps) => Promise<void>>;
-    changeCategoryState?: HierarchyVisibilityHandlerOverridableMethod<(props: ChangeCategoryVisibilityStateProps) => Promise<void>>;
-    changeElementGroupingNodeState?: HierarchyVisibilityHandlerOverridableMethod<(props: {
-        node: GroupingHierarchyNode;
-        on: boolean;
-    }) => Promise<void>>;
-    changeElementsState?: HierarchyVisibilityHandlerOverridableMethod<(props: ChangeGeometricElementsDisplayStateProps) => Promise<void>>;
-}
-/**
- * Props for `createModelsTreeVisibilityHandler`.
- * @internal
- */
-export interface ModelsTreeVisibilityHandlerProps {
-    viewport: Viewport;
-    idsCache: ModelsTreeIdsCache;
-    imodelAccess: ECClassHierarchyInspector;
-    overrides?: ModelsTreeVisibilityHandlerOverrides;
-    filteredPaths?: HierarchyFilteringPath[];
-}
-/**
- * Creates an instance if `ModelsTreeVisibilityHandler`.
- * @internal
- */
-export declare function createModelsTreeVisibilityHandler(props: ModelsTreeVisibilityHandlerProps): HierarchyVisibilityHandler;
-/**
- * Enables display of all given models. Also enables display of all categories and clears always and
- * never drawn lists in the viewport.
- * @public
- */
-export declare function showAllModels(models: string[], viewport: Viewport): Promise<void>;
-/**
- * Disables display of all given models.
- * @public
- */
-export declare function hideAllModels(models: string[], viewport: Viewport): Promise<void>;
-/**
- * Inverts display of all given models.
- * @public
- */
-export declare function invertAllModels(models: string[], viewport: Viewport): Promise<void>;
-/**
- * Based on the value of `enable` argument, either enables or disables display of given models.
- * @public
- */
-export declare function toggleModels(models: string[], enable: boolean, viewport: Viewport): Promise<void>;
-/**
- * Checks if all given models are displayed in given viewport.
- * @public
- */
-export declare function areAllModelsVisible(models: string[], viewport: Viewport): boolean;
-export {};
-//# sourceMappingURL=ModelsTreeVisibilityHandler.d.ts.map