1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
5 import { unionize, ofType, UnionOf } from "~/common/unionize";
6 import { TreeNode, initTreeNode, getNodeDescendants, getNodeDescendantsIds, getNodeValue, TreeNodeStatus, getNode } from '~/models/tree';
7 import { Dispatch } from 'redux';
8 import { RootState } from '~/store/store';
9 import { ServiceRepository } from '~/services/services';
10 import { FilterBuilder } from '~/services/api/filter-builder';
11 import { pipe, map, values, mapValues } from 'lodash/fp';
12 import { ResourceKind } from '~/models/resource';
13 import { GroupContentsResource } from '../../services/groups-service/groups-service';
14 import { CollectionDirectory, CollectionFile } from '../../models/collection-file';
15 import { getTreePicker, TreePicker } from './tree-picker';
16 import { ProjectsTreePickerItem } from '~/views-components/projects-tree-picker/generic-projects-tree-picker';
18 export const treePickerActions = unionize({
19 LOAD_TREE_PICKER_NODE: ofType<{ id: string, pickerId: string }>(),
20 LOAD_TREE_PICKER_NODE_SUCCESS: ofType<{ id: string, nodes: Array<TreeNode<any>>, pickerId: string }>(),
21 TOGGLE_TREE_PICKER_NODE_COLLAPSE: ofType<{ id: string, pickerId: string }>(),
22 ACTIVATE_TREE_PICKER_NODE: ofType<{ id: string, pickerId: string }>(),
23 DEACTIVATE_TREE_PICKER_NODE: ofType<{ pickerId: string }>(),
24 TOGGLE_TREE_PICKER_NODE_SELECTION: ofType<{ id: string, pickerId: string }>(),
25 SELECT_TREE_PICKER_NODE: ofType<{ id: string | string[], pickerId: string }>(),
26 DESELECT_TREE_PICKER_NODE: ofType<{ id: string | string[], pickerId: string }>(),
27 EXPAND_TREE_PICKER_NODES: ofType<{ ids: string[], pickerId: string }>(),
28 RESET_TREE_PICKER: ofType<{ pickerId: string }>()
31 export type TreePickerAction = UnionOf<typeof treePickerActions>;
33 export const getProjectsTreePickerIds = (pickerId: string) => ({
34 home: `${pickerId}_home`,
35 shared: `${pickerId}_shared`,
36 favorites: `${pickerId}_favorites`,
39 export const getAllNodes = <Value>(pickerId: string, filter = (node: TreeNode<Value>) => true) => (state: TreePicker) =>
41 () => values(getProjectsTreePickerIds(pickerId)),
44 .map(id => getTreePicker<Value>(id)(state)),
47 .map(getNodeDescendants(''))
48 .reduce((allNodes, nodes) => allNodes.concat(nodes), []),
51 .reduce((map, node) =>
53 ? map.set(node.id, node)
54 : map, new Map<string, TreeNode<Value>>())
57 uniqueNodes => Array.from(uniqueNodes),
59 export const getSelectedNodes = <Value>(pickerId: string) => (state: TreePicker) =>
60 getAllNodes<Value>(pickerId, node => node.selected)(state);
62 export const initProjectsTreePicker = (pickerId: string) =>
63 async (dispatch: Dispatch, _: () => RootState, services: ServiceRepository) => {
64 const { home, shared, favorites } = getProjectsTreePickerIds(pickerId);
65 dispatch<any>(initUserProject(home));
66 dispatch<any>(initSharedProject(shared));
67 dispatch<any>(initFavoritesProject(favorites));
70 interface ReceiveTreePickerDataParams<T> {
72 extractNodeData: (value: T) => { id: string, value: T, status?: TreeNodeStatus };
76 export const receiveTreePickerData = <T>(params: ReceiveTreePickerDataParams<T>) =>
77 (dispatch: Dispatch) => {
78 const { data, extractNodeData, id, pickerId, } = params;
79 dispatch(treePickerActions.LOAD_TREE_PICKER_NODE_SUCCESS({
81 nodes: data.map(item => initTreeNode(extractNodeData(item))),
84 dispatch(treePickerActions.TOGGLE_TREE_PICKER_NODE_COLLAPSE({ id, pickerId }));
87 interface LoadProjectParams {
90 includeCollections?: boolean;
91 includeFiles?: boolean;
94 export const loadProject = (params: LoadProjectParams) =>
95 async (dispatch: Dispatch, _: () => RootState, services: ServiceRepository) => {
96 const { id, pickerId, includeCollections = false, includeFiles = false, loadShared = false } = params;
98 dispatch(treePickerActions.LOAD_TREE_PICKER_NODE({ id, pickerId }));
100 const filters = pipe(
101 (fb: FilterBuilder) => includeCollections
102 ? fb.addIsA('uuid', [ResourceKind.PROJECT, ResourceKind.COLLECTION])
103 : fb.addIsA('uuid', [ResourceKind.PROJECT]),
104 fb => fb.getFilters(),
105 )(new FilterBuilder());
107 const { items } = await services.groupsService.contents(loadShared ? '' : id, { filters, excludeHomeProject: loadShared || undefined });
109 dispatch<any>(receiveTreePickerData<GroupContentsResource>({
113 extractNodeData: item => ({
116 status: item.kind === ResourceKind.PROJECT
117 ? TreeNodeStatus.INITIAL
119 ? TreeNodeStatus.INITIAL
120 : TreeNodeStatus.LOADED
125 export const loadCollection = (id: string, pickerId: string) =>
126 async (dispatch: Dispatch, getState: () => RootState, services: ServiceRepository) => {
127 dispatch(treePickerActions.LOAD_TREE_PICKER_NODE({ id, pickerId }));
129 const picker = getTreePicker<ProjectsTreePickerItem>(pickerId)(getState().treePicker);
132 const node = getNode(id)(picker);
133 if (node && 'kind' in node.value && node.value.kind === ResourceKind.COLLECTION) {
135 const files = await services.collectionService.files(node.value.portableDataHash);
136 const data = getNodeDescendants('')(files).map(node => node.value);
138 dispatch<any>(receiveTreePickerData<CollectionDirectory | CollectionFile>({
142 extractNodeData: value => ({
144 status: TreeNodeStatus.LOADED,
153 export const initUserProject = (pickerId: string) =>
154 async (dispatch: Dispatch<any>, getState: () => RootState, services: ServiceRepository) => {
155 const uuid = services.authService.getUuid();
157 dispatch(receiveTreePickerData({
160 data: [{ uuid, name: 'Projects' }],
161 extractNodeData: value => ({
163 status: TreeNodeStatus.INITIAL,
169 export const loadUserProject = (pickerId: string, includeCollections = false, includeFiles = false) =>
170 async (dispatch: Dispatch<any>, getState: () => RootState, services: ServiceRepository) => {
171 const uuid = services.authService.getUuid();
173 dispatch(loadProject({ id: uuid, pickerId, includeCollections, includeFiles }));
178 export const initSharedProject = (pickerId: string) =>
179 async (dispatch: Dispatch<any>, getState: () => RootState, services: ServiceRepository) => {
180 dispatch(receiveTreePickerData({
183 data: [{ uuid: 'Shared with me', name: 'Shared with me' }],
184 extractNodeData: value => ({
186 status: TreeNodeStatus.INITIAL,
192 export const initFavoritesProject = (pickerId: string) =>
193 async (dispatch: Dispatch<any>, getState: () => RootState, services: ServiceRepository) => {
194 dispatch(receiveTreePickerData({
197 data: [{ uuid: 'Favorites', name: 'Favorites' }],
198 extractNodeData: value => ({
200 status: TreeNodeStatus.INITIAL,
206 interface LoadFavoritesProjectParams {
208 includeCollections?: boolean;
209 includeFiles?: boolean;
211 export const loadFavoritesProject = (params: LoadFavoritesProjectParams) =>
212 async (dispatch: Dispatch<any>, getState: () => RootState, services: ServiceRepository) => {
213 const { pickerId, includeCollections = false, includeFiles = false } = params;
214 const uuid = services.authService.getUuid();
217 const filters = pipe(
218 (fb: FilterBuilder) => includeCollections
219 ? fb.addIsA('headUuid', [ResourceKind.PROJECT, ResourceKind.COLLECTION])
220 : fb.addIsA('headUuid', [ResourceKind.PROJECT]),
221 fb => fb.getFilters(),
222 )(new FilterBuilder());
224 const { items } = await services.favoriteService.list(uuid, { filters });
226 dispatch<any>(receiveTreePickerData<GroupContentsResource>({
230 extractNodeData: item => ({
233 status: item.kind === ResourceKind.PROJECT
234 ? TreeNodeStatus.INITIAL
236 ? TreeNodeStatus.INITIAL
237 : TreeNodeStatus.LOADED