#pragma once

#include <cstdint>
#include <set>

#include "cycles_t.h"
#include "cycle_t.h"
#include "test_t.h"

class checker_t {
public:
private:
    static std::set<uint8_t> m_undocumented_opcodes;
    static bool m_undocumented_opcodes_initialised;

    const test_t m_test;

    std::ostringstream m_os;
    std::vector<std::string> m_messages;

    typedef std::tuple<uint16_t, uint8_t, std::string> actual_cycle_t;
    typedef std::vector<actual_cycle_t> actual_cycles_t;
    actual_cycles_t m_actualCycles;
    bool m_cycle_count_mismatch = false;

    //int m_cycles = 0;
    //bool m_valid = true;
    //bool m_undocumented = false;

    [[nodiscard]] auto& os() { return m_os; }

    void seedUndocumentedOpcodes();
    [[nodiscard]] bool checkState(EightBit::MOS6502& cpu, EightBit::Ram& ram);

    void pushCurrentMessage();

    void raise(std::string what, uint16_t expected, uint16_t actual);
    void raise(std::string what, uint8_t expected, uint8_t actual);
    void raise(std::string what, std::string_view expected, std::string_view actual);

    template<class T>
    bool check(std::string what, T expected, T actual) {
        const auto success = actual == expected;
        if (!success)
            raise(what, expected, actual);
        return success;
    }

    bool check(std::string what, uint16_t address, uint8_t expected, uint8_t actual);

    //void addActualCycle(const actual_cycle_t& value);
    void addActualCycle(uint16_t address, uint8_t value, std::string action);
    void addActualCycle(EightBit::register16_t address, uint8_t value, std::string action);

    void addActualReadCycle(EightBit::register16_t address, uint8_t value);
    void addActualWriteCycle(EightBit::register16_t address, uint8_t value);

    //void disassemble(uint16_t address);

    template<class T>
    void dumpCycle(const uint16_t address, const uint8_t value, const T action) {
        os()
            << std::setfill('0') << std::hex
            << "Address: " << std::setw(4) << (int)address
            << ", value: " << std::setw(2) << (int)value
            << ", action: " << action;
        pushCurrentMessage();
    }

    void dumpCycles(std::string which, cycles_t cycles);
    void dumpCycles(cycles_t cycles);
    void dumpCycle(cycle_t cycle);

    void dumpCycles(std::string which, const actual_cycles_t& cycles);
    void dumpCycles(const actual_cycles_t& cycles);
    void dumpCycle(const actual_cycle_t& cycle);






    //[[nodiscard]] auto test() const noexcept { return m_test; }
    //[[nodiscard]] constexpr const auto& messages() const noexcept { return m_messages; }

    //[[nodiscard]] constexpr auto cycles() const noexcept { return m_cycles; }
    //[[nodiscard]] constexpr auto valid() const noexcept { return m_valid; }
    //[[nodiscard]] constexpr auto invalid() const noexcept { return !valid(); }
    //[[nodiscard]] constexpr auto unimplemented() const noexcept { return invalid() && m_cycle_count_mismatch && (cycles() == 1); }
    //[[nodiscard]] constexpr auto implemented() const noexcept { return !unimplemented(); }
    //[[nodiscard]] constexpr auto undocumented() const noexcept { return m_undocumented; }
    //[[nodiscard]] constexpr auto documented() const noexcept { return !undocumented(); }

    //void check();


public:
    checker_t(const test_t test);

    [[nodiscard]] auto test() const noexcept { return m_test; }


    void initialise(EightBit::Bus& bus);
    void initialiseState(EightBit::MOS6502& cpu, EightBit::Ram& ram);

};