@itwin/tree-widget-react 0.5.0 → 0.6.1

package/lib/cjs/components/trees/models-tree/Hierarchy.json

@@ -0,0 +1,387 @@
+{
+  "$schema": "../../../../node_modules/@itwin/presentation-common/Ruleset.schema.json",
+  "id": "tree-widget-react/ModelsTree",
+  "requiredSchemas": [
+    {
+      "name": "BisCore"
+    }
+  ],
+  "rules": [
+    {
+      "ruleType": "RootNodes",
+      "autoExpand": true,
+      "specifications": [
+        {
+          "specType": "InstanceNodesOfSpecificClasses",
+          "classes": [
+            {
+              "schemaName": "BisCore",
+              "classNames": [
+                "Subject"
+              ]
+            }
+          ],
+          "instanceFilter": "this.Parent = NULL",
+          "arePolymorphic": false,
+          "groupByClass": false,
+          "groupByLabel": false
+        }
+      ],
+      "customizationRules": [
+        {
+          "ruleType": "ExtendedData",
+          "items": {
+            "isSubject": "true"
+          }
+        }
+      ]
+    },
+    {
+      "ruleType": "ChildNodes",
+      "condition": "ParentNode.IsOfClass(\"Subject\", \"BisCore\")",
+      "specifications": [
+        {
+          "specType": "RelatedInstanceNodes",
+          "relationshipPaths": [
+            {
+              "relationship": {
+                "schemaName": "BisCore",
+                "className": "SubjectOwnsSubjects"
+              },
+              "direction": "Forward",
+              "targetClass": {
+                "schemaName": "BisCore",
+                "className": "Subject"
+              }
+            }
+          ],
+          "instanceFilter": "json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") <> NULL OR ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") = \"Hierarchy\"",
+          "hideNodesInHierarchy": true,
+          "groupByClass": false,
+          "groupByLabel": false
+        },
+        {
+          "specType": "RelatedInstanceNodes",
+          "relationshipPaths": [
+            {
+              "relationship": {
+                "schemaName": "BisCore",
+                "className": "SubjectOwnsSubjects"
+              },
+              "direction": "Forward",
+              "targetClass": {
+                "schemaName": "BisCore",
+                "className": "Subject"
+              }
+            }
+          ],
+          "instanceFilter": "json_extract(this.JsonProperties, \"$.Subject.Job.Bridge\") = NULL AND ifnull(json_extract(this.JsonProperties, \"$.Subject.Model.Type\"), \"\") <> \"Hierarchy\"",
+          "hideIfNoChildren": true,
+          "groupByClass": false,
+          "groupByLabel": false
+        }
+      ],
+      "customizationRules": [
+        {
+          "ruleType": "ExtendedData",
+          "items": {
+            "isSubject": "true"
+          }
+        }
+      ]
+    },
+    {
+      "ruleType": "ChildNodes",
+      "condition": "ParentNode.IsOfClass(\"Subject\", \"BisCore\")",
+      "specifications": [
+        {
+          "specType": "InstanceNodesOfSpecificClasses",
+          "classes": {
+            "schemaName": "BisCore",
+            "classNames": [
+              "GeometricModel3d"
+            ]
+          },
+          "arePolymorphic": true,
+          "relatedInstances": [
+            {
+              "relationshipPath": {
+                "relationship": {
+                  "schemaName": "BisCore",
+                  "className": "ModelModelsElement"
+                },
+                "direction": "Forward",
+                "targetClass": {
+                  "schemaName": "BisCore",
+                  "className": "InformationPartitionElement"
+                }
+              },
+              "alias": "partition",
+              "isRequired": true
+            }
+          ],
+          "instanceFilter": "(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") = NULL AND json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") = NULL AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"BisCore:GeometricElement3d\")",
+          "hasChildren": "Always",
+          "hideIfNoChildren": true,
+          "groupByClass": false,
+          "groupByLabel": false
+        },
+        {
+          "specType": "InstanceNodesOfSpecificClasses",
+          "classes": {
+            "schemaName": "BisCore",
+            "classNames": [
+              "GeometricModel3d"
+            ]
+          },
+          "arePolymorphic": true,
+          "relatedInstances": [
+            {
+              "relationshipPath": {
+                "relationship": {
+                  "schemaName": "BisCore",
+                  "className": "ModelModelsElement"
+                },
+                "direction": "Forward",
+                "targetClass": {
+                  "schemaName": "BisCore",
+                  "className": "InformationPartitionElement"
+                }
+              },
+              "alias": "partition",
+              "isRequired": true
+            }
+          ],
+          "instanceFilter": "(parent.ECInstanceId = partition.Parent.Id OR json_extract(parent.JsonProperties, \"$.Subject.Model.TargetPartition\") = printf(\"0x%x\", partition.ECInstanceId)) AND NOT this.IsPrivate AND (json_extract(partition.JsonProperties, \"$.PhysicalPartition.Model.Content\") <> NULL OR json_extract(partition.JsonProperties, \"$.GraphicalPartition3d.Model.Content\") <> NULL) AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"BisCore:GeometricElement3d\")",
+          "hasChildren": "Always",
+          "hideNodesInHierarchy": true,
+          "groupByClass": false,
+          "groupByLabel": false
+        }
+      ],
+      "customizationRules": [
+        {
+          "ruleType": "ExtendedData",
+          "items": {
+            "isModel": "true"
+          }
+        }
+      ]
+    },
+    {
+      "ruleType": "ChildNodes",
+      "condition": "ParentNode.IsOfClass(\"ISubModeledElement\", \"BisCore\")",
+      "specifications": [
+        {
+          "specType": "RelatedInstanceNodes",
+          "relationshipPaths": [
+            {
+              "relationship": {
+                "schemaName": "BisCore",
+                "className": "ModelModelsElement"
+              },
+              "direction": "Backward"
+            }
+          ],
+          "instanceFilter": "NOT this.IsPrivate AND this.HasRelatedInstance(\"BisCore:ModelContainsElements\", \"Forward\", \"BisCore:GeometricElement3d\")",
+          "hideNodesInHierarchy": true,
+          "groupByClass": false,
+          "groupByLabel": false
+        }
+      ],
+      "customizationRules": [
+        {
+          "ruleType": "ExtendedData",
+          "items": {
+            "isModel": "true"
+          }
+        }
+      ]
+    },
+    {
+      "ruleType": "ChildNodes",
+      "condition": "ParentNode.IsOfClass(\"GeometricModel3d\", \"BisCore\")",
+      "specifications": [
+        {
+          "specType": "RelatedInstanceNodes",
+          "relationshipPaths": [
+            [
+              {
+                "relationship": {
+                  "schemaName": "BisCore",
+                  "className": "ModelContainsElements"
+                },
+                "direction": "Forward",
+                "targetClass": {
+                  "schemaName": "BisCore",
+                  "className": "GeometricElement3d"
+                }
+              },
+              {
+                "relationship": {
+                  "schemaName": "BisCore",
+                  "className": "GeometricElement3dIsInCategory"
+                },
+                "direction": "Forward",
+                "targetClass": {
+                  "schemaName": "BisCore",
+                  "className": "SpatialCategory"
+                }
+              }
+            ]
+          ],
+          "instanceFilter": "NOT this.IsPrivate",
+          "suppressSimilarAncestorsCheck": true,
+          "hideIfNoChildren": true,
+          "groupByClass": false,
+          "groupByLabel": false
+        }
+      ],
+      "customizationRules": [
+        {
+          "ruleType": "ExtendedData",
+          "items": {
+            "isCategory": "true",
+            "modelId": "ParentNode.InstanceId"
+          }
+        }
+      ]
+    },
+    {
+      "ruleType": "ChildNodes",
+      "condition": "ParentNode.IsOfClass(\"SpatialCategory\", \"BisCore\")",
+      "specifications": [
+        {
+          "specType": "RelatedInstanceNodes",
+          "relationshipPaths": [
+            {
+              "relationship": {
+                "schemaName": "BisCore",
+                "className": "GeometricElement3dIsInCategory"
+              },
+              "direction": "Backward",
+              "targetClass": {
+                "schemaName": "BisCore",
+                "className": "GeometricElement3d"
+              }
+            }
+          ],
+          "instanceFilter": "this.Model.Id = parent.parent.ECInstanceId ANDALSO this.Parent = NULL ANDALSO (NOT HasVariable(\"filtered-element-ids\") OR GetVariableIntValues(\"filtered-element-ids\").AnyMatch(id => id = this.ECInstanceId))",
+          "groupByClass": false,
+          "groupByLabel": false
+        }
+      ],
+      "customizationRules": [
+        {
+          "ruleType": "ExtendedData",
+          "items": {
+            "modelId": "this.Model.Id",
+            "categoryId": "this.Category.Id"
+          }
+        }
+      ]
+    },
+    {
+      "ruleType": "ChildNodes",
+      "condition": "ParentNode.IsOfClass(\"GeometricElement3d\", \"BisCore\")",
+      "specifications": [
+        {
+          "specType": "RelatedInstanceNodes",
+          "relationshipPaths": [
+            {
+              "relationship": {
+                "schemaName": "BisCore",
+                "className": "ElementOwnsChildElements"
+              },
+              "direction": "Forward",
+              "targetClass": {
+                "schemaName": "BisCore",
+                "className": "GeometricElement3d"
+              }
+            }
+          ],
+          "instanceFilter": "NOT HasVariable(\"filtered-element-ids\") OR GetVariableIntValues(\"filtered-element-ids\").AnyMatch(id => id = this.ECInstanceId)",
+          "groupByClass": false,
+          "groupByLabel": false
+        }
+      ],
+      "customizationRules": [
+        {
+          "ruleType": "ExtendedData",
+          "items": {
+            "modelId": "this.Model.Id",
+            "categoryId": "this.Category.Id"
+          }
+        }
+      ]
+    },
+    {
+      "ruleType": "Grouping",
+      "class": {
+        "schemaName": "BisCore",
+        "className": "Subject"
+      },
+      "groups": [
+        {
+          "specType": "SameLabelInstance",
+          "applicationStage": "PostProcess"
+        }
+      ]
+    },
+    {
+      "ruleType": "Grouping",
+      "class": {
+        "schemaName": "BisCore",
+        "className": "SpatialCategory"
+      },
+      "groups": [
+        {
+          "specType": "SameLabelInstance",
+          "applicationStage": "PostProcess"
+        }
+      ]
+    },
+    {
+      "ruleType": "ImageIdOverride",
+      "condition": "ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Subject\", \"BisCore\")",
+      "imageIdExpression": "IIF(this.Parent.Id = NULL, \"icon-imodel-hollow-2\", \"icon-folder\")"
+    },
+    {
+      "ruleType": "ImageIdOverride",
+      "condition": "ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Model\", \"BisCore\")",
+      "imageIdExpression": "\"icon-model\""
+    },
+    {
+      "ruleType": "ImageIdOverride",
+      "condition": "ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Category\", \"BisCore\")",
+      "imageIdExpression": "\"icon-layers\""
+    },
+    {
+      "ruleType": "ImageIdOverride",
+      "condition": "ThisNode.IsInstanceNode ANDALSO ThisNode.IsOfClass(\"Element\", \"BisCore\")",
+      "imageIdExpression": "\"icon-item\""
+    },
+    {
+      "ruleType": "Content",
+      "condition": "ContentDisplayType = \"AssemblyElementsRequest\"",
+      "specifications": [
+        {
+          "specType": "SelectedNodeInstances"
+        },
+        {
+          "specType": "ContentRelatedInstances",
+          "relationshipPaths": [
+            {
+              "relationship": {
+                "schemaName": "BisCore",
+                "className": "ElementOwnsChildElements"
+              },
+              "direction": "Forward",
+              "count": "*"
+            }
+          ]
+        }
+      ]
+    }
+  ]
+}

