src/store/project/project-reducer.ts

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 import { Project } from "../../models/project";
   6 import actions, { ProjectAction } from "./project-action";
   7 import { TreeItem } from "../../components/tree/tree";
   8 import * as _ from "lodash";
   9
  10 export type ProjectState = Array<TreeItem<Project>>;
  11
  12 function findTreeItem<T>(tree: Array<TreeItem<T>>, itemId: string): TreeItem<T> | undefined {
  13     let item;
  14     for (const t of tree) {
  15         item = t.id === itemId
  16             ? t
  17             : findTreeItem(t.items ? t.items : [], itemId);
  18         if (item) {
  19             break;
  20         }
  21     }
  22     return item;
  23 }
  24
  25 function resetTreeActivity<T>(tree: Array<TreeItem<T>>) {
  26     for (const t of tree) {
  27         t.active = false;
  28         resetTreeActivity(t.items ? t.items : []);
  29     }
  30 }
  31
  32 function resetTreeLoader<T>(tree: Array<TreeItem<T>>) {
  33     for (const t of tree) {
  34         t.isLoaded = true;
  35         resetTreeLoader(t.items ? t.items : []);
  36     }
  37 }
  38
  39 function updateProjectTree(tree: Array<TreeItem<Project>>, projects: Project[], parentItemId?: string): Array<TreeItem<Project>> {
  40     resetTreeLoader(tree)
  41     let treeItem;
  42     if (parentItemId) {
  43         treeItem = findTreeItem(tree, parentItemId);
  44     }
  45     const items = projects.map((p, idx) => ({
  46         id: p.uuid,
  47         open: false,
  48         active: false,
  49         isLoaded: true,
  50         data: p,
  51         items: []
  52     } as TreeItem<Project>));
  53
  54     if (treeItem) {
  55         treeItem.items = items;
  56         return tree;
  57     }
  58
  59     return items;
  60 }
  61
  62
  63 const projectsReducer = (state: ProjectState = [], action: ProjectAction) => {
  64     return actions.match(action, {
  65         CREATE_PROJECT: project => [...state, project],
  66         REMOVE_PROJECT: () => state,
  67         PROJECTS_REQUEST: () => state,
  68         PROJECTS_SUCCESS: ({ projects, parentItemId }) => {
  69             return updateProjectTree(state, projects, parentItemId);
  70         },
  71         TOGGLE_PROJECT_TREE_ITEM: itemId => {
  72             const tree = _.cloneDeep(state);
  73             resetTreeActivity(tree);
  74             const item = findTreeItem(tree, itemId);
  75             if (item) {
  76                 item.open = !item.open;
  77                 item.active = true;
  78                 item.isLoaded = false;
  79             }
  80             return tree;
  81         },
  82         default: () => state
  83     });
  84 };
  85
  86 export default projectsReducer;