/* file tokenizer.test.cpp
 *
 * author: Roland Conybeare
 */

#include "tokenizer.hpp"
#include <catch2/catch.hpp>

namespace xo {
    using xo::tok::tokentype;
    using token = xo::tok::token<char>;
    using xo::tok::span;

    namespace ut {
        namespace {
            struct testcase_tkz {
                std::string input_;
                bool expect_throw_;
                token expected_tk_;
                bool consume_all_;
            };

            std::vector<testcase_tkz>
            s_testcase_v = {
                {"<", false, token::leftangle(), true},
                {">", false, token::rightangle(), true},

                {"(", false, token::leftparen(), true},
                {")", false, token::rightparen(), true},

                {"[", false, token::leftbracket(), true},
                {"]", false, token::rightbracket(), true},

                {"{", false, token::leftbrace(), true},
                {" {", false, token::leftbrace(), true},

                {"\t{", false, token::leftbrace(), true},
                {"\n{", false, token::leftbrace(), true},
                {"}", false, token::rightbrace(), true},

                {"0",  false, token::i64_token("0"), true},
                {"1",  false, token::i64_token("1"), true},
                {"12",  false, token::i64_token("12"), true},
                {"123",  false, token::i64_token("123"), true},
                {"1234",  false, token::i64_token("1234"), true},

                {"0 ", false, token::i64_token("0"), false},
                {"1 ", false, token::i64_token("1"), false},
                {"12 ", false, token::i64_token("12"), false},
                {"123 ", false, token::i64_token("123"), false},
                {"1234 ", false, token::i64_token("1234"), false},

                {"1<", false, token::i64_token("1"), false},
                {"1>", false, token::i64_token("1"), false},
                {"1(", false, token::i64_token("1"), false},
                {"1)", false, token::i64_token("1"), false},
                {"1[", false, token::i64_token("1"), false},
                {"1]", false, token::i64_token("1"), false},
                {"1{", false, token::i64_token("1"), false},
                {"1}", false, token::i64_token("1"), false},
                {"1;", false, token::i64_token("1"), false},
                {"1:", false, token::i64_token("1"), false},
                {"1,", false, token::i64_token("1"), false},

                {".1", false, token::f64_token(".1"), true},
                {".12", false, token::f64_token(".12"), true},
                {".123", false, token::f64_token(".123"), true},

                {"+.1", false, token::f64_token("+.1"), true},
                {"+.12", false, token::f64_token("+.12"), true},
                {"+.123", false, token::f64_token("+.123"), true},

                {"-.1", false, token::f64_token("-.1"), true},
                {"-.12", false, token::f64_token("-.12"), true},
                {"-.123", false, token::f64_token("-.123"), true},

                {"1.", false, token::f64_token("1."), true},
                {"1.2", false, token::f64_token("1.2"), true},
                {"1.23", false, token::f64_token("1.23"), true},

                {"1e0", false, token::f64_token("1e0"), true},
                {"1e-1", false, token::f64_token("1e-1"), true},
                {"1e1", false, token::f64_token("1e1"), true},
                {"1e+1", false, token::f64_token("1e+1"), true},

                {"\"hello\"", false, token::string_token("hello"), true},
                /* tokenizer sees this input:
                 *   "\"hi\", she said"
                 */
                {"\"\\\"hi\\\", she said\"", false, token::string_token("\"hi\", she said"), true},
                /* tokenizer sees this input:
                 *   "look ma, newline ->\n<- "
                 */
                {"\"look ma, newline ->\\n<- \"", false,
                 token::string_token("look ma, newline ->\n<- "), true},
                /* tokenizer sees this input:
                 *   "tab to the right [\t], to the right [\t]"
                 */
                {"\"tab to the right [\\t], to the right [\\t]\"", false,
                 token::string_token("tab to the right [\t], to the right [\t]"), true},

                {"symbol", false, token::symbol_token("symbol"), true},
            };
        }

        TEST_CASE("tokenizer", "[tokenizer]") {
            for (std::size_t i_tc = 0, n_tc = s_testcase_v.size(); i_tc < n_tc; ++i_tc) {
                const testcase_tkz & testcase = s_testcase_v[i_tc];

                INFO(xtag("input", testcase.input_));
                INFO(xtag("i_tc", i_tc));

                using tokenizer
                    = xo::tok::tokenizer<char>;

                tokenizer tkz;
                tokenizer::span_type
                    in_span(testcase.input_.c_str(),
                            testcase.input_.c_str() + testcase.input_.size());

                auto out = tkz.scan(in_span);

                auto tk = out.first;

                if (tk.is_invalid())
                    tk = tkz.notify_eof();

                REQUIRE(tk.tk_type() == testcase.expected_tk_.tk_type());
                if (tk.tk_type() == tokentype::tk_i64)
                {
                    REQUIRE(!tk.text().empty());
                    REQUIRE(tk.i64_value() == testcase.expected_tk_.i64_value());
                } else if (tk.tk_type() == tokentype::tk_f64)
                {
                    REQUIRE(!tk.text().empty());
                    REQUIRE(tk.f64_value() == testcase.expected_tk_.f64_value());
                } else if(tk.tk_type() == tokentype::tk_string)
                {
                    /* tk.text() can be empty, consider input "" */
                    REQUIRE(tk.text() == testcase.expected_tk_.text());
                } else if(tk.tk_type() == tokentype::tk_symbol)
                {
                    REQUIRE(!tk.text().empty());
                    REQUIRE(tk.text() == testcase.expected_tk_.text());
                } else {
                    REQUIRE(tk.text().empty());
                }

                /* must consume all input for tests we're doing here */
                if (testcase.consume_all_)
                    REQUIRE(out.second == in_span);
                else
                    REQUIRE(out.second != in_span);
            }
        }

    } /*namespace ut*/
} /*namespace xo*/

/* end tokenizer.test.cpp */