package/lib/cjs/components/trees/models-tree/ModelsTree.d.ts

@@ -0,0 +1,82 @@
+import "./ModelsTree.scss";
+import * as React from "react";
+import { SelectionMode } from "@itwin/components-react";
+import { ClassGroupingOption } from "../Common";
+import { ModelsVisibilityHandler } from "./ModelsVisibilityHandler";
+import type { VisibilityTreeFilterInfo } from "../Common";
+import type { ModelsTreeSelectionPredicate } from "./ModelsVisibilityHandler";
+import type { Id64Array } from "@itwin/core-bentley";
+import type { IModelConnection, Viewport } from "@itwin/core-frontend";
+import type { Ruleset } from "@itwin/presentation-common";
+import type { IPresentationTreeDataProvider } from "@itwin/presentation-components";
+/** @internal */
+export declare const RULESET_MODELS: Ruleset;
+/** @internal */
+export declare const RULESET_MODELS_GROUPED_BY_CLASS: Ruleset;
+/** Props for [[ModelsTree]] component
+ * @public
+ */
+export interface ModelsTreeProps {
+    /**
+     * An IModel to pull data from
+     */
+    iModel: IModelConnection;
+    /** Width of the component */
+    width: number;
+    /** Height of the component */
+    height: number;
+    /**
+     * Selection mode in the tree
+     */
+    selectionMode?: SelectionMode;
+    /**
+     * Predicate which indicates whether node can be selected or no
+     * @alpha
+     */
+    selectionPredicate?: ModelsTreeSelectionPredicate;
+    /**
+     * Active view used to determine and control visibility
+     */
+    activeView?: Viewport;
+    /**
+     * Ref to the root HTML element used by this component
+     */
+    rootElementRef?: React.Ref<HTMLDivElement>;
+    /**
+     * Information for tree filtering.
+     * @alpha
+     */
+    filterInfo?: VisibilityTreeFilterInfo;
+    /**
+     * Filter the hierarchy by given element IDs.
+     * @alpha
+     */
+    filteredElementIds?: Id64Array;
+    /**
+     * Callback invoked when tree is filtered.
+     */
+    onFilterApplied?: (filteredDataProvider: IPresentationTreeDataProvider, matchesCount: number) => void;
+    /**
+     * Should the tree group displayed element nodes by class.
+     * @beta
+     */
+    enableElementsClassGrouping?: ClassGroupingOption;
+    /**
+     * Auto-update the hierarchy when data in the iModel changes.
+     * @alpha
+     */
+    enableHierarchyAutoUpdate?: boolean;
+    /**
+     * Custom visibility handler.
+     * @alpha
+     */
+    modelsVisibilityHandler?: ModelsVisibilityHandler;
+}
+/**
+ * A tree component that shows a subject - model - category - element
+ * hierarchy along with checkboxes that represent and allow changing
+ * the display of those instances.
+ * @public
+ */
+export declare function ModelsTree(props: ModelsTreeProps): JSX.Element;
+//# sourceMappingURL=ModelsTree.d.ts.map

