foris-js/src/common/RebootButton.js

/*
 * 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, { useState, useEffect } from "react";

import PropTypes from "prop-types";

import { useAPIPost } from "../api/hooks";
import { API_STATE } from "../api/utils";
import Button from "../bootstrap/Button";
import { Modal, ModalHeader, ModalBody, ModalFooter } from "../bootstrap/Modal";
import { useAlert } from "../context/alertContext/AlertContext";
import { ForisURLs } from "../utils/forisUrls";

RebootButton.propTypes = {
    /** Additional props to be passed to the button */
    props: PropTypes.object,
};

function RebootButton(props) {
    const [triggered, setTriggered] = useState(false);
    const [modalShown, setModalShown] = useState(false);
    const [triggerRebootStatus, triggerReboot] = useAPIPost(ForisURLs.reboot);

    const [setAlert] = useAlert();
    useEffect(() => {
        if (triggerRebootStatus.state === API_STATE.ERROR) {
            setAlert(_("Reboot request failed."));
        }
    });

    const rebootHandler = () => {
        setTriggered(true);
        triggerReboot();
        setModalShown(false);
    };

    return (
        <>
            <RebootModal
                shown={modalShown}
                setShown={setModalShown}
                onReboot={rebootHandler}
            />
            <Button
                className="btn-danger"
                loading={triggered}
                disabled={triggered}
                onClick={() => setModalShown(true)}
                {...props}
            >
                {_("Reboot")}
            </Button>
        </>
    );
}

RebootModal.propTypes = {
    shown: PropTypes.bool.isRequired,
    setShown: PropTypes.func.isRequired,
    onReboot: PropTypes.func.isRequired,
};

function RebootModal({ shown, setShown, onReboot }) {
    return (
        <Modal shown={shown} setShown={setShown}>
            <ModalHeader setShown={setShown} title={_("Warning!")} />
            <ModalBody>
                <p>{_("Are you sure you want to restart the router?")}</p>
            </ModalBody>
            <ModalFooter>
                <Button
                    className="btn-secondary"
                    onClick={() => setShown(false)}
                >
                    {_("Cancel")}
                </Button>
                <Button className="btn-danger" onClick={onReboot}>
                    {_("Confirm reboot")}
                </Button>
            </ModalFooter>
        </Modal>
    );
}

export default RebootButton;