/*
 * Copyright (C) 2019-2024 CZ.NIC z.s.p.o. (https://www.nic.cz/)
 *
 * This is free software, licensed under the GNU General Public License v3.
 * See /LICENSE for more information.
 */

import React from "react";

import ErrorMessage from "./ErrorMessage";
import { API_STATE } from "../api/utils";
import { Spinner } from "../bootstrap/Spinner";

function withEither(conditionalFn, Either) {
    return (Component) => {
        function WithEither(props) {
            if (conditionalFn(props)) {
                return <Either {...props} />;
            }
            return <Component {...props} />;
        }

        // Setting displayName for better debugging
        WithEither.displayName = `WithEither(${Component.displayName || Component.name || "Component"})`;

        return WithEither;
    };
}

// Loading

function isSending(props) {
    if (Array.isArray(props.apiState)) {
        return props.apiState.some((state) =>
            [API_STATE.INIT, API_STATE.SENDING].includes(state)
        );
    }
    return [API_STATE.INIT, API_STATE.SENDING].includes(props.apiState);
}

const withSpinner = (conditionalFn) => withEither(conditionalFn, Spinner);
const withSending = (Either) => withEither(isSending, Either);
const withSpinnerOnSending = withSpinner(isSending);

// Error handling

const withError = (conditionalFn) => withEither(conditionalFn, ErrorMessage);
const withErrorMessage = withError((props) => {
    if (Array.isArray(props.apiState)) {
        return props.apiState.includes(API_STATE.ERROR);
    }
    return props.apiState === API_STATE.ERROR;
});

export {
    withEither,
    withSpinner,
    withSending,
    withSpinnerOnSending,
    withError,
    withErrorMessage,
};