@itwin/tree-widget-react 3.0.0-dev.0 → 3.0.0-dev.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@itwin/tree-widget-react",
-  "version": "3.0.0-dev.0",
+  "version": "3.0.0-dev.2",
   "description": "Tree Widget React",
   "keywords": [
     "Bentley",
@@ -33,12 +33,14 @@
     "copy:esm": "cpx \"./src/**/*.{scss,json}\" ./lib/esm",
     "cover": "nyc npm run test",
     "extract-api": "betools extract-api --entry=tree-widget-react --apiReportFolder=./api --apiReportTempFolder=./api/temp --apiSummaryFolder=./api",
+    "check-internal": "node ../../../scripts/checkInternal.js --apiSummary ./api/tree-widget-react.api.md",
     "lint": "npm run lint:eslint && npm run lint:stylelint",
     "lint:eslint": "eslint -f visualstudio \"./src/**/*.{ts,tsx}\" 1>&2",
     "lint:stylelint": "stylelint \"./src/**/*.scss\"",
     "lint:fix": "npm run lint:eslint -- --fix && npm run lint:stylelint -- --fix",
     "pseudolocalize": "betools pseudolocalize --englishDir ./public/locales/en --out ./lib/public/locales/en-PSEUDO",
     "test": "mocha \"./lib/cjs/test/**/*.test.js\"",
+    "test:unit": "mocha \"./lib/cjs/test/**/*.test.js\" -i -g \"#integration\"",
     "test:e2e": "node ../../../scripts/run-e2e-with-docker.js tree-widget",
     "test:e2e:local": "node ../../../scripts/run-e2e-tests.js",
     "test:e2e:debug": "cross-env PWDEBUG=1 playwright test --ui",
@@ -60,12 +62,12 @@
     "@bentley/icons-generic": "^1.0.34",
     "@itwin/itwinui-icons-react": "^2.8.0",
     "@itwin/itwinui-illustrations-react": "^2.1.0",
-    "@itwin/itwinui-react": "^3.5.0",
-    "@itwin/presentation-core-interop": "^0.1.2",
-    "@itwin/presentation-hierarchies": "^0.1.1",
-    "@itwin/presentation-hierarchies-react": "^0.2.0",
-    "@itwin/presentation-shared": "^0.1.1",
-    "@itwin/unified-selection": "^0.4.1",
+    "@itwin/itwinui-react": "^3.11.1",
+    "@itwin/presentation-core-interop": "^0.2.3",
+    "@itwin/presentation-hierarchies": "^0.2.0",
+    "@itwin/presentation-hierarchies-react": "^0.5.3",
+    "@itwin/presentation-shared": "^0.3.1",
+    "@itwin/unified-selection": "^0.4.4",
     "classnames": "^2.3.1",
     "i18next": "^10.2.2",
     "react-error-boundary": "^4.0.10",
@@ -88,19 +90,22 @@
     "@itwin/core-react": "^4.10.0",
     "@itwin/core-telemetry": "^4.5.1",
     "@itwin/ecschema-metadata": "^4.5.1",
+    "@itwin/ecschema-rpcinterface-common": "^4.5.1",
+    "@itwin/ecschema-rpcinterface-impl": "^4.5.1",
     "@itwin/eslint-plugin": "^4.0.0-dev.38",
     "@itwin/imodel-components-react": "^4.10.0",
     "@itwin/presentation-backend": "^4.5.1",
     "@itwin/presentation-common": "^4.5.1",
-    "@itwin/presentation-components": "^5.2.0",
+    "@itwin/presentation-components": "^5.3.3",
     "@itwin/presentation-frontend": "^4.5.1",
-    "@itwin/presentation-testing": "^5.0.3",
+    "@itwin/presentation-testing": "^5.0.13",
     "@itwin/webgl-compatibility": "^4.5.1",
     "@playwright/test": "^1.41.2",
     "@testing-library/dom": "^9.3.3",
     "@testing-library/react": "^14.1.2",
     "@testing-library/user-event": "^14.5.2",
     "@types/chai": "4.3.5",
+    "@types/chai-as-promised": "^7.1.1",
     "@types/chai-jest-snapshot": "^1.3.4",
     "@types/chai-subset": "1.3.3",
     "@types/deep-equal": "^1.0.1",
@@ -115,6 +120,7 @@
     "@typescript-eslint/eslint-plugin": "^5.59.8",
     "@typescript-eslint/parser": "^5.59.8",
     "chai": "^4.3.7",
+    "chai-as-promised": "^7.1.1",
     "chai-jest-snapshot": "^2.0.0",
     "chai-subset": "1.6.0",
     "cpx2": "^3.0.0",
@@ -126,6 +132,7 @@
     "eslint-plugin-import": "^2.29.0",
     "eslint-plugin-react": "^7.33.2",
     "eslint-plugin-unused-imports": "^2.0.0",
+    "fast-xml-parser": "^4.3.6",
     "global-jsdom": "^9.2.0",
     "ignore-styles": "^5.0.1",
     "jsdom": "^23.1.0",
@@ -138,6 +145,7 @@
     "react-redux": "^7.2.9",
     "redux": "^4.1.0",
     "rimraf": "^3.0.2",
+    "sanitize-filename": "^1.6.3",
     "sinon": "^17.0.1",
     "sinon-chai": "^3.7.0",
     "source-map-support": "^0.5.6",
@@ -145,8 +153,7 @@
     "stylelint-config-standard-scss": "^11.1.0",
     "typemoq": "^2.1.0",
     "typescript": "~5.0.0",
-    "xmlhttprequest": "^1.8.0",
-    "sanitize-filename": "^1.6.3"
+    "xmlhttprequest": "^1.8.0"
   },
   "eslintConfig": {
     "extends": [

package/lib/cjs/components/TreeFilteringState.d.ts

@@ -1,18 +0,0 @@
-import type { IPresentationTreeDataProvider } from "@itwin/presentation-components";
-/** @internal */
-export interface SearchOptions {
-    isFiltering: boolean;
-    onFilterCancel: () => void;
-    onFilterStart: (newFilter: string) => void;
-    onResultSelectedChanged: (index: number) => void;
-    matchedResultCount: number | undefined;
-    activeMatchIndex: number | undefined;
-}
-/** @internal */
-export declare const useTreeFilteringState: () => {
-    searchOptions: SearchOptions;
-    filterString: string;
-    onFilterApplied: (provider: IPresentationTreeDataProvider, matches: number) => void;
-    filteredProvider: IPresentationTreeDataProvider | undefined;
-};
-//# sourceMappingURL=TreeFilteringState.d.ts.map

package/lib/cjs/components/TreeFilteringState.js

@@ -1,46 +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.useTreeFilteringState = void 0;
-const react_1 = require("react");
-/** @internal */
-const useTreeFilteringState = () => {
-    const [{ filterString, matchedResultCount, activeMatchIndex, filteredProvider }, setState] = (0, react_1.useState)({ filterString: "" });
-    const onFilterCancel = (0, react_1.useCallback)(() => {
-        setState({ filterString: "" });
-    }, []);
-    const onFilterStart = (0, react_1.useCallback)((newFilter) => {
-        setState({ filterString: newFilter });
-    }, []);
-    const onResultSelectedChanged = (0, react_1.useCallback)((index) => {
-        setState((prev) => ({ ...prev, activeMatchIndex: index }));
-    }, []);
-    const onFilterApplied = (0, react_1.useCallback)((provider, matches) => {
-        setState((prev) => ({
-            ...prev,
-            activeMatchIndex: prev.activeMatchIndex === undefined ? 1 : Math.min(prev.activeMatchIndex, matches),
-            matchedResultCount: matches,
-            filteredProvider: provider,
-        }));
-    }, []);
-    const isFiltering = !!filterString && matchedResultCount === undefined;
-    const searchOptions = {
-        isFiltering,
-        onFilterCancel,
-        onFilterStart,
-        onResultSelectedChanged,
-        matchedResultCount,
-        activeMatchIndex,
-    };
-    return {
-        searchOptions,
-        filterString,
-        onFilterApplied,
-        filteredProvider,
-    };
-};
-exports.useTreeFilteringState = useTreeFilteringState;
-//# sourceMappingURL=TreeFilteringState.js.map

package/lib/cjs/components/TreeFilteringState.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"TreeFilteringState.js","sourceRoot":"","sources":["../../../src/components/TreeFilteringState.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAA8C;AAqB9C,gBAAgB;AACT,MAAM,qBAAqB,GAAG,GAAG,EAAE;IACxC,MAAM,CAAC,EAAE,YAAY,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAAqB,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;IAEhJ,MAAM,cAAc,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QACtC,QAAQ,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;IACjC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,aAAa,GAAG,IAAA,mBAAW,EAAC,CAAC,SAAiB,EAAE,EAAE;QACtD,QAAQ,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC;IACxC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,uBAAuB,GAAG,IAAA,mBAAW,EAAC,CAAC,KAAa,EAAE,EAAE;QAC5D,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IAC7D,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,eAAe,GAAG,IAAA,mBAAW,EAAC,CAAC,QAAuC,EAAE,OAAe,EAAE,EAAE;QAC/F,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YAClB,GAAG,IAAI;YACP,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE,OAAO,CAAC;YACpG,kBAAkB,EAAE,OAAO;YAC3B,gBAAgB,EAAE,QAAQ;SAC3B,CAAC,CAAC,CAAC;IACN,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,CAAC,CAAC,YAAY,IAAI,kBAAkB,KAAK,SAAS,CAAC;IACvE,MAAM,aAAa,GAAkB;QACnC,WAAW;QACX,cAAc;QACd,aAAa;QACb,uBAAuB;QACvB,kBAAkB;QAClB,gBAAgB;KACjB,CAAC;IAEF,OAAO;QACL,aAAa;QACb,YAAY;QACZ,eAAe;QACf,gBAAgB;KACjB,CAAC;AACJ,CAAC,CAAC;AAxCW,QAAA,qBAAqB,yBAwChC","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 { useCallback, useState } from \"react\";\n\nimport type { IPresentationTreeDataProvider } from \"@itwin/presentation-components\";\n\n/** @internal */\nexport interface SearchOptions {\n  isFiltering: boolean;\n  onFilterCancel: () => void;\n  onFilterStart: (newFilter: string) => void;\n  onResultSelectedChanged: (index: number) => void;\n  matchedResultCount: number | undefined;\n  activeMatchIndex: number | undefined;\n}\n\ninterface TreeFilteringState {\n  filterString: string;\n  matchedResultCount?: number;\n  activeMatchIndex?: number;\n  filteredProvider?: IPresentationTreeDataProvider;\n}\n\n/** @internal */\nexport const useTreeFilteringState = () => {\n  const [{ filterString, matchedResultCount, activeMatchIndex, filteredProvider }, setState] = useState<TreeFilteringState>({ filterString: \"\" });\n\n  const onFilterCancel = useCallback(() => {\n    setState({ filterString: \"\" });\n  }, []);\n\n  const onFilterStart = useCallback((newFilter: string) => {\n    setState({ filterString: newFilter });\n  }, []);\n\n  const onResultSelectedChanged = useCallback((index: number) => {\n    setState((prev) => ({ ...prev, activeMatchIndex: index }));\n  }, []);\n\n  const onFilterApplied = useCallback((provider: IPresentationTreeDataProvider, matches: number) => {\n    setState((prev) => ({\n      ...prev,\n      activeMatchIndex: prev.activeMatchIndex === undefined ? 1 : Math.min(prev.activeMatchIndex, matches),\n      matchedResultCount: matches,\n      filteredProvider: provider,\n    }));\n  }, []);\n\n  const isFiltering = !!filterString && matchedResultCount === undefined;\n  const searchOptions: SearchOptions = {\n    isFiltering,\n    onFilterCancel,\n    onFilterStart,\n    onResultSelectedChanged,\n    matchedResultCount,\n    activeMatchIndex,\n  };\n\n  return {\n    searchOptions,\n    filterString,\n    onFilterApplied,\n    filteredProvider,\n  };\n};\n"]}

package/lib/cjs/components/trees/CategoriesVisibilityUtils.d.ts

@@ -1,25 +0,0 @@
-import type { CategoryInfo } from "./category-tree/CategoriesTreeButtons";
-import type { IModelConnection, ViewManager, Viewport } from "@itwin/core-frontend";
-/**
- * Toggles visibility of categories to show or hide.
- * @internal
- */
-export declare function toggleAllCategories(viewManager: ViewManager, imodel: IModelConnection, display: boolean, viewport?: Viewport, forAllViewports?: boolean): Promise<void>;
-/**
- * Gets ids of all categories from specified imodel and viewport.
- * @internal
- */
-export declare function getCategories(imodel: IModelConnection, viewport?: Viewport): Promise<string[]>;
-/**
- * Changes category display in the viewport.
- * @internal
- */
-export declare function enableCategory(viewManager: ViewManager, imodel: IModelConnection, ids: string[], enabled: boolean, forAllViewports: boolean, enableAllSubCategories?: boolean): Promise<void>;
-/**
- * Changes subcategory display in the viewport
- * @internal
- */
-export declare function enableSubCategory(viewManager: ViewManager, key: string, enabled: boolean, forAllViewports?: boolean): void;
-/** @internal */
-export declare function loadCategoriesFromViewport(iModel?: IModelConnection, vp?: Viewport): Promise<CategoryInfo[]>;
-//# sourceMappingURL=CategoriesVisibilityUtils.d.ts.map

package/lib/cjs/components/trees/CategoriesVisibilityUtils.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"CategoriesVisibilityUtils.js","sourceRoot":"","sources":["../../../../src/components/trees/CategoriesVisibilityUtils.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,oDAAoD;AACpD,wDAAkE;AAKlE,MAAM,sBAAsB,GAAmB,EAAE,CAAC;AAElD;;;GAGG;AACI,KAAK,UAAU,mBAAmB,CACvC,WAAwB,EACxB,MAAwB,EACxB,OAAgB,EAChB,QAAmB,EACnB,eAAyB;IAEzB,uBAAuB;IACvB,MAAM,UAAU,GAAG,QAAQ,IAAI,WAAW,CAAC,gBAAgB,EAAE,CAAC;IAC9D,MAAM,GAAG,GAAG,MAAM,aAAa,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IAEpD,qBAAqB;IACrB,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE;QACpB,OAAO;KACR;IAED,MAAM,cAAc,CAAC,WAAW,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,eAAe,IAAI,KAAK,CAAC,CAAC;AACpF,CAAC;AAjBD,kDAiBC;AAED;;;GAGG;AACI,KAAK,UAAU,aAAa,CAAC,MAAwB,EAAE,QAAmB;IAC/E,MAAM,UAAU,GAAG,MAAM,0BAA0B,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACtE,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;AAC3D,CAAC;AAHD,sCAGC;AAED;;;GAGG;AACI,KAAK,UAAU,cAAc,CAClC,WAAwB,EACxB,MAAwB,EACxB,GAAa,EACb,OAAgB,EAChB,eAAwB,EACxB,sBAAsB,GAAG,IAAI;IAE7B,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;QAC7B,OAAO;KACR;IAED,MAAM,cAAc,GAAG,KAAK,EAAE,EAAY,EAAE,EAAE;QAC5C,2GAA2G;QAC3G,oCAAoC;QACpC,IAAI,WAAW,CAAC,YAAY,IAAI,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE;YACvF,EAAE,CAAC,qBAAqB,CAAC,GAAG,EAAE,OAAO,EAAE,sBAAsB,CAAC,CAAC;YAE/D,sCAAsC;YACtC,MAAM,yBAAyB,GAAa,EAAE,CAAC;YAC/C,KAAK,MAAM,GAAG,IAAI,EAAE,CAAC,0BAA0B,EAAE;gBAC/C,uBAAuB;gBACvB,IAAI,GAAG,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,KAAK,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE;oBACvD,yBAAyB,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;iBAC7C;aACF;YACD,EAAE,CAAC,0BAA0B,CAAC,WAAW,CAAC,yBAAyB,EAAE,GAAG,EAAE,0CAA0B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAEpH,6GAA6G;YAC7G,IAAI,KAAK,KAAK,OAAO,EAAE;gBACrB,CAAC,MAAM,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,YAAY,EAAE,EAAE;oBACtE,YAAY,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,EAAE,eAAe,CAAC,CAAC,CAAC;gBAClH,CAAC,CAAC,CAAC;aACJ;SACF;IACH,CAAC,CAAC;IAEF,iGAAiG;IACjG,IAAI,eAAe,EAAE;QACnB,KAAK,MAAM,QAAQ,IAAI,WAAW,EAAE;YAClC,MAAM,cAAc,CAAC,QAAQ,CAAC,CAAC;SAChC;KACF;SAAM;QACL,MAAM,cAAc,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;KAChD;AACH,CAAC;AA7CD,wCA6CC;AAED;;;GAGG;AACH,SAAgB,iBAAiB,CAAC,WAAwB,EAAE,GAAW,EAAE,OAAgB,EAAE,eAAyB;IAClH,IAAI,CAAC,WAAW,CAAC,YAAY,EAAE;QAC7B,OAAO;KACR;IAED,MAAM,cAAc,GAAG,CAAC,EAAY,EAAE,EAAE;QACtC,2GAA2G;QAC3G,oCAAoC;QACpC,IAAI,WAAW,CAAC,YAAY,IAAI,WAAW,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE;YACvF,EAAE,CAAC,wBAAwB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;SAC3C;IACH,CAAC,CAAC;IAEF,iGAAiG;IACjG,IAAI,eAAe,EAAE;QACnB,KAAK,MAAM,QAAQ,IAAI,WAAW,EAAE;YAClC,cAAc,CAAC,QAAQ,CAAC,CAAC;SAC1B;KACF;SAAM;QACL,cAAc,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;KAC1C;AACH,CAAC;AArBD,8CAqBC;AAED,gBAAgB;AACT,KAAK,UAAU,0BAA0B,CAAC,MAAyB,EAAE,EAAa;IACvF,IAAI,CAAC,EAAE,EAAE;QACP,OAAO,sBAAsB,CAAC;KAC/B;IAED,yCAAyC;IACzC,MAAM,4BAA4B,GAChC,2IAA2I,CAAC;IAC9I,MAAM,4BAA4B,GAChC,0JAA0J,CAAC;IAC7J,MAAM,KAAK,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,4BAA4B,CAAC;IAC3F,MAAM,MAAM,GAAG,kCAAkC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,yBAAyB,2BAA2B,KAAK,GAAG,CAAC;IAE3J,MAAM,UAAU,GAAmB,EAAE,CAAC;IAEtC,uBAAuB;IACvB,IAAI,MAAM,EAAE;QACV,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,MAAM,EAAE,SAAS,EAAE,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;QAC3H,CAAC,MAAM,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE;YACnF,UAAU,CAAC,IAAI,CAAC,EAAE,UAAU,EAAE,GAAG,CAAC,EAAE,EAAE,cAAc,EAAE,GAAG,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;QAC9H,CAAC,CAAC,CAAC;KACJ;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAvBD,gEAuBC","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 { QueryRowFormat } from \"@itwin/core-common\";\nimport { PerModelCategoryVisibility } from \"@itwin/core-frontend\";\n\nimport type { CategoryInfo } from \"./category-tree/CategoriesTreeButtons\";\nimport type { IModelConnection, ViewManager, Viewport } from \"@itwin/core-frontend\";\n\nconst EMPTY_CATEGORIES_ARRAY: CategoryInfo[] = [];\n\n/**\n * Toggles visibility of categories to show or hide.\n * @internal\n */\nexport async function toggleAllCategories(\n  viewManager: ViewManager,\n  imodel: IModelConnection,\n  display: boolean,\n  viewport?: Viewport,\n  forAllViewports?: boolean,\n) {\n  // istanbul ignore next\n  const activeView = viewport ?? viewManager.getFirstOpenView();\n  const ids = await getCategories(imodel, activeView);\n\n  // istanbul ignore if\n  if (ids.length === 0) {\n    return;\n  }\n\n  await enableCategory(viewManager, imodel, ids, display, forAllViewports ?? false);\n}\n\n/**\n * Gets ids of all categories from specified imodel and viewport.\n * @internal\n */\nexport async function getCategories(imodel: IModelConnection, viewport?: Viewport) {\n  const categories = await loadCategoriesFromViewport(imodel, viewport);\n  return categories.map((category) => category.categoryId);\n}\n\n/**\n * Changes category display in the viewport.\n * @internal\n */\nexport async function enableCategory(\n  viewManager: ViewManager,\n  imodel: IModelConnection,\n  ids: string[],\n  enabled: boolean,\n  forAllViewports: boolean,\n  enableAllSubCategories = true,\n) {\n  if (!viewManager.selectedView) {\n    return;\n  }\n\n  const updateViewport = async (vp: Viewport) => {\n    // Only act on viewports that are both 3D or both 2D. Important if we have multiple viewports opened and we\n    // are using 'allViewports' property\n    if (viewManager.selectedView && viewManager.selectedView.view.is3d() === vp.view.is3d()) {\n      vp.changeCategoryDisplay(ids, enabled, enableAllSubCategories);\n\n      // remove category overrides per model\n      const modelsContainingOverrides: string[] = [];\n      for (const ovr of vp.perModelCategoryVisibility) {\n        // istanbul ignore else\n        if (ids.findIndex((id) => id === ovr.categoryId) !== -1) {\n          modelsContainingOverrides.push(ovr.modelId);\n        }\n      }\n      vp.perModelCategoryVisibility.setOverride(modelsContainingOverrides, ids, PerModelCategoryVisibility.Override.None);\n\n      // changeCategoryDisplay only enables subcategories, it does not disabled them. So we must do that ourselves.\n      if (false === enabled) {\n        (await imodel.categories.getCategoryInfo(ids)).forEach((categoryInfo) => {\n          categoryInfo.subCategories.forEach((value) => enableSubCategory(viewManager, value.id, false, forAllViewports));\n        });\n      }\n    }\n  };\n\n  // This property let us act on all viewports or just on the selected one, configurable by the app\n  if (forAllViewports) {\n    for (const viewport of viewManager) {\n      await updateViewport(viewport);\n    }\n  } else {\n    await updateViewport(viewManager.selectedView);\n  }\n}\n\n/**\n * Changes subcategory display in the viewport\n * @internal\n */\nexport function enableSubCategory(viewManager: ViewManager, key: string, enabled: boolean, forAllViewports?: boolean) {\n  if (!viewManager.selectedView) {\n    return;\n  }\n\n  const updateViewport = (vp: Viewport) => {\n    // Only act on viewports that are both 3D or both 2D. Important if we have multiple viewports opened and we\n    // are using 'allViewports' property\n    if (viewManager.selectedView && viewManager.selectedView.view.is3d() === vp.view.is3d()) {\n      vp.changeSubCategoryDisplay(key, enabled);\n    }\n  };\n\n  // This property let us act on all viewports or just on the selected one, configurable by the app\n  if (forAllViewports) {\n    for (const viewport of viewManager) {\n      updateViewport(viewport);\n    }\n  } else {\n    updateViewport(viewManager.selectedView);\n  }\n}\n\n/** @internal */\nexport async function loadCategoriesFromViewport(iModel?: IModelConnection, vp?: Viewport) {\n  if (!vp) {\n    return EMPTY_CATEGORIES_ARRAY;\n  }\n\n  // Query categories and add them to state\n  const selectUsedSpatialCategoryIds =\n    \"SELECT DISTINCT Category.Id as id from BisCore.GeometricElement3d WHERE Category.Id IN (SELECT ECInstanceId from BisCore.SpatialCategory)\";\n  const selectUsedDrawingCategoryIds =\n    \"SELECT DISTINCT Category.Id as id from BisCore.GeometricElement2d WHERE Model.Id=? AND Category.Id IN (SELECT ECInstanceId from BisCore.DrawingCategory)\";\n  const ecsql = vp.view.is3d() ? selectUsedSpatialCategoryIds : selectUsedDrawingCategoryIds;\n  const ecsql2 = `SELECT ECInstanceId as id FROM ${vp.view.is3d() ? \"BisCore.SpatialCategory\" : \"BisCore.DrawingCategory\"} WHERE ECInstanceId IN (${ecsql})`;\n\n  const categories: CategoryInfo[] = [];\n\n  // istanbul ignore else\n  if (iModel) {\n    const rows = await iModel.createQueryReader(ecsql2, undefined, { rowFormat: QueryRowFormat.UseJsPropertyNames }).toArray();\n    (await iModel.categories.getCategoryInfo(rows.map((row) => row.id))).forEach((val) => {\n      categories.push({ categoryId: val.id, subCategoryIds: val.subCategories.size ? [...val.subCategories.keys()] : undefined });\n    });\n  }\n  return categories;\n}\n"]}