package/lib/cjs/components/trees/models-tree/ModelsTree.js

@@ -0,0 +1,144 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    Object.defineProperty(o, k2, { enumerable: true, get: function() { return m[k]; } });
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelsTree = exports.RULESET_MODELS_GROUPED_BY_CLASS = exports.RULESET_MODELS = void 0;
+require("./ModelsTree.scss");
+const React = __importStar(require("react"));
+const components_react_1 = require("@itwin/components-react");
+const core_react_1 = require("@itwin/core-react");
+const presentation_components_1 = require("@itwin/presentation-components");
+const presentation_frontend_1 = require("@itwin/presentation-frontend");
+const TreeWidget_1 = require("../../../TreeWidget");
+const Common_1 = require("../Common");
+const VisibilityTreeEventHandler_1 = require("../VisibilityTreeEventHandler");
+const VisibilityTreeRenderer_1 = require("../VisibilityTreeRenderer");
+const ModelsVisibilityHandler_1 = require("./ModelsVisibilityHandler");
+const PAGING_SIZE = 20;
+/** @internal */
+exports.RULESET_MODELS = require("./Hierarchy.json"); // eslint-disable-line @typescript-eslint/no-var-requires
+/** @internal */
+exports.RULESET_MODELS_GROUPED_BY_CLASS = require("./Hierarchy.GroupedByClass.json"); // eslint-disable-line @typescript-eslint/no-var-requires
+const RULESET_MODELS_SEARCH = require("./ModelsTreeSearch.json"); // eslint-disable-line @typescript-eslint/no-var-requires
+/**
+ * A tree component that shows a subject - model - category - element
+ * hierarchy along with checkboxes that represent and allow changing
+ * the display of those instances.
+ * @public
+ */
+function ModelsTree(props) {
+    const { nodeLoader, onItemsRendered } = useModelsTreeNodeLoader(props);
+    const { filteredNodeLoader, isFiltering, nodeHighlightingProps } = VisibilityTreeRenderer_1.useVisibilityTreeFiltering(nodeLoader, props.filterInfo, props.onFilterApplied);
+    const filterApplied = filteredNodeLoader !== nodeLoader;
+    const { activeView, modelsVisibilityHandler, selectionPredicate } = props;
+    const nodeSelectionPredicate = React.useCallback((key, node) => {
+        return !selectionPredicate ? true : selectionPredicate(key, ModelsVisibilityHandler_1.ModelsVisibilityHandler.getNodeType(node, nodeLoader.dataProvider));
+    }, [selectionPredicate, nodeLoader.dataProvider]);
+    const visibilityHandler = useVisibilityHandler(nodeLoader.dataProvider.rulesetId, props.iModel, activeView, modelsVisibilityHandler, getFilteredDataProvider(filteredNodeLoader.dataProvider), props.enableHierarchyAutoUpdate);
+    const eventHandler = core_react_1.useDisposable(React.useCallback(() => new VisibilityTreeEventHandler_1.VisibilityTreeEventHandler({
+        nodeLoader: filteredNodeLoader,
+        visibilityHandler,
+        collapsedChildrenDisposalEnabled: true,
+        selectionPredicate: nodeSelectionPredicate,
+    }), [filteredNodeLoader, visibilityHandler, nodeSelectionPredicate]));
+    const treeModel = components_react_1.useTreeModel(filteredNodeLoader.modelSource);
+    const treeRenderer = VisibilityTreeRenderer_1.useVisibilityTreeRenderer(true, false);
+    const overlay = isFiltering ? React.createElement("div", { className: "filteredTreeOverlay" }) : undefined;
+    // istanbul ignore next
+    const noFilteredDataRenderer = React.useCallback(() => {
+        return React.createElement(VisibilityTreeRenderer_1.VisibilityTreeNoFilteredData, { title: TreeWidget_1.TreeWidget.translate("modelTree.noModelFound"), message: TreeWidget_1.TreeWidget.translate("modelTree.noMatchingModelNames") });
+    }, []);
+    return (React.createElement("div", { className: "tree-widget-models-tree", ref: props.rootElementRef },
+        React.createElement(components_react_1.ControlledTree, { nodeLoader: filteredNodeLoader, model: treeModel, selectionMode: props.selectionMode || components_react_1.SelectionMode.None, eventsHandler: eventHandler, treeRenderer: treeRenderer, nodeHighlightingProps: nodeHighlightingProps, noDataRenderer: filterApplied ? noFilteredDataRenderer : undefined, onItemsRendered: onItemsRendered, width: props.width, height: props.height }),
+        overlay));
+}
+exports.ModelsTree = ModelsTree;
+function useModelsTreeNodeLoader(props) {
+    var _a, _b;
+    // note: this is a temporary workaround for auto-update not working on ruleset variable changes - instead
+    // of auto-updating we just re-create the node loader by re-creating the ruleset
+    const rulesets = React.useMemo(() => {
+        return {
+            general: (!props.enableElementsClassGrouping) ? { ...exports.RULESET_MODELS } : /* istanbul ignore next */ { ...exports.RULESET_MODELS_GROUPED_BY_CLASS },
+            search: { ...RULESET_MODELS_SEARCH },
+        };
+    }, [props.filteredElementIds]); // eslint-disable-line react-hooks/exhaustive-deps
+    // const rulesets = {
+    //   general: (!props.enableElementsClassGrouping) ? RULESET_MODELS : /* istanbul ignore next */ RULESET_MODELS_GROUPED_BY_CLASS,
+    //   search: RULESET_MODELS_SEARCH,
+    // };
+    const { nodeLoader, onItemsRendered } = presentation_components_1.usePresentationTreeNodeLoader({
+        imodel: props.iModel,
+        ruleset: rulesets.general,
+        appendChildrenCountForGroupingNodes: (props.enableElementsClassGrouping === Common_1.ClassGroupingOption.YesWithCounts),
+        pagingSize: PAGING_SIZE,
+        enableHierarchyAutoUpdate: props.enableHierarchyAutoUpdate,
+    });
+    const { nodeLoader: searchNodeLoader, onItemsRendered: onSearchItemsRendered } = presentation_components_1.usePresentationTreeNodeLoader({
+        imodel: props.iModel,
+        ruleset: rulesets.search,
+        pagingSize: PAGING_SIZE,
+        enableHierarchyAutoUpdate: props.enableHierarchyAutoUpdate,
+    });
+    const activeNodeLoader = ((_a = props.filterInfo) === null || _a === void 0 ? void 0 : _a.filter) ? searchNodeLoader : nodeLoader;
+    const activeItemsRenderedCallback = ((_b = props.filterInfo) === null || _b === void 0 ? void 0 : _b.filter) ? onSearchItemsRendered : onItemsRendered;
+    const vars = presentation_frontend_1.Presentation.presentation.vars(activeNodeLoader.dataProvider.rulesetId);
+    if (props.filteredElementIds) {
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        vars.setId64s("filtered-element-ids", props.filteredElementIds);
+    }
+    else {
+        // eslint-disable-next-line @typescript-eslint/no-floating-promises
+        vars.unset("filtered-element-ids");
+    }
+    return {
+        nodeLoader: activeNodeLoader,
+        onItemsRendered: activeItemsRenderedCallback,
+    };
+}
+function useVisibilityHandler(rulesetId, iModel, activeView, visibilityHandler, filteredDataProvider, hierarchyAutoUpdateEnabled) {
+    const subjectModelIdsCache = React.useMemo(() => new ModelsVisibilityHandler_1.SubjectModelIdsCache(iModel), [iModel]);
+    const defaultVisibilityHandler = core_react_1.useOptionalDisposable(React.useCallback(() => {
+        if (activeView)
+            return createVisibilityHandler(rulesetId, activeView, subjectModelIdsCache, hierarchyAutoUpdateEnabled);
+        return undefined;
+    }, [rulesetId, activeView, subjectModelIdsCache, hierarchyAutoUpdateEnabled]));
+    const handler = visibilityHandler !== null && visibilityHandler !== void 0 ? visibilityHandler : defaultVisibilityHandler;
+    React.useEffect(() => {
+        handler && handler.setFilteredDataProvider(filteredDataProvider);
+    }, [handler, filteredDataProvider]);
+    return handler;
+}
+const createVisibilityHandler = (rulesetId, activeView, subjectModelIdsCache, hierarchyAutoUpdateEnabled) => {
+    // istanbul ignore next
+    return new ModelsVisibilityHandler_1.ModelsVisibilityHandler({ rulesetId, viewport: activeView, hierarchyAutoUpdateEnabled, subjectModelIdsCache });
+};
+const isFilteredDataProvider = (dataProvider) => {
+    const filteredProvider = dataProvider;
+    return filteredProvider.nodeMatchesFilter !== undefined && filteredProvider.getActiveMatch !== undefined && filteredProvider.countFilteringResults !== undefined;
+};
+const getFilteredDataProvider = (dataProvider) => {
+    return isFilteredDataProvider(dataProvider) ? dataProvider : undefined;
+};
+//# sourceMappingURL=ModelsTree.js.map

