/*
 * Copyright (c) 2011 Alex Hornung <alex@alexhornung.com>.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE
 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/resource.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/wait.h>

#include <errno.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <unistd.h>

#include <err.h>

#include <libprop/proplib.h>

#include "parser.h"
#include "testcase.h"
#include "runlist.h"
#include <dfregress.h>

static int count_by_result[10];

static void
print_row(char c)
{
        int i;

        for (i = 0; i < 80; i++)
                printf("%c", c);
        printf("\n");
}

static void
print_header(void)
{
        int i, j;

        i = printf("Test case");
        for (j = 60 - i; j > 0; j--)
                printf(" ");
        printf("Result\n");

        print_row('-');
}

static char *get_arg_string(prop_dictionary_t testcase)
{
        static char buf[2048];
        const char **argv;
        int i;

        buf[0] = '\0';

        argv = testcase_get_args(testcase);

        /* Skip the first argument, since it's just the test's name */
        for (i = 1; argv[i] != NULL; i++) {
                strcat(buf, argv[i]);
                if (argv[i+1] != NULL)
                        strcat(buf, " ");
        }

        return (i > 1) ? buf : NULL;
}

static void
print_summary(prop_array_t runlist)
{
        float total_run;
        float total_tests;

        total_tests = 0.0 + prop_array_count(runlist);
        total_run = 0.0 + total_tests - count_by_result[RESULT_BUILDFAIL] -
            count_by_result[RESULT_PREFAIL] - count_by_result[RESULT_NOTRUN] -
            count_by_result[RESULT_UNKNOWN];

        printf("\n\n");
        print_row('=');
        printf("Summary:\n\n");


        printf("Tests not built:\t%d\n", count_by_result[RESULT_BUILDFAIL]);

        printf("Tests not run:\t\t%.0f\n", total_tests - total_run);

        printf("Tests pre-failed:\t%d\n", count_by_result[RESULT_PREFAIL]);

        printf("Tests post-failed:\t%d\n", count_by_result[RESULT_POSTFAIL]);

        printf("Tests passed:\t\t%d\n", count_by_result[RESULT_PASS]);

        printf("Tests failed:\t\t%d\n", count_by_result[RESULT_FAIL]);

        printf("Tests crashed:\t\t%d\n", count_by_result[RESULT_SIGNALLED]);

        printf("Tests timed out:\t%d\n", count_by_result[RESULT_TIMEOUT]);


        printf("------\n");

        printf("Run rate:\t\t%.2f\n", total_run/total_tests);
        printf("Pass rate:\t\t%.2f\n", count_by_result[RESULT_PASS]/total_run);
}

static int
runlist_parse_summary(void *arg __unused, prop_dictionary_t testcase)
{
        char *args;
        int i, j;

        ++count_by_result[testcase_get_result(testcase)];
        args = get_arg_string(testcase);

        i = printf("%s", testcase_get_name(testcase));
        if (args != NULL)
                i+= printf(" (%s)", args);

        for (j = 60 - i; j > 0; j--)
                printf(" ");

        printf("%s\n", testcase_get_result_desc(testcase));

        return 0;
}

static int
runlist_parse_detail(void *arg __unused, prop_dictionary_t testcase)
{
        char *args;

        args = get_arg_string(testcase);

        printf("\n");
        print_row('=');

        printf("Test: %s\n", testcase_get_name(testcase));
        if (args != NULL)
                printf("Command line arguments: %s\n", args);

        printf("Type: %s\n", testcase_get_type_desc(testcase));
        printf("Result: %s\n", testcase_get_result_desc(testcase));

        switch (testcase_get_result(testcase)) {
        case RESULT_FAIL:
                printf("Exit code: %d\n", testcase_get_exit_value(testcase));
                break;

        case RESULT_SIGNALLED:
                printf("Signal: %d\n", testcase_get_signal(testcase));
                break;
        }

        print_row('-');
        printf("driver sysbuf:\n%s\n", testcase_get_sys_buf(testcase));
        print_row('-');
        printf("build log:\n%s\n", testcase_get_build_buf(testcase));
        print_row('-');
        printf("'pre' log:\n%s\n", testcase_get_precmd_buf(testcase));
        print_row('-');
        printf("testcase stdout:\n%s\n", testcase_get_stdout_buf(testcase));
        print_row('-');
        printf("testcase stderr:\n%s\n", testcase_get_stderr_buf(testcase));
        print_row('-');
        printf("'post' log:\n%s\n", testcase_get_postcmd_buf(testcase));
        print_row('-');
        printf("cleanup log:\n%s\n", testcase_get_cleanup_buf(testcase));

        return 0;
}

int
main(int argc, char *argv[])
{
        if (argc != 2) {
                fprintf(stderr, "Usage: dfr2text <foo.plist>\n");
                exit(1);
        }

        prop_array_t runlist = runlist_load(argv[1]);

        memset(count_by_result, 0, sizeof(count_by_result));
        print_header();
        runlist_iterate(runlist, runlist_parse_summary, runlist);
        print_summary(runlist);
        printf("\n\nDETAILED RESULTS:\n");

        runlist_iterate(runlist, runlist_parse_detail, runlist);
        return 0;
}