#include "AvrGdbRsp.hpp"

#include "src/Exceptions/Exception.hpp"

using namespace Bloom::DebugServers::Gdb;
using namespace Bloom::Exceptions;

using Bloom::Targets::TargetRegisterDescriptor;
using Bloom::Targets::TargetRegisterType;

void AvrGdbRsp::loadRegisterNumberToDescriptorMapping() {
    auto& registerDescriptorsByType = this->targetDescriptor.registerDescriptorsByType;
    if (!registerDescriptorsByType.contains(TargetRegisterType::STATUS_REGISTER)) {
        throw Exception("Missing status register descriptor");
    }

    if (!registerDescriptorsByType.contains(TargetRegisterType::STACK_POINTER)) {
        throw Exception("Missing stack pointer register descriptor");
    }

    if (!registerDescriptorsByType.contains(TargetRegisterType::PROGRAM_COUNTER)) {
        throw Exception("Missing program counter register descriptor");
    }

    if (!registerDescriptorsByType.contains(TargetRegisterType::GENERAL_PURPOSE_REGISTER)
        || registerDescriptorsByType.at(TargetRegisterType::GENERAL_PURPOSE_REGISTER).size() != 32) {
        throw Exception("Unexpected general purpose register count");
    }

    auto& gpRegisterDescriptors = registerDescriptorsByType.at(TargetRegisterType::GENERAL_PURPOSE_REGISTER);

    std::size_t descriptorIndex = 0;
    for (auto& descriptor : gpRegisterDescriptors) {
        this->registerNumberToDescriptorMapping.insert(std::pair(
            static_cast<GdbRegisterNumber>(descriptorIndex),
            descriptor
        ));

        descriptorIndex++;
    }

    this->registerNumberToDescriptorMapping.insert(std::pair(
        static_cast<GdbRegisterNumber>(32),
        *(registerDescriptorsByType.at(TargetRegisterType::STATUS_REGISTER).begin())
    ));

    this->registerNumberToDescriptorMapping.insert(std::pair(
        static_cast<GdbRegisterNumber>(33),
        *(registerDescriptorsByType.at(TargetRegisterType::STACK_POINTER).begin())
    ));

    this->registerNumberToDescriptorMapping.insert(std::pair(
        static_cast<GdbRegisterNumber>(34),
        *(registerDescriptorsByType.at(TargetRegisterType::PROGRAM_COUNTER).begin())
    ));
}

void AvrGdbRsp::init() {
    this->loadRegisterNumberToDescriptorMapping();

    GdbRspDebugServer::init();
}