// Copyright (c) 2021-present The Bitcoin Core developers

// Distributed under the MIT software license, see the accompanying

// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#include <bitcoin-build-config.h> // IWYU pragma: keep

#include <init/common.h>

#include <clientversion.h>

#include <common/args.h>

#include <logging.h>

#include <node/interface_ui.h>

#include <tinyformat.h>

#include <util/fs.h>

#include <util/fs_helpers.h>

#include <util/result.h>

#include <util/string.h>

#include <util/time.h>

#include <util/translation.h>

#include <algorithm>

#include <ranges>

#include <string>

#include <vector>

using util::SplitString;

namespace init {

void AddLoggingArgs(ArgsManager& argsman)

{

    argsman.AddArg("-debuglogfile=<file>", strprintf("Specify location of debug log file (default: %s). Relative paths will be prefixed by a net-specific datadir location. Pass -nodebuglogfile to disable writing the log to a file.", DEFAULT_DEBUGLOGFILE), ArgsManager::ALLOW_ANY, OptionsCategory::OPTIONS);

    argsman.AddArg("-debug=<category>", "Output debug and trace logging (default: -nodebug, supplying <category> is optional). "

        "If <category> is not supplied or if <category> is 1 or \"all\", output all debug logging. If <category> is 0 or \"none\", any other categories are ignored. Other valid values for <category> are: " + LogInstance().LogCategoriesString() + ". This option can be specified multiple times to output multiple categories.",

        ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-debugexclude=<category>", "Exclude debug and trace logging for a category. Can be used in conjunction with -debug=1 to output debug and trace logging for all categories except the specified category. This option can be specified multiple times to exclude multiple categories. This takes priority over \"-debug\"", ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-logips", strprintf("Include IP addresses in log output (default: %u)", DEFAULT_LOGIPS), ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-loglevel=<level>|<category>:<level>", strprintf("Set the global or per-category severity level for logging categories enabled with the -debug configuration option or the logging RPC. Possible values are %s (default=%s). The following levels are always logged: error, warning, info. If <category>:<level> is supplied, the setting will override the global one and may be specified multiple times to set multiple category-specific levels. <category> can be: %s.", LogInstance().LogLevelsString(), LogInstance().LogLevelToStr(BCLog::DEFAULT_LOG_LEVEL), LogInstance().LogCategoriesString()), ArgsManager::DISALLOW_NEGATION | ArgsManager::DISALLOW_ELISION | ArgsManager::DEBUG_ONLY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-logtimestamps", strprintf("Prepend debug output with timestamp (default: %u)", DEFAULT_LOGTIMESTAMPS), ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-logthreadnames", strprintf("Prepend debug output with name of the originating thread (default: %u)", DEFAULT_LOGTHREADNAMES), ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-logsourcelocations", strprintf("Prepend debug output with name of the originating source location (source file, line number and function name) (default: %u)", DEFAULT_LOGSOURCELOCATIONS), ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-logtimemicros", strprintf("Add microsecond precision to debug timestamps (default: %u)", DEFAULT_LOGTIMEMICROS), ArgsManager::ALLOW_ANY | ArgsManager::DEBUG_ONLY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-loglevelalways", strprintf("Always prepend a category and level (default: %u)", DEFAULT_LOGLEVELALWAYS), ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-logratelimit", strprintf("Apply rate limiting to unconditional logging to mitigate disk-filling attacks (default: %u)", BCLog::DEFAULT_LOGRATELIMIT), ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-printtoconsole", "Send trace/debug info to console (default: 1 when no -daemon. To disable logging to file, set -nodebuglogfile)", ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

    argsman.AddArg("-shrinkdebugfile", "Shrink debug log file on client startup (default: 1 when no -debug)", ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);

}

void SetLoggingOptions(const ArgsManager& args)

{

    LogInstance().m_print_to_file = !args.IsArgNegated("-debuglogfile");

    LogInstance().m_file_path = AbsPathForConfigVal(args, args.GetPathArg("-debuglogfile", DEFAULT_DEBUGLOGFILE));

    LogInstance().m_print_to_console = args.GetBoolArg("-printtoconsole", !args.GetBoolArg("-daemon", false));

    LogInstance().m_log_timestamps = args.GetBoolArg("-logtimestamps", DEFAULT_LOGTIMESTAMPS);

    LogInstance().m_log_time_micros = args.GetBoolArg("-logtimemicros", DEFAULT_LOGTIMEMICROS);

    LogInstance().m_log_threadnames = args.GetBoolArg("-logthreadnames", DEFAULT_LOGTHREADNAMES);

    LogInstance().m_log_sourcelocations = args.GetBoolArg("-logsourcelocations", DEFAULT_LOGSOURCELOCATIONS);

    LogInstance().m_always_print_category_level = args.GetBoolArg("-loglevelalways", DEFAULT_LOGLEVELALWAYS);

    fLogIPs = args.GetBoolArg("-logips", DEFAULT_LOGIPS);

}

util::Result<void> SetLoggingLevel(const ArgsManager& args)

{

        for (const std::string& level_str : args.GetArgs("-loglevel")) {

            if (level_str.find_first_of(':', 3) == std::string::npos) {

                // user passed a global log level, i.e. -loglevel=<level>

                if (!LogInstance().SetLogLevel(level_str)) {

                    return util::Error{strprintf(_("Unsupported global logging level %s=%s. Valid values: %s."), "-loglevel", level_str, LogInstance().LogLevelsString())};

                }

            } else {

                // user passed a category-specific log level, i.e. -loglevel=<category>:<level>

                const auto& toks = SplitString(level_str, ':');

                if (!(toks.size() == 2 && LogInstance().SetCategoryLogLevel(toks[0], toks[1]))) {

                    return util::Error{strprintf(_("Unsupported category-specific logging level %1$s=%2$s. Expected %1$s=<category>:<loglevel>. Valid categories: %3$s. Valid loglevels: %4$s."), "-loglevel", level_str, LogInstance().LogCategoriesString(), LogInstance().LogLevelsString())};

                }

            }

        }

    return {};

}

util::Result<void> SetLoggingCategories(const ArgsManager& args)

{

    const std::vector<std::string> categories = args.GetArgs("-debug");

    // Special-case: Disregard any debugging categories appearing before -debug=0/none

    const auto last_negated = std::find_if(categories.rbegin(), categories.rend(),

                                           [](const std::string& cat) { return cat == "0" || cat == "none"; });

    const auto categories_to_process = (last_negated == categories.rend()) ? categories : std::ranges::subrange(last_negated.base(), categories.end());

    for (const auto& cat : categories_to_process) {

        if (!LogInstance().EnableCategory(cat)) {

            return util::Error{strprintf(_("Unsupported logging category %s=%s."), "-debug", cat)};

        }

    }

    // Now remove the logging categories which were explicitly excluded

    for (const std::string& cat : args.GetArgs("-debugexclude")) {

        if (!LogInstance().DisableCategory(cat)) {

            return util::Error{strprintf(_("Unsupported logging category %s=%s."), "-debugexclude", cat)};

        }

    }

    LogInfo("Log output may contain privacy-sensitive information. Be cautious when sharing logs.");

    return {};

}

bool StartLogging(const ArgsManager& args)

{

    if (LogInstance().m_print_to_file) {

        if (args.GetBoolArg("-shrinkdebugfile", LogInstance().DefaultShrinkDebugFile())) {

            // Do this first since it both loads a bunch of debug.log into memory,

            // and because this needs to happen before any other debug.log printing

            LogInstance().ShrinkDebugFile();

        }

    }

    if (!LogInstance().StartLogging()) {

            return InitError(Untranslated(strprintf("Could not open debug log file %s",

                fs::PathToString(LogInstance().m_file_path))));

    }

    if (!LogInstance().m_log_timestamps) {

        LogInfo("Startup time: %s", FormatISO8601DateTime(GetTime()));

    }

    LogInfo("Default data directory %s", fs::PathToString(GetDefaultDataDir()));

    LogInfo("Using data directory %s", fs::PathToString(gArgs.GetDataDirNet()));

    // Only log conf file usage message if conf file actually exists.

    fs::path config_file_path = args.GetConfigFilePath();

    if (args.IsArgNegated("-conf")) {

        LogInfo("Config file: <disabled>");

    } else if (fs::is_directory(config_file_path)) {

        LogWarning("Config file: %s (is directory, not file)", fs::PathToString(config_file_path));

    } else if (fs::exists(config_file_path)) {

        LogInfo("Config file: %s", fs::PathToString(config_file_path));

    } else if (args.IsArgSet("-conf")) {

        InitWarning(strprintf(_("The specified config file %s does not exist"), fs::PathToString(config_file_path)));

    } else {

        // Not categorizing as "Warning" because it's the default behavior

        LogInfo("Config file: %s (not found, skipping)", fs::PathToString(config_file_path));

    }

    // Log the config arguments to debug.log

    args.LogArgs();

    return true;

}

void LogPackageVersion()

{

    std::string version_string = FormatFullVersion();

#ifdef DEBUG

    version_string += " (debug build)";

#else

    version_string += " (release build)";

#endif

    LogInfo(CLIENT_NAME " version %s", version_string);

}

} // namespace init