package/lib/cjs/components/trees/models-tree/ModelsTree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ModelsTree.js","sourceRoot":"","sources":["../../../../../src/components/trees/models-tree/ModelsTree.tsx"],"names":[],"mappings":";AAAA;;;+FAG+F;;;;;;;;;;;;;;;;;;;;;;AAE/F,6BAA2B;AAC3B,6CAA+B;AAC/B,8DAAsF;AACtF,kDAAyE;AACzE,4EAA+E;AAC/E,wEAA4D;AAC5D,oDAAiD;AACjD,sCAAgD;AAChD,8EAA2E;AAC3E,sEAAgI;AAChI,uEAA0F;AAS1F,MAAM,WAAW,GAAG,EAAE,CAAC;AAEvB,gBAAgB;AACH,QAAA,cAAc,GAAY,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,yDAAyD;AAC7H,gBAAgB;AACH,QAAA,+BAA+B,GAAY,OAAO,CAAC,iCAAiC,CAAC,CAAC,CAAC,yDAAyD;AAE7J,MAAM,qBAAqB,GAAY,OAAO,CAAC,yBAAyB,CAAC,CAAC,CAAC,yDAAyD;AA8DpI;;;;;GAKG;AACH,SAAgB,UAAU,CAAC,KAAsB;IAC/C,MAAM,EAAE,UAAU,EAAE,eAAe,EAAE,GAAG,uBAAuB,CAAC,KAAK,CAAC,CAAC;IACvE,MAAM,EAAE,kBAAkB,EAAE,WAAW,EAAE,qBAAqB,EAAE,GAAG,mDAA0B,CAAC,UAAU,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,eAAe,CAAC,CAAC;IACnJ,MAAM,aAAa,GAAG,kBAAkB,KAAK,UAAU,CAAC;IAExD,MAAM,EAAE,UAAU,EAAE,uBAAuB,EAAE,kBAAkB,EAAE,GAAG,KAAK,CAAC;IAC1E,MAAM,sBAAsB,GAAG,KAAK,CAAC,WAAW,CAAC,CAAC,GAAY,EAAE,IAAkB,EAAE,EAAE;QACpF,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,iDAAuB,CAAC,WAAW,CAAC,IAAI,EAAE,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC;IAClI,CAAC,EAAE,CAAC,kBAAkB,EAAE,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC;IAElD,MAAM,iBAAiB,GAAG,oBAAoB,CAC5C,UAAU,CAAC,YAAY,CAAC,SAAS,EACjC,KAAK,CAAC,MAAM,EACZ,UAAU,EACV,uBAAuB,EACvB,uBAAuB,CAAC,kBAAkB,CAAC,YAAY,CAAC,EACxD,KAAK,CAAC,yBAAyB,CAAC,CAAC;IACnC,MAAM,YAAY,GAAG,0BAAa,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,IAAI,uDAA0B,CAAC;QACxF,UAAU,EAAE,kBAAkB;QAC9B,iBAAiB;QACjB,gCAAgC,EAAE,IAAI;QACtC,kBAAkB,EAAE,sBAAsB;KAC3C,CAAC,EAAE,CAAC,kBAAkB,EAAE,iBAAiB,EAAE,sBAAsB,CAAC,CAAC,CAAC,CAAC;IAEtE,MAAM,SAAS,GAAG,+BAAY,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;IAC/D,MAAM,YAAY,GAAG,kDAAyB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAE5D,MAAM,OAAO,GAAG,WAAW,CAAC,CAAC,CAAC,6BAAK,SAAS,EAAC,qBAAqB,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;IAElF,uBAAuB;IACvB,MAAM,sBAAsB,GAAG,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE;QACpD,OAAO,oBAAC,qDAA4B,IAClC,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,wBAAwB,CAAC,EACrD,OAAO,EAAE,uBAAU,CAAC,SAAS,CAAC,gCAAgC,CAAC,GAC/D,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,CACL,6BAAK,SAAS,EAAC,yBAAyB,EAAC,GAAG,EAAE,KAAK,CAAC,cAAc;QAChE,oBAAC,iCAAc,IACb,UAAU,EAAE,kBAAkB,EAC9B,KAAK,EAAE,SAAS,EAChB,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,gCAAa,CAAC,IAAI,EACxD,aAAa,EAAE,YAAY,EAC3B,YAAY,EAAE,YAAY,EAC1B,qBAAqB,EAAE,qBAAqB,EAC5C,cAAc,EAAE,aAAa,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,SAAS,EAClE,eAAe,EAAE,eAAe,EAChC,KAAK,EAAE,KAAK,CAAC,KAAK,EAClB,MAAM,EAAE,KAAK,CAAC,MAAM,GACpB;QACD,OAAO,CACJ,CACP,CAAC;AACJ,CAAC;AAtDD,gCAsDC;AAED,SAAS,uBAAuB,CAAC,KAAsB;;IACrD,yGAAyG;IACzG,gFAAgF;IAChF,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE;QAClC,OAAO;YACL,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,sBAAc,EAAE,CAAC,CAAC,CAAC,0BAA0B,CAAC,EAAE,GAAG,uCAA+B,EAAE;YACzI,MAAM,EAAE,EAAE,GAAG,qBAAqB,EAAE;SACrC,CAAC;IACJ,CAAC,EAAE,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,kDAAkD;IAClF,qBAAqB;IACrB,iIAAiI;IACjI,mCAAmC;IACnC,KAAK;IAEL,MAAM,EAAE,UAAU,EAAE,eAAe,EAAE,GAAG,uDAA6B,CAAC;QACpE,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,OAAO,EAAE,QAAQ,CAAC,OAAO;QACzB,mCAAmC,EAAE,CAAC,KAAK,CAAC,2BAA2B,KAAK,4BAAmB,CAAC,aAAa,CAAC;QAC9G,UAAU,EAAE,WAAW;QACvB,yBAAyB,EAAE,KAAK,CAAC,yBAAyB;KAC3D,CAAC,CAAC;IACH,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,eAAe,EAAE,qBAAqB,EAAE,GAAG,uDAA6B,CAAC;QAC7G,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,OAAO,EAAE,QAAQ,CAAC,MAAM;QACxB,UAAU,EAAE,WAAW;QACvB,yBAAyB,EAAE,KAAK,CAAC,yBAAyB;KAC3D,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,CAAA,MAAA,KAAK,CAAC,UAAU,0CAAE,MAAM,EAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,UAAU,CAAC;IAClF,MAAM,2BAA2B,GAAG,CAAA,MAAA,KAAK,CAAC,UAAU,0CAAE,MAAM,EAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,eAAe,CAAC;IAEvG,MAAM,IAAI,GAAG,oCAAY,CAAC,YAAY,CAAC,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;IACrF,IAAI,KAAK,CAAC,kBAAkB,EAAE;QAC5B,mEAAmE;QACnE,IAAI,CAAC,QAAQ,CAAC,sBAAsB,EAAE,KAAK,CAAC,kBAAkB,CAAC,CAAC;KACjE;SAAM;QACL,mEAAmE;QACnE,IAAI,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;KACpC;IAED,OAAO;QACL,UAAU,EAAE,gBAAgB;QAC5B,eAAe,EAAE,2BAA2B;KAC7C,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAC3B,SAAiB,EACjB,MAAwB,EACxB,UAAqB,EACrB,iBAA2C,EAC3C,oBAA4D,EAC5D,0BAAoC;IAEpC,MAAM,oBAAoB,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,8CAAoB,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7F,MAAM,wBAAwB,GAAG,kCAAqB,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE;QAC5E,IAAI,UAAU;YACZ,OAAO,uBAAuB,CAAC,SAAS,EAAE,UAAU,EAAE,oBAAoB,EAAE,0BAA0B,CAAC,CAAC;QAC1G,OAAO,SAAS,CAAC;IACnB,CAAC,EAAE,CAAC,SAAS,EAAE,UAAU,EAAE,oBAAoB,EAAE,0BAA0B,CAAC,CAAC,CAAC,CAAC;IAE/E,MAAM,OAAO,GAAG,iBAAiB,aAAjB,iBAAiB,cAAjB,iBAAiB,GAAI,wBAAwB,CAAC;IAE9D,KAAK,CAAC,SAAS,CAAC,GAAG,EAAE;QACnB,OAAO,IAAI,OAAO,CAAC,uBAAuB,CAAC,oBAAoB,CAAC,CAAC;IACnE,CAAC,EAAE,CAAC,OAAO,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAEpC,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,uBAAuB,GAAG,CAAC,SAAiB,EAAE,UAAoB,EAAE,oBAA0C,EAAE,0BAAoC,EAAuC,EAAE;IACjM,uBAAuB;IACvB,OAAO,IAAI,iDAAuB,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,UAAU,EAAE,0BAA0B,EAAE,oBAAoB,EAAE,CAAC,CAAC;AAC5H,CAAC,CAAC;AAEF,MAAM,sBAAsB,GAAG,CAAC,YAAmF,EAAyD,EAAE;IAC5K,MAAM,gBAAgB,GAAG,YAAqD,CAAC;IAC/E,OAAO,gBAAgB,CAAC,iBAAiB,KAAK,SAAS,IAAI,gBAAgB,CAAC,cAAc,KAAK,SAAS,IAAI,gBAAgB,CAAC,qBAAqB,KAAK,SAAS,CAAC;AACnK,CAAC,CAAC;AAEF,MAAM,uBAAuB,GAAG,CAAC,YAAmF,EAAqD,EAAE;IACzK,OAAO,sBAAsB,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC;AACzE,CAAC,CAAC","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 \"./ModelsTree.scss\";\nimport * as React from \"react\";\nimport { ControlledTree, SelectionMode, useTreeModel } from \"@itwin/components-react\";\nimport { useDisposable, useOptionalDisposable } from \"@itwin/core-react\";\nimport { usePresentationTreeNodeLoader } from \"@itwin/presentation-components\";\nimport { Presentation } from \"@itwin/presentation-frontend\";\nimport { TreeWidget } from \"../../../TreeWidget\";\nimport { ClassGroupingOption } from \"../Common\";\nimport { VisibilityTreeEventHandler } from \"../VisibilityTreeEventHandler\";\nimport { useVisibilityTreeFiltering, useVisibilityTreeRenderer, VisibilityTreeNoFilteredData } from \"../VisibilityTreeRenderer\";\nimport { ModelsVisibilityHandler, SubjectModelIdsCache } from \"./ModelsVisibilityHandler\";\nimport type { VisibilityTreeFilterInfo } from \"../Common\";\nimport type { ModelsTreeSelectionPredicate } from \"./ModelsVisibilityHandler\";\nimport type { TreeNodeItem } from \"@itwin/components-react\";\nimport type { Id64Array } from \"@itwin/core-bentley\";\nimport type { IModelConnection, Viewport } from \"@itwin/core-frontend\";\nimport type { NodeKey, Ruleset } from \"@itwin/presentation-common\";\nimport type { IFilteredPresentationTreeDataProvider, IPresentationTreeDataProvider } from \"@itwin/presentation-components\";\n\nconst PAGING_SIZE = 20;\n\n/** @internal */\nexport const RULESET_MODELS: Ruleset = require(\"./Hierarchy.json\"); // eslint-disable-line @typescript-eslint/no-var-requires\n/** @internal */\nexport const RULESET_MODELS_GROUPED_BY_CLASS: Ruleset = require(\"./Hierarchy.GroupedByClass.json\"); // eslint-disable-line @typescript-eslint/no-var-requires\n\nconst RULESET_MODELS_SEARCH: Ruleset = require(\"./ModelsTreeSearch.json\"); // eslint-disable-line @typescript-eslint/no-var-requires\n\n/** Props for [[ModelsTree]] component\n * @public\n */\nexport interface ModelsTreeProps {\n  /**\n   * An IModel to pull data from\n   */\n  iModel: IModelConnection;\n  /** Width of the component */\n  width: number;\n  /** Height of the component */\n  height: number;\n  /**\n   * Selection mode in the tree\n   */\n  selectionMode?: SelectionMode;\n  /**\n   * Predicate which indicates whether node can be selected or no\n   * @alpha\n   */\n  selectionPredicate?: ModelsTreeSelectionPredicate;\n  /**\n   * Active view used to determine and control visibility\n   */\n  activeView?: Viewport;\n  /**\n   * Ref to the root HTML element used by this component\n   */\n  rootElementRef?: React.Ref<HTMLDivElement>;\n  /**\n   * Information for tree filtering.\n   * @alpha\n   */\n  filterInfo?: VisibilityTreeFilterInfo;\n  /**\n   * Filter the hierarchy by given element IDs.\n   * @alpha\n   */\n  filteredElementIds?: Id64Array;\n  /**\n   * Callback invoked when tree is filtered.\n   */\n  onFilterApplied?: (filteredDataProvider: IPresentationTreeDataProvider, matchesCount: number) => void;\n  /**\n   * Should the tree group displayed element nodes by class.\n   * @beta\n   */\n  enableElementsClassGrouping?: ClassGroupingOption;\n  /**\n   * Auto-update the hierarchy when data in the iModel changes.\n   * @alpha\n   */\n  enableHierarchyAutoUpdate?: boolean;\n  /**\n   * Custom visibility handler.\n   * @alpha\n   */\n  modelsVisibilityHandler?: ModelsVisibilityHandler;\n}\n\n/**\n * A tree component that shows a subject - model - category - element\n * hierarchy along with checkboxes that represent and allow changing\n * the display of those instances.\n * @public\n */\nexport function ModelsTree(props: ModelsTreeProps) {\n  const { nodeLoader, onItemsRendered } = useModelsTreeNodeLoader(props);\n  const { filteredNodeLoader, isFiltering, nodeHighlightingProps } = useVisibilityTreeFiltering(nodeLoader, props.filterInfo, props.onFilterApplied);\n  const filterApplied = filteredNodeLoader !== nodeLoader;\n\n  const { activeView, modelsVisibilityHandler, selectionPredicate } = props;\n  const nodeSelectionPredicate = React.useCallback((key: NodeKey, node: TreeNodeItem) => {\n    return !selectionPredicate ? true : selectionPredicate(key, ModelsVisibilityHandler.getNodeType(node, nodeLoader.dataProvider));\n  }, [selectionPredicate, nodeLoader.dataProvider]);\n\n  const visibilityHandler = useVisibilityHandler(\n    nodeLoader.dataProvider.rulesetId,\n    props.iModel,\n    activeView,\n    modelsVisibilityHandler,\n    getFilteredDataProvider(filteredNodeLoader.dataProvider),\n    props.enableHierarchyAutoUpdate);\n  const eventHandler = useDisposable(React.useCallback(() => new VisibilityTreeEventHandler({\n    nodeLoader: filteredNodeLoader,\n    visibilityHandler,\n    collapsedChildrenDisposalEnabled: true,\n    selectionPredicate: nodeSelectionPredicate,\n  }), [filteredNodeLoader, visibilityHandler, nodeSelectionPredicate]));\n\n  const treeModel = useTreeModel(filteredNodeLoader.modelSource);\n  const treeRenderer = useVisibilityTreeRenderer(true, false);\n\n  const overlay = isFiltering ? <div className=\"filteredTreeOverlay\" /> : undefined;\n\n  // istanbul ignore next\n  const noFilteredDataRenderer = React.useCallback(() => {\n    return <VisibilityTreeNoFilteredData\n      title={TreeWidget.translate(\"modelTree.noModelFound\")}\n      message={TreeWidget.translate(\"modelTree.noMatchingModelNames\")}\n    />;\n  }, []);\n\n  return (\n    <div className=\"tree-widget-models-tree\" ref={props.rootElementRef}>\n      <ControlledTree\n        nodeLoader={filteredNodeLoader}\n        model={treeModel}\n        selectionMode={props.selectionMode || SelectionMode.None}\n        eventsHandler={eventHandler}\n        treeRenderer={treeRenderer}\n        nodeHighlightingProps={nodeHighlightingProps}\n        noDataRenderer={filterApplied ? noFilteredDataRenderer : undefined}\n        onItemsRendered={onItemsRendered}\n        width={props.width}\n        height={props.height}\n      />\n      {overlay}\n    </div>\n  );\n}\n\nfunction useModelsTreeNodeLoader(props: ModelsTreeProps) {\n  // note: this is a temporary workaround for auto-update not working on ruleset variable changes - instead\n  // of auto-updating we just re-create the node loader by re-creating the ruleset\n  const rulesets = React.useMemo(() => {\n    return {\n      general: (!props.enableElementsClassGrouping) ? { ...RULESET_MODELS } : /* istanbul ignore next */ { ...RULESET_MODELS_GROUPED_BY_CLASS },\n      search: { ...RULESET_MODELS_SEARCH },\n    };\n  }, [props.filteredElementIds]); // eslint-disable-line react-hooks/exhaustive-deps\n  // const rulesets = {\n  //   general: (!props.enableElementsClassGrouping) ? RULESET_MODELS : /* istanbul ignore next */ RULESET_MODELS_GROUPED_BY_CLASS,\n  //   search: RULESET_MODELS_SEARCH,\n  // };\n\n  const { nodeLoader, onItemsRendered } = usePresentationTreeNodeLoader({\n    imodel: props.iModel,\n    ruleset: rulesets.general,\n    appendChildrenCountForGroupingNodes: (props.enableElementsClassGrouping === ClassGroupingOption.YesWithCounts),\n    pagingSize: PAGING_SIZE,\n    enableHierarchyAutoUpdate: props.enableHierarchyAutoUpdate,\n  });\n  const { nodeLoader: searchNodeLoader, onItemsRendered: onSearchItemsRendered } = usePresentationTreeNodeLoader({\n    imodel: props.iModel,\n    ruleset: rulesets.search,\n    pagingSize: PAGING_SIZE,\n    enableHierarchyAutoUpdate: props.enableHierarchyAutoUpdate,\n  });\n\n  const activeNodeLoader = props.filterInfo?.filter ? searchNodeLoader : nodeLoader;\n  const activeItemsRenderedCallback = props.filterInfo?.filter ? onSearchItemsRendered : onItemsRendered;\n\n  const vars = Presentation.presentation.vars(activeNodeLoader.dataProvider.rulesetId);\n  if (props.filteredElementIds) {\n    // eslint-disable-next-line @typescript-eslint/no-floating-promises\n    vars.setId64s(\"filtered-element-ids\", props.filteredElementIds);\n  } else {\n    // eslint-disable-next-line @typescript-eslint/no-floating-promises\n    vars.unset(\"filtered-element-ids\");\n  }\n\n  return {\n    nodeLoader: activeNodeLoader,\n    onItemsRendered: activeItemsRenderedCallback,\n  };\n}\n\nfunction useVisibilityHandler(\n  rulesetId: string,\n  iModel: IModelConnection,\n  activeView?: Viewport,\n  visibilityHandler?: ModelsVisibilityHandler,\n  filteredDataProvider?: IFilteredPresentationTreeDataProvider,\n  hierarchyAutoUpdateEnabled?: boolean,\n) {\n  const subjectModelIdsCache = React.useMemo(() => new SubjectModelIdsCache(iModel), [iModel]);\n  const defaultVisibilityHandler = useOptionalDisposable(React.useCallback(() => {\n    if (activeView)\n      return createVisibilityHandler(rulesetId, activeView, subjectModelIdsCache, hierarchyAutoUpdateEnabled);\n    return undefined;\n  }, [rulesetId, activeView, subjectModelIdsCache, hierarchyAutoUpdateEnabled]));\n\n  const handler = visibilityHandler ?? defaultVisibilityHandler;\n\n  React.useEffect(() => {\n    handler && handler.setFilteredDataProvider(filteredDataProvider);\n  }, [handler, filteredDataProvider]);\n\n  return handler;\n}\n\nconst createVisibilityHandler = (rulesetId: string, activeView: Viewport, subjectModelIdsCache: SubjectModelIdsCache, hierarchyAutoUpdateEnabled?: boolean): ModelsVisibilityHandler | undefined => {\n  // istanbul ignore next\n  return new ModelsVisibilityHandler({ rulesetId, viewport: activeView, hierarchyAutoUpdateEnabled, subjectModelIdsCache });\n};\n\nconst isFilteredDataProvider = (dataProvider: IPresentationTreeDataProvider | IFilteredPresentationTreeDataProvider): dataProvider is IFilteredPresentationTreeDataProvider => {\n  const filteredProvider = dataProvider as IFilteredPresentationTreeDataProvider;\n  return filteredProvider.nodeMatchesFilter !== undefined && filteredProvider.getActiveMatch !== undefined && filteredProvider.countFilteringResults !== undefined;\n};\n\nconst getFilteredDataProvider = (dataProvider: IPresentationTreeDataProvider | IFilteredPresentationTreeDataProvider): IFilteredPresentationTreeDataProvider | undefined => {\n  return isFilteredDataProvider(dataProvider) ? dataProvider : undefined;\n};\n"]}

package/lib/cjs/components/trees/{ModelsTree.scss → models-tree/ModelsTree.scss}

@@ -2,8 +2,12 @@
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
-
 @import "~@itwin/core-react/lib/cjs/core-react/index";
+@import "../VisibilityTreeBase";
+
+.tree-widget-models-tree {
+  @include tree-widget-visibility-tree-base;
+}
 
 .tree-widget-models-tree-container {
   flex: 1;

package/lib/{esm/components/trees/ModelsTree.d.ts → cjs/components/trees/models-tree/ModelsTreeComponent.d.ts}

@@ -1,5 +1,5 @@
 /// <reference types="react" />
 import "./ModelsTree.scss";
-import type { ModelTreeProps } from "../../types";
+import type { ModelTreeProps } from "../../../types";
 export declare const ModelsTreeComponent: (props: ModelTreeProps) => JSX.Element;
-//# sourceMappingURL=ModelsTree.d.ts.map
+//# sourceMappingURL=ModelsTreeComponent.d.ts.map