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

import { ContainerRequestResource } from '../../models/container-request';
import { ContainerResource } from '../../models/container';
import { ResourcesState, getResource } from '~/store/resources/resources';
import { filterResources } from '../resources/resources';
import { ResourceKind, Resource } from '~/models/resource';

export interface Process {
    containerRequest: ContainerRequestResource;
    container?: ContainerResource;
}

export const getProcess = (uuid: string) => (resources: ResourcesState): Process | undefined => {
    const containerRequest = getResource<ContainerRequestResource>(uuid)(resources);
    if (containerRequest) {
        if (containerRequest.containerUuid) {
            const container = getResource<ContainerResource>(containerRequest.containerUuid)(resources);
            if (container) {
                return { containerRequest, container };
            }
        }
        return { containerRequest };
    }
    return;
};

export const getSubprocesses = (uuid: string) => (resources: ResourcesState) => {
    const containerRequests = filterResources(isSubprocess(uuid))(resources) as ContainerRequestResource[];
    return containerRequests.reduce((subprocesses, { uuid }) => {
        const process = getProcess(uuid)(resources);
        return process
            ? [...subprocesses, process]
            : subprocesses;
    }, []);
};

export const getProcessStatus = (process: Process) =>
    process.container
        ? process.container.state
        : process.containerRequest.state;

const isSubprocess = (uuid: string) => (resource: Resource) =>
    resource.kind === ResourceKind.CONTAINER_REQUEST
    && (resource as ContainerRequestResource).requestingContainerUuid === uuid;