package/lib/cjs/components/trees/VisibilityTreeBase.scss

@@ -1,82 +0,0 @@
-/*---------------------------------------------------------------------------------------------
-* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
-* See LICENSE.md in the project root for license terms and full copyright notice.
-*--------------------------------------------------------------------------------------------*/
-
-.tree-widget-visibility-tree-base {
-  .tree-widget-tree-nodes-list {
-    &.enlarge {
-      .without-expander {
-        .visibility-tree-checkbox-container + * {
-          margin-left: var(--enlarged-node-expander-additional-padding);
-        }
-      }
-
-      .visibility-tree-checkbox-container {
-        height: var(--enlarged-node-height);
-        width: var(--enlarged-node-height);
-
-        .visibility-tree-checkbox {
-          --iui-checkbox-target-size: var(--enlarged-node-height);
-        }
-      }
-    }
-
-    .core-tree-node {
-      &.with-checkbox {
-        > .contents {
-          margin-left: 0 !important;
-          padding-left: calc(var(--iui-size-3xs) / 2);
-        }
-      }
-
-      &.disable-expander {
-        > .contents {
-          /* stylelint-disable-next-line selector-class-pattern */
-          > .core-tree-expansionToggle {
-            display: none;
-          }
-        }
-      }
-
-      .visibility-tree-checkbox-container {
-        height: var(--iui-size-l);
-        width: var(--iui-size-l);
-        position: sticky;
-        order: -1;
-        left: 0;
-        background-color: var(--iui-color-background);
-        z-index: 1;
-        display: flex;
-        align-items: center;
-        justify-content: center;
-
-        .visibility-tree-checkbox {
-          &:disabled {
-            background-color: transparent;
-          }
-
-          &:hover {
-            background-color: transparent;
-          }
-        }
-      }
-
-      &.is-selected {
-        > .contents {
-          > .visibility-tree-checkbox-container {
-            background-color: var(--iui-color-background-accent-muted);
-          }
-        }
-      }
-
-      &:not(.is-selected) {
-        > .contents:hover {
-          > .visibility-tree-checkbox-container {
-            background-color: var(--iui-color-background-hover);
-          }
-        }
-      }
-    }
-  }
-}

