#include "agent/humanus.h"
#include "logger.h"
#include "prompt.h"
#include "flow/flow_factory.h"

#if defined (__unix__) || (defined (__APPLE__) && defined (__MACH__))
#include <signal.h>
#include <unistd.h>
#elif defined (_WIN32)
#define WIN32_LEAN_AND_MEAN
#ifndef NOMINMAX
#define NOMINMAX
#endif
#include <windows.h>
#include <signal.h>
#endif

using namespace humanus;

#if defined (__unix__) || (defined (__APPLE__) && defined (__MACH__)) || defined (_WIN32)
static void sigint_handler(int signo) {
    if (signo == SIGINT) {
        logger->info("Interrupted by user\n");
        exit(0);
    }
}
#endif

int main() {

    // ctrl+C handling
    {
#if defined (__unix__) || (defined (__APPLE__) && defined (__MACH__))
        struct sigaction sigint_action;
        sigint_action.sa_handler = sigint_handler;
        sigemptyset (&sigint_action.sa_mask);
        sigint_action.sa_flags = 0;
        sigaction(SIGINT, &sigint_action, NULL);
#elif defined (_WIN32)
        auto console_ctrl_handler = +[](DWORD ctrl_type) -> BOOL {
            return (ctrl_type == CTRL_C_EVENT) ? (sigint_handler(SIGINT), true) : false;
        };
        SetConsoleCtrlHandler(reinterpret_cast<PHANDLER_ROUTINE>(console_ctrl_handler), true);
#endif
    }
    
    Humanus agent = Humanus();
    while (true) {
        if (agent.current_step == agent.max_steps) {
            std::cout << "Automatically paused after " << agent.max_steps << " steps." << std::endl;
            std::cout << "Enter your prompt (enter an empty line to resume or 'exit' to quit): ";
            agent.reset(false);
        } else {
            std::cout << "Enter your prompt (or 'exit' to quit): ";
        }
        std::string prompt;
        std::getline(std::cin, prompt);
        if (prompt == "exit") {
            logger->info("Goodbye!");
            break;
        }
        logger->info("Processing your request...");
        agent.run(prompt);
    }

    // std::shared_ptr<BaseAgent> agent_ptr = std::make_shared<Humanus>();
    // std::map<std::string, std::shared_ptr<BaseAgent>> agents;
    // agents["default"] = agent_ptr;

    // auto flow = FlowFactory::create_flow(
    //     FlowType::PLANNING,
    //     nullptr,  // llm
    //     nullptr,  // planning_tool
    //     std::vector<std::string>{},  // executor_keys
    //     "",  // active_plan_id
    //     agents,  // agents
    //     std::vector<std::shared_ptr<BaseTool>>{},  // tools
    //     "default"  // primary_agent_key
    // );

    // while (true) {
    //     if (agent_ptr->current_step == agent_ptr->max_steps) {
    //         std::cout << "Automatically paused after " << agent_ptr->current_step << " steps." << std::endl;
    //         std::cout << "Enter your prompt (enter an empty line to resume or 'exit' to quit): ";
    //         agent_ptr->reset(false);
    //     } else {
    //         std::cout << "Enter your prompt (or 'exit' to quit): ";
    //     }

    //     if (agent_ptr->state != AgentState::IDLE) {
    //         break;
    //     }
        
    //     std::string prompt;
    //     std::getline(std::cin, prompt);
    //     if (prompt == "exit") {
    //         logger->info("Goodbye!");
    //         break;
    //     }

    //     std::cout << "Processing your request..." << std::endl;
    //     auto result = flow->execute(prompt);
    //     std::cout << result << std::endl;
    // }
}