@plannotator/ui 0.49.0 → 0.50.0

package/hooks/useDiffFileTreeExpansion.ts

@@ -0,0 +1,92 @@
+import { useCallback, useEffect, useMemo, useState } from 'react';
+import {
+  getAllFolderPaths,
+  getAncestorPaths,
+  type DiffFileTreeFile,
+  type DiffFileTreeNode,
+} from '../utils/diffFileTree';
+
+/** Which folders start open: every folder (the review's behavior), none, or
+ * an explicit list of folder paths (`DiffFileTreeNode.path`). */
+export type DiffFileTreeDefaultExpanded = 'all' | 'none' | readonly string[];
+
+function initialSet(allFolderPaths: string[], defaultExpanded: DiffFileTreeDefaultExpanded): Set<string> {
+  if (defaultExpanded === 'all') return new Set(allFolderPaths);
+  if (defaultExpanded === 'none') return new Set();
+  return new Set(defaultExpanded);
+}
+
+/**
+ * Folder open/closed state for a diff file tree, shared by the review's
+ * FileTree and the embeddable DiffFileTree:
+ * - starts from `defaultExpanded` and resets to it whenever `tree` changes
+ *   identity (a new diff), so pass a memoized tree;
+ * - keeps every ancestor of `activePath` open, so moving the selection always
+ *   reveals the selected file.
+ */
+export function useDiffFileTreeExpansion<F extends DiffFileTreeFile>(
+  tree: DiffFileTreeNode<F>[],
+  activePath: string | null | undefined,
+  defaultExpanded: DiffFileTreeDefaultExpanded = 'all',
+) {
+  const allFolderPaths = useMemo(() => getAllFolderPaths(tree), [tree]);
+  const [expandedFolders, setExpandedFolders] = useState<Set<string>>(() => initialSet(allFolderPaths, defaultExpanded));
+  const [prevTree, setPrevTree] = useState(tree);
+
+  // Reset when the tree changes (initial render + diff switch).
+  if (tree !== prevTree) {
+    setPrevTree(tree);
+    setExpandedFolders(initialSet(allFolderPaths, defaultExpanded));
+  }
+
+  // Auto-expand ancestors of the active file so keyboard nav always reveals the target.
+  useEffect(() => {
+    if (activePath) {
+      const ancestors = getAncestorPaths(activePath);
+      setExpandedFolders((prev) => {
+        const missing = ancestors.filter((p) => !prev.has(p));
+        if (missing.length === 0) return prev;
+        const next = new Set(prev);
+        for (const p of missing) next.add(p);
+        return next;
+      });
+    }
+  }, [activePath, tree]);
+
+  const toggleFolder = useCallback((path: string) => {
+    setExpandedFolders((prev) => {
+      const next = new Set(prev);
+      if (next.has(path)) {
+        next.delete(path);
+      } else {
+        next.add(path);
+      }
+      return next;
+    });
+  }, []);
+
+  const setFolderExpanded = useCallback((path: string, expanded: boolean) => {
+    setExpandedFolders((prev) => {
+      if (prev.has(path) === expanded) return prev;
+      const next = new Set(prev);
+      if (expanded) next.add(path);
+      else next.delete(path);
+      return next;
+    });
+  }, []);
+
+  const areAllFoldersExpanded = allFolderPaths.length > 0 && allFolderPaths.every((path) => expandedFolders.has(path));
+
+  const toggleAllFolders = useCallback(() => {
+    setExpandedFolders(areAllFoldersExpanded ? new Set() : new Set(allFolderPaths));
+  }, [allFolderPaths, areAllFoldersExpanded]);
+
+  return {
+    expandedFolders,
+    allFolderPaths,
+    areAllFoldersExpanded,
+    toggleFolder,
+    setFolderExpanded,
+    toggleAllFolders,
+  };
+}

package/package.json

@@ -1,6 +1,7 @@
 {
   "name": "@plannotator/ui",
-  "version": "0.49.0",
+  "version": "0.50.0",
+  "license": "Apache-2.0",
   "type": "module",
   "exports": {
     "./components/*": "./components/*.tsx",
@@ -76,7 +77,7 @@
     "@lezer/highlight": "^1.2.3",
     "@pierre/diffs": "1.3.6",
     "@plannotator/atomic-editor": "^0.9.0",
-    "@plannotator/core": "0.25.9",
+    "@plannotator/core": "0.25.10",
     "@plannotator/markdown-editor": "^0.5.0",
     "@plannotator/web-highlighter": "^0.8.1",
     "@tanstack/react-table": "^8.21.3",