Checkbox actions on selected nodes (#833)

* Toggle visibility for all selected nodes

* Remove no longer needed refs

* Cleanup

packages/itwin/tree-widget/src/components/trees/experimental/common/UseMultiCheckboxHandler.ts

@@ -0,0 +1,43 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+
+import { PresentationHierarchyNode, PresentationTreeNode, isPresentationHierarchyNode } from "@itwin/presentation-hierarchies-react";
+
+interface UseMultiCheckboxHandlerProps {
+  rootNodes: PresentationTreeNode[] | undefined;
+  isNodeSelected: (nodeId: string) => boolean;
+  onClick: (node: PresentationHierarchyNode, checked: boolean) => void;
+}
+
+interface UseMultiCheckboxHandlerResult {
+  onCheckboxClicked: (node: PresentationHierarchyNode, checked: boolean) => void;
+}
+
+/** @internal */
+export function useMultiCheckboxHandler({ rootNodes, isNodeSelected, onClick }: UseMultiCheckboxHandlerProps): UseMultiCheckboxHandlerResult {
+  const clickSelectedNodes = (nodes: Array<PresentationTreeNode>, checked: boolean) => {
+    nodes.forEach((node) => {
+      if (!isPresentationHierarchyNode(node)) {
+        return;
+      }
+      if (isNodeSelected(node.id)) {
+        onClick(node, checked);
+      }
+      if (node.isExpanded && typeof node.children !== "boolean") {
+        clickSelectedNodes(node.children, checked);
+      }
+    });
+  };
+
+  const onCheckboxClicked = (node: PresentationHierarchyNode, checked: boolean) => {
+    if (!isNodeSelected(node.id)) {
+      onClick(node, checked);
+      return;
+    }
+    rootNodes && clickSelectedNodes(rootNodes, checked);
+  };
+
+  return { onCheckboxClicked };
+}

packages/itwin/tree-widget/src/components/trees/experimental/common/VisibilityTree.tsx

@@ -14,6 +14,7 @@ import { SchemaContext } from "@itwin/ecschema-metadata";
 import { createECSchemaProvider, createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
 import { createLimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
 import { createCachingECClassHierarchyInspector } from "@itwin/presentation-shared";
+import { useMultiCheckboxHandler } from "./UseMultiCheckboxHandler";
 
 interface VisibilityTreeOwnProps {
   imodel: IModelConnection;
@@ -74,7 +75,8 @@ function VisibilityTreeImpl({
     sourceName: "ExperimentalModelsTree",
   });
 
-  const nodesVisibility = useHierarchyVisibility({ visibilityHandlerFactory });
+  const { getCheckboxStatus, onCheckboxClicked: onClick } = useHierarchyVisibility({ visibilityHandlerFactory });
+  const { onCheckboxClicked } = useMultiCheckboxHandler({ rootNodes, isNodeSelected: treeProps.isNodeSelected, onClick });
   const { filteringDialog, onFilterClick } = useHierarchyFiltering({
     imodel,
     hierarchyProvider,
@@ -107,7 +109,8 @@ function VisibilityTreeImpl({
         <VisibilityTreeRenderer
           rootNodes={rootNodes}
           {...treeProps}
-          {...nodesVisibility}
+          getCheckboxStatus={getCheckboxStatus}
+          onCheckboxClicked={onCheckboxClicked}
           onFilterClick={onFilterClick}
           getIcon={getIcon}
           size={density === "enlarged" ? "default" : "small"}