#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <kernaux/cmdline.h>

#include <assert.h>
#include <stdbool.h>
#include <stddef.h>
#include <string.h>

static const unsigned int ARGV_COUNT_MAX = 100;
static const unsigned int ARG_SIZE_MAX = 4096;

static void test(
    const char *cmdline,
    unsigned int argv_count_max,
    unsigned int arg_size_max,

    bool expected_result,
    const char *expected_error_msg,
    unsigned int expected_argc,
    const char *const *const expected_argv
);

static const char *const argv_empty[] = {""};
static const char *const argv_empty_X2[] = {"", ""};
static const char *const argv_empty_X3[] = {"", "", ""};

static const char *const argv_foo[] = {"foo"};
static const char *const argv_foo_bar[] = {"foo", "bar"};
static const char *const argv_foo_bar_car[] = {"foo", "bar", "car"};

static const char *const argv_spaceX3_X3[] = {"   ", "   ", "   "};
static const char *const argv_backslashX3_X3[] = {"\\\\\\", "\\\\\\", "\\\\\\"};
static const char *const argv_quotmarkX3_X3[] = {"\"\"\"", "\"\"\"", "\"\"\""};

static const char *const argv_foospacebar_car[] = {"foo bar", "car"};
static const char *const argv_foo_barspacecar[] = {"foo", "bar car"};

static const char *const argv_space[] = {" "};
static const char *const argv_backslash[] = {"\\"};
static const char *const argv_quotmark[] = {"\""};
static const char *const argv_foospace[] = {"foo "};
static const char *const argv_foobackslash[] = {"foo\\"};
static const char *const argv_fooquotmark[] = {"foo\""};
static const char *const argv_spacefoo[] = {" foo"};
static const char *const argv_backslashfoo[] = {"\\foo"};
static const char *const argv_quotmarkfoo[] = {"\"foo"};
static const char *const argv_spacefoospace[] = {" foo "};
static const char *const argv_backslashfoobackslash[] = {"\\foo\\"};
static const char *const argv_quotmarkfooquotmark[] = {"\"foo\""};
static const char *const argv_foospacebar[] = {"foo bar"};
static const char *const argv_foobackslashbar[] = {"foo\\bar"};
static const char *const argv_fooquotmarkbar[] = {"foo\"bar"};
static const char *const argv_spacefoo_bar[] = {" foo", "bar"};
static const char *const argv_backslashfoo_bar[] = {"\\foo", "bar"};
static const char *const argv_quotmarkfoo_bar[] = {"\"foo", "bar"};
static const char *const argv_foospace_bar[] = {"foo ", "bar"};
static const char *const argv_foobackslash_bar[] = {"foo\\", "bar"};
static const char *const argv_fooquotmark_bar[] = {"foo\"", "bar"};
static const char *const argv_spacefoospace_bar[] = {" foo ", "bar"};
static const char *const argv_backslashfoobackslash_bar[] = {"\\foo\\", "bar"};
static const char *const argv_quotmarkfooquotmark_bar[] = {"\"foo\"", "bar"};
static const char *const argv_foo_spacebar[] = {"foo", " bar"};
static const char *const argv_foo_backslashbar[] = {"foo", "\\bar"};
static const char *const argv_foo_quotmarkbar[] = {"foo", "\"bar"};
static const char *const argv_foo_barspace[] = {"foo", "bar "};
static const char *const argv_foo_barbackslash[] = {"foo", "bar\\"};
static const char *const argv_foo_barquotmark[] = {"foo", "bar\""};
static const char *const argv_foo_spacebarspace[] = {"foo", " bar "};
static const char *const argv_foo_backslashbarbackslash[] = {"foo", "\\bar\\"};
static const char *const argv_foo_quotmarkbarquotmark[] = {"foo", "\"bar\""};

