#include <AK/Assertions.h>
#include <errno.h>
#include <sched.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <unistd.h>
#include <LibCore/CFile.h>

void start_process(const char* prog, int prio)
{
    pid_t pid = 0;

    while (true) {
        dbgprintf("Forking for %s...\n", prog);
        int pid = fork();
        if (pid < 0) {
            dbgprintf("Fork %s failed! %s\n", prog, strerror(errno));
            continue;
        } else if (pid > 0) {
            // parent...
            dbgprintf("Process %s hopefully started with priority %d...\n", prog, prio);
            return;
        }
        break;
    }

    while (true) {
        if (pid == 0) {
            dbgprintf("Executing for %s... at prio %d\n", prog, prio);
            struct sched_param p;
            p.sched_priority = prio;
            int ret = sched_setparam(pid, &p);
            ASSERT(ret == 0);

            char* progv[256];
            progv[0] = (char*)prog;
            progv[1] = nullptr;
            ret = execv(prog, progv);
            if (ret < 0) {
                dbgprintf("Exec %s failed! %s", prog, strerror(errno));
                continue;
            }
            break;
        } else {
            break;
        }
    }
}

static void check_for_test_mode()
{
    CFile f("/proc/cmdline");
    if (!f.open(CIODevice::ReadOnly)) {
        dbgprintf("Failed to read command line: %s\n", f.error_string());
        ASSERT(false);
    }
    const String cmd = String::copy(f.read_all());
    dbgprintf("Read command line: %s\n", cmd.characters());
    if (cmd.matches("*testmode=1*")) {
        // Eventually, we should run a test binary and wait for it to finish
        // before shutting down. But this is good enough for now.
        dbgprintf("Waiting for testmode shutdown...\n");
        sleep(5);
        dbgprintf("Shutting down due to testmode...\n");
        if (fork() == 0) {
            execl("/bin/shutdown", "/bin/shutdown", "-n", nullptr);
            ASSERT_NOT_REACHED();
        }
    } else {
        dbgprintf("Continuing normally\n");
    }
}

int main(int, char**)
{
    int lowest_prio = sched_get_priority_min(SCHED_OTHER);
    int highest_prio = sched_get_priority_max(SCHED_OTHER);
    start_process("/bin/LookupServer", lowest_prio);
    start_process("/bin/WindowServer", highest_prio);
    start_process("/bin/Taskbar", highest_prio);
    start_process("/bin/Terminal", highest_prio - 1);
    start_process("/bin/Launcher", highest_prio);

    // This won't return if we're in test mode.
    check_for_test_mode();

    while (1) {
        sleep(1);
    }
}