#include <drogon/drogon.h>
#include <drogon/HttpAppFramework.h>
#include "controllers/StreamController.h"
#include "controllers/UserController.h"
#include "controllers/AdminController.h"
#include "controllers/RealmController.h"
#include "services/DatabaseService.h"
#include "services/StatsService.h"
#include "services/AuthService.h"
#include <exception>
#include <csignal>
#include <sys/stat.h>

using namespace drogon;

int main() {
    // Simplified signal handlers
    signal(SIGSEGV, [](int s){ LOG_ERROR << "Signal " << s; exit(s); });
    signal(SIGABRT, [](int s){ LOG_ERROR << "Signal " << s; exit(s); });
    
    try {
        LOG_INFO << "Starting streaming backend server...";
        
        // Create upload directories
        mkdir("./uploads", 0755);
        mkdir("./uploads/avatars", 0755);
        
        // Initialize DatabaseService
        LOG_INFO << "Initializing DatabaseService...";
        DatabaseService::getInstance().initialize();
        
        // Load config
        LOG_INFO << "Loading configuration...";
        app().loadConfigFile("config.json");
        
        // Register a pre-routing advice to handle CORS
        app().registerPreRoutingAdvice([](const HttpRequestPtr &req,
                                         AdviceCallback &&acb,
                                         AdviceChainCallback &&accb) {
            // Handle CORS preflight requests
            if (req->getMethod() == Options) {
                auto resp = HttpResponse::newHttpResponse();
                resp->setStatusCode(k204NoContent);
                
                // Get origin from request
                std::string origin = req->getHeader("Origin");
                if (origin.empty()) {
                    origin = "*";
                }
                
                resp->addHeader("Access-Control-Allow-Origin", origin);
                resp->addHeader("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS");
                resp->addHeader("Access-Control-Allow-Headers", "Content-Type, Authorization");
                resp->addHeader("Access-Control-Allow-Credentials", "true");
                resp->addHeader("Access-Control-Max-Age", "86400");
                acb(resp);
                return;
            }
            accb();
        });
        
        // Register post-handling advice to add CORS headers to all responses
        app().registerPostHandlingAdvice([](const HttpRequestPtr &req,
                                           const HttpResponsePtr &resp) {
            // Get origin from request
            std::string origin = req->getHeader("Origin");
            if (origin.empty()) {
                origin = "*";
            }
            
            resp->addHeader("Access-Control-Allow-Origin", origin);
            resp->addHeader("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS");
            resp->addHeader("Access-Control-Allow-Headers", "Content-Type, Authorization");
            resp->addHeader("Access-Control-Allow-Credentials", "true");
        });
        
        // Register beginning advice to initialize StatsService after app starts
        app().registerBeginningAdvice([]() {
            LOG_INFO << "Application started successfully";
            
            // Initialize StatsService after app is running
            LOG_INFO << "Initializing StatsService...";
            StatsService::getInstance().initialize();
        });
        
        app().setTermSignalHandler([]() {
            LOG_INFO << "Received termination signal, shutting down...";
            StatsService::getInstance().shutdown();
            app().quit();
        });
        
        // Start the application
        LOG_INFO << "Starting Drogon framework...";
        app().run();
        
    } catch (const std::exception& e) {
        LOG_ERROR << "Exception caught in main: " << e.what();
        return 1;
    } catch (...) {
        LOG_ERROR << "Unknown exception caught in main";
        return 1;
    }
    
    return 0;
}