#include <unistd.h>

#include <assert.h>

#include <stdio.h>

#include "main.h"

extern char *optarg;

extern int optind, opterr, optopt;

int kill_server = 0;

int need_server = 0;

int clear_finished = 0;

int list_jobs = 0;

int store_output = 1;

int server_socket;

/* Allocated in get_command() */

char *new_command;

void get_command(int index, int argc, char **argv)

{

    int size;

    int i;

    size = 0;

    /* Count bytes needed */

    for (i = index; i < argc; ++i)

    {

        /* The '1' is for spaces, and at the last i,

         * for the null character */

        size = size + strlen(argv[i]) + 1;

    }

    /* Alloc */

    new_command = malloc(size);

    assert(new_command != NULL);

    /* Build the command */

    strcpy(new_command, argv[index]);

    for (i = index+1; i < argc; ++i)

    {

        strcat(new_command, " ");

        strcat(new_command, argv[i]);

    }

}

void parse_opts(int argc, char **argv)

{

    int c;

    /* Parse options */

    while(1) {

        c = getopt(argc, argv, "klcn");

        if (c == -1)

            break;

        switch(c)

        {

            case 'k':

                kill_server = 1;

                break;

            case 'l':

                list_jobs = 1;

                break;

            case 'c':

                clear_finished = 1;

                break;

            case 'n':

                store_output = 0;

                break;

        }

    }

    new_command = 0;

    if (optind < argc)

        get_command(optind, argc, argv);

    if (list_jobs || kill_server || (new_command != 0)

            || clear_finished)

        need_server = 1;

}

static int go_background()

{

    int pid;

    pid = fork();

    switch(pid)

    {

        case -1:

            perror("fork failed");

            exit(-1);

            break;

        case 0:

            break;

        default:

            exit(0);

    }

}

int main(int argc, char **argv)

{

    parse_opts(argc, argv);

    if (need_server)

        ensure_server_up();

    if (new_command != 0)

    {

        go_background();

        assert(need_server);

        c_new_job(new_command);

        c_wait_server_commands(new_command, store_output);

        free(new_command);

    }

    if (list_jobs != 0)

    {

        assert(need_server);

        c_list_jobs();

        c_wait_server_lines();

    }

    if (kill_server)

    {

        assert(need_server);

        c_shutdown_server();

    }

    if (clear_finished)

    {

        assert(need_server);

        c_clear_finished();

    }

    if (need_server)

    {

        close(server_socket);

    }

    return 0;

}