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 { projectActions, ProjectAction } from "./project-action";
   8 import { TreeItem, TreeItemStatus } from "~/components/tree/tree";
   9 import { ProjectResource } from "~/models/project";
  10
  11 export type ProjectState = {
  12     items: Array<TreeItem<ProjectResource>>,
  13     currentItemId: string
  14 };
  15
  16 interface ProjectUpdater {
  17     opened: boolean;
  18     uuid: string;
  19 }
  20
  21 export function findTreeItem<T>(tree: Array<TreeItem<T>>, itemId: string): TreeItem<T> | undefined {
  22     let item;
  23     for (const t of tree) {
  24         item = t.id === itemId
  25             ? t
  26             : findTreeItem(t.items ? t.items : [], itemId);
  27         if (item) {
  28             break;
  29         }
  30     }
  31     return item;
  32 }
  33
  34 export function getActiveTreeItem<T>(tree: Array<TreeItem<T>>): TreeItem<T> | undefined {
  35     let item;
  36     for (const t of tree) {
  37         item = t.active
  38             ? t
  39             : getActiveTreeItem(t.items ? t.items : []);
  40         if (item) {
  41             break;
  42         }
  43     }
  44     return item;
  45 }
  46
  47 export function getTreePath<T>(tree: Array<TreeItem<T>>, itemId: string): Array<TreeItem<T>> {
  48     for (const item of tree) {
  49         if (item.id === itemId) {
  50             return [item];
  51         } else {
  52             const branch = getTreePath(item.items || [], itemId);
  53             if (branch.length > 0) {
  54                 return [item, ...branch];
  55             }
  56         }
  57     }
  58     return [];
  59 }
  60
  61 function resetTreeActivity<T>(tree: Array<TreeItem<T>>) {
  62     for (const t of tree) {
  63         t.active = false;
  64         resetTreeActivity(t.items ? t.items : []);
  65     }
  66 }
  67
  68 function updateProjectTree(tree: Array<TreeItem<ProjectResource>>, projects: ProjectResource[], parentItemId?: string): Array<TreeItem<ProjectResource>> {
  69     let treeItem;
  70     if (parentItemId) {
  71         treeItem = findTreeItem(tree, parentItemId);
  72         if (treeItem) {
  73             treeItem.status = TreeItemStatus.LOADED;
  74         }
  75     }
  76     const items = projects.map(p => ({
  77         id: p.uuid,
  78         open: false,
  79         active: false,
  80         status: TreeItemStatus.INITIAL,
  81         data: p,
  82         items: []
  83     } as TreeItem<ProjectResource>));
  84
  85     if (treeItem) {
  86         treeItem.items = items;
  87         return tree;
  88     }
  89
  90     return items;
  91 }
  92
  93 const initialState: ProjectState = {
  94     items: [],
  95     currentItemId: ""
  96 };
  97
  98
  99 export const projectsReducer = (state: ProjectState = initialState, action: ProjectAction) => {
 100     return projectActions.match(action, {
 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, items };
 110         },
 111         PROJECTS_SUCCESS: ({ projects, parentItemId }) => {
 112             const items = _.cloneDeep(state.items);
 113             return {
 114                 ...state,
 115                 items: updateProjectTree(items, projects, parentItemId)
 116             };
 117         },
 118         TOGGLE_PROJECT_TREE_ITEM_OPEN: itemId => {
 119             const items = _.cloneDeep(state.items);
 120             const item = findTreeItem(items, itemId);
 121             if (item) {
 122                 item.open = !item.open;
 123             }
 124             return {
 125                 ...state,
 126                 items,
 127                 currentItemId: itemId
 128             };
 129         },
 130         TOGGLE_PROJECT_TREE_ITEM_ACTIVE: itemId => {
 131             const items = _.cloneDeep(state.items);
 132             resetTreeActivity(items);
 133             const item = findTreeItem(items, itemId);
 134             if (item) {
 135                 item.active = true;
 136             }
 137             return {
 138                 ...state,
 139                 items,
 140                 currentItemId: itemId
 141             };
 142         },
 143         RESET_PROJECT_TREE_ACTIVITY: () => {
 144             const items = _.cloneDeep(state.items);
 145             resetTreeActivity(items);
 146             return {
 147                 ...state,
 148                 items,
 149                 currentItemId: ""
 150             };
 151         },
 152         default: () => state
 153     });
 154 };