package/lib/cjs/components/trees/VisibilityTreeEventHandler.d.ts

@@ -1,67 +0,0 @@
-/** @packageDocumentation
- * @module IModelComponents
- */
-import { UnifiedSelectionTreeEventHandler } from "@itwin/presentation-components";
-import type { BeEvent, IDisposable } from "@itwin/core-bentley";
-import type { TreeCheckboxStateChangeEventArgs, TreeNodeItem, TreeSelectionModificationEventArgs, TreeSelectionReplacementEventArgs } from "@itwin/components-react";
-import type { UnifiedSelectionTreeEventHandlerParams } from "@itwin/presentation-components";
-/**
- * Data structure that describes instance visibility status.
- * @public
- */
-export interface VisibilityStatus {
-    state: "visible" | "partial" | "hidden";
-    isDisabled?: boolean;
-    tooltip?: string;
-}
-/**
- * Type definition of visibility change event listener.
- * @public
- */
-export type VisibilityChangeListener = (nodeIds?: string[], visibilityStatus?: Map<string, VisibilityStatus>) => void;
-/**
- * Visibility handler used to change or get visibility of instances represented by the tree node.
- * @public
- */
-export interface IVisibilityHandler extends IDisposable {
-    getVisibilityStatus(node: TreeNodeItem): VisibilityStatus | Promise<VisibilityStatus>;
-    changeVisibility(node: TreeNodeItem, shouldDisplay: boolean): Promise<void>;
-    onVisibilityChange: BeEvent<VisibilityChangeListener>;
-}
-/**
- * Type definition of predicate used to decide if node can be selected.
- * @public
- */
-export type VisibilityTreeSelectionPredicate = (node: TreeNodeItem) => boolean;
-/**
- * Parameters for [[VisibilityTreeEventHandler]]
- * @public
- */
-export interface VisibilityTreeEventHandlerParams extends UnifiedSelectionTreeEventHandlerParams {
-    visibilityHandler: IVisibilityHandler;
-    selectionPredicate?: VisibilityTreeSelectionPredicate;
-}
-/**
- * Base event handler for visibility tree.
- * @public
- */
-export declare class VisibilityTreeEventHandler extends UnifiedSelectionTreeEventHandler {
-    private _visibilityHandler;
-    private _selectionPredicate?;
-    private _listeners;
-    private _isChangingVisibility;
-    constructor(params: VisibilityTreeEventHandlerParams);
-    dispose(): void;
-    private filterSelectionItems;
-    onSelectionModified({ modifications }: TreeSelectionModificationEventArgs): import("@itwin/components-react").Subscription | undefined;
-    onSelectionReplaced({ replacements }: TreeSelectionReplacementEventArgs): import("@itwin/components-react").Subscription | undefined;
-    onCheckboxStateChanged(event: TreeCheckboxStateChangeEventArgs): undefined;
-    private changeVisibility;
-    private updateCheckboxes;
-    private updateModel;
-    private collectAffectedNodesCheckboxInfos;
-    private collectAllNodesCheckboxInfos;
-    private getNodeCheckBoxInfo;
-    private createCheckboxInfo;
-}
-//# sourceMappingURL=VisibilityTreeEventHandler.d.ts.map

