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 export function findTreeBranch<T>(tree: Array<TreeItem<T>>, itemId: string): Array<TreeItem<T>> {
  26     for(const item of tree){
  27         if(item.id === itemId){
  28             return [item];
  29         } else {
  30             const branch = findTreeBranch(item.items || [], itemId);
  31             if(branch.length > 0){
  32                 return [item, ...branch];
  33             }
  34         }
  35     }
  36     return [];
  37 }
  38
  39 function resetTreeActivity<T>(tree: Array<TreeItem<T>>) {
  40     for (const t of tree) {
  41         t.active = false;
  42         resetTreeActivity(t.items ? t.items : []);
  43     }
  44 }
  45
  46 function updateProjectTree(tree: Array<TreeItem<Project>>, projects: Project[], parentItemId?: string): Array<TreeItem<Project>> {
  47     let treeItem;
  48     if (parentItemId) {
  49         treeItem = findTreeItem(tree, parentItemId);
  50         if (treeItem) {
  51             treeItem.status = TreeItemStatus.Loaded;
  52         }
  53     }
  54     const items = projects.map((p, idx) => ({
  55         id: p.uuid,
  56         open: false,
  57         active: false,
  58         status: TreeItemStatus.Initial,
  59         data: p,
  60         items: []
  61     } as TreeItem<Project>));
  62
  63     if (treeItem) {
  64         treeItem.items = items;
  65         return tree;
  66     }
  67
  68     return items;
  69 }
  70
  71 const projectsReducer = (state: ProjectState = [], action: ProjectAction) => {
  72     return actions.match(action, {
  73         CREATE_PROJECT: project => [...state, project],
  74         REMOVE_PROJECT: () => state,
  75         PROJECTS_REQUEST: itemId => {
  76             const tree = _.cloneDeep(state);
  77             const item = findTreeItem(tree, itemId);
  78             if (item) {
  79                 item.status = TreeItemStatus.Pending;
  80             }
  81             return tree;
  82         },
  83         PROJECTS_SUCCESS: ({ projects, parentItemId }) => {
  84             return updateProjectTree(state, projects, parentItemId);
  85         },
  86         TOGGLE_PROJECT_TREE_ITEM: itemId => {
  87             const tree = _.cloneDeep(state);
  88             resetTreeActivity(tree);
  89             const item = findTreeItem(tree, itemId);
  90             if (item) {
  91                 item.open = !item.open;
  92                 item.active = true;
  93                 item.toggled = true;
  94             }
  95             return tree;
  96         },
  97         default: () => state
  98     });
  99 };
 100
 101 export default projectsReducer;