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

import { ServiceRepository } from '~/services/services';
import { MiddlewareAPI, Dispatch } from 'redux';
import {
    DataExplorerMiddlewareService, dataExplorerToListParams, listResultsToDataExplorerItemsMeta, getDataExplorerColumnFilters
} from '~/store/data-explorer/data-explorer-middleware-service';
import { RootState } from '~/store/store';
import { snackbarActions, SnackbarKind } from '~/store/snackbar/snackbar-actions';
import { DataExplorer, getDataExplorer } from '~/store/data-explorer/data-explorer-reducer';
import { updateResources } from '~/store/resources/resources-actions';
import { SortDirection } from '~/components/data-table/data-column';
import { OrderDirection, OrderBuilder } from '~/services/api/order-builder';
import { ListResults } from '~/services/common-service/common-service';
import { getSortColumn } from "~/store/data-explorer/data-explorer-reducer";
import { ProcessResource } from '~/models/process';
import { SubprocessPanelColumnNames } from '~/views/subprocess-panel/subprocess-panel-root';
import { FilterBuilder, joinFilters } from '~/services/api/filter-builder';
import { subprocessPanelActions } from './subprocess-panel-actions';
import { DataColumns } from '~/components/data-table/data-table';
import { ProcessStatusFilter } from '../resource-type-filters/resource-type-filters';
import { ContainerRequestResource } from '~/models/container-request';
import { progressIndicatorActions } from '../progress-indicator/progress-indicator-actions';
import { loadMissingProcessesInformation } from '../project-panel/project-panel-middleware-service';

export class SubprocessMiddlewareService extends DataExplorerMiddlewareService {
    constructor(private services: ServiceRepository, id: string) {
        super(id);
    }

    async requestItems(api: MiddlewareAPI<Dispatch, RootState>) {
        const state = api.getState();
        const dataExplorer = getDataExplorer(state.dataExplorer, this.getId());

        try {
            api.dispatch(progressIndicatorActions.START_WORKING(this.getId()));
            const parentContainerRequestUuid = state.processPanel.containerRequestUuid;
            const parentContainerRequest = await this.services.containerRequestService.get(parentContainerRequestUuid);
            const containerRequests = await this.services.containerRequestService.list(
                { ...getParams(dataExplorer, parentContainerRequest) });

            api.dispatch(progressIndicatorActions.PERSIST_STOP_WORKING(this.getId()));
            api.dispatch(updateResources(containerRequests.items));
            await api.dispatch<any>(loadMissingProcessesInformation(containerRequests.items));
            // Populate the actual user view
            api.dispatch(setItems(containerRequests));
        } catch {
            api.dispatch(progressIndicatorActions.PERSIST_STOP_WORKING(this.getId()));
            api.dispatch(couldNotFetchSubprocesses());
        }
    }
}

export const getParams = (
    dataExplorer: DataExplorer,
    parentContainerRequest: ContainerRequestResource) => ({
        ...dataExplorerToListParams(dataExplorer),
        order: getOrder(dataExplorer),
        filters: getFilters(dataExplorer, parentContainerRequest)
    });

const getOrder = (dataExplorer: DataExplorer) => {
    const sortColumn = getSortColumn(dataExplorer);
    const order = new OrderBuilder<ProcessResource>();
    if (sortColumn) {
        const sortDirection = sortColumn && sortColumn.sortDirection === SortDirection.ASC
            ? OrderDirection.ASC
            : OrderDirection.DESC;

        const columnName = sortColumn && sortColumn.name === SubprocessPanelColumnNames.NAME ? "name" : "modifiedAt";
        return order
            .addOrder(sortDirection, columnName)
            .getOrder();
    } else {
        return order.getOrder();
    }
};

export const getFilters = (
    dataExplorer: DataExplorer,
    parentContainerRequest: ContainerRequestResource) => {
        const columns = dataExplorer.columns as DataColumns<string>;
        const statusColumnFilters = getDataExplorerColumnFilters(columns, 'Status');
        const activeStatusFilter = Object.keys(statusColumnFilters).find(
            filterName => statusColumnFilters[filterName].selected
        );

        // Get all the subprocess' container requests and containers.
        const fb = new FilterBuilder().addEqual('requesting_container_uuid', parentContainerRequest.containerUuid);
        switch (activeStatusFilter) {
            case ProcessStatusFilter.COMPLETED: {
                fb.addEqual('container.state', 'Complete');
                fb.addEqual('container.exit_code', '0');
                break;
            }
            case ProcessStatusFilter.FAILED: {
                fb.addEqual('container.state', 'Complete');
                fb.addDistinct('container.exit_code', '0');
                break;
            }
            case ProcessStatusFilter.CANCELLED:
            case ProcessStatusFilter.FAILED:
            case ProcessStatusFilter.LOCKED:
            case ProcessStatusFilter.QUEUED:
            case ProcessStatusFilter.RUNNING: {
                fb.addEqual('container.state', activeStatusFilter);
                break;
            }
        }
        const statusFilters = fb.getFilters();

        const nameFilters = dataExplorer.searchValue
            ? new FilterBuilder()
                .addILike("name", dataExplorer.searchValue)
                .getFilters()
            : '';

        return joinFilters(
            nameFilters,
            statusFilters
        );
    };

export const setItems = (listResults: ListResults<ProcessResource>) =>
    subprocessPanelActions.SET_ITEMS({
        ...listResultsToDataExplorerItemsMeta(listResults),
        items: listResults.items.map(resource => resource.uuid),
    });

const couldNotFetchSubprocesses = () =>
    snackbarActions.OPEN_SNACKBAR({
        message: 'Could not fetch subprocesses.',
        kind: SnackbarKind.ERROR
    });