X-Git-Url: https://git.arvados.org/arvados-workbench2.git/blobdiff_plain/a5984c9693267f41237746af6b37d8402effdc39..eeb82d50816250cc6287928e6d958affa73880ee:/src/store/project/project-reducer.ts?ds=sidebyside

diff --git a/src/store/project/project-reducer.ts b/src/store/project/project-reducer.ts
index de11d51f..458177c0 100644
--- a/src/store/project/project-reducer.ts
+++ b/src/store/project/project-reducer.ts
@@ -4,13 +4,52 @@
 
 import { Project } from "../../models/project";
 import actions, { ProjectAction } from "./project-action";
+import { TreeItem } from "../../components/tree/tree";
+import * as _ from "lodash";
 
-export type ProjectState = Project[];
+export type ProjectState = Array<TreeItem<Project>>;
+
+function findTreeItem<T>(tree: Array<TreeItem<T>>, itemId: string): TreeItem<T> | undefined {
+    let item;
+    for (const t of tree) {
+        item = t.id === itemId
+            ? t
+            : findTreeItem(t.items ? t.items : [], itemId);
+        if (item) {
+            break;
+        }
+    }
+    return item;
+}
+
+function resetTreeActivity<T>(tree: Array<TreeItem<T>>): boolean | undefined {
+    let item;
+    for (const leaf of tree) {
+        item = leaf.active === true
+            ? leaf.active = false
+            : resetTreeActivity(leaf.items ? leaf.items : []);
+    }
+    return item;
+}
 
 const projectsReducer = (state: ProjectState = [], action: ProjectAction) => {
     return actions.match(action, {
-        CREATE_PROJECT: (project) => [...state, project],
+        CREATE_PROJECT: project => [...state, project],
         REMOVE_PROJECT: () => state,
+        PROJECTS_REQUEST: () => state,
+        PROJECTS_SUCCESS: projects => {
+            return projects;
+        },
+        TOGGLE_PROJECT_TREE_ITEM: itemId => {
+            const tree = _.cloneDeep(state);
+            resetTreeActivity(tree);
+            const item = findTreeItem(tree, itemId);
+            if (item) {
+                item.open = !item.open;
+                item.active = true;
+            }
+            return tree;
+        },
         default: () => state
     });
 };