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, TreeItemStatus } from "../../components/tree/tree";
   8 import * as _ from "lodash";
   9
  10 export type ProjectState = Array<TreeItem<Project>>;
  11
  12 export 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 updateProjectTree(tree: Array<TreeItem<Project>>, projects: Project[], parentItemId?: string): Array<TreeItem<Project>> {
  33     let treeItem;
  34     if (parentItemId) {
  35         treeItem = findTreeItem(tree, parentItemId);
  36         if (treeItem) {
  37             treeItem.status = TreeItemStatus.Loaded;
  38         }
  39     }
  40     const items = projects.map((p, idx) => ({
  41         id: p.uuid,
  42         open: false,
  43         active: false,
  44         status: TreeItemStatus.Initial,
  45         data: p,
  46         items: []
  47     } as TreeItem<Project>));
  48
  49     if (treeItem) {
  50         treeItem.items = items;
  51         return tree;
  52     }
  53
  54     return items;
  55 }
  56
  57 const projectsReducer = (state: ProjectState = [], action: ProjectAction) => {
  58     return actions.match(action, {
  59         CREATE_PROJECT: project => [...state, project],
  60         REMOVE_PROJECT: () => state,
  61         PROJECTS_REQUEST: itemId => {
  62             const tree = _.cloneDeep(state);
  63             const item = findTreeItem(tree, itemId);
  64             if (item) {
  65                 item.status = TreeItemStatus.Pending;
  66             }
  67             return tree;
  68         },
  69         PROJECTS_SUCCESS: ({ projects, parentItemId }) => {
  70             return updateProjectTree(state, projects, parentItemId);
  71         },
  72         TOGGLE_PROJECT_TREE_ITEM: itemId => {
  73             const tree = _.cloneDeep(state);
  74             resetTreeActivity(tree);
  75             const item = findTreeItem(tree, itemId);
  76             if (item) {
  77                 item.open = !item.open;
  78                 item.active = true;
  79                 item.toggled = true;
  80             }
  81             return tree;
  82         },
  83         default: () => state
  84     });
  85 };
  86
  87 export default projectsReducer;