#include <stdio.h>
#include <stdlib.h>

#ifdef _WIN32
# include <Windows.h>
#else
# include <unistd.h>
#endif

#include <libmicrovmi.h>

void pause_vm(void* driver, unsigned long sleep_duration_sec) {
    if (microvmi_pause(driver)) {
        printf("Paused.\n");
    } else {
        printf("Unable to pause VM.\n");
        return;
    }
    sleep(sleep_duration_sec);
    if (microvmi_resume(driver)) {
            printf("Resumed.\n");
    } else {
        printf("Unable to resume VM.\n");
    }
}

int main(int argc, char* argv[]) {
    if (argc < 3) {
        printf("Usage: pause <vm_name> <sleep_seconds>.\n");
        return 1;
    }
    unsigned long sleep_duration_sec = strtoul(argv[2], NULL, 0);
    if (sleep_duration_sec == 0) {
        printf("Unable to parse sleep duration or zero provided.\n");
        return 1;
    }
    microvmi_envlogger_init();
    const char* init_error = NULL;
        void* vm_name = argv[1];
        DriverInitParamsFFI init_params = {
            .common = {
                .vm_name = vm_name
            }
        };
    void* driver = microvmi_init(NULL, &init_params, &init_error);
    if (!driver) {
        fprintf(stderr, "%s\n", init_error);
        rs_cstring_free((char*)init_error);
        return 1;
    }
    pause_vm(driver, sleep_duration_sec);
    microvmi_destroy(driver);
    return 0;
}