1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
5 import { createTree, TreeNode, setNode, Tree, TreeNodeStatus, setNodeStatus, expandNode, deactivateNode } from '~/models/tree';
6 import { TreePicker } from "./tree-picker";
7 import { treePickerActions, TreePickerAction } from "./tree-picker-actions";
8 import { compose } from "redux";
9 import { activateNode, getNode, toggleNodeCollapse, toggleNodeSelection } from '~/models/tree';
11 export const treePickerReducer = (state: TreePicker = {}, action: TreePickerAction) =>
12 treePickerActions.match(action, {
13 LOAD_TREE_PICKER_NODE: ({ id, pickerId }) =>
14 updateOrCreatePicker(state, pickerId, setNodeStatus(id)(TreeNodeStatus.PENDING)),
15 LOAD_TREE_PICKER_NODE_SUCCESS: ({ id, nodes, pickerId }) =>
16 updateOrCreatePicker(state, pickerId, compose(receiveNodes(nodes)(id), setNodeStatus(id)(TreeNodeStatus.LOADED))),
17 TOGGLE_TREE_PICKER_NODE_COLLAPSE: ({ id, pickerId }) =>
18 updateOrCreatePicker(state, pickerId, toggleNodeCollapse(id)),
19 ACTIVATE_TREE_PICKER_NODE: ({ id, pickerId }) =>
20 updateOrCreatePicker(state, pickerId, activateNode(id)),
21 DEACTIVATE_TREE_PICKER_NODE: ({ pickerId }) =>
22 updateOrCreatePicker(state, pickerId, deactivateNode),
23 TOGGLE_TREE_PICKER_NODE_SELECTION: ({ id, pickerId }) =>
24 updateOrCreatePicker(state, pickerId, toggleNodeSelection(id)),
25 RESET_TREE_PICKER: ({ pickerId }) =>
26 updateOrCreatePicker(state, pickerId, createTree),
27 EXPAND_TREE_PICKER_NODES: ({ pickerId, ids }) =>
28 updateOrCreatePicker(state, pickerId, expandNode(...ids)),
32 const updateOrCreatePicker = <V>(state: TreePicker, pickerId: string, func: (value: Tree<V>) => Tree<V>) => {
33 const picker = state[pickerId] || createTree();
34 const updatedPicker = func(picker);
35 return { ...state, [pickerId]: updatedPicker };
38 const receiveNodes = <V>(nodes: Array<TreeNode<V>>) => (parent: string) => (state: Tree<V>) => {
39 const parentNode = getNode(parent)(state);
42 newState = setNode({ ...parentNode, children: [] })(state);
44 return nodes.reduce((tree, node) => {
45 return setNode({ ...node, parent })(tree);