int main()
{
    test("",             0, 0, true, "", 0, NULL);
    test("   ",          0, 0, true, "", 0, NULL);
    test("foo",          0, 0, true, "", 1, argv_foo);
    test("foo bar",      0, 0, true, "", 2, argv_foo_bar);
    test(" foo bar",     0, 0, true, "", 2, argv_foo_bar);
    test("foo bar ",     0, 0, true, "", 2, argv_foo_bar);
    test(" foo bar ",    0, 0, true, "", 2, argv_foo_bar);
    test("foo  bar",     0, 0, true, "", 2, argv_foo_bar);
    test("  foo  bar",   0, 0, true, "", 2, argv_foo_bar);
    test("foo  bar  ",   0, 0, true, "", 2, argv_foo_bar);
    test("  foo  bar  ", 0, 0, true, "", 2, argv_foo_bar);
    test("foo bar car",  0, 0, true, "", 3, argv_foo_bar_car);

    test("\"\"",                    0, 0, true, "", 1, argv_empty);
    test("\"\" \"\"",               0, 0, true, "", 2, argv_empty_X2);
    test("\"\" \"\" \"\"",          0, 0, true, "", 3, argv_empty_X3);
    test("\"foo\"",                 0, 0, true, "", 1, argv_foo);
    test("\"foo\" \"bar\"",         0, 0, true, "", 2, argv_foo_bar);
    test(" \"foo\" \"bar\"",        0, 0, true, "", 2, argv_foo_bar);
    test("\"foo\" \"bar\" ",        0, 0, true, "", 2, argv_foo_bar);
    test(" \"foo\" \"bar\" ",       0, 0, true, "", 2, argv_foo_bar);
    test("\"foo\"  \"bar\"",        0, 0, true, "", 2, argv_foo_bar);
    test("\"foo\"  \"bar\"  ",      0, 0, true, "", 2, argv_foo_bar);
    test("  \"foo\"  \"bar\"",      0, 0, true, "", 2, argv_foo_bar);
    test("  \"foo\"  \"bar\"  ",    0, 0, true, "", 2, argv_foo_bar);
    test("\"foo\" \"bar\" \"car\"", 0, 0, true, "", 3, argv_foo_bar_car);

    test("foo bar car", 3, 0, true, "", 3, argv_foo_bar_car);
    test("foo bar car", 0, 4, true, "", 3, argv_foo_bar_car);
    test("foo bar car", 3, 4, true, "", 3, argv_foo_bar_car);

    test("foo bar car", 2, 0, false, "too many args", 0, NULL);
    test("foo bar car", 0, 3, false, "arg too long",  0, NULL);
    test("foo bar car", 2, 3, false, "arg too long",  0, NULL);

    test("\"foo\" \"bar\" \"car\"", 3, 0, true, "", 3, argv_foo_bar_car);
    test("\"foo\" \"bar\" \"car\"", 0, 4, true, "", 3, argv_foo_bar_car);
    test("\"foo\" \"bar\" \"car\"", 3, 4, true, "", 3, argv_foo_bar_car);

    test("\"foo\" \"bar\" \"car\"", 2, 0, false, "too many args", 0, NULL);
    test("\"foo\" \"bar\" \"car\"", 0, 3, false, "arg too long",  0, NULL);
    test("\"foo\" \"bar\" \"car\"", 2, 3, false, "arg too long",  0, NULL);

    test("\\ ",                 0, 0, true, "", 1, argv_space);
    test("\"\\ \"",             0, 0, true, "", 1, argv_space);
    test("\\\\",                0, 0, true, "", 1, argv_backslash);
    test("\"\\\\\"",            0, 0, true, "", 1, argv_backslash);
    test("\\\"",                0, 0, true, "", 1, argv_quotmark);
    test("\"\\\"\"",            0, 0, true, "", 1, argv_quotmark);
    test("foo\\ ",              0, 0, true, "", 1, argv_foospace);
    test("\"foo\\ \"",          0, 0, true, "", 1, argv_foospace);
    test("foo\\\\",             0, 0, true, "", 1, argv_foobackslash);
    test("\"foo\\\\\"",         0, 0, true, "", 1, argv_foobackslash);
    test("foo\\\"",             0, 0, true, "", 1, argv_fooquotmark);
    test("\"foo\\\"\"",         0, 0, true, "", 1, argv_fooquotmark);
    test("\\ foo",              0, 0, true, "", 1, argv_spacefoo);
    test("\"\\ foo\"",          0, 0, true, "", 1, argv_spacefoo);
    test("\\\\foo",             0, 0, true, "", 1, argv_backslashfoo);
    test("\"\\\\foo\"",         0, 0, true, "", 1, argv_backslashfoo);
    test("\\\"foo",             0, 0, true, "", 1, argv_quotmarkfoo);
    test("\"\\\"foo\"",         0, 0, true, "", 1, argv_quotmarkfoo);
    test("\\ foo\\ ",           0, 0, true, "", 1, argv_spacefoospace);
    test("\"\\ foo\\ \"",       0, 0, true, "", 1, argv_spacefoospace);
    test("\\\\foo\\\\",         0, 0, true, "", 1, argv_backslashfoobackslash);
    test("\"\\\\foo\\\\\"",     0, 0, true, "", 1, argv_backslashfoobackslash);
    test("\\\"foo\\\"",         0, 0, true, "", 1, argv_quotmarkfooquotmark);
    test("\"\\\"foo\\\"\"",     0, 0, true, "", 1, argv_quotmarkfooquotmark);
    test("foo\\ bar",           0, 0, true, "", 1, argv_foospacebar);
    test("\"foo\\ bar\"",       0, 0, true, "", 1, argv_foospacebar);
    test("foo\\\\bar",          0, 0, true, "", 1, argv_foobackslashbar);
    test("\"foo\\\\bar\"",      0, 0, true, "", 1, argv_foobackslashbar);
    test("foo\\\"bar",          0, 0, true, "", 1, argv_fooquotmarkbar);
    test("\"foo\\\"bar\"",      0, 0, true, "", 1, argv_fooquotmarkbar);
    test("\\ foo bar",          0, 0, true, "", 2, argv_spacefoo_bar);
    test("\"\\ foo\" bar",      0, 0, true, "", 2, argv_spacefoo_bar);
    test("\\\\foo bar",         0, 0, true, "", 2, argv_backslashfoo_bar);
    test("\"\\\\foo\" bar",     0, 0, true, "", 2, argv_backslashfoo_bar);
    test("\\\"foo bar",         0, 0, true, "", 2, argv_quotmarkfoo_bar);
    test("\"\\\"foo\" bar",     0, 0, true, "", 2, argv_quotmarkfoo_bar);
    test("foo\\  bar",          0, 0, true, "", 2, argv_foospace_bar);
    test("\"foo\\ \" bar",      0, 0, true, "", 2, argv_foospace_bar);
    test("foo\\\\ bar",         0, 0, true, "", 2, argv_foobackslash_bar);
    test("\"foo\\\\\" bar",     0, 0, true, "", 2, argv_foobackslash_bar);
    test("foo\\\" bar",         0, 0, true, "", 2, argv_fooquotmark_bar);
    test("\"foo\\\"\" bar",     0, 0, true, "", 2, argv_fooquotmark_bar);
    test("\\ foo\\  bar",       0, 0, true, "", 2, argv_spacefoospace_bar);
    test("\"\\ foo\\ \" bar",   0, 0, true, "", 2, argv_spacefoospace_bar);
    test("\\\\foo\\\\ bar",     0, 0, true, "", 2, argv_backslashfoobackslash_bar);
    test("\"\\\\foo\\\\\" bar", 0, 0, true, "", 2, argv_backslashfoobackslash_bar);
    test("\\\"foo\\\" bar",     0, 0, true, "", 2, argv_quotmarkfooquotmark_bar);
    test("\"\\\"foo\\\"\" bar", 0, 0, true, "", 2, argv_quotmarkfooquotmark_bar);
    test("foo \\ bar",          0, 0, true, "", 2, argv_foo_spacebar);
    test("foo \"\\ bar\"",      0, 0, true, "", 2, argv_foo_spacebar);
    test("foo \\\\bar",         0, 0, true, "", 2, argv_foo_backslashbar);
    test("foo \"\\\\bar\"",     0, 0, true, "", 2, argv_foo_backslashbar);
    test("foo \\\"bar",         0, 0, true, "", 2, argv_foo_quotmarkbar);
    test("foo \"\\\"bar\"",     0, 0, true, "", 2, argv_foo_quotmarkbar);
    test("foo bar\\ ",          0, 0, true, "", 2, argv_foo_barspace);
    test("foo \"bar\\ \"",      0, 0, true, "", 2, argv_foo_barspace);
    test("foo bar\\\\",         0, 0, true, "", 2, argv_foo_barbackslash);
    test("foo \"bar\\\\\"",     0, 0, true, "", 2, argv_foo_barbackslash);
    test("foo bar\\\"",         0, 0, true, "", 2, argv_foo_barquotmark);
    test("foo \"bar\\\"\"",     0, 0, true, "", 2, argv_foo_barquotmark);
    test("foo \\ bar\\ ",       0, 0, true, "", 2, argv_foo_spacebarspace);
    test("foo \"\\ bar\\ \"",   0, 0, true, "", 2, argv_foo_spacebarspace);
    test("foo \\\\bar\\\\",     0, 0, true, "", 2, argv_foo_backslashbarbackslash);
    test("foo \"\\\\bar\\\\\"", 0, 0, true, "", 2, argv_foo_backslashbarbackslash);
    test("foo \\\"bar\\\"",     0, 0, true, "", 2, argv_foo_quotmarkbarquotmark);
    test("foo \"\\\"bar\\\"\"", 0, 0, true, "", 2, argv_foo_quotmarkbarquotmark);

    test("foo\\ bar car",       0, 0, true, "", 2, argv_foospacebar_car);
    test("\"foo bar\" car",     0, 0, true, "", 2, argv_foospacebar_car);
    test("\"foo bar\" \"car\"", 0, 0, true, "", 2, argv_foospacebar_car);
    test("foo bar\\ car",       0, 0, true, "", 2, argv_foo_barspacecar);
    test("foo \"bar car\"",     0, 0, true, "", 2, argv_foo_barspacecar);
    test("\"foo\" \"bar car\"", 0, 0, true, "", 2, argv_foo_barspacecar);

    test("\\ \\ \\  \\ \\ \\  \\ \\ \\ ",          3, 0, true, "", 3, argv_spaceX3_X3);
    test("\\\\\\\\\\\\ \\\\\\\\\\\\ \\\\\\\\\\\\", 3, 0, true, "", 3, argv_backslashX3_X3);
    test("\\\"\\\"\\\" \\\"\\\"\\\" \\\"\\\"\\\"", 3, 0, true, "", 3, argv_quotmarkX3_X3);
    test("\\ \\ \\  \\ \\ \\  \\ \\ \\ ",          0, 4, true, "", 3, argv_spaceX3_X3);
    test("\\\\\\\\\\\\ \\\\\\\\\\\\ \\\\\\\\\\\\", 0, 4, true, "", 3, argv_backslashX3_X3);
    test("\\\"\\\"\\\" \\\"\\\"\\\" \\\"\\\"\\\"", 0, 4, true, "", 3, argv_quotmarkX3_X3);
    test("\\ \\ \\  \\ \\ \\  \\ \\ \\ ",          3, 4, true, "", 3, argv_spaceX3_X3);
    test("\\\\\\\\\\\\ \\\\\\\\\\\\ \\\\\\\\\\\\", 3, 4, true, "", 3, argv_backslashX3_X3);
    test("\\\"\\\"\\\" \\\"\\\"\\\" \\\"\\\"\\\"", 3, 4, true, "", 3, argv_quotmarkX3_X3);

    test("\\ \\ \\  \\ \\ \\  \\ \\ \\ ",          2, 0, false, "too many args", 0, NULL);
    test("\\\\\\\\\\\\ \\\\\\\\\\\\ \\\\\\\\\\\\", 2, 0, false, "too many args", 0, NULL);
    test("\\\"\\\"\\\" \\\"\\\"\\\" \\\"\\\"\\\"", 2, 0, false, "too many args", 0, NULL);
    test("\\ \\ \\  \\ \\ \\  \\ \\ \\ ",          0, 3, false, "arg too long",  0, NULL);
    test("\\\\\\\\\\\\ \\\\\\\\\\\\ \\\\\\\\\\\\", 0, 3, false, "arg too long",  0, NULL);
    test("\\\"\\\"\\\" \\\"\\\"\\\" \\\"\\\"\\\"", 0, 3, false, "arg too long",  0, NULL);
    test("\\ \\ \\  \\ \\ \\  \\ \\ \\ ",          2, 3, false, "arg too long",  0, NULL);
    test("\\\\\\\\\\\\ \\\\\\\\\\\\ \\\\\\\\\\\\", 2, 3, false, "arg too long",  0, NULL);
    test("\\\"\\\"\\\" \\\"\\\"\\\" \\\"\\\"\\\"", 2, 3, false, "arg too long",  0, NULL);

    test("\\",     0, 0, false, "EOL after backslash", 0, NULL);
    test(" \\",    0, 0, false, "EOL after backslash", 0, NULL);
    test("\\ \\",  0, 0, false, "EOL after backslash", 0, NULL);
    test("\\\\\\", 0, 0, false, "EOL after backslash", 0, NULL);
    test("\\\"\\", 0, 0, false, "EOL after backslash", 0, NULL);
    test("foo\\",  0, 0, false, "EOL after backslash", 0, NULL);

    test("\"\\",     0, 0, false, "EOL after backslash inside quote", 0, NULL);
    test("\" \\",    0, 0, false, "EOL after backslash inside quote", 0, NULL);
    test("\"\\ \\",  0, 0, false, "EOL after backslash inside quote", 0, NULL);
    test("\"\\\\\\", 0, 0, false, "EOL after backslash inside quote", 0, NULL);
    test("\"\\\"\\", 0, 0, false, "EOL after backslash inside quote", 0, NULL);
    test("\"foo\\",  0, 0, false, "EOL after backslash inside quote", 0, NULL);

    test("foo\"",    0, 0, false, "unescaped quotation mark", 0, NULL);
    test("foo\"bar", 0, 0, false, "unescaped quotation mark", 0, NULL);

    test("\"",    0, 0, false, "EOL inside quote", 0, NULL);
    test("\"foo", 0, 0, false, "EOL inside quote", 0, NULL);

    return 0;
}

void test(
    const char *const cmdline,
    unsigned int argv_count_max,
    unsigned int arg_size_max,

    const bool expected_result,
    const char *const expected_error_msg,
    unsigned int expected_argc,
    const char *const *const expected_argv
) {
    if (argv_count_max == 0) {
        argv_count_max = ARGV_COUNT_MAX;
    }

    if (arg_size_max == 0) {
        arg_size_max = ARG_SIZE_MAX;
    }

    char error_msg[KERNAUX_CMDLINE_ERROR_MSG_SIZE_MAX];
    unsigned int argc = 1234;
    char *argv[argv_count_max];
    char buffer[argv_count_max * arg_size_max];

    assert(
        kernaux_cmdline_parse(
            cmdline,
            error_msg,
            &argc,
            argv,
            buffer,
            argv_count_max,
            arg_size_max
        ) == !!expected_result
    );

    assert(strcmp(error_msg, expected_error_msg) == 0);
    assert(argc == expected_argc);

    for (unsigned int index = 0; index < argc; ++index) {
        assert(strcmp(argv[index], expected_argv[index]) == 0);
    }

    for (unsigned int index = argc; index < argv_count_max; ++index) {
        assert(argv[index] == NULL);
    }
}