package/lib/cjs/components/trees/VisibilityTreeEventHandler.js

@@ -1,158 +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.
- *--------------------------------------------------------------------------------------------*/
-/** @packageDocumentation
- * @module IModelComponents
- */
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.VisibilityTreeEventHandler = void 0;
-const rxjs_1 = require("rxjs");
-const core_react_1 = require("@itwin/core-react");
-const presentation_components_1 = require("@itwin/presentation-components");
-const IsPromiseLike_1 = require("../utils/IsPromiseLike");
-/**
- * Base event handler for visibility tree.
- * @public
- */
-class VisibilityTreeEventHandler extends presentation_components_1.UnifiedSelectionTreeEventHandler {
-    constructor(params) {
-        super(params);
-        this._listeners = new Array();
-        this._visibilityHandler = params.visibilityHandler;
-        this._selectionPredicate = params.selectionPredicate;
-        this._isChangingVisibility = false;
-        this._listeners.push(this._visibilityHandler.onVisibilityChange.addListener(async (nodeIds, visibilityStatus) => {
-            if (this._isChangingVisibility) {
-                return;
-            }
-            void this.updateCheckboxes(nodeIds, visibilityStatus);
-        }));
-        this._listeners.push(this.modelSource.onModelChanged.addListener(async ([_, changes]) => {
-            void this.updateCheckboxes([...changes.addedNodeIds, ...changes.modifiedNodeIds]);
-        }));
-        this.updateCheckboxes(); // eslint-disable-line @typescript-eslint/no-floating-promises
-    }
-    dispose() {
-        super.dispose();
-        this._listeners.forEach((disposeFunc) => disposeFunc());
-    }
-    filterSelectionItems(items) {
-        // istanbul ignore if
-        if (!this._selectionPredicate) {
-            return items;
-        }
-        return items.filter((item) => this._selectionPredicate(item));
-    }
-    onSelectionModified({ modifications }) {
-        const filteredModification = new rxjs_1.Observable((subscriber) => modifications.subscribe(subscriber)).pipe((0, rxjs_1.map)(({ selectedNodeItems, deselectedNodeItems }) => {
-            return {
-                selectedNodeItems: this.filterSelectionItems(selectedNodeItems),
-                deselectedNodeItems: this.filterSelectionItems(deselectedNodeItems),
-            };
-        }));
-        return super.onSelectionModified({ modifications: filteredModification });
-    }
-    onSelectionReplaced({ replacements }) {
-        const filteredReplacements = new rxjs_1.Observable((subscriber) => replacements.subscribe(subscriber)).pipe((0, rxjs_1.map)(({ selectedNodeItems }) => {
-            return {
-                selectedNodeItems: this.filterSelectionItems(selectedNodeItems),
-            };
-        }));
-        return super.onSelectionReplaced({ replacements: filteredReplacements });
-    }
-    onCheckboxStateChanged(event) {
-        const handleStateChanged = () => {
-            this._isChangingVisibility = false;
-            void this.updateCheckboxes();
-        };
-        new rxjs_1.Observable((subscriber) => event.stateChanges.subscribe(subscriber))
-            .pipe((0, rxjs_1.mergeMap)((changes) => this.changeVisibility(changes)))
-            .subscribe({
-            next: (nodeId) => {
-                void this.updateCheckboxes([nodeId]);
-            },
-            complete: handleStateChanged,
-            error: handleStateChanged,
-        });
-        return undefined;
-    }
-    changeVisibility(changes) {
-        return (0, rxjs_1.from)(changes).pipe((0, rxjs_1.mergeMap)(({ nodeItem, newState }) => {
-            this._isChangingVisibility = true;
-            return (0, rxjs_1.from)(this._visibilityHandler.changeVisibility(nodeItem, newState === core_react_1.CheckBoxState.On)).pipe((0, rxjs_1.ignoreElements)(), (0, rxjs_1.endWith)(nodeItem.id));
-        }));
-    }
-    async updateCheckboxes(affectedNodes, visibilityStatus) {
-        const changes = await (affectedNodes
-            ? this.collectAffectedNodesCheckboxInfos(affectedNodes, visibilityStatus)
-            : this.collectAllNodesCheckboxInfos(visibilityStatus));
-        this.updateModel(changes);
-    }
-    updateModel(changes) {
-        this.modelSource.modifyModel((model) => {
-            for (const [nodeId, checkboxInfo] of changes) {
-                const node = model.getNode(nodeId);
-                // istanbul ignore if
-                if (!node) {
-                    continue;
-                }
-                node.checkbox.isDisabled = checkboxInfo.isDisabled;
-                node.checkbox.isVisible = checkboxInfo.isVisible;
-                node.checkbox.state = checkboxInfo.state;
-                node.checkbox.tooltip = checkboxInfo.tooltip;
-            }
-        });
-    }
-    async collectAffectedNodesCheckboxInfos(affectedNodes, visibilityStatus) {
-        const nodeStates = new Map();
-        if (affectedNodes.length === 0) {
-            return nodeStates;
-        }
-        await Promise.all(affectedNodes.map(async (nodeId) => {
-            const node = this.modelSource.getModel().getNode(nodeId);
-            // istanbul ignore else
-            if (node) {
-                nodeStates.set(nodeId, await this.getNodeCheckBoxInfo(node, visibilityStatus));
-            }
-        }));
-        return nodeStates;
-    }
-    async collectAllNodesCheckboxInfos(visibilityStatus) {
-        const nodeStates = new Map();
-        for (const node of this.modelSource.getModel().iterateTreeModelNodes()) {
-            nodeStates.set(node.id, await this.getNodeCheckBoxInfo(node, visibilityStatus));
-        }
-        return nodeStates;
-    }
-    async getNodeCheckBoxInfo(node, visibilityStatus) {
-        const result = visibilityStatus?.get(node.id) ?? this._visibilityHandler.getVisibilityStatus(node.item);
-        if ((0, IsPromiseLike_1.isPromiseLike)(result)) {
-            return this.createCheckboxInfo(await result);
-        }
-        return this.createCheckboxInfo(result);
-    }
-    createCheckboxInfo(status) {
-        return {
-            state: visibilityStateToCheckboxState(status),
-            isDisabled: status.isDisabled || false,
-            isVisible: true,
-            tooltip: status.tooltip,
-        };
-    }
-}
-exports.VisibilityTreeEventHandler = VisibilityTreeEventHandler;
-const visibilityStateToCheckboxState = (status) => {
-    switch (status.state) {
-        case "visible":
-            return core_react_1.CheckBoxState.On;
-        // istanbul ignore next
-        case "partial":
-            return core_react_1.CheckBoxState.Partial;
-        case "hidden":
-        default:
-            return core_react_1.CheckBoxState.Off;
-    }
-};
-//# sourceMappingURL=VisibilityTreeEventHandler.js.map

