@itwin/tree-widget-react 0.10.0-dev.1 → 1.0.0-dev.0

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

@@ -1 +1 @@
-{"version":3,"file":"TreeFilteringState.js","sourceRoot":"","sources":["../../../src/components/TreeFilteringState.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAAA;;;+FAG+F;AAC/F,+CAAwC;AAWjC,MAAM,qBAAqB,GAAG,GAAG,EAAE;IACxC,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAAC,EAAE,CAAC,CAAC;IACrD,MAAM,CAAC,kBAAkB,EAAE,qBAAqB,CAAC,GAAG,IAAA,gBAAQ,GAAU,CAAC;IACvE,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,GAAU,CAAC;IACnE,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,GAAiC,CAAC;IAE1F,MAAM,cAAc,GAAG,eAAK,CAAC,WAAW,CAAC,GAAG,EAAE;QAC5C,eAAe,CAAC,EAAE,CAAC,CAAC;QACpB,qBAAqB,CAAC,SAAS,CAAC,CAAC;QACjC,mBAAmB,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,aAAa,GAAG,eAAK,CAAC,WAAW,CAAC,CAAC,SAAiB,EAAE,EAAE;QAC5D,eAAe,CAAC,SAAS,CAAC,CAAC;QAC3B,qBAAqB,CAAC,SAAS,CAAC,CAAC;QACjC,mBAAmB,CAAC,SAAS,CAAC,CAAC;IACjC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,uBAAuB,GAAG,eAAK,CAAC,WAAW,CAAC,CAAC,KAAa,EAAE,EAAE;QAClE,mBAAmB,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,eAAe,GAAG,eAAK,CAAC,WAAW,CAAC,CAAC,QAAuC,EAAE,OAAe,EAAE,EAAE;QACrG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;QAC9B,qBAAqB,CAAC,OAAO,CAAC,CAAC;IACjC,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;KACnB,CAAC;IAEF,OAAO;QACL,aAAa;QACb,YAAY;QACZ,gBAAgB;QAChB,eAAe;QACf,gBAAgB;KACjB,CAAC;AACJ,CAAC,CAAC;AA3CW,QAAA,qBAAqB,yBA2ChC","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*--------------------------------------------------------------------------------------------*/\nimport React, { useState } from \"react\";\nimport type { IPresentationTreeDataProvider } from \"@itwin/presentation-components\";\n\nexport interface SearchOptions {\n  isFiltering: boolean;\n  onFilterCancel: () => void;\n  onFilterStart: (newFilter: string) => void;\n  onResultSelectedChanged: (index: number) => void;\n  matchedResultCount: number | undefined;\n}\n\nexport const useTreeFilteringState = () => {\n  const [filterString, setFilterString] = useState(\"\");\n  const [matchedResultCount, setMatchedResultCount] = useState<number>();\n  const [activeMatchIndex, setActiveMatchIndex] = useState<number>();\n  const [filteredProvider, setFilteredProvider] = useState<IPresentationTreeDataProvider>();\n\n  const onFilterCancel = React.useCallback(() => {\n    setFilterString(\"\");\n    setMatchedResultCount(undefined);\n    setFilteredProvider(undefined);\n  }, []);\n\n  const onFilterStart = React.useCallback((newFilter: string) => {\n    setFilterString(newFilter);\n    setMatchedResultCount(undefined);\n    setFilteredProvider(undefined);\n  }, []);\n\n  const onResultSelectedChanged = React.useCallback((index: number) => {\n    setActiveMatchIndex(index);\n  }, []);\n\n  const onFilterApplied = React.useCallback((provider: IPresentationTreeDataProvider, matches: number) => {\n    setFilteredProvider(provider);\n    setMatchedResultCount(matches);\n  }, []);\n\n  const isFiltering = !!filterString && matchedResultCount === undefined;\n  const searchOptions: SearchOptions = {\n    isFiltering,\n    onFilterCancel,\n    onFilterStart,\n    onResultSelectedChanged,\n    matchedResultCount,\n  };\n\n  return {\n    searchOptions,\n    filterString,\n    activeMatchIndex,\n    onFilterApplied,\n    filteredProvider,\n  };\n};\n"]}
+{"version":3,"file":"TreeFilteringState.js","sourceRoot":"","sources":["../../../src/components/TreeFilteringState.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;;;AAE/F,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/TreeWidgetComponent.d.ts

@@ -1,9 +1,33 @@
 /// <reference types="react" />
 import "./TreeWidgetComponent.scss";
-import { SelectableContentDefinition } from "@itwin/components-react";
-interface TreeWidgetComponentProps {
-    trees?: SelectableContentDefinition[];
+import type { IModelConnection } from "@itwin/core-frontend";
+/**
+ * Definition of a tree component displayed in [[TreeWidgetComponent]]
+ * @public
+ */
+export interface TreeDefinition {
+    /** Id of the tree */
+    id: string;
+    /** Callback that is used to get tree label */
+    getLabel: () => string;
+    /** Callback that is used to render tree component */
+    render: () => React.ReactNode;
+    /**
+     * Callback that is used to determine if tree should be shown for current active iModel connection.
+     * If callback is `undefined` tree is shown for all iModel connections.
+     */
+    shouldShow?: (imodel: IModelConnection) => Promise<boolean>;
 }
-export declare function TreeWidgetComponent(props: TreeWidgetComponentProps): JSX.Element;
-export {};
+/**
+ * Props for [[TreeWidgetComponent]]
+ * @public
+ */
+export interface TreeWidgetComponentProps {
+    trees: TreeDefinition[];
+}
+/**
+ * A component that renders a tree (combo box) selector and the selected tree component.
+ * @public
+ */
+export declare function TreeWidgetComponent(props: TreeWidgetComponentProps): JSX.Element | null;
 //# sourceMappingURL=TreeWidgetComponent.d.ts.map

package/lib/cjs/components/TreeWidgetComponent.js

@@ -1,41 +1,97 @@
 "use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TreeWidgetComponent = void 0;
+const jsx_runtime_1 = require("react/jsx-runtime");
 /*---------------------------------------------------------------------------------------------
 * 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.TreeWidgetComponent = void 0;
 require("./TreeWidgetComponent.scss");
-const React = __importStar(require("react"));
+const react_1 = require("react");
+const appui_react_1 = require("@itwin/appui-react");
 const components_react_1 = require("@itwin/components-react");
+const core_react_1 = require("@itwin/core-react");
+const itwinui_react_1 = require("@itwin/itwinui-react");
+const TreeWidget_1 = require("../TreeWidget");
+/**
+ * A component that renders a tree (combo box) selector and the selected tree component.
+ * @public
+ */
 function TreeWidgetComponent(props) {
-    const trees = [];
-    if (props.trees && props.trees.length !== 0) {
-        for (const entry of props.trees) {
-            trees.push(entry);
+    const imodel = (0, appui_react_1.useActiveIModelConnection)();
+    if (!imodel)
+        return null;
+    return (0, jsx_runtime_1.jsx)(TreeWidgetComponentContent, { ...props, imodel: imodel });
+}
+exports.TreeWidgetComponent = TreeWidgetComponent;
+function TreeWidgetComponentContent(props) {
+    const { trees: treeDefinitions, imodel } = props;
+    const trees = useActiveTrees(treeDefinitions, imodel);
+    return ((0, jsx_runtime_1.jsx)("div", { className: "tree-widget-visibility-widget", children: (0, jsx_runtime_1.jsx)(components_react_1.SelectableContent, { ...getSelectableContentProps(trees) }) }));
+}
+function useActiveTrees(treeDefinitions, imodel) {
+    const [trees, setTrees] = (0, react_1.useState)();
+    (0, react_1.useEffect)(() => {
+        let disposed = false;
+        (async () => {
+            const visibleTrees = await getActiveTrees(treeDefinitions, imodel);
+            // istanbul ignore else
+            if (!disposed) {
+                setTrees(visibleTrees);
+            }
+        })();
+        return () => { disposed = true; };
+    }, [treeDefinitions, imodel]);
+    return trees;
+}
+async function getActiveTrees(treeDefinitions, imodel) {
+    const handleDefinition = async (treeDef) => {
+        if (treeDef.shouldShow !== undefined && !(await treeDef.shouldShow(imodel))) {
+            return undefined;
         }
+        return {
+            id: treeDef.id,
+            label: treeDef.getLabel(),
+            render: treeDef.render,
+        };
+    };
+    return (await Promise.all(treeDefinitions.map(handleDefinition))).filter((tree) => tree !== undefined);
+}
+function getSelectableContentProps(trees) {
+    if (trees === undefined) {
+        return {
+            defaultSelectedContentId: "loading",
+            children: [{
+                    id: "loading",
+                    label: "",
+                    render: () => ((0, jsx_runtime_1.jsx)(Delayed, { children: (0, jsx_runtime_1.jsx)(itwinui_react_1.ProgressLinear, { indeterminate: true }) })),
+                }],
+        };
+    }
+    if (trees.length === 0) {
+        return {
+            defaultSelectedContentId: "no-trees",
+            children: [{
+                    id: "no-trees",
+                    label: "",
+                    render: () => ((0, jsx_runtime_1.jsx)(core_react_1.FillCentered, { children: TreeWidget_1.TreeWidget.translate("noTrees") })),
+                }],
+        };
     }
-    return (React.createElement("div", { className: "tree-widget-visibility-widget" },
-        React.createElement(components_react_1.SelectableContent, { children: trees, defaultSelectedContentId: trees[0].id })));
+    return {
+        defaultSelectedContentId: trees[0].id,
+        children: trees,
+    };
+}
+function Delayed({ delay = 200, children }) {
+    const [show, setShow] = (0, react_1.useState)(false);
+    (0, react_1.useEffect)(() => {
+        const id = setTimeout(() => { setShow(true); }, delay);
+        return () => { clearTimeout(id); };
+    }, [delay]);
+    if (!show) {
+        return null;
+    }
+    return (0, jsx_runtime_1.jsx)(jsx_runtime_1.Fragment, { children: children });
 }
-exports.TreeWidgetComponent = TreeWidgetComponent;
 //# sourceMappingURL=TreeWidgetComponent.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"TreeWidgetComponent.js","sourceRoot":"","sources":["../../../src/components/TreeWidgetComponent.tsx"],"names":[],"mappings":";AAAA;;;+FAG+F;;;;;;;;;;;;;;;;;;;;;;AAE/F,sCAAoC;AACpC,6CAA+B;AAC/B,8DAAyF;AAMzF,SAAgB,mBAAmB,CAAC,KAA+B;IACjE,MAAM,KAAK,GAAkC,EAAE,CAAC;IAEhD,IAAI,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;QAC3C,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,KAAK,EAAE;YAC/B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;SACnB;KACF;IAED,OAAO,CACL,6BAAK,SAAS,EAAC,+BAA+B;QAC5C,oBAAC,oCAAiB,IAChB,QAAQ,EAAE,KAAK,EACf,wBAAwB,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,GACrC,CACE,CACP,CAAC;AACJ,CAAC;AAjBD,kDAiBC","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 \"./TreeWidgetComponent.scss\";\nimport * as React from \"react\";\nimport { SelectableContent, SelectableContentDefinition } from \"@itwin/components-react\";\n\ninterface TreeWidgetComponentProps {\n  trees?: SelectableContentDefinition[];\n}\n\nexport function TreeWidgetComponent(props: TreeWidgetComponentProps) {\n  const trees: SelectableContentDefinition[] = [];\n\n  if (props.trees && props.trees.length !== 0) {\n    for (const entry of props.trees) {\n      trees.push(entry);\n    }\n  }\n\n  return (\n    <div className=\"tree-widget-visibility-widget\">\n      <SelectableContent\n        children={trees} // eslint-disable-line react/no-children-prop\n        defaultSelectedContentId={trees[0].id}\n      />\n    </div>\n  );\n}\n"]}
+{"version":3,"file":"TreeWidgetComponent.js","sourceRoot":"","sources":["../../../src/components/TreeWidgetComponent.tsx"],"names":[],"mappings":";;;;AAAA;;;+FAG+F;AAE/F,sCAAoC;AACpC,iCAA4C;AAC5C,oDAA+D;AAC/D,8DAA4D;AAC5D,kDAAiD;AACjD,wDAAsD;AACtD,8CAA2C;AAgC3C;;;GAGG;AACH,SAAgB,mBAAmB,CAAC,KAA+B;IACjE,MAAM,MAAM,GAAG,IAAA,uCAAyB,GAAE,CAAC;IAE3C,IAAI,CAAC,MAAM;QACT,OAAO,IAAI,CAAC;IAEd,OAAO,uBAAC,0BAA0B,OAAK,KAAK,EAAE,MAAM,EAAE,MAAM,GAAI,CAAC;AACnE,CAAC;AAPD,kDAOC;AAED,SAAS,0BAA0B,CAAC,KAA6D;IAC/F,MAAM,EAAE,KAAK,EAAE,eAAe,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IACjD,MAAM,KAAK,GAAG,cAAc,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;IAEtD,OAAO,CACL,gCAAK,SAAS,EAAC,+BAA+B,YAC5C,uBAAC,oCAAiB,OAAK,yBAAyB,CAAC,KAAK,CAAC,GAAI,GACvD,CACP,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,eAAiC,EAAE,MAAwB;IACjF,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,GAAiC,CAAC;IAEpE,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,CAAC,KAAK,IAAI,EAAE;YACV,MAAM,YAAY,GAAG,MAAM,cAAc,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;YACnE,uBAAuB;YACvB,IAAI,CAAC,QAAQ,EAAE;gBACb,QAAQ,CAAC,YAAY,CAAC,CAAC;aACxB;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,GAAG,EAAE,GAAG,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC,EAAE,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC,CAAC;IAE9B,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,eAAiC,EAAE,MAAwB;IACvF,MAAM,gBAAgB,GAAG,KAAK,EAAE,OAAuB,EAAE,EAAE;QACzD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,EAAE;YAC3E,OAAO,SAAS,CAAC;SAClB;QACD,OAAO;YACL,EAAE,EAAE,OAAO,CAAC,EAAE;YACd,KAAK,EAAE,OAAO,CAAC,QAAQ,EAAE;YACzB,MAAM,EAAE,OAAO,CAAC,MAAM;SACvB,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,CAAC,MAAM,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,SAAS,CAAkC,CAAC;AAC1I,CAAC;AAED,SAAS,yBAAyB,CAAC,KAAqC;IACtE,IAAI,KAAK,KAAK,SAAS,EAAE;QACvB,OAAO;YACL,wBAAwB,EAAE,SAAS;YACnC,QAAQ,EAAE,CAAC;oBACT,EAAE,EAAE,SAAS;oBACb,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,GAAG,EAAE,CAAC,CACZ,uBAAC,OAAO,cACN,uBAAC,8BAAc,IAAC,aAAa,EAAE,IAAI,GAAI,GAC/B,CACX;iBACF,CAAC;SACH,CAAC;KACH;IAED,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;QACtB,OAAO;YACL,wBAAwB,EAAE,UAAU;YACpC,QAAQ,EAAE,CAAC;oBACT,EAAE,EAAE,UAAU;oBACd,KAAK,EAAE,EAAE;oBACT,MAAM,EAAE,GAAG,EAAE,CAAC,CACZ,uBAAC,yBAAY,cACV,uBAAU,CAAC,SAAS,CAAC,SAAS,CAAC,GACnB,CAChB;iBACF,CAAC;SACH,CAAC;KACH;IAED,OAAO;QACL,wBAAwB,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE;QACrC,QAAQ,EAAE,KAAK;KAChB,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,EAAE,KAAK,GAAG,GAAG,EAAE,QAAQ,EAAyC;IAC/E,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,GAAG,IAAA,gBAAQ,EAAC,KAAK,CAAC,CAAC;IAExC,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,GAAG,EAAE,GAAG,YAAY,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACrC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IAEZ,IAAI,CAAC,IAAI,EAAE;QACT,OAAO,IAAI,CAAC;KACb;IAED,OAAO,2DAAG,QAAQ,GAAI,CAAC;AACzB,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 \"./TreeWidgetComponent.scss\";\nimport { useEffect, useState } from \"react\";\nimport { useActiveIModelConnection } from \"@itwin/appui-react\";\nimport { SelectableContent } from \"@itwin/components-react\";\nimport { FillCentered } from \"@itwin/core-react\";\nimport { ProgressLinear } from \"@itwin/itwinui-react\";\nimport { TreeWidget } from \"../TreeWidget\";\n\nimport type { PropsWithChildren } from \"react\";\nimport type { SelectableContentDefinition, SelectableContentProps } from \"@itwin/components-react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\n\n/**\n * Definition of a tree component displayed in [[TreeWidgetComponent]]\n * @public\n */\nexport interface TreeDefinition {\n  /** Id of the tree */\n  id: string;\n  /** Callback that is used to get tree label */\n  getLabel: () => string;\n  /** Callback that is used to render tree component */\n  render: () => React.ReactNode;\n  /**\n   * Callback that is used to determine if tree should be shown for current active iModel connection.\n   * If callback is `undefined` tree is shown for all iModel connections.\n   */\n  shouldShow?: (imodel: IModelConnection) => Promise<boolean>;\n}\n\n/**\n * Props for [[TreeWidgetComponent]]\n * @public\n */\nexport interface TreeWidgetComponentProps {\n  trees: TreeDefinition[];\n}\n\n/**\n * A component that renders a tree (combo box) selector and the selected tree component.\n * @public\n */\nexport function TreeWidgetComponent(props: TreeWidgetComponentProps) {\n  const imodel = useActiveIModelConnection();\n\n  if (!imodel)\n    return null;\n\n  return <TreeWidgetComponentContent {...props} imodel={imodel} />;\n}\n\nfunction TreeWidgetComponentContent(props: TreeWidgetComponentProps & { imodel: IModelConnection}) {\n  const { trees: treeDefinitions, imodel } = props;\n  const trees = useActiveTrees(treeDefinitions, imodel);\n\n  return (\n    <div className=\"tree-widget-visibility-widget\">\n      <SelectableContent {...getSelectableContentProps(trees)} />\n    </div>\n  );\n}\n\nfunction useActiveTrees(treeDefinitions: TreeDefinition[], imodel: IModelConnection) {\n  const [trees, setTrees] = useState<SelectableContentDefinition[]>();\n\n  useEffect(() => {\n    let disposed = false;\n    (async () => {\n      const visibleTrees = await getActiveTrees(treeDefinitions, imodel);\n      // istanbul ignore else\n      if (!disposed) {\n        setTrees(visibleTrees);\n      }\n    })();\n\n    return () => { disposed = true; };\n  }, [treeDefinitions, imodel]);\n\n  return trees;\n}\n\nasync function getActiveTrees(treeDefinitions: TreeDefinition[], imodel: IModelConnection): Promise<SelectableContentDefinition[]> {\n  const handleDefinition = async (treeDef: TreeDefinition) => {\n    if (treeDef.shouldShow !== undefined && !(await treeDef.shouldShow(imodel))) {\n      return undefined;\n    }\n    return {\n      id: treeDef.id,\n      label: treeDef.getLabel(),\n      render: treeDef.render,\n    };\n  };\n\n  return (await Promise.all(treeDefinitions.map(handleDefinition))).filter((tree) => tree !== undefined) as SelectableContentDefinition[];\n}\n\nfunction getSelectableContentProps(trees?: SelectableContentDefinition[]): SelectableContentProps {\n  if (trees === undefined) {\n    return {\n      defaultSelectedContentId: \"loading\",\n      children: [{\n        id: \"loading\",\n        label: \"\",\n        render: () => (\n          <Delayed>\n            <ProgressLinear indeterminate={true} />\n          </Delayed>\n        ),\n      }],\n    };\n  }\n\n  if (trees.length === 0) {\n    return {\n      defaultSelectedContentId: \"no-trees\",\n      children: [{\n        id: \"no-trees\",\n        label: \"\",\n        render: () => (\n          <FillCentered>\n            {TreeWidget.translate(\"noTrees\")}\n          </FillCentered>\n        ),\n      }],\n    };\n  }\n\n  return {\n    defaultSelectedContentId: trees[0].id,\n    children: trees,\n  };\n}\n\nfunction Delayed({ delay = 200, children }: PropsWithChildren<{ delay?: number }>) {\n  const [show, setShow] = useState(false);\n\n  useEffect(() => {\n    const id = setTimeout(() => { setShow(true); }, delay);\n    return () => { clearTimeout(id); };\n  }, [delay]);\n\n  if (!show) {\n    return null;\n  }\n\n  return <>{children}</>;\n}\n"]}

package/lib/cjs/components/TreeWidgetComponent.scss

@@ -14,16 +14,11 @@
 
   .components-selectable-content {
     flex: 1;
+    overflow: hidden;
   }
 
   .components-selectable-content-header {
-    display: flex;
-    align-items: center;
-    margin-bottom: var(--iui-size-2xs);
-
-    .uicore-reactSelectTop {
-      flex: 1;
-    }
+    padding-bottom: var(--iui-size-2xs);
   }
 
   .components-selectable-content-wrapper {
@@ -32,25 +27,10 @@
     flex-grow: 1;
     height: 100%;
   }
-
-  .visibility-tree-toolbar {
-    display: flex;
-    flex-direction: row;
-    align-items: center;
-    margin-bottom: var(--iui-size-2xs);
-
-    .tree-toolbar-searchbox {
-      margin-left: auto;
-
-      .components-filtering-input {
-        margin: 0;
-      }
-    }
-  }
 }
 
 .tree-widget-select {
-  margin-bottom: var(--iui-size-2xs);
+  padding-bottom: var(--iui-size-2xs);
 }
 
 .tree-widget-tree-container {
@@ -60,5 +40,5 @@
 }
 
 .tree-widget-search-bar {
-  margin-bottom: var(--iui-size-2xs);
+  padding-bottom: var(--iui-size-2xs);
 }

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

@@ -1,6 +1,29 @@
-import { StagePanelLocation, StagePanelSection, UiItemsProvider, Widget } from "@itwin/appui-react";
-import { TreeWidgetOptions } from "../types";
+import { StagePanelLocation, StagePanelSection } from "@itwin/appui-react";
+import type { UiItemsProvider, Widget } from "@itwin/appui-react";
+import type { TreeDefinition } from "./TreeWidgetComponent";
+/**
+ * Parameters for creating a [[TreeWidgetUiItemsProvider]].
+ * @public
+ */
+export interface TreeWidgetOptions {
+    /** The stage panel to place the widget in. Defaults to `StagePanelLocation.Right`. */
+    defaultPanelLocation?: StagePanelLocation;
+    /** The stage panel section to place the widget in. Defaults to `StagePanelSection.Start`. */
+    defaultPanelSection?: StagePanelSection;
+    /** Widget priority in the stage panel. */
+    defaultTreeWidgetPriority?: number;
+    /** Trees to show in the widget. Defaults to [[ModelsTreeComponent]] and [[CategoriesTreeComponent]]. */
+    trees?: TreeDefinition[];
+}
+/**
+ * Id of the tree widget created by [[TreeWidgetUiItemsProvider]].
+ * @public
+ */
 export declare const TreeWidgetId = "tree-widget-react:trees";
+/**
+ * A [[UiItemsProvider]] implementation that provides a [[TreeWidgetComponent]] into a stage panel.
+ * @public
+ */
 export declare class TreeWidgetUiItemsProvider implements UiItemsProvider {
     private _treeWidgetOptions?;
     readonly id = "TreeWidgetUiItemsProvider";

package/lib/cjs/components/TreeWidgetUiItemsProvider.js

@@ -1,71 +1,56 @@
 "use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TreeWidgetUiItemsProvider = exports.TreeWidgetId = void 0;
+const jsx_runtime_1 = require("react/jsx-runtime");
 /*---------------------------------------------------------------------------------------------
 * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
 * See LICENSE.md in the project root for license terms and full copyright notice.
 *--------------------------------------------------------------------------------------------*/
-/* eslint-disable react/display-name */
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.TreeWidgetUiItemsProvider = exports.TreeWidgetId = void 0;
-const react_1 = __importDefault(require("react"));
 const appui_react_1 = require("@itwin/appui-react");
+const itwinui_icons_react_1 = require("@itwin/itwinui-icons-react");
 const TreeWidget_1 = require("../TreeWidget");
-const types_1 = require("../types");
 const CategoriesTreeComponent_1 = require("./trees/category-tree/CategoriesTreeComponent");
-const Common_1 = require("./trees/Common");
 const ModelsTreeComponent_1 = require("./trees/models-tree/ModelsTreeComponent");
 const TreeWidgetComponent_1 = require("./TreeWidgetComponent");
+/**
+ * Id of the tree widget created by [[TreeWidgetUiItemsProvider]].
+ * @public
+ */
 exports.TreeWidgetId = "tree-widget-react:trees";
+/**
+ * A [[UiItemsProvider]] implementation that provides a [[TreeWidgetComponent]] into a stage panel.
+ * @public
+ */
 class TreeWidgetUiItemsProvider {
     constructor(_treeWidgetOptions) {
         this._treeWidgetOptions = _treeWidgetOptions;
         this.id = "TreeWidgetUiItemsProvider";
     }
     provideWidgets(_stageId, stageUsage, location, section) {
-        const widgets = [];
         const preferredLocation = this._treeWidgetOptions?.defaultPanelLocation ?? appui_react_1.StagePanelLocation.Right;
         const preferredPanelSection = this._treeWidgetOptions?.defaultPanelSection ?? appui_react_1.StagePanelSection.Start;
-        if (location === preferredLocation &&
-            section === preferredPanelSection &&
-            stageUsage === appui_react_1.StageUsage.General) {
-            const trees = [];
-            if (!this._treeWidgetOptions?.hideTrees?.modelsTree) {
-                trees.push({
-                    label: TreeWidget_1.TreeWidget.translate("models"),
-                    id: types_1.ModelsTreeId,
-                    render: () => (react_1.default.createElement(ModelsTreeComponent_1.ModelsTreeComponent, { enableElementsClassGrouping: this._treeWidgetOptions?.enableElementsClassGrouping
-                            ? Common_1.ClassGroupingOption.YesWithCounts
-                            : Common_1.ClassGroupingOption.No, ...this._treeWidgetOptions?.modelsTreeProps })),
-                });
-            }
-            if (!this._treeWidgetOptions?.hideTrees?.categoriesTree) {
-                trees.push({
-                    label: TreeWidget_1.TreeWidget.translate("categories"),
-                    id: types_1.CategoriesTreeId,
-                    render: () => (react_1.default.createElement(CategoriesTreeComponent_1.CategoriesTreeComponent, { ...this._treeWidgetOptions?.categoriesTreeProps })),
-                });
-            }
-            if (this._treeWidgetOptions?.additionalTrees) {
-                trees.push(...this._treeWidgetOptions.additionalTrees);
-            }
-            if (this._treeWidgetOptions?.defaultTreeId && trees.length !== 0) {
-                // Adding the defaultTree to first index
-                const { defaultTreeId } = this._treeWidgetOptions;
-                const extractedDefaultTree = trees.filter((tree) => tree.id === defaultTreeId)[0];
-                const index = trees.indexOf(extractedDefaultTree);
-                trees.unshift(trees.splice(index, 1)[0]);
-            }
-            widgets.push({
+        if (location !== preferredLocation || section !== preferredPanelSection || stageUsage !== appui_react_1.StageUsage.General) {
+            return [];
+        }
+        const trees = this._treeWidgetOptions?.trees ?? [
+            {
+                id: ModelsTreeComponent_1.ModelsTreeComponent.id,
+                getLabel: ModelsTreeComponent_1.ModelsTreeComponent.getLabel,
+                render: () => (0, jsx_runtime_1.jsx)(ModelsTreeComponent_1.ModelsTreeComponent, {}),
+            },
+            {
+                id: CategoriesTreeComponent_1.CategoriesTreeComponent.id,
+                getLabel: CategoriesTreeComponent_1.CategoriesTreeComponent.getLabel,
+                render: () => (0, jsx_runtime_1.jsx)(CategoriesTreeComponent_1.CategoriesTreeComponent, {}),
+            },
+        ];
+        return [{
                 id: exports.TreeWidgetId,
                 label: TreeWidget_1.TreeWidget.translate("treeview"),
-                content: react_1.default.createElement(TreeWidgetComponent_1.TreeWidgetComponent, { trees: trees }),
-                icon: "icon-hierarchy-tree",
+                content: (0, jsx_runtime_1.jsx)(TreeWidgetComponent_1.TreeWidgetComponent, { trees: trees }),
+                icon: (0, jsx_runtime_1.jsx)(itwinui_icons_react_1.SvgHierarchyTree, {}),
                 priority: this._treeWidgetOptions?.defaultTreeWidgetPriority,
-            });
-        }
-        return widgets;
+            }];
     }
 }
 exports.TreeWidgetUiItemsProvider = TreeWidgetUiItemsProvider;

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

@@ -1 +1 @@
-{"version":3,"file":"TreeWidgetUiItemsProvider.js","sourceRoot":"","sources":["../../../src/components/TreeWidgetUiItemsProvider.tsx"],"names":[],"mappings":";AAAA;;;+FAG+F;AAC/F,uCAAuC;;;;;;AAEvC,kDAA0B;AAC1B,oDAAgH;AAEhH,8CAA2C;AAC3C,oCAA6E;AAC7E,2FAAwF;AACxF,2CAAqD;AACrD,iFAA8E;AAC9E,+DAA4D;AAE/C,QAAA,YAAY,GAAG,yBAAyB,CAAC;AACtD,MAAa,yBAAyB;IAGpC,YAAoB,kBAAsC;QAAtC,uBAAkB,GAAlB,kBAAkB,CAAoB;QAF1C,OAAE,GAAG,2BAA2B,CAAC;IAEa,CAAC;IAExD,cAAc,CACnB,QAAgB,EAChB,UAAkB,EAClB,QAA4B,EAC5B,OAA2B;QAE3B,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,iBAAiB,GAAG,IAAI,CAAC,kBAAkB,EAAE,oBAAoB,IAAI,gCAAkB,CAAC,KAAK,CAAC;QACpG,MAAM,qBAAqB,GAAG,IAAI,CAAC,kBAAkB,EAAE,mBAAmB,IAAI,+BAAiB,CAAC,KAAK,CAAC;QACtG,IACE,QAAQ,KAAK,iBAAiB;YAC9B,OAAO,KAAK,qBAAqB;YACjC,UAAU,KAAK,wBAAU,CAAC,OAAO,EACjC;YACA,MAAM,KAAK,GAAkC,EAAE,CAAC;YAEhD,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,SAAS,EAAE,UAAU,EAAE;gBACnD,KAAK,CAAC,IAAI,CAAC;oBACT,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,QAAQ,CAAC;oBACrC,EAAE,EAAE,oBAAY;oBAChB,MAAM,EAAE,GAAG,EAAE,CAAC,CACZ,8BAAC,yCAAmB,IAClB,2BAA2B,EACzB,IAAI,CAAC,kBAAkB,EAAE,2BAA2B;4BAClD,CAAC,CAAC,4BAAmB,CAAC,aAAa;4BACnC,CAAC,CAAC,4BAAmB,CAAC,EAAE,KAExB,IAAI,CAAC,kBAAkB,EAAE,eAAe,GAC5C,CACH;iBACF,CAAC,CAAC;aACJ;YAED,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,SAAS,EAAE,cAAc,EAAE;gBACvD,KAAK,CAAC,IAAI,CAAC;oBACT,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,YAAY,CAAC;oBACzC,EAAE,EAAE,wBAAgB;oBACpB,MAAM,EAAE,GAAG,EAAE,CAAC,CACZ,8BAAC,iDAAuB,OAAK,IAAI,CAAC,kBAAkB,EAAE,mBAAmB,GAAI,CAC9E;iBACF,CAAC,CAAC;aACJ;YAED,IAAI,IAAI,CAAC,kBAAkB,EAAE,eAAe,EAAE;gBAC5C,KAAK,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC;aACxD;YAED,IAAI,IAAI,CAAC,kBAAkB,EAAE,aAAa,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE;gBAChE,wCAAwC;gBACxC,MAAM,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC,kBAAkB,CAAC;gBAClD,MAAM,oBAAoB,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;gBAClF,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC;gBAClD,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;aAC1C;YAED,OAAO,CAAC,IAAI,CAAC;gBACX,EAAE,EAAE,oBAAY;gBAChB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,UAAU,CAAC;gBACvC,OAAO,EAAE,8BAAC,yCAAmB,IAAC,KAAK,EAAE,KAAK,GAAI;gBAC9C,IAAI,EAAE,qBAAqB;gBAC3B,QAAQ,EAAE,IAAI,CAAC,kBAAkB,EAAE,yBAAyB;aAC7D,CAAC,CAAC;SACJ;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAvED,8DAuEC","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/* eslint-disable react/display-name */\n\nimport React from \"react\";\nimport { StagePanelLocation, StagePanelSection, StageUsage, UiItemsProvider, Widget } from \"@itwin/appui-react\";\nimport { SelectableContentDefinition } from \"@itwin/components-react\";\nimport { TreeWidget } from \"../TreeWidget\";\nimport { CategoriesTreeId, ModelsTreeId, TreeWidgetOptions } from \"../types\";\nimport { CategoriesTreeComponent } from \"./trees/category-tree/CategoriesTreeComponent\";\nimport { ClassGroupingOption } from \"./trees/Common\";\nimport { ModelsTreeComponent } from \"./trees/models-tree/ModelsTreeComponent\";\nimport { TreeWidgetComponent } from \"./TreeWidgetComponent\";\n\nexport const TreeWidgetId = \"tree-widget-react:trees\";\nexport class TreeWidgetUiItemsProvider implements UiItemsProvider {\n  public readonly id = \"TreeWidgetUiItemsProvider\";\n\n  constructor(private _treeWidgetOptions?: TreeWidgetOptions) { }\n\n  public provideWidgets(\n    _stageId: string,\n    stageUsage: string,\n    location: StagePanelLocation,\n    section?: StagePanelSection\n  ): ReadonlyArray<Widget> {\n    const widgets: Widget[] = [];\n    const preferredLocation = this._treeWidgetOptions?.defaultPanelLocation ?? StagePanelLocation.Right;\n    const preferredPanelSection = this._treeWidgetOptions?.defaultPanelSection ?? StagePanelSection.Start;\n    if (\n      location === preferredLocation &&\n      section === preferredPanelSection &&\n      stageUsage === StageUsage.General\n    ) {\n      const trees: SelectableContentDefinition[] = [];\n\n      if (!this._treeWidgetOptions?.hideTrees?.modelsTree) {\n        trees.push({\n          label: TreeWidget.translate(\"models\"),\n          id: ModelsTreeId,\n          render: () => (\n            <ModelsTreeComponent\n              enableElementsClassGrouping={\n                this._treeWidgetOptions?.enableElementsClassGrouping\n                  ? ClassGroupingOption.YesWithCounts\n                  : ClassGroupingOption.No\n              }\n              {...this._treeWidgetOptions?.modelsTreeProps}\n            />\n          ),\n        });\n      }\n\n      if (!this._treeWidgetOptions?.hideTrees?.categoriesTree) {\n        trees.push({\n          label: TreeWidget.translate(\"categories\"),\n          id: CategoriesTreeId,\n          render: () => (\n            <CategoriesTreeComponent {...this._treeWidgetOptions?.categoriesTreeProps} />\n          ),\n        });\n      }\n\n      if (this._treeWidgetOptions?.additionalTrees) {\n        trees.push(...this._treeWidgetOptions.additionalTrees);\n      }\n\n      if (this._treeWidgetOptions?.defaultTreeId && trees.length !== 0) {\n        // Adding the defaultTree to first index\n        const { defaultTreeId } = this._treeWidgetOptions;\n        const extractedDefaultTree = trees.filter((tree) => tree.id === defaultTreeId)[0];\n        const index = trees.indexOf(extractedDefaultTree);\n        trees.unshift(trees.splice(index, 1)[0]);\n      }\n\n      widgets.push({\n        id: TreeWidgetId,\n        label: TreeWidget.translate(\"treeview\"),\n        content: <TreeWidgetComponent trees={trees} />,\n        icon: \"icon-hierarchy-tree\",\n        priority: this._treeWidgetOptions?.defaultTreeWidgetPriority,\n      });\n    }\n\n    return widgets;\n  }\n}\n"]}
+{"version":3,"file":"TreeWidgetUiItemsProvider.js","sourceRoot":"","sources":["../../../src/components/TreeWidgetUiItemsProvider.tsx"],"names":[],"mappings":";;;;AAAA;;;+FAG+F;AAE/F,oDAAuF;AACvF,oEAA8D;AAC9D,8CAA2C;AAC3C,2FAAwF;AACxF,iFAA8E;AAC9E,+DAA4D;AAoB5D;;;GAGG;AACU,QAAA,YAAY,GAAG,yBAAyB,CAAC;AAEtD;;;GAGG;AACH,MAAa,yBAAyB;IAGpC,YAAoB,kBAAsC;QAAtC,uBAAkB,GAAlB,kBAAkB,CAAoB;QAF1C,OAAE,GAAG,2BAA2B,CAAC;IAEa,CAAC;IAExD,cAAc,CACnB,QAAgB,EAChB,UAAkB,EAClB,QAA4B,EAC5B,OAA2B;QAE3B,MAAM,iBAAiB,GAAG,IAAI,CAAC,kBAAkB,EAAE,oBAAoB,IAAI,gCAAkB,CAAC,KAAK,CAAC;QACpG,MAAM,qBAAqB,GAAG,IAAI,CAAC,kBAAkB,EAAE,mBAAmB,IAAI,+BAAiB,CAAC,KAAK,CAAC;QAEtG,IAAI,QAAQ,KAAK,iBAAiB,IAAI,OAAO,KAAK,qBAAqB,IAAI,UAAU,KAAK,wBAAU,CAAC,OAAO,EAAE;YAC5G,OAAO,EAAE,CAAC;SACX;QAED,MAAM,KAAK,GAAqB,IAAI,CAAC,kBAAkB,EAAE,KAAK,IAAI;YAChE;gBACE,EAAE,EAAE,yCAAmB,CAAC,EAAE;gBAC1B,QAAQ,EAAE,yCAAmB,CAAC,QAAQ;gBACtC,MAAM,EAAE,GAAG,EAAE,CAAC,uBAAC,yCAAmB,KAAG;aACtC;YACD;gBACE,EAAE,EAAE,iDAAuB,CAAC,EAAE;gBAC9B,QAAQ,EAAE,iDAAuB,CAAC,QAAQ;gBAC1C,MAAM,EAAE,GAAG,EAAE,CAAC,uBAAC,iDAAuB,KAAG;aAC1C;SACF,CAAC;QAEF,OAAO,CAAC;gBACN,EAAE,EAAE,oBAAY;gBAChB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,UAAU,CAAC;gBACvC,OAAO,EAAE,uBAAC,yCAAmB,IAAC,KAAK,EAAE,KAAK,GAAI;gBAC9C,IAAI,EAAE,uBAAC,sCAAgB,KAAG;gBAC1B,QAAQ,EAAE,IAAI,CAAC,kBAAkB,EAAE,yBAAyB;aAC7D,CAAC,CAAC;IACL,CAAC;CACF;AAvCD,8DAuCC","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 { StagePanelLocation, StagePanelSection, StageUsage } from \"@itwin/appui-react\";\nimport { SvgHierarchyTree } from \"@itwin/itwinui-icons-react\";\nimport { TreeWidget } from \"../TreeWidget\";\nimport { CategoriesTreeComponent } from \"./trees/category-tree/CategoriesTreeComponent\";\nimport { ModelsTreeComponent } from \"./trees/models-tree/ModelsTreeComponent\";\nimport { TreeWidgetComponent } from \"./TreeWidgetComponent\";\n\nimport type { UiItemsProvider, Widget } from \"@itwin/appui-react\";\nimport type { TreeDefinition } from \"./TreeWidgetComponent\";\n\n/**\n * Parameters for creating a [[TreeWidgetUiItemsProvider]].\n * @public\n */\nexport interface TreeWidgetOptions {\n  /** The stage panel to place the widget in. Defaults to `StagePanelLocation.Right`. */\n  defaultPanelLocation?: StagePanelLocation;\n  /** The stage panel section to place the widget in. Defaults to `StagePanelSection.Start`. */\n  defaultPanelSection?: StagePanelSection;\n  /** Widget priority in the stage panel. */\n  defaultTreeWidgetPriority?: number;\n  /** Trees to show in the widget. Defaults to [[ModelsTreeComponent]] and [[CategoriesTreeComponent]]. */\n  trees?: TreeDefinition[];\n}\n\n/**\n * Id of the tree widget created by [[TreeWidgetUiItemsProvider]].\n * @public\n */\nexport const TreeWidgetId = \"tree-widget-react:trees\";\n\n/**\n * A [[UiItemsProvider]] implementation that provides a [[TreeWidgetComponent]] into a stage panel.\n * @public\n */\nexport class TreeWidgetUiItemsProvider implements UiItemsProvider {\n  public readonly id = \"TreeWidgetUiItemsProvider\";\n\n  constructor(private _treeWidgetOptions?: TreeWidgetOptions) { }\n\n  public provideWidgets(\n    _stageId: string,\n    stageUsage: string,\n    location: StagePanelLocation,\n    section?: StagePanelSection\n  ): ReadonlyArray<Widget> {\n    const preferredLocation = this._treeWidgetOptions?.defaultPanelLocation ?? StagePanelLocation.Right;\n    const preferredPanelSection = this._treeWidgetOptions?.defaultPanelSection ?? StagePanelSection.Start;\n\n    if (location !== preferredLocation || section !== preferredPanelSection || stageUsage !== StageUsage.General) {\n      return [];\n    }\n\n    const trees: TreeDefinition[] = this._treeWidgetOptions?.trees ?? [\n      {\n        id: ModelsTreeComponent.id,\n        getLabel: ModelsTreeComponent.getLabel,\n        render: () => <ModelsTreeComponent />,\n      },\n      {\n        id: CategoriesTreeComponent.id,\n        getLabel: CategoriesTreeComponent.getLabel,\n        render: () => <CategoriesTreeComponent />,\n      },\n    ];\n\n    return [{\n      id: TreeWidgetId,\n      label: TreeWidget.translate(\"treeview\"),\n      content: <TreeWidgetComponent trees={trees} />,\n      icon: <SvgHierarchyTree />,\n      priority: this._treeWidgetOptions?.defaultTreeWidgetPriority,\n    }];\n  }\n}\n"]}

package/lib/cjs/components/tree-header/TreeHeader.d.ts

@@ -0,0 +1,26 @@
+/// <reference types="react" />
+import "./TreeHeader.scss";
+import type { Viewport } from "@itwin/core-frontend";
+import type { CommonProps } from "@itwin/core-react";
+/** @internal */
+export interface TreeHeaderButtonProps {
+    viewport: Viewport;
+}
+/** @internal */
+export interface TreeHeaderProps extends CommonProps {
+    /** Filtering is cleared after everything's loaded */
+    onFilterStart: (newFilter: string) => void;
+    /** listens for onClick event for Clear (x) icon */
+    onFilterClear: () => void;
+    /** Total number of results/entries */
+    resultCount?: number;
+    /** Current selected result index */
+    selectedIndex?: number;
+    /** Callback to currently selected result/entry change */
+    onSelectedChanged: (index: number) => void;
+    /** Header buttons */
+    children?: React.ReactNode;
+}
+/** @internal */
+export declare function TreeHeader(props: TreeHeaderProps): JSX.Element;
+//# sourceMappingURL=TreeHeader.d.ts.map

package/lib/cjs/components/tree-header/TreeHeader.js

@@ -0,0 +1,43 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TreeHeader = void 0;
+const jsx_runtime_1 = require("react/jsx-runtime");
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+require("./TreeHeader.scss");
+const classnames_1 = __importDefault(require("classnames"));
+const react_1 = require("react");
+const itwinui_icons_react_1 = require("@itwin/itwinui-icons-react");
+const itwinui_react_1 = require("@itwin/itwinui-react");
+const TreeWidget_1 = require("../../TreeWidget");
+/** @internal */
+function TreeHeader(props) {
+    const { onFilterStart, onFilterClear, resultCount, selectedIndex, onSelectedChanged, children, className } = props;
+    const [isSearchOpen, setIsSearchOpen] = (0, react_1.useState)(false);
+    return ((0, jsx_runtime_1.jsxs)("div", { className: (0, classnames_1.default)("tree-widget-tree-header", className), children: [(0, jsx_runtime_1.jsx)(HeaderButtons, { contracted: isSearchOpen, children: children }), (0, jsx_runtime_1.jsxs)(itwinui_react_1.SearchBox, { expandable: true, onExpand: () => setIsSearchOpen(true), onCollapse: () => setIsSearchOpen(false), className: (0, classnames_1.default)("tree-widget-search-box", !isSearchOpen && "contracted"), children: [(0, jsx_runtime_1.jsx)(itwinui_react_1.SearchBox.CollapsedState, { children: (0, jsx_runtime_1.jsx)(itwinui_react_1.SearchBox.ExpandButton, { title: TreeWidget_1.TreeWidget.translate("searchBox.searchForSomething"), "aria-label": TreeWidget_1.TreeWidget.translate("searchBox.open"), size: "small" }) }), (0, jsx_runtime_1.jsxs)(itwinui_react_1.SearchBox.ExpandedState, { children: [(0, jsx_runtime_1.jsx)(itwinui_react_1.SearchBox.Input, { placeholder: TreeWidget_1.TreeWidget.translate("searchBox.search"), onChange: (e) => e.currentTarget.value ? onFilterStart(e.currentTarget.value) : onFilterClear(), className: "search-input" }), (0, jsx_runtime_1.jsx)(SearchResultStepper, { selectedIndex: selectedIndex, total: resultCount, onStep: onSelectedChanged }), (0, jsx_runtime_1.jsx)(itwinui_react_1.SearchBox.CollapseButton, { onClick: onFilterClear, size: "small", "aria-label": TreeWidget_1.TreeWidget.translate("searchBox.close") })] })] })] }));
+}
+exports.TreeHeader = TreeHeader;
+function HeaderButtons(props) {
+    const className = (0, classnames_1.default)("button-container", props.contracted && "contracted");
+    return ((0, jsx_runtime_1.jsx)(itwinui_react_1.ButtonGroup, { className: className, overflowButton: (overflowStart) => ((0, jsx_runtime_1.jsx)(itwinui_react_1.DropdownMenu, { menuItems: () => react_1.Children.toArray(props.children)
+                .slice(overflowStart === 0 ? overflowStart : overflowStart - 1)
+                .map((btn, index) => (0, jsx_runtime_1.jsx)("li", { className: "dropdown-item", role: "menuitem", children: btn }, index)), className: "tree-header-button-dropdown-container", children: (0, jsx_runtime_1.jsx)(itwinui_react_1.IconButton, { styleType: "borderless", size: "small", children: (0, jsx_runtime_1.jsx)(itwinui_icons_react_1.SvgMore, {}) }) })), children: props.children }));
+}
+function SearchResultStepper(props) {
+    const { selectedIndex = 1, total, onStep } = props;
+    if (!total)
+        return null;
+    return ((0, jsx_runtime_1.jsxs)(jsx_runtime_1.Fragment, { children: [(0, jsx_runtime_1.jsx)("span", { className: "searchbox-stepping-count", children: `${selectedIndex}/${total}` }), (0, jsx_runtime_1.jsx)(itwinui_react_1.Divider, { orientation: "vertical" }), (0, jsx_runtime_1.jsx)(itwinui_react_1.SearchBox.Button, { title: TreeWidget_1.TreeWidget.translate("searchBox.previous"), size: "small", onClick: () => {
+                    if (selectedIndex > 1)
+                        onStep(selectedIndex - 1);
+                }, children: (0, jsx_runtime_1.jsx)(itwinui_icons_react_1.SvgCaretUpSmall, {}) }), (0, jsx_runtime_1.jsx)(itwinui_react_1.SearchBox.Button, { title: TreeWidget_1.TreeWidget.translate("searchBox.next"), size: "small", onClick: () => {
+                    if (selectedIndex < total)
+                        onStep(selectedIndex + 1);
+                }, children: (0, jsx_runtime_1.jsx)(itwinui_icons_react_1.SvgCaretDownSmall, {}) })] }));
+}
+//# sourceMappingURL=TreeHeader.js.map

package/lib/cjs/components/tree-header/TreeHeader.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TreeHeader.js","sourceRoot":"","sources":["../../../../src/components/tree-header/TreeHeader.tsx"],"names":[],"mappings":";;;;;;;AAAA;;;+FAG+F;AAE/F,6BAA2B;AAC3B,4DAAoC;AACpC,iCAA2C;AAC3C,oEAAyF;AACzF,wDAAiG;AACjG,iDAA8C;AA0B9C,gBAAgB;AAChB,SAAgB,UAAU,CAAC,KAAsB;IAC/C,MAAM,EAAE,aAAa,EAAE,aAAa,EAAE,WAAW,EAAE,aAAa,EAAE,iBAAiB,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,KAAK,CAAC;IACnH,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAAU,KAAK,CAAC,CAAC;IACjE,OAAO,CACL,iCAAK,SAAS,EAAE,IAAA,oBAAU,EAAC,yBAAyB,EAAE,SAAS,CAAC,aAC9D,uBAAC,aAAa,IAAC,UAAU,EAAE,YAAY,YACpC,QAAQ,GACK,EAChB,wBAAC,yBAAS,IACR,UAAU,QACV,QAAQ,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,EACrC,UAAU,EAAE,GAAG,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,EACxC,SAAS,EAAE,IAAA,oBAAU,EAAC,wBAAwB,EAAE,CAAC,YAAY,IAAI,YAAY,CAAC,aAE9E,uBAAC,yBAAS,CAAC,cAAc,cACvB,uBAAC,yBAAS,CAAC,YAAY,IACrB,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,8BAA8B,CAAC,gBAC/C,uBAAU,CAAC,SAAS,CAAC,gBAAgB,CAAC,EAClD,IAAI,EAAC,OAAO,GACZ,GACuB,EAC3B,wBAAC,yBAAS,CAAC,aAAa,eACtB,uBAAC,yBAAS,CAAC,KAAK,IACd,WAAW,EAAE,uBAAU,CAAC,SAAS,CAAC,kBAAkB,CAAC,EACrD,QAAQ,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,aAAa,EAAE,EAC/F,SAAS,EAAC,cAAc,GACxB,EACF,uBAAC,mBAAmB,IAClB,aAAa,EAAE,aAAa,EAC5B,KAAK,EAAE,WAAW,EAClB,MAAM,EAAE,iBAAiB,GACzB,EACF,uBAAC,yBAAS,CAAC,cAAc,IACvB,OAAO,EAAE,aAAa,EACtB,IAAI,EAAC,OAAO,gBACA,uBAAU,CAAC,SAAS,CAAC,iBAAiB,CAAC,GACnD,IACsB,IAChB,IACR,CACP,CAAC;AACJ,CAAC;AAzCD,gCAyCC;AAOD,SAAS,aAAa,CAAC,KAAyB;IAC9C,MAAM,SAAS,GAAG,IAAA,oBAAU,EAC1B,kBAAkB,EAClB,KAAK,CAAC,UAAU,IAAI,YAAY,CACjC,CAAC;IAEF,OAAO,CACL,uBAAC,2BAAW,IACV,SAAS,EAAE,SAAS,EACpB,cAAc,EAAE,CAAC,aAAa,EAAE,EAAE,CAAC,CACjC,uBAAC,4BAAY,IACX,SAAS,EAAE,GAAG,EAAE,CACd,gBAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC;iBAC7B,KAAK,CAAC,aAAa,KAAK,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,aAAa,GAAG,CAAC,CAAC;iBAC9D,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,+BAAgB,SAAS,EAAC,eAAe,EAAC,IAAI,EAAC,UAAU,YAAE,GAAG,IAArD,KAAK,CAAsD,CAAC,EAE9F,SAAS,EAAC,uCAAuC,YAEjD,uBAAC,0BAAU,IAAC,SAAS,EAAC,YAAY,EAAC,IAAI,EAAC,OAAO,YAC7C,uBAAC,6BAAO,KAAG,GACA,GACA,CAChB,YAEA,KAAK,CAAC,QAAQ,GACH,CACf,CAAC;AACJ,CAAC;AAQD,SAAS,mBAAmB,CAAC,KAA+B;IAC1D,MAAM,EAAE,aAAa,GAAG,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;IACnD,IAAI,CAAC,KAAK;QACR,OAAO,IAAI,CAAC;IAEd,OAAO,CACL,6DACE,iCAAM,SAAS,EAAC,0BAA0B,YAAE,GAAG,aAAa,IAAI,KAAK,EAAE,GAAQ,EAC/E,uBAAC,uBAAO,IAAC,WAAW,EAAC,UAAU,GAAG,EAClC,uBAAC,yBAAS,CAAC,MAAM,IACf,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,oBAAoB,CAAC,EACjD,IAAI,EAAC,OAAO,EACZ,OAAO,EAAE,GAAG,EAAE;oBACZ,IAAI,aAAa,GAAG,CAAC;wBACnB,MAAM,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;gBAC9B,CAAC,YAED,uBAAC,qCAAe,KAAG,GACF,EACnB,uBAAC,yBAAS,CAAC,MAAM,IACf,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,gBAAgB,CAAC,EAC7C,IAAI,EAAC,OAAO,EACZ,OAAO,EAAE,GAAG,EAAE;oBACZ,IAAI,aAAa,GAAG,KAAK;wBACvB,MAAM,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC;gBAC9B,CAAC,YAED,uBAAC,uCAAiB,KAAG,GACJ,IAClB,CACJ,CAAC;AACJ,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 \"./TreeHeader.scss\";\nimport classnames from \"classnames\";\nimport { Children, useState } from \"react\";\nimport { SvgCaretDownSmall, SvgCaretUpSmall, SvgMore } from \"@itwin/itwinui-icons-react\";\nimport { ButtonGroup, Divider, DropdownMenu, IconButton, SearchBox } from \"@itwin/itwinui-react\";\nimport { TreeWidget } from \"../../TreeWidget\";\n\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { CommonProps } from \"@itwin/core-react\";\n\n/** @internal */\nexport interface TreeHeaderButtonProps {\n  viewport: Viewport;\n}\n\n/** @internal */\nexport interface TreeHeaderProps extends CommonProps {\n  /** Filtering is cleared after everything's loaded */\n  onFilterStart: (newFilter: string) => void;\n  /** listens for onClick event for Clear (x) icon */\n  onFilterClear: () => void;\n  /** Total number of results/entries */\n  resultCount?: number;\n  /** Current selected result index */\n  selectedIndex?: number;\n  /** Callback to currently selected result/entry change */\n  onSelectedChanged: (index: number) => void;\n  /** Header buttons */\n  children?: React.ReactNode;\n}\n\n/** @internal */\nexport function TreeHeader(props: TreeHeaderProps) {\n  const { onFilterStart, onFilterClear, resultCount, selectedIndex, onSelectedChanged, children, className } = props;\n  const [isSearchOpen, setIsSearchOpen] = useState<boolean>(false);\n  return (\n    <div className={classnames(\"tree-widget-tree-header\", className)}>\n      <HeaderButtons contracted={isSearchOpen}>\n        {children}\n      </HeaderButtons>\n      <SearchBox\n        expandable\n        onExpand={() => setIsSearchOpen(true)}\n        onCollapse={() => setIsSearchOpen(false)}\n        className={classnames(\"tree-widget-search-box\", !isSearchOpen && \"contracted\")}\n      >\n        <SearchBox.CollapsedState>\n          <SearchBox.ExpandButton\n            title={TreeWidget.translate(\"searchBox.searchForSomething\")}\n            aria-label={TreeWidget.translate(\"searchBox.open\")}\n            size=\"small\"\n          />\n        </SearchBox.CollapsedState>\n        <SearchBox.ExpandedState >\n          <SearchBox.Input\n            placeholder={TreeWidget.translate(\"searchBox.search\")}\n            onChange={(e) => e.currentTarget.value ? onFilterStart(e.currentTarget.value) : onFilterClear()}\n            className=\"search-input\"\n          />\n          <SearchResultStepper\n            selectedIndex={selectedIndex}\n            total={resultCount}\n            onStep={onSelectedChanged}\n          />\n          <SearchBox.CollapseButton\n            onClick={onFilterClear}\n            size=\"small\"\n            aria-label={TreeWidget.translate(\"searchBox.close\")}\n          />\n        </SearchBox.ExpandedState>\n      </SearchBox>\n    </div>\n  );\n}\n\ninterface HeaderButtonsProps {\n  contracted: boolean;\n  children?: React.ReactNode;\n}\n\nfunction HeaderButtons(props: HeaderButtonsProps) {\n  const className = classnames(\n    \"button-container\",\n    props.contracted && \"contracted\",\n  );\n\n  return (\n    <ButtonGroup\n      className={className}\n      overflowButton={(overflowStart) => (\n        <DropdownMenu\n          menuItems={() =>\n            Children.toArray(props.children)\n              .slice(overflowStart === 0 ? overflowStart : overflowStart - 1)\n              .map((btn, index) => <li key={index} className=\"dropdown-item\" role=\"menuitem\">{btn}</li>)\n          }\n          className=\"tree-header-button-dropdown-container\"\n        >\n          <IconButton styleType=\"borderless\" size=\"small\">\n            <SvgMore />\n          </IconButton>\n        </DropdownMenu>\n      )}\n    >\n      {props.children}\n    </ButtonGroup>\n  );\n}\n\ninterface SearchResultStepperProps {\n  total?: number;\n  onStep: (newIndex: number) => void;\n  selectedIndex?: number;\n}\n\nfunction SearchResultStepper(props: SearchResultStepperProps) {\n  const { selectedIndex = 1, total, onStep } = props;\n  if (!total)\n    return null;\n\n  return (\n    <>\n      <span className=\"searchbox-stepping-count\">{`${selectedIndex}/${total}`}</span>\n      <Divider orientation=\"vertical\" />\n      <SearchBox.Button\n        title={TreeWidget.translate(\"searchBox.previous\")}\n        size=\"small\"\n        onClick={() => {\n          if (selectedIndex > 1)\n            onStep(selectedIndex - 1);\n        }}\n      >\n        <SvgCaretUpSmall />\n      </SearchBox.Button>\n      <SearchBox.Button\n        title={TreeWidget.translate(\"searchBox.next\")}\n        size=\"small\"\n        onClick={() => {\n          if (selectedIndex < total)\n            onStep(selectedIndex + 1);\n        }}\n      >\n        <SvgCaretDownSmall />\n      </SearchBox.Button>\n    </>\n  );\n}\n"]}

