firmware/components/rpc/wireless/mDNSDiscoveryService.cpp

//
// Created by Johnathon Slightham on 2025-05-25.
//


#include <string>
#include <format>
#include <sstream>

#include "mdns.h"
#include "ConfigManager.h"
#include "wireless/mDNSDiscoveryService.h"
#include "constants/tcp.h"

mDNSDiscoveryService::mDNSDiscoveryService() {
    mdns_init();

    const auto& config_manager = ConfigManager::get_instance();

    mdns_hostname_set(std::format("botchain-{}", config_manager.get_module_id()).c_str());
    mdns_instance_name_set(std::format("BotChain Robot Module {}", config_manager.get_module_id()).c_str());

    mdns_service_add(nullptr, "_robotcontrol", "_tcp", TCP_PORT, nullptr, 0);
    mdns_service_instance_name_set("_robotcontrol", "_tcp", "Robot Control TCP Server");

    m_module_id = std::to_string(config_manager.get_module_id());
    m_module_type = std::to_string(config_manager.get_module_type());

    mdns_txt_item_t service_txt_data[3] = {
        {"module_id", m_module_id.c_str()},
        {"module_type",m_module_type.c_str()},
           {"connected_modules",""},
    };

    mdns_service_txt_set("_robotcontrol", "_tcp", service_txt_data, 3);
}

mDNSDiscoveryService::~mDNSDiscoveryService() {
    mdns_free();
}

void mDNSDiscoveryService::set_connected_boards(const std::vector<int>& boards) {
    std::stringstream ss;

    const auto& config_manager = ConfigManager::get_instance();

    for (int i = 0; i < boards.size(); i++) {
        ss << std::to_string(boards[i]);
        if (i < boards.size() - 1) {
            ss << ',';
        }
    }

    m_module_id = std::to_string(config_manager.get_module_id());
    m_module_type = std::to_string(config_manager.get_module_type());
    m_connected_modules = ss.str();

    mdns_txt_item_t service_txt_data[3] = {
        {"module_id", m_module_id.c_str()},
        {"module_type", m_module_type.c_str()},
           {"connected_modules", m_connected_modules.c_str()},
    };
    mdns_service_txt_set("_robotcontrol", "_tcp", service_txt_data, 3);
}