// Copyright Contributors to the DNF5 project.
// Copyright Contributors to the libdnf project.
// SPDX-License-Identifier: LGPL-2.1-or-later
//
// This file is part of libdnf: https://github.com/rpm-software-management/libdnf/
//
// Libdnf is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 2.1 of the License, or
// (at your option) any later version.
//
// Libdnf is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with libdnf.  If not, see <https://www.gnu.org/licenses/>.

#ifndef LIBDNF5_UTILS_FORMAT_LOCALE_HPP
#define LIBDNF5_UTILS_FORMAT_LOCALE_HPP

#include "bgettext/bgettext-mark-common.h"
#include "locale.hpp"

#include <fmt/format.h>


namespace libdnf5::utils {

/// Format `args` according to the `format_string`, and return the result as a string.
///
/// @param translate     If `true`, it will attempt to translate the message to the requested locale.
/// @param format_string message contains formatting string
/// @param plural_form   number is used to select the appropriate plural form of the format string
/// @param args          arguments to be formatted
/// @return A string object holding the formatted result.
template <typename... Args>
std::string format(bool translate, BgettextMessage format_string, unsigned long plural_form, Args &&... args) {
    const char * final_format_string{nullptr};

    if (translate) {
        final_format_string = b_dmgettext(NULL, format_string, plural_form);
    } else {
        if (plural_form > 1) {
            final_format_string = b_gettextmsg_get_plural_id(format_string);
        }
        if (!final_format_string) {
            final_format_string = b_gettextmsg_get_id(format_string);
        }
    }

    return fmt::vformat(final_format_string, fmt::make_format_args(args...));
}


/// Format `args` according to the `format_string`, and return the result as a string.
///
/// @param locale        requested locale for translation and argument formatting, nullptr = use global/thread locale
/// @param translate     If `true`, it will attempt to translate the message to the requested locale.
/// @param format_string message contains formatting string
/// @param plural_form   number is used to select the appropriate plural form of the format string
/// @param args          arguments to be formatted
/// @return A string object holding the formatted result.
template <typename... Args>
std::string format(
    const Locale * locale, bool translate, BgettextMessage format_string, unsigned long plural_form, Args &&... args) {
    const char * final_format_string{nullptr};

    if (!locale) {
        return utils::format(translate, format_string, plural_form, args...);
    }

    if (translate) {
        // sets the requested C locale for this thread, needed for bgettext
        ::locale_t previous_locale = uselocale(locale->get_c_locale());

        final_format_string = b_dmgettext(NULL, format_string, plural_form);

        // restores the previous C locale for this thread
        uselocale(previous_locale);
    } else {
        if (plural_form > 1) {
            final_format_string = b_gettextmsg_get_plural_id(format_string);
        }
        if (!final_format_string) {
            final_format_string = b_gettextmsg_get_id(format_string);
        }
    }

    return fmt::vformat(locale->get_cpp_locale(), final_format_string, fmt::make_format_args(args...));
}

}  // namespace libdnf5::utils

#endif  // LIBDNF5_UTILS_FORMAT_LOCALE_HPP
