#include <gtest/gtest.h>
#include <yycc.hpp>
#include <yycc/carton/wcwidth.hpp>
#include <yycc/carton/termcolor.hpp>

#define WCWDITH ::yycc::carton::wcwidth
#define TERMCOLOR ::yycc::carton::termcolor

namespace yycctest::carton::wcwidth {

#define TEST_SUCCESS(strl, len) \
    { \
        auto rv = WCWDITH::wcswidth(strl); \
        ASSERT_TRUE(rv.has_value()); \
        EXPECT_EQ(rv.value(), len); \
    }

#define TEST_FAIL(strl) \
    { \
        auto rv = WCWDITH::wcswidth(strl); \
        EXPECT_FALSE(rv.has_value()); \
    }

    TEST(CartonWcwdith, BadAnsi) {
        TEST_FAIL(u8"\033?");
    }

    TEST(CartonWcwdith, BadCsi) {
        TEST_FAIL(u8"\033[\t");
    }

    TEST(CartonWcwdith, English) {
        TEST_SUCCESS(u8"abc", 3);
    }

    TEST(CartonWcwdith, Chinese) {
        TEST_SUCCESS(u8"中文", 4);
        TEST_SUCCESS(u8"中a文", 5);
    }

    TEST(CartonWcwdith, Japanese) {
        TEST_SUCCESS(u8"ありがとう", 10);
        TEST_SUCCESS(u8"アリガトウ", 10);
        TEST_SUCCESS(u8"ｱﾘｶﾞﾄｳ", 6);
    }

    TEST(CartonWcwdith, Termcolor) {
        using Color = TERMCOLOR::Color;
        TEST_SUCCESS(TERMCOLOR::colored(u8"abc", Color::Red), 3);
        TEST_SUCCESS(TERMCOLOR::colored(u8"中文", Color::Red), 4);
        TEST_SUCCESS(TERMCOLOR::colored(u8"ありがとう", Color::Red), 10);
    }

} // namespace yycctest::carton::wcwidth