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 * as _ from "lodash";
   6
   7 import { Project } from "../../models/project";
   8 import actions, { ProjectAction } from "./project-action";
   9 import { TreeItem, TreeItemStatus } from "../../components/tree/tree";
  10
  11 export type ProjectState = Array<TreeItem<Project>>;
  12
  13 export function findTreeItem<T>(tree: Array<TreeItem<T>>, itemId: string): TreeItem<T> | undefined {
  14     let item;
  15     for (const t of tree) {
  16         item = t.id === itemId
  17             ? t
  18             : findTreeItem(t.items ? t.items : [], itemId);
  19         if (item) {
  20             break;
  21         }
  22     }
  23     return item;
  24 }
  25
  26 export function getTreePath<T>(tree: Array<TreeItem<T>>, itemId: string): Array<TreeItem<T>> {
  27     for(const item of tree){
  28         if(item.id === itemId){
  29             return [item];
  30         } else {
  31             const branch = getTreePath(item.items || [], itemId);
  32             if(branch.length > 0){
  33                 return [item, ...branch];
  34             }
  35         }
  36     }
  37     return [];
  38 }
  39
  40 function resetTreeActivity<T>(tree: Array<TreeItem<T>>) {
  41     for (const t of tree) {
  42         t.active = false;
  43         resetTreeActivity(t.items ? t.items : []);
  44     }
  45 }
  46
  47 function updateProjectTree(tree: Array<TreeItem<Project>>, projects: Project[], parentItemId?: string): Array<TreeItem<Project>> {
  48     let treeItem;
  49     if (parentItemId) {
  50         treeItem = findTreeItem(tree, parentItemId);
  51         if (treeItem) {
  52             treeItem.status = TreeItemStatus.Loaded;
  53         }
  54     }
  55     const items = projects.map((p, idx) => ({
  56         id: p.uuid,
  57         open: false,
  58         active: false,
  59         status: TreeItemStatus.Initial,
  60         data: p,
  61         items: []
  62     } as TreeItem<Project>));
  63
  64     if (treeItem) {
  65         treeItem.items = items;
  66         return tree;
  67     }
  68
  69     return items;
  70 }
  71
  72 const projectsReducer = (state: ProjectState = [], action: ProjectAction) => {
  73     return actions.match(action, {
  74         CREATE_PROJECT: project => [...state, project],
  75         REMOVE_PROJECT: () => state,
  76         PROJECTS_REQUEST: itemId => {
  77             const tree = _.cloneDeep(state);
  78             const item = findTreeItem(tree, itemId);
  79             if (item) {
  80                 item.status = TreeItemStatus.Pending;
  81             }
  82             return tree;
  83         },
  84         PROJECTS_SUCCESS: ({ projects, parentItemId }) => {
  85             return updateProjectTree(state, projects, parentItemId);
  86         },
  87         TOGGLE_PROJECT_TREE_ITEM_OPEN: itemId => {
  88             const tree = _.cloneDeep(state);
  89             const item = findTreeItem(tree, itemId);
  90             if (item) {
  91                 item.toggled = true;
  92                 item.open = !item.open;
  93             }
  94             return tree;
  95         },
  96         TOGGLE_PROJECT_TREE_ITEM_ACTIVE: itemId => {
  97             const tree = _.cloneDeep(state);
  98             resetTreeActivity(tree);
  99             const item = findTreeItem(tree, itemId);
 100             if (item) {
 101                 item.active = true;
 102             }
 103             return tree;
 104         },
 105         RESET_PROJECT_TREE_ACTIVITY: () => {
 106             const tree = _.cloneDeep(state);
 107             resetTreeActivity(tree);
 108             return tree;
 109         },
 110         default: () => state
 111     });
 112 };
 113
 114 export default projectsReducer;