package/lib/cjs/components/tree-header/TreeHeader.scss

@@ -0,0 +1,69 @@
+/*---------------------------------------------------------------------------------------------
+* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+* See LICENSE.md in the project root for license terms and full copyright notice.
+*--------------------------------------------------------------------------------------------*/
+
+// Buttons dropdown container
+.tree-header-button-dropdown-container {
+  display: flex;
+  flex-direction: column;
+  align-items: center;
+  margin-top: calc(var(--iui-size-2xs) + var(--iui-size-3xs));
+
+  .dropdown-item {
+    width: 100%;
+    height: var(--iui-size-xl);
+
+    > * {
+      width: 100%;
+      height: 100%;
+    }
+
+    &:hover {
+      background-color: transparent;
+      cursor: default;
+    }
+  }
+}
+
+.tree-widget-tree-header {
+  display: flex;
+  align-items: center;
+  height: var(--iui-size-xl);
+  justify-content: space-between;
+
+  .tree-widget-search-box {
+    height: calc(var(--iui-size-l) + var(--iui-size-3xs) * 3);
+    padding: 0;
+    width: calc(100% - var(--iui-size-xl));
+
+    &.contracted {
+      outline: none;
+      width: var(--iui-size-l);
+
+      &::before {
+        border: none;
+      }
+    }
+
+    .searchbox-stepping-count {
+      font-size: var(--iui-size-s);
+      color: var(--iui-color-text);
+      margin-right: var(--iui-size-2xs);
+    }
+
+    .search-input {
+      margin-right: var(--iui-size-xs);
+      padding: 0 var(--iui-size-xs);
+    }
+  }
+
+  .button-container {
+    height: 100%;
+    width: calc(100% - var(--iui-size-l));
+
+    &.contracted {
+      width: var(--iui-size-xl);
+    }
+  }
+}

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

@@ -1,18 +1,24 @@
-import { IModelConnection, ViewManager, Viewport } from "@itwin/core-frontend";
-import { CategoryInfo } from "./category-tree/CategoryVisibilityHandler";
+import type { IModelConnection, ViewManager, Viewport } from "@itwin/core-frontend";
+import type { CategoryInfo } from "./category-tree/CategoryVisibilityHandler";
 /**
  * Toggles visibility of categories to show or hide.
- * @alpha
+ * @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.
- * @alpha
+ * @internal
  */
 export declare function getCategories(imodel: IModelConnection, viewport?: Viewport): Promise<string[]>;
-/** Changes category display in the viewport */
+/**
+ * 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 */
+/**
+ * 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[]>;