main.cpp

#include "vfs/Volume.h"
#include "logic/PersistenceManager.h"
#include "logic/UserManager.h"
#include "logic/ProblemBank.h"
#include "logic/ContestManager.h"
#include "logic/EvaluationEngine.h"
#include "logic/EasyContest.h"
#include "logic/MediumContest.h"
#include "logic/HardContest.h"
#include "server/HttpServer.h"
#include "server/Router.h"
#include "server/ApiHandlers.h"
#include <cstdio>
#include <ctime>
#include <csignal>
#include <cstdlib>

#ifdef _WIN32
#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif
#include <windows.h>
#else
#include <sys/stat.h>
#include <sys/types.h>
#endif

static const char *VFS_PATH = "data/volume.vfs";
static const char *TEMP_DIR = "temp";
static const int PORT = 8080;

static HttpServer *g_server = 0;
static Volume *g_volume = 0; // global ptrs for signal handler

void signalHandler(int sig) {
    (void)sig;
    if (g_server) g_server->stop();
}

#ifdef _WIN32
BOOL WINAPI consoleHandler(DWORD signal) {
    if (signal == CTRL_C_EVENT || signal == CTRL_CLOSE_EVENT) {
        if (g_server) g_server->stop();
        return TRUE;
    }
    return FALSE;
}
#endif

void ensureDataDir() {
#ifdef _WIN32
    CreateDirectoryA("data", NULL);
    CreateDirectoryA("temp", NULL);
#else
    mkdir("data", 0777);
    mkdir("temp", 0777);
#endif
}

Volume *initVolume() {
    Volume *vol = new Volume("OnlineJudge");

    FILE *fp = fopen(VFS_PATH, "r");
    if (fp) {
        fclose(fp);
        if (vol->load(VFS_PATH)) {
            return vol;
        }
    }

    StorageMedium *disk = new StorageMedium("disk0", 1024 * 1024, SSD);
    vol->addMedia(disk);

    Permission rootPerm(true, true, true, true, false, true, true, false, true);
    Partition *part = new Partition("main", 1024 * 1024, NTFS, disk);
    disk->addPartition(part);
    vol->setPrimaryPartition(part);

    Directory *rootDir = part->getRootDirectory();

    User *su = new User("admin", 0, "/", SUPERUSER, 999);
    vol->registerUser(su);

    vol->createDirectory(rootDir, "users", su, rootPerm);
    vol->createDirectory(rootDir, "problems", su, rootPerm);
    vol->createDirectory(rootDir, "system", su, rootPerm);

    FileSystemEntity *sysEnt = vol->getEntity("/system");
    Directory *sysDir = (Directory *)sysEnt;
    vol->createDirectory(sysDir, "tokens", su, rootPerm);
    vol->createDirectory(sysDir, "sessions", su, rootPerm);

    FileSystemEntity *probEnt = vol->getEntity("/problems");
    Directory *probDir = (Directory *)probEnt;
    for (int d = 1; d <= 5; d++) {
        char name[4];
        name[0] = 'd';
        name[1] = '0' + d;
        name[2] = '\0';
        vol->createDirectory(probDir, name, su, rootPerm);
    }

    return vol;
}

int main() {
    ensureDataDir();

    Volume *volume = initVolume();
    g_volume = volume;
    User *su = volume->findUser("admin");

    PersistenceManager persistence(volume, su);
    EvaluationEngine evaluator(TEMP_DIR);
    ProblemBank bank(volume, &persistence);
    UserManager users(volume, &persistence);
    ContestManager contests(&bank, &evaluator, &persistence, volume);

    if (!evaluator.init()) {
        printf("Warning: compiler (g++) not found. Code evaluation disabled.\n");
    }

    // seed problems if first run
    if (bank.loadFromVfs() == 0) {
        printf("Seeding problem bank...\n");
        bank.seedToVfs();
        bank.loadFromVfs();
    }

    int loadedUsers = users.loadAllFromVfs();
    printf("Loaded %d users from VFS.\n", loadedUsers);
    printf("Loaded %d problems.\n", bank.getCount());

    Router router;
    ApiHandlers handlers(&users, &contests, &bank, &evaluator, &persistence, volume);
    handlers.registerAll(&router);
    router.addStaticDirectory("/", "frontend/");

    int port = 8080;
    const char *envPort = getenv("PORT");
    if (envPort != NULL) {
        port = atoi(envPort);
    }
    HttpServer server(port, &router, &handlers);
    g_server = &server;

    signal(SIGINT, signalHandler);
    signal(SIGTERM, signalHandler);
#ifdef _WIN32
    SetConsoleCtrlHandler(consoleHandler, TRUE);
#endif

    printf("Starting server...\n");
    int result = server.start();

    // IMPORTANT: save before exit otherwise data lost
    printf("Saving VFS...\n");
    volume->save(VFS_PATH);

    delete volume;
    printf("Shutdown complete.\n");
    return result;
}