FEX-Emu--FEX/ThunkLibs/Generator/diagnostics.h

#pragma once

#include <clang/AST/ASTContext.h>
#include <clang/Basic/Diagnostic.h>
#include <clang/Basic/SourceLocation.h>

#include <utility>
#include <vector>

struct ClangDiagnosticAsException {
    std::pair<clang::SourceLocation, unsigned> diagnostic;

    std::vector<ClangDiagnosticAsException> notes;

    // List of callbacks that add an argument to a clang::DiagnosticBuilder
    std::vector<std::function<void(clang::DiagnosticBuilder&)>> args;

    ClangDiagnosticAsException& AddString(std::string str) {
        args.push_back([arg=std::move(str)](clang::DiagnosticBuilder& db) {
            db.AddString(arg);
        });
        return *this;
    }

    ClangDiagnosticAsException& AddTaggedVal(clang::QualType type) {
        args.push_back([val=type](clang::DiagnosticBuilder& db) {
            db.AddTaggedVal(reinterpret_cast<uintptr_t>(val.getAsOpaquePtr()), clang::DiagnosticsEngine::ak_qualtype);
        });
        return *this;
    }

    ClangDiagnosticAsException& addNote(ClangDiagnosticAsException diagnostic) {
        notes.push_back(std::move(diagnostic));
        return *this;
    }

    void Report(clang::DiagnosticsEngine& diagnostics) const {
        {
            auto builder = diagnostics.Report(diagnostic.first, diagnostic.second);
            for (auto& arg_appender : args) {
                arg_appender(builder);
            }
        }
        for (auto& note : notes) {
            note.Report(diagnostics);
        }
    }
};

// Helper class to build a custom DiagID from the given message and store it in a throwable object
struct ErrorReporter {
    clang::ASTContext& context;

    template<std::size_t N>
    [[nodiscard]] ClangDiagnosticAsException operator()(clang::SourceLocation loc, const char (&message)[N],
                                                        clang::DiagnosticsEngine::Level level = clang::DiagnosticsEngine::Error) {
        auto id = context.getDiagnostics().getCustomDiagID(level, message);
        return { std::pair(loc, id) };
    }
};