// main.cpp
#include "StardustBridge.hpp"
#include "OverteClient.hpp"
#include "SceneSync.Hpp"
#include "InputHandler.hpp"
#include "DomainDiscovery.hpp"

#include <iostream>
#include <thread>
#include <chrono>

int main(int argc, char** argv) {
    // Simple CLI: --socket=/path/to.sock or --abstract=name
    std::string socketOverride;
    for (int i = 1; i < argc; ++i) {
        std::string arg = argv[i];
        const std::string so = "--socket=";
        const std::string ab = "--abstract=";
        if (arg.rfind(so, 0) == 0) socketOverride = arg.substr(so.size());
        else if (arg.rfind(ab, 0) == 0) socketOverride = '@' + arg.substr(ab.size());
    }
    StardustBridge stardust;
    if (!stardust.connect(socketOverride)) {
        std::cerr << "Failed to connect to StardustXR compositor.\n";
        return 1;
    }

    // Overte localhost default assumption (can override via OVERTE_URL env or --overte=ws://host:port)
    std::string overteUrl = "ws://127.0.0.1:40102";
    bool useDiscovery = false;
    for (int i = 1; i < argc; ++i) {
        std::string arg = argv[i];
        const std::string ov = "--overte=";
        const std::string disc = "--discover";
        if (arg.rfind(ov, 0) == 0) overteUrl = arg.substr(ov.size());
        else if (arg == disc) useDiscovery = true;
    }
    if (const char* envOv = std::getenv("OVERTE_URL")) {
        overteUrl = envOv;
    }
    if (const char* envDisc = std::getenv("OVERTE_DISCOVER")) {
        if (std::string(envDisc) == "1" || std::string(envDisc) == "true") useDiscovery = true;
    }

    if (useDiscovery) {
        std::cout << "[Discovery] Querying metaverse directories for public domains..." << std::endl;
        auto domains = discoverDomains(25);
        if (domains.empty()) {
            std::cout << "[Discovery] No domains found via directory; using provided URL: " << overteUrl << std::endl;
        } else {
            std::cout << "[Discovery] Found " << domains.size() << " candidate domain(s):" << std::endl;
            int idx = 0;
            for (const auto& d : domains) {
                std::cout << "  [" << idx++ << "] "
                          << (d.name.empty() ? d.networkHost : d.name)
                          << " -> ws://" << d.networkHost << ":" << d.httpPort
                          << " (udp:" << d.udpPort << ")" << std::endl;
            }
            
            // Probe for reachability unless disabled
            bool probeEnabled = true;
            if (const char* envProbe = std::getenv("OVERTE_DISCOVER_PROBE")) {
                if (std::string(envProbe) == "0" || std::string(envProbe) == "false") probeEnabled = false;
            }
            
            int choice = -1;
            if (probeEnabled) {
                std::cout << "[Discovery] Probing domains for reachability..." << std::endl;
                for (size_t i = 0; i < domains.size(); ++i) {
                    std::cout << "[Discovery] Probing [" << i << "] " << domains[i].networkHost << ":" << domains[i].httpPort << "... " << std::flush;
                    if (probeDomain(domains[i])) {
                        std::cout << "REACHABLE" << std::endl;
                        choice = static_cast<int>(i);
                        break;
                    } else {
                        std::cout << "unreachable" << std::endl;
                    }
                }
                if (choice < 0) {
                    std::cout << "[Discovery] No reachable domains found; using first candidate anyway." << std::endl;
                    choice = 0;
                }
            } else {
                std::cout << "[Discovery] Probing disabled; selecting first candidate." << std::endl;
                choice = 0;
            }
            
            // Allow override index via env (supersedes probe)
            if (const char* envIdx = std::getenv("OVERTE_DISCOVER_INDEX")) {
                try { 
                    int manualChoice = std::stoi(envIdx);
                    if (manualChoice >= 0 && manualChoice < (int)domains.size()) {
                        choice = manualChoice;
                        std::cout << "[Discovery] Manual override: selecting index " << choice << std::endl;
                    }
                } catch (...) {}
            }
            
            if (choice < 0 || choice >= (int)domains.size()) choice = 0;
            
            const auto& pick = domains[choice];
            overteUrl = std::string("ws://") + pick.networkHost + ":" + std::to_string(pick.httpPort);
            // Pass UDP override via env for this process lifetime
            setenv("OVERTE_UDP_PORT", std::to_string(pick.udpPort).c_str(), 1);
            std::cout << "[Discovery] Selected: " << overteUrl << std::endl;
        }
    }
    OverteClient overte(overteUrl);
    // Overte is optional; warn if unreachable but continue in offline mode.
    if (!overte.connect()) {
        std::cerr << "[Overte] Domain unreachable; running in offline mode.\n";
    }

    InputHandler input(stardust, overte);

    // Main loop
    while (stardust.running()) {
    overte.poll();
        stardust.poll();

        // Sync avatars/entities
        SceneSync::update(stardust, overte);

        // Simple input mapping
        input.update(1.0f / 90.0f);

        // Small sleep to avoid busy-spin in the stub
        std::this_thread::sleep_for(std::chrono::milliseconds(11));
    }

    return 0;
}