/*
 * anopa - Copyright (C) 2015-2017 Olivier Brunel
 *
 * aa-echo.c
 * Copyright (C) 2015-2018 Olivier Brunel <jjk@jjacky.com>
 *
 * This file is part of anopa.
 *
 * anopa is free software: you can redistribute it and/or modify it under the
 * terms of the GNU General Public License as published by the Free Software
 * Foundation, either version 3 of the License, or (at your option) any later
 * version.
 *
 * anopa is distributed in the hope that it will be useful, but WITHOUT ANY
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * anopa. If not, see http://www.gnu.org/licenses/
 */

#include <getopt.h>
#include <skalibs/bytestr.h>
#include <anopa/common.h>
#include <anopa/output.h>

typedef void (*put_fn) (const char *name, const char *msg, int end);

static void
put_title (const char *name, const char *msg, int end)
{
    aa_put_title (1, name, msg, end);
}

static void
put_title2 (const char *name, const char *msg, int end)
{
    aa_put_title (0, name, msg, end);
}

static void
dieusage (int rc)
{
    aa_die_usage (rc, "[OPTION...] MESSAGE...",
            " -D, --double-output           Enable double-output mode\n"
            " -O, --log-file FILE|FD        Write log to FILE|FD\n"
            " -B, --blank-first             Print a blank line (LF) first\n"
            " -T, --title                   Show a main title (default)\n"
            " -t, --title2                  Show a secondary title\n"
            " -w, --warning                 Show a warning\n"
            " -e, --error                   Show an error\n"
            " -n, --normal                  Show \"normal\" text\n"
            " -h, --help                    Show this help screen and exit\n"
            " -V, --version                 Show version information and exit\n"
            "\n"
            "MESSAGE can be used to set the text color:\n"
            "\n"
            " +g, +green                    Set color to green\n"
            " +b, +blue                     Set color to blue\n"
            " +y, +yellow                   Set color to yellow\n"
            " +r, +red                      Set color to red\n"
            " +w, +white                    Set color to white\n"
            " +n, +normal                   Set color to normal\n"
            " ++TEXT                        To just print +TEXT\n"
            );
}

int
main (int argc, char * const argv[])
{
    PROG = "aa-echo";
    int blank = 0;
    put_fn put = put_title;
    int where = AA_OUT;
    int i;

    for (;;)
    {
        struct option longopts[] = {
            { "blank-first",        no_argument,        NULL,   'B' },
            { "double-output",      no_argument,        NULL,   'D' },
            { "error",              no_argument,        NULL,   'e' },
            { "help",               no_argument,        NULL,   'h' },
            { "normal",             no_argument,        NULL,   'n' },
            { "log-file",           required_argument,  NULL,   'O' },
            { "title",              no_argument,        NULL,   'T' },
            { "title2",             no_argument,        NULL,   't' },
            { "version",            no_argument,        NULL,   'V' },
            { "warning",            no_argument,        NULL,   'w' },
            { NULL, 0, 0, 0 }
        };
        int c;

        c = getopt_long (argc, argv, "BDehnO:TtVw", longopts, NULL);
        if (c == -1)
            break;
        switch (c)
        {
            case 'B':
                blank = 1;
                break;

            case 'D':
                aa_set_double_output (1);
                break;

            case 'e':
                put = aa_put_err;
                where = AA_ERR;
                break;

            case 'h':
                dieusage (RC_OK);

            case 'n':
                put = NULL;
                where = AA_OUT;
                break;

            case 'O':
                aa_set_log_file_or_die (optarg);
                break;

            case 'T':
                put = put_title;
                where = AA_OUT;
                break;

            case 't':
                put = put_title2;
                where = AA_OUT;
                break;

            case 'V':
                aa_die_version ();

            case 'w':
                put = aa_put_warn;
                where = AA_ERR;
                break;

            default:
                dieusage (RC_FATAL_USAGE);
        }
    }
    argc -= optind;
    argv += optind;

    if (argc < 1)
        dieusage (RC_FATAL_USAGE);

    if (blank)
        aa_bs (where, "\n");
    if (put)
        put ("", NULL, 0);
    for (i = 0; i < argc; ++i)
    {
        if (*argv[i] == '+')
        {
            if (str_equal (argv[i], "+g") || str_equal (argv[i], "+green"))
                aa_is (where, ANSI_HIGHLIGHT_GREEN_ON);
            else if (str_equal (argv[i], "+b") || str_equal (argv[i], "+blue"))
                aa_is (where, ANSI_HIGHLIGHT_BLUE_ON);
            else if (str_equal (argv[i], "+y") || str_equal (argv[i], "+yellow"))
                aa_is (where, ANSI_HIGHLIGHT_YELLOW_ON);
            else if (str_equal (argv[i], "+r") || str_equal (argv[i], "+red"))
                aa_is (where, ANSI_HIGHLIGHT_RED_ON);
            else if (str_equal (argv[i], "+w") || str_equal (argv[i], "+white"))
                aa_is (where, ANSI_HIGHLIGHT_ON);
            else if (str_equal (argv[i], "+n") || str_equal (argv[i], "+normal"))
                aa_is (where, ANSI_HIGHLIGHT_OFF);
            else
                aa_bs (where, argv[i] + 1);
        }
        else
            aa_bs (where, argv[i]);
    }
    aa_bs_end (where);

    return RC_OK;
}