/*
 * Copyright (C) Contributors to the Suwayomi project
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

import { useLocation, useNavigate } from 'react-router-dom';
import { useContext } from 'react';
import { useHistory } from '@/util/useHistory.ts';
import { NavBarContext } from '@/components/context/NavbarContext.tsx';

export const useBackButton = () => {
    const navigate = useNavigate();
    const history = useHistory();
    const location = useLocation();
    const { defaultBackTo: backToUrl } = useContext(NavBarContext);

    return () => {
        const isLastPageInHistory = location.key === 'default';
        const wasPreviousPageReader = history[history.length - 2]?.match(/\/manga\/[0-9]+\/chapter\/[0-9]+.*/g);

        if (isLastPageInHistory || wasPreviousPageReader) {
            navigate(backToUrl ?? '');
            return;
        }

        navigate(-1);
    };
};