/*
    Task Spooler - a task queue system for the unix user
    Copyright (C) 2007-2009  Lluís Batlle i Rossell

    Please find the license in the provided COPYING file.
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/time.h>
#include "main.h"

/* From jobs.c */
extern int busy_slots;
extern int max_slots;

char * joblistdump_headers()
{
    char * line;

    line = malloc(600);
    snprintf(line, 600, "#!/bin/sh\n# - task spooler (ts) job dump\n"
            "# This file has been created because a SIGTERM killed\n"
            "# your queue server.\n"
            "# The finished commands are listed first.\n"
            "# The commands running or to be run are stored as you would\n"
            "# probably run them. Take care - some quotes may have got"
            " broken\n\n");

    return line;
}

char * joblist_headers()
{
    char * line;

    line = malloc(100);
    snprintf(line, 100, "%-4s %-10s %-20s %-8s %-14s %s [run=%i/%i]\n",
            "ID",
            "State",
            "Output",
            "E-Level",
            "Times(r/u/s)",
            "Command",
            busy_slots,
            max_slots);

    return line;
}

static int max(int a, int b)
{
    if (a > b)
        return a;
    return b;
}

static const char * ofilename_shown(const struct Job *p)
{
    const char * output_filename;

    if (p->state == SKIPPED)
    {
        output_filename = "(no output)";
    } else if (p->store_output)
    {
        if (p->state == QUEUED)
        {
            output_filename = "(file)";
        } else
        {
            if (p->output_filename == 0)
                /* This may happen due to concurrency
                 * problems */
                output_filename = "(...)";
            else
                output_filename = p->output_filename;
        }
    } else
        output_filename = "stdout";


    return output_filename;
}

static char * print_noresult(const struct Job *p)
{
    const char * jobstate;
    const char * output_filename;
    int maxlen;
    char * line;
    /* 18 chars should suffice for a string like "[int]&& " */
    char dependstr[18] = "";

    jobstate = jstate2string(p->state);
    output_filename = ofilename_shown(p);

    maxlen = 4 + 1 + 10 + 1 + max(20, strlen(output_filename)) + 1 + 8 + 1
        + 14 + 1 + strlen(p->command) + 20; /* 20 is the margin for errors */

    if (p->label)
        maxlen += 3 + strlen(p->label);
    if (p->do_depend)
    {
        maxlen += sizeof(dependstr);
        if (p->depend_on == -1)
            snprintf(dependstr, sizeof(dependstr), "&& ");
        else
            snprintf(dependstr, sizeof(dependstr), "[%i]&& ", p->depend_on);
    }

    line = (char *) malloc(maxlen);
    if (line == NULL)
        error("Malloc for %i failed.\n", maxlen);

    if (p->label)
        snprintf(line, maxlen, "%-4i %-10s %-20s %-8s %14s %s[%s]%s\n",
                p->jobid,
                jobstate,
                output_filename,
                "",
                "",
		        dependstr,
                p->label,
                p->command);
    else
        snprintf(line, maxlen, "%-4i %-10s %-20s %-8s %14s %s%s\n",
                p->jobid,
                jobstate,
                output_filename,
                "",
                "",
		        dependstr,
                p->command);

    return line;
}

static char * print_result(const struct Job *p)
{
    const char * jobstate;
    int maxlen;
    char * line;
    const char * output_filename;
    /* 18 chars should suffice for a string like "[int]&& " */
    char dependstr[18] = "";

    jobstate = jstate2string(p->state);
    output_filename = ofilename_shown(p);

    maxlen = 4 + 1 + 10 + 1 + max(20, strlen(output_filename)) + 1 + 8 + 1
        + 14 + 1 + strlen(p->command) + 20; /* 20 is the margin for errors */

    if (p->label)
        maxlen += 3 + strlen(p->label);
    if (p->do_depend)
    {
        maxlen += sizeof(dependstr);
        if (p->depend_on == -1)
            snprintf(dependstr, sizeof(dependstr), "&& ");
        else
            snprintf(dependstr, sizeof(dependstr), "[%i]&& ", p->depend_on);
    }

    line = (char *) malloc(maxlen);
    if (line == NULL)
        error("Malloc for %i failed.\n", maxlen);

    if (p->label)
        snprintf(line, maxlen, "%-4i %-10s %-20s %-8i %0.2f/%0.2f/%0.2f %s[%s]"
                "%s\n",
                p->jobid,
                jobstate,
                output_filename,
                p->result.errorlevel,
                p->result.real_ms,
                p->result.user_ms,
                p->result.system_ms,
                dependstr,
                p->label,
                p->command);
    else
        snprintf(line, maxlen, "%-4i %-10s %-20s %-8i %0.2f/%0.2f/%0.2f %s%s\n",
                p->jobid,
                jobstate,
                output_filename,
                p->result.errorlevel,
                p->result.real_ms,
                p->result.user_ms,
                p->result.system_ms,
                dependstr,
                p->command);

    return line;
}

char * joblist_line(const struct Job *p)
{
    char * line;

    if (p->state == FINISHED)
        line = print_result(p);
    else
        line = print_noresult(p);

    return line;
}

char * joblistdump_torun(const struct Job *p)
{
    int maxlen;
    char * line;

    maxlen = 10 + strlen(p->command) + 20; /* 20 is the margin for errors */

    line = (char *) malloc(maxlen);
    if (line == NULL)
        error("Malloc for %i failed.\n", maxlen);

    snprintf(line, maxlen, "ts %s\n", p->command);

    return line;
}