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