package/lib/cjs/components/trees/VisibilityTreeEventHandler.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"VisibilityTreeEventHandler.js","sourceRoot":"","sources":["../../../../src/components/trees/VisibilityTreeEventHandler.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;AAEH,+BAAgF;AAChF,kDAAkD;AAClD,4EAAkF;AAClF,0DAAuD;AAuDvD;;;GAGG;AACH,MAAa,0BAA2B,SAAQ,0DAAgC;IAM9E,YAAY,MAAwC;QAClD,KAAK,CAAC,MAAM,CAAC,CAAC;QAJR,eAAU,GAAG,IAAI,KAAK,EAAc,CAAC;QAK3C,IAAI,CAAC,kBAAkB,GAAG,MAAM,CAAC,iBAAiB,CAAC;QACnD,IAAI,CAAC,mBAAmB,GAAG,MAAM,CAAC,kBAAkB,CAAC;QACrD,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC;QACnC,IAAI,CAAC,UAAU,CAAC,IAAI,CAClB,IAAI,CAAC,kBAAkB,CAAC,kBAAkB,CAAC,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,gBAAgB,EAAE,EAAE;YACzF,IAAI,IAAI,CAAC,qBAAqB,EAAE;gBAC9B,OAAO;aACR;YACD,KAAK,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,gBAAgB,CAAC,CAAC;QACxD,CAAC,CAAC,CACH,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,IAAI,CAClB,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,EAAE;YACjE,KAAK,IAAI,CAAC,gBAAgB,CAAC,CAAC,GAAG,OAAO,CAAC,YAAY,EAAE,GAAG,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC;QACpF,CAAC,CAAC,CACH,CAAC;QACF,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAC,8DAA8D;IACzF,CAAC;IAEe,OAAO;QACrB,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;IAC1D,CAAC;IAEO,oBAAoB,CAAC,KAAqB;QAChD,qBAAqB;QACrB,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE;YAC7B,OAAO,KAAK,CAAC;SACd;QAED,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,mBAAoB,CAAC,IAAI,CAAC,CAAC,CAAC;IACjE,CAAC;IAEe,mBAAmB,CAAC,EAAE,aAAa,EAAsC;QACvF,MAAM,oBAAoB,GAAG,IAAI,iBAAU,CAAsB,CAAC,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CACxH,IAAA,UAAG,EAAC,CAAC,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,EAAE,EAAE;YACjD,OAAO;gBACL,iBAAiB,EAAE,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC;gBAC/D,mBAAmB,EAAE,IAAI,CAAC,oBAAoB,CAAC,mBAAmB,CAAC;aACpE,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QACF,OAAO,KAAK,CAAC,mBAAmB,CAAC,EAAE,aAAa,EAAE,oBAAoB,EAAE,CAAC,CAAC;IAC5E,CAAC;IAEe,mBAAmB,CAAC,EAAE,YAAY,EAAqC;QACrF,MAAM,oBAAoB,GAAG,IAAI,iBAAU,CAAwC,CAAC,UAAU,EAAE,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CACzI,IAAA,UAAG,EAAC,CAAC,EAAE,iBAAiB,EAAE,EAAE,EAAE;YAC5B,OAAO;gBACL,iBAAiB,EAAE,IAAI,CAAC,oBAAoB,CAAC,iBAAiB,CAAC;aAChE,CAAC;QACJ,CAAC,CAAC,CACH,CAAC;QACF,OAAO,KAAK,CAAC,mBAAmB,CAAC,EAAE,YAAY,EAAE,oBAAoB,EAAE,CAAC,CAAC;IAC3E,CAAC;IAEe,sBAAsB,CAAC,KAAuC;QAC5E,MAAM,kBAAkB,GAAG,GAAG,EAAE;YAC9B,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC;YACnC,KAAK,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC/B,CAAC,CAAC;QAEF,IAAI,iBAAU,CAAwB,CAAC,UAAU,EAAE,EAAE,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;aAC5F,IAAI,CAAC,IAAA,eAAQ,EAAC,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC;aAC3D,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;gBACf,KAAK,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC;YACvC,CAAC;YACD,QAAQ,EAAE,kBAAkB;YAC5B,KAAK,EAAE,kBAAkB;SAC1B,CAAC,CAAC;QACL,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,gBAAgB,CAAC,OAA8B;QACrD,OAAO,IAAA,WAAI,EAAC,OAAO,CAAC,CAAC,IAAI,CACvB,IAAA,eAAQ,EAAC,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,EAAE;YAClC,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC;YAClC,OAAO,IAAA,WAAI,EAAC,IAAI,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,QAAQ,EAAE,QAAQ,KAAK,0BAAa,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAA,qBAAc,GAAE,EAAE,IAAA,cAAO,EAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;QAC9I,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,aAAwB,EAAE,gBAAgD;QACvG,MAAM,OAAO,GAAG,MAAM,CAAC,aAAa;YAClC,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,aAAa,EAAE,gBAAgB,CAAC;YACzE,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACzD,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC5B,CAAC;IAEO,WAAW,CAAC,OAAkC;QACpD,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,KAAK,EAAE,EAAE;YACrC,KAAK,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,IAAI,OAAO,EAAE;gBAC5C,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACnC,qBAAqB;gBACrB,IAAI,CAAC,IAAI,EAAE;oBACT,SAAS;iBACV;gBAED,IAAI,CAAC,QAAQ,CAAC,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;gBACnD,IAAI,CAAC,QAAQ,CAAC,SAAS,GAAG,YAAY,CAAC,SAAS,CAAC;gBACjD,IAAI,CAAC,QAAQ,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC;gBACzC,IAAI,CAAC,QAAQ,CAAC,OAAO,GAAG,YAAY,CAAC,OAAO,CAAC;aAC9C;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,iCAAiC,CAAC,aAAuB,EAAE,gBAAgD;QACvH,MAAM,UAAU,GAAG,IAAI,GAAG,EAAwB,CAAC;QACnD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE;YAC9B,OAAO,UAAU,CAAC;SACnB;QAED,MAAM,OAAO,CAAC,GAAG,CACf,aAAa,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;YACjC,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACzD,uBAAuB;YACvB,IAAI,IAAI,EAAE;gBACR,UAAU,CAAC,GAAG,CAAC,MAAM,EAAE,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC;aAChF;QACH,CAAC,CAAC,CACH,CAAC;QACF,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,4BAA4B,CAAC,gBAAgD;QACzF,MAAM,UAAU,GAAG,IAAI,GAAG,EAAwB,CAAC;QACnD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,CAAC,qBAAqB,EAAE,EAAE;YACtE,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC;SACjF;QACD,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,IAAmB,EAAE,gBAAgD;QACrG,MAAM,MAAM,GAAG,gBAAgB,EAAE,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExG,IAAI,IAAA,6BAAa,EAAC,MAAM,CAAC,EAAE;YACzB,OAAO,IAAI,CAAC,kBAAkB,CAAC,MAAM,MAAM,CAAC,CAAC;SAC9C;QACD,OAAO,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;IACzC,CAAC;IAEO,kBAAkB,CAAC,MAAwB;QACjD,OAAO;YACL,KAAK,EAAE,8BAA8B,CAAC,MAAM,CAAC;YAC7C,UAAU,EAAE,MAAM,CAAC,UAAU,IAAI,KAAK;YACtC,SAAS,EAAE,IAAI;YACf,OAAO,EAAE,MAAM,CAAC,OAAO;SACxB,CAAC;IACJ,CAAC;CACF;AA9JD,gEA8JC;AAED,MAAM,8BAA8B,GAAG,CAAC,MAAwB,EAAE,EAAE;IAClE,QAAQ,MAAM,CAAC,KAAK,EAAE;QACpB,KAAK,SAAS;YACZ,OAAO,0BAAa,CAAC,EAAE,CAAC;QAC1B,uBAAuB;QACvB,KAAK,SAAS;YACZ,OAAO,0BAAa,CAAC,OAAO,CAAC;QAC/B,KAAK,QAAQ,CAAC;QACd;YACE,OAAO,0BAAa,CAAC,GAAG,CAAC;KAC5B;AACH,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/** @packageDocumentation\n * @module IModelComponents\n */\n\nimport { endWith, from, ignoreElements, map, mergeMap, Observable } from \"rxjs\";\nimport { CheckBoxState } from \"@itwin/core-react\";\nimport { UnifiedSelectionTreeEventHandler } from \"@itwin/presentation-components\";\nimport { isPromiseLike } from \"../utils/IsPromiseLike\";\n\nimport type { BeEvent, IDisposable } from \"@itwin/core-bentley\";\nimport type {\n  CheckBoxInfo,\n  CheckboxStateChange,\n  TreeCheckboxStateChangeEventArgs,\n  TreeModelNode,\n  TreeNodeItem,\n  TreeSelectionChange,\n  TreeSelectionModificationEventArgs,\n  TreeSelectionReplacementEventArgs,\n} from \"@itwin/components-react\";\nimport type { UnifiedSelectionTreeEventHandlerParams } from \"@itwin/presentation-components\";\n/**\n * Data structure that describes instance visibility status.\n * @public\n */\nexport interface VisibilityStatus {\n  state: \"visible\" | \"partial\" | \"hidden\";\n  isDisabled?: boolean;\n  tooltip?: string;\n}\n\n/**\n * Type definition of visibility change event listener.\n * @public\n */\nexport type VisibilityChangeListener = (nodeIds?: string[], visibilityStatus?: Map<string, VisibilityStatus>) => void;\n\n/**\n * Visibility handler used to change or get visibility of instances represented by the tree node.\n * @public\n */\nexport interface IVisibilityHandler extends IDisposable {\n  getVisibilityStatus(node: TreeNodeItem): VisibilityStatus | Promise<VisibilityStatus>;\n  changeVisibility(node: TreeNodeItem, shouldDisplay: boolean): Promise<void>;\n  onVisibilityChange: BeEvent<VisibilityChangeListener>;\n}\n\n/**\n * Type definition of predicate used to decide if node can be selected.\n * @public\n */\nexport type VisibilityTreeSelectionPredicate = (node: TreeNodeItem) => boolean;\n\n/**\n * Parameters for [[VisibilityTreeEventHandler]]\n * @public\n */\nexport interface VisibilityTreeEventHandlerParams extends UnifiedSelectionTreeEventHandlerParams {\n  visibilityHandler: IVisibilityHandler;\n  selectionPredicate?: VisibilityTreeSelectionPredicate;\n}\n\n/**\n * Base event handler for visibility tree.\n * @public\n */\nexport class VisibilityTreeEventHandler extends UnifiedSelectionTreeEventHandler {\n  private _visibilityHandler: IVisibilityHandler;\n  private _selectionPredicate?: VisibilityTreeSelectionPredicate;\n  private _listeners = new Array<() => void>();\n  private _isChangingVisibility: boolean;\n\n  constructor(params: VisibilityTreeEventHandlerParams) {\n    super(params);\n    this._visibilityHandler = params.visibilityHandler;\n    this._selectionPredicate = params.selectionPredicate;\n    this._isChangingVisibility = false;\n    this._listeners.push(\n      this._visibilityHandler.onVisibilityChange.addListener(async (nodeIds, visibilityStatus) => {\n        if (this._isChangingVisibility) {\n          return;\n        }\n        void this.updateCheckboxes(nodeIds, visibilityStatus);\n      }),\n    );\n    this._listeners.push(\n      this.modelSource.onModelChanged.addListener(async ([_, changes]) => {\n        void this.updateCheckboxes([...changes.addedNodeIds, ...changes.modifiedNodeIds]);\n      }),\n    );\n    this.updateCheckboxes(); // eslint-disable-line @typescript-eslint/no-floating-promises\n  }\n\n  public override dispose() {\n    super.dispose();\n    this._listeners.forEach((disposeFunc) => disposeFunc());\n  }\n\n  private filterSelectionItems(items: TreeNodeItem[]) {\n    // istanbul ignore if\n    if (!this._selectionPredicate) {\n      return items;\n    }\n\n    return items.filter((item) => this._selectionPredicate!(item));\n  }\n\n  public override onSelectionModified({ modifications }: TreeSelectionModificationEventArgs) {\n    const filteredModification = new Observable<TreeSelectionChange>((subscriber) => modifications.subscribe(subscriber)).pipe(\n      map(({ selectedNodeItems, deselectedNodeItems }) => {\n        return {\n          selectedNodeItems: this.filterSelectionItems(selectedNodeItems),\n          deselectedNodeItems: this.filterSelectionItems(deselectedNodeItems),\n        };\n      }),\n    );\n    return super.onSelectionModified({ modifications: filteredModification });\n  }\n\n  public override onSelectionReplaced({ replacements }: TreeSelectionReplacementEventArgs) {\n    const filteredReplacements = new Observable<{ selectedNodeItems: TreeNodeItem[] }>((subscriber) => replacements.subscribe(subscriber)).pipe(\n      map(({ selectedNodeItems }) => {\n        return {\n          selectedNodeItems: this.filterSelectionItems(selectedNodeItems),\n        };\n      }),\n    );\n    return super.onSelectionReplaced({ replacements: filteredReplacements });\n  }\n\n  public override onCheckboxStateChanged(event: TreeCheckboxStateChangeEventArgs) {\n    const handleStateChanged = () => {\n      this._isChangingVisibility = false;\n      void this.updateCheckboxes();\n    };\n\n    new Observable<CheckboxStateChange[]>((subscriber) => event.stateChanges.subscribe(subscriber))\n      .pipe(mergeMap((changes) => this.changeVisibility(changes)))\n      .subscribe({\n        next: (nodeId) => {\n          void this.updateCheckboxes([nodeId]);\n        },\n        complete: handleStateChanged,\n        error: handleStateChanged,\n      });\n    return undefined;\n  }\n\n  private changeVisibility(changes: CheckboxStateChange[]) {\n    return from(changes).pipe(\n      mergeMap(({ nodeItem, newState }) => {\n        this._isChangingVisibility = true;\n        return from(this._visibilityHandler.changeVisibility(nodeItem, newState === CheckBoxState.On)).pipe(ignoreElements(), endWith(nodeItem.id));\n      }),\n    );\n  }\n\n  private async updateCheckboxes(affectedNodes?: string[], visibilityStatus?: Map<string, VisibilityStatus>) {\n    const changes = await (affectedNodes\n      ? this.collectAffectedNodesCheckboxInfos(affectedNodes, visibilityStatus)\n      : this.collectAllNodesCheckboxInfos(visibilityStatus));\n    this.updateModel(changes);\n  }\n\n  private updateModel(changes: Map<string, CheckBoxInfo>) {\n    this.modelSource.modifyModel((model) => {\n      for (const [nodeId, checkboxInfo] of changes) {\n        const node = model.getNode(nodeId);\n        // istanbul ignore if\n        if (!node) {\n          continue;\n        }\n\n        node.checkbox.isDisabled = checkboxInfo.isDisabled;\n        node.checkbox.isVisible = checkboxInfo.isVisible;\n        node.checkbox.state = checkboxInfo.state;\n        node.checkbox.tooltip = checkboxInfo.tooltip;\n      }\n    });\n  }\n\n  private async collectAffectedNodesCheckboxInfos(affectedNodes: string[], visibilityStatus?: Map<string, VisibilityStatus>) {\n    const nodeStates = new Map<string, CheckBoxInfo>();\n    if (affectedNodes.length === 0) {\n      return nodeStates;\n    }\n\n    await Promise.all(\n      affectedNodes.map(async (nodeId) => {\n        const node = this.modelSource.getModel().getNode(nodeId);\n        // istanbul ignore else\n        if (node) {\n          nodeStates.set(nodeId, await this.getNodeCheckBoxInfo(node, visibilityStatus));\n        }\n      }),\n    );\n    return nodeStates;\n  }\n\n  private async collectAllNodesCheckboxInfos(visibilityStatus?: Map<string, VisibilityStatus>) {\n    const nodeStates = new Map<string, CheckBoxInfo>();\n    for (const node of this.modelSource.getModel().iterateTreeModelNodes()) {\n      nodeStates.set(node.id, await this.getNodeCheckBoxInfo(node, visibilityStatus));\n    }\n    return nodeStates;\n  }\n\n  private async getNodeCheckBoxInfo(node: TreeModelNode, visibilityStatus?: Map<string, VisibilityStatus>): Promise<CheckBoxInfo> {\n    const result = visibilityStatus?.get(node.id) ?? this._visibilityHandler.getVisibilityStatus(node.item);\n\n    if (isPromiseLike(result)) {\n      return this.createCheckboxInfo(await result);\n    }\n    return this.createCheckboxInfo(result);\n  }\n\n  private createCheckboxInfo(status: VisibilityStatus): CheckBoxInfo {\n    return {\n      state: visibilityStateToCheckboxState(status),\n      isDisabled: status.isDisabled || false,\n      isVisible: true,\n      tooltip: status.tooltip,\n    };\n  }\n}\n\nconst visibilityStateToCheckboxState = (status: VisibilityStatus) => {\n  switch (status.state) {\n    case \"visible\":\n      return CheckBoxState.On;\n    // istanbul ignore next\n    case \"partial\":\n      return CheckBoxState.Partial;\n    case \"hidden\":\n    default:\n      return CheckBoxState.Off;\n  }\n};\n"]}