// SPDX-FileCopyrightText: Copyright 2014 Citra Emulator Project
// SPDX-FileCopyrightText: Copyright 2024-2026 shadPS4 Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#include <algorithm>

#include "common/assert.h"
#include "common/logging/filter.h"

namespace Common::Log {
namespace {
template <typename It>
Level GetLevelByName(const It begin, const It end) {
    for (u8 i = 0; i < static_cast<u8>(Level::Count); ++i) {
        const char* level_name = GetLevelName(static_cast<Level>(i));
        if (std::string_view(begin, end).compare(level_name) == 0) {
            return static_cast<Level>(i);
        }
    }
    return Level::Count;
}

template <typename It>
Class GetClassByName(const It begin, const It end) {
    for (u8 i = 0; i < static_cast<u8>(Class::Count); ++i) {
        const char* level_name = GetLogClassName(static_cast<Class>(i));
        if (std::string_view(begin, end).compare(level_name) == 0) {
            return static_cast<Class>(i);
        }
    }
    return Class::Count;
}

template <typename Iterator>
bool ParseFilterRule(Filter& instance, Iterator begin, Iterator end) {
    auto level_separator = std::find(begin, end, ':');
    if (level_separator == end) {
        LOG_ERROR(Log, "Invalid log filter. Must specify a log level after `:`: {}",
                  std::string_view(begin, end));
        return false;
    }

    const Level level = GetLevelByName(level_separator + 1, end);
    if (level == Level::Count) {
        LOG_ERROR(Log, "Unknown log level in filter: {}", std::string_view(begin, end));
        return false;
    }

    if (std::string_view(begin, level_separator).compare("*") == 0) {
        instance.ResetAll(level);
        return true;
    }

    const Class log_class = GetClassByName(begin, level_separator);
    if (log_class == Class::Count) {
        LOG_ERROR(Log, "Unknown log class in filter: {}", std::string(begin, end));
        return false;
    }

    instance.SetClassLevel(log_class, level);
    return true;
}
} // Anonymous namespace

/// Macro listing all log classes. Code should define CLS and SUB as desired before invoking this.
#define ALL_LOG_CLASSES()                                                                          \
    CLS(Log)                                                                                       \
    CLS(Common)                                                                                    \
    SUB(Common, Filesystem)                                                                        \
    SUB(Common, Memory)                                                                            \
    CLS(Core)                                                                                      \
    SUB(Core, Linker)                                                                              \
    SUB(Core, Devices)                                                                             \
    CLS(Config)                                                                                    \
    CLS(Debug)                                                                                     \
    CLS(Kernel)                                                                                    \
    SUB(Kernel, Pthread)                                                                           \
    SUB(Kernel, Vmm)                                                                               \
    SUB(Kernel, Fs)                                                                                \
    SUB(Kernel, Event)                                                                             \
    SUB(Kernel, Sce)                                                                               \
    CLS(Lib)                                                                                       \
    SUB(Lib, LibC)                                                                                 \
    SUB(Lib, LibcInternal)                                                                         \
    SUB(Lib, Kernel)                                                                               \
    SUB(Lib, Pad)                                                                                  \
    SUB(Lib, SystemGesture)                                                                        \
    SUB(Lib, GnmDriver)                                                                            \
    SUB(Lib, SystemService)                                                                        \
    SUB(Lib, UserService)                                                                          \
    SUB(Lib, VideoOut)                                                                             \
    SUB(Lib, CommonDlg)                                                                            \
    SUB(Lib, MsgDlg)                                                                               \
    SUB(Lib, AudioOut)                                                                             \
    SUB(Lib, AudioIn)                                                                              \
    SUB(Lib, Net)                                                                                  \
    SUB(Lib, NetCtl)                                                                               \
    SUB(Lib, SaveData)                                                                             \
    SUB(Lib, SaveDataDialog)                                                                       \
    SUB(Lib, Http)                                                                                 \
    SUB(Lib, Http2)                                                                                \
    SUB(Lib, Ssl)                                                                                  \
    SUB(Lib, Ssl2)                                                                                 \
    SUB(Lib, SysModule)                                                                            \
    SUB(Lib, Move)                                                                                 \
    SUB(Lib, NpAuth)                                                                               \
    SUB(Lib, NpCommon)                                                                             \
    SUB(Lib, NpCommerce)                                                                           \
    SUB(Lib, NpManager)                                                                            \
    SUB(Lib, NpScore)                                                                              \
    SUB(Lib, NpTrophy)                                                                             \
    SUB(Lib, NpWebApi)                                                                             \
    SUB(Lib, NpWebApi2)                                                                            \
    SUB(Lib, NpProfileDialog)                                                                      \
    SUB(Lib, NpSnsFacebookDialog)                                                                  \
    SUB(Lib, Screenshot)                                                                           \
    SUB(Lib, LibCInternal)                                                                         \
    SUB(Lib, AppContent)                                                                           \
    SUB(Lib, Rtc)                                                                                  \
    SUB(Lib, Rudp)                                                                                 \
    SUB(Lib, DiscMap)                                                                              \
    SUB(Lib, Png)                                                                                  \
    SUB(Lib, Jpeg)                                                                                 \
    SUB(Lib, PlayGo)                                                                               \
    SUB(Lib, PlayGoDialog)                                                                         \
    SUB(Lib, Random)                                                                               \
    SUB(Lib, Usbd)                                                                                 \
    SUB(Lib, Ajm)                                                                                  \
    SUB(Lib, ErrorDialog)                                                                          \
    SUB(Lib, ImeDialog)                                                                            \
    SUB(Lib, AvPlayer)                                                                             \
    SUB(Lib, Ngs2)                                                                                 \
    SUB(Lib, Audio3d)                                                                              \
    SUB(Lib, Ime)                                                                                  \
    SUB(Lib, GameLiveStreaming)                                                                    \
    SUB(Lib, Remoteplay)                                                                           \
    SUB(Lib, SharePlay)                                                                            \
    SUB(Lib, Fiber)                                                                                \
    SUB(Lib, Vdec2)                                                                                \
    SUB(Lib, Videodec)                                                                             \
    SUB(Lib, RazorCpu)                                                                             \
    SUB(Lib, Mouse)                                                                                \
    SUB(Lib, WebBrowserDialog)                                                                     \
    SUB(Lib, NpParty)                                                                              \
    SUB(Lib, Zlib)                                                                                 \
    SUB(Lib, Hmd)                                                                                  \
    SUB(Lib, Font)                                                                                 \
    SUB(Lib, FontFt)                                                                               \
    SUB(Lib, HmdSetupDialog)                                                                       \
    SUB(Lib, SigninDialog)                                                                         \
    SUB(Lib, Camera)                                                                               \
    SUB(Lib, CompanionHttpd)                                                                       \
    SUB(Lib, CompanionUtil)                                                                        \
    SUB(Lib, Voice)                                                                                \
    SUB(Lib, VrTracker)                                                                            \
    CLS(Frontend)                                                                                  \
    CLS(Render)                                                                                    \
    SUB(Render, Vulkan)                                                                            \
    SUB(Render, Recompiler)                                                                        \
    CLS(ImGui)                                                                                     \
    CLS(Input)                                                                                     \
    CLS(Tty)                                                                                       \
    CLS(KeyManager)                                                                                \
    CLS(Loader)

// GetClassName is a macro defined by Windows.h, grrr...
const char* GetLogClassName(Class log_class) {
    switch (log_class) {
#define CLS(x)                                                                                     \
    case Class::x:                                                                                 \
        return #x;
#define SUB(x, y)                                                                                  \
    case Class::x##_##y:                                                                           \
        return #x "." #y;
        ALL_LOG_CLASSES()
#undef CLS
#undef SUB
    case Class::Count:
    default:
        break;
    }
    UNREACHABLE();
}

const char* GetLevelName(Level log_level) {
#define LVL(x)                                                                                     \
    case Level::x:                                                                                 \
        return #x
    switch (log_level) {
        LVL(Trace);
        LVL(Debug);
        LVL(Info);
        LVL(Warning);
        LVL(Error);
        LVL(Critical);
    case Level::Count:
    default:
        break;
    }
#undef LVL
    UNREACHABLE();
}

Filter::Filter(Level default_level) {
    ResetAll(default_level);
}

void Filter::ResetAll(Level level) {
    class_levels.fill(level);
}

void Filter::SetClassLevel(Class log_class, Level level) {
    class_levels[static_cast<std::size_t>(log_class)] = level;
}

void Filter::ParseFilterString(std::string_view filter_view) {
    auto clause_begin = filter_view.cbegin();
    while (clause_begin != filter_view.cend()) {
        auto clause_end = std::find(clause_begin, filter_view.cend(), ' ');

        // If clause isn't empty
        if (clause_end != clause_begin) {
            ParseFilterRule(*this, clause_begin, clause_end);
        }

        if (clause_end != filter_view.cend()) {
            // Skip over the whitespace
            ++clause_end;
        }
        clause_begin = clause_end;
    }
}

bool Filter::CheckMessage(Class log_class, Level level) const {
    return static_cast<u8>(level) >=
           static_cast<u8>(class_levels[static_cast<std::size_t>(log_class)]);
}

bool Filter::IsDebug() const {
    return std::any_of(class_levels.begin(), class_levels.end(), [](const Level& l) {
        return static_cast<u8>(l) <= static_cast<u8>(Level::Debug);
    });
}

} // namespace Common::Log
