// Copyright (C) The Arvados Authors. All rights reserved.
//
// SPDX-License-Identifier: AGPL-3.0

import {
    createTree, TreeNode, setNode, Tree, TreeNodeStatus, setNodeStatus,
    expandNode, deactivateNode, selectNodes, deselectNodes,
    activateNode, getNode, toggleNodeCollapse, toggleNodeSelection, appendSubtree, expandNodeAncestors
} from 'models/tree';
import { TreePicker } from "./tree-picker";
import { treePickerActions, treePickerSearchActions, TreePickerAction, TreePickerSearchAction, LoadProjectParams } from "./tree-picker-actions";
import { compose } from "redux";
import { pipe } from 'lodash/fp';

export const treePickerReducer = (state: TreePicker = {}, action: TreePickerAction) =>
    treePickerActions.match(action, {
        LOAD_TREE_PICKER_NODE: ({ id, pickerId }) =>
            updateOrCreatePicker(state, pickerId, setNodeStatus(id)(TreeNodeStatus.PENDING)),

        LOAD_TREE_PICKER_NODE_SUCCESS: ({ id, nodes, pickerId }) =>
            updateOrCreatePicker(state, pickerId, compose(receiveNodes(nodes)(id), setNodeStatus(id)(TreeNodeStatus.LOADED))),

        APPEND_TREE_PICKER_NODE_SUBTREE: ({ id, subtree, pickerId }) =>
            updateOrCreatePicker(state, pickerId, compose(appendSubtree(id, subtree), setNodeStatus(id)(TreeNodeStatus.LOADED))),

        TOGGLE_TREE_PICKER_NODE_COLLAPSE: ({ id, pickerId }) =>
            updateOrCreatePicker(state, pickerId, toggleNodeCollapse(id)),

        EXPAND_TREE_PICKER_NODE: ({ id, pickerId }) =>
            updateOrCreatePicker(state, pickerId, expandNode(id)),

        EXPAND_TREE_PICKER_NODE_ANCESTORS: ({ id, pickerId }) =>
            updateOrCreatePicker(state, pickerId, expandNodeAncestors(id)),

        ACTIVATE_TREE_PICKER_NODE: ({ id, pickerId, relatedTreePickers = [] }) =>
            pipe(
                () => relatedTreePickers.reduce(
                    (state, relatedPickerId) => updateOrCreatePicker(state, relatedPickerId, deactivateNode),
                    state
                ),
                state => updateOrCreatePicker(state, pickerId, activateNode(id))
            )(),

        DEACTIVATE_TREE_PICKER_NODE: ({ pickerId }) =>
            updateOrCreatePicker(state, pickerId, deactivateNode),

        TOGGLE_TREE_PICKER_NODE_SELECTION: ({ id, pickerId, cascade }) =>
            updateOrCreatePicker(state, pickerId, toggleNodeSelection(id, cascade)),

        SELECT_TREE_PICKER_NODE: ({ id, pickerId, cascade }) =>
            updateOrCreatePicker(state, pickerId, selectNodes(id, cascade)),

        DESELECT_TREE_PICKER_NODE: ({ id, pickerId, cascade }) =>
            updateOrCreatePicker(state, pickerId, deselectNodes(id, cascade)),

        RESET_TREE_PICKER: ({ pickerId }) =>
            updateOrCreatePicker(state, pickerId, createTree),

        EXPAND_TREE_PICKER_NODES: ({ pickerId, ids }) =>
            updateOrCreatePicker(state, pickerId, expandNode(...ids)),

        default: () => state
    });

const updateOrCreatePicker = <V>(state: TreePicker, pickerId: string, func: (value: Tree<V>) => Tree<V>) => {
    const picker = state[pickerId] || createTree();
    const updatedPicker = func(picker);
    return { ...state, [pickerId]: updatedPicker };
};

const receiveNodes = <V>(nodes: Array<TreeNode<V>>) => (parent: string) => (state: Tree<V>) => {
    const parentNode = getNode(parent)(state);
    let newState = state;
    if (parentNode) {
        newState = setNode({ ...parentNode, children: [] })(state);
    }
    return nodes.reduce((tree, node) => {
        const preexistingNode = getNode(node.id)(state);
        if (preexistingNode) {
            node = { ...preexistingNode, value: node.value };
        }
        return setNode({ ...node, parent })(tree);
    }, newState);
};

interface TreePickerSearch {
    projectSearchValues: { [pickerId: string]: string };
    collectionFilterValues: { [pickerId: string]: string };
    loadProjectParams: { [pickerId: string]: LoadProjectParams };
}

export const treePickerSearchReducer = (state: TreePickerSearch = { projectSearchValues: {}, collectionFilterValues: {}, loadProjectParams: {} }, action: TreePickerSearchAction) =>
    treePickerSearchActions.match(action, {
        SET_TREE_PICKER_PROJECT_SEARCH: ({ pickerId, projectSearchValue }) => ({
            ...state, projectSearchValues: { ...state.projectSearchValues, [pickerId]: projectSearchValue }
        }),

        SET_TREE_PICKER_COLLECTION_FILTER: ({ pickerId, collectionFilterValue }) => ({
            ...state, collectionFilterValues: { ...state.collectionFilterValues, [pickerId]: collectionFilterValue }
        }),

        SET_TREE_PICKER_LOAD_PARAMS: ({ pickerId, params }) => ({
            ...state, loadProjectParams: { ...state.loadProjectParams, [pickerId]: params }
        }),

        default: () => state
    });