/* Unit tests for i18n language selection and text lookup */

#include "../../include/i18n.h"
#include "text_assert.h"
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define TEST(name) static void test_##name()
#define RUN_TEST(name) do { \
    printf("Running %s... ", #name); \
    test_##name(); \
    printf("✓\n"); \
    tests_passed++; \
} while(0)

static int tests_passed = 0;

TEST(parse_explicit_languages) {
    ui_lang_t lang;

    assert(i18n_parse_ui_lang("zh", UI_LANG_EN) == UI_LANG_ZH);
    assert(i18n_parse_ui_lang("zh_CN.UTF-8", UI_LANG_EN) == UI_LANG_ZH);
    assert(i18n_parse_ui_lang("en", UI_LANG_ZH) == UI_LANG_EN);
    assert(i18n_parse_ui_lang("en_US.UTF-8", UI_LANG_ZH) == UI_LANG_EN);
    assert(i18n_parse_ui_lang("C", UI_LANG_ZH) == UI_LANG_EN);
    assert(i18n_parse_ui_lang("POSIX", UI_LANG_ZH) == UI_LANG_EN);

    assert(i18n_try_parse_ui_lang("zh", &lang) == true);
    assert(lang == UI_LANG_ZH);
    assert(i18n_try_parse_ui_lang("en", &lang) == true);
    assert(lang == UI_LANG_EN);
    assert(i18n_try_parse_ui_lang("cn", &lang) == false);
    assert(i18n_try_parse_ui_lang("english", &lang) == false);
    assert(i18n_try_parse_ui_lang("chinese", &lang) == false);
    assert(i18n_try_parse_ui_lang("中文", &lang) == false);
    assert(i18n_try_parse_ui_lang("英文", &lang) == false);
    assert(i18n_try_parse_ui_lang("fr", &lang) == false);
}

TEST(parse_unknown_uses_fallback) {
    assert(i18n_parse_ui_lang(NULL, UI_LANG_ZH) == UI_LANG_ZH);
    assert(i18n_parse_ui_lang("", UI_LANG_EN) == UI_LANG_EN);
    assert(i18n_parse_ui_lang("fr_FR.UTF-8", UI_LANG_ZH) == UI_LANG_ZH);
}

TEST(parse_ignores_surrounding_whitespace) {
    ui_lang_t lang;

    assert(i18n_try_parse_ui_lang("  zh  ", &lang) == true);
    assert(lang == UI_LANG_ZH);
    assert(i18n_parse_ui_lang("\ten_US.UTF-8\n", UI_LANG_ZH) == UI_LANG_EN);
    assert(i18n_try_parse_ui_lang(" english ", &lang) == false);
    assert(i18n_try_parse_ui_lang("zh CN", &lang) == false);

    setenv("TNT_LANG", " zh ", 1);
    setenv("LC_ALL", "en_US.UTF-8", 1);
    assert(i18n_default_ui_lang() == UI_LANG_ZH);
}

TEST(default_prefers_tnt_lang) {
    setenv("TNT_LANG", "zh_CN.UTF-8", 1);
    setenv("LC_ALL", "en_US.UTF-8", 1);
    assert(i18n_default_ui_lang() == UI_LANG_ZH);

    setenv("TNT_LANG", "en", 1);
    setenv("LC_ALL", "zh_CN.UTF-8", 1);
    assert(i18n_default_ui_lang() == UI_LANG_EN);
}

TEST(default_uses_locale_when_no_tnt_lang) {
    unsetenv("TNT_LANG");
    setenv("LC_ALL", "zh_CN.UTF-8", 1);
    assert(i18n_default_ui_lang() == UI_LANG_ZH);

    setenv("LC_ALL", "C", 1);
    assert(i18n_default_ui_lang() == UI_LANG_EN);
}

TEST(text_lookup_matches_language) {
    i18n_string_t sample = I18N_STRING("fallback", "替代");

    assert(strcmp(i18n_string(sample, UI_LANG_EN), "fallback") == 0);
    assert(strcmp(i18n_string(sample, UI_LANG_ZH), "替代") == 0);
    assert(strcmp(i18n_string(sample, (ui_lang_t)99), "fallback") == 0);

    assert(strstr(i18n_text(UI_LANG_EN, I18N_USERNAME_PROMPT),
                  "display name") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_USERNAME_PROMPT),
                  "用户名") != NULL);
    assert(strstr(i18n_text((ui_lang_t)99, I18N_USERNAME_PROMPT),
                  "display name") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_WELCOME_SUBTITLE),
                  "anonymous chat") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_WELCOME_SUBTITLE),
                  "匿名聊天室") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_HELP_STATUS_FORMAT),
                  "KEY REFERENCE") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_HELP_STATUS_FORMAT),
                  "l:lang") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_HELP_STATUS_FORMAT),
                  "按键参考") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_HELP_STATUS_FORMAT),
                  "l:语言") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_COMMAND_OUTPUT_TITLE),
                  "COMMAND") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_COMMAND_OUTPUT_TITLE),
                  "命令输出") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_COMMAND_OUTPUT_STATUS_FORMAT),
                  "q:close") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_COMMAND_OUTPUT_STATUS_FORMAT),
                  "q:关闭") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_MOTD_CONTINUE_HINT),
                  "Press any key") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_MOTD_CONTINUE_HINT),
                  "按任意键") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_TITLE_ONLINE_FORMAT),
                  "online") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_TITLE_ONLINE_FORMAT),
                  "在线") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_IDLE_TIMEOUT_FORMAT),
                  "idle timeout") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_IDLE_TIMEOUT_FORMAT),
                  "空闲超时") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_MSG_SENT_FORMAT),
                  "Private message sent") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_MSG_SENT_FORMAT),
                  "私信已发送") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_INBOX_TITLE),
                  "Private messages") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_INBOX_TITLE),
                  "私信") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_SEARCH_HEADER_FORMAT),
                  "Search") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_SEARCH_HEADER_FORMAT),
                  "搜索") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_LANG_CURRENT_FORMAT),
                  "lang <en|zh>") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_LANG_CURRENT_FORMAT),
                  "lang <en|zh>") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_UNKNOWN_COMMAND_FORMAT),
                  "Unknown command") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_UNKNOWN_COMMAND_FORMAT),
                  "未知命令") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_EXEC_POST_EMPTY),
                  "message cannot be empty") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_EXEC_POST_EMPTY),
                  "消息不能为空") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_EXEC_COMMAND_TOO_LONG),
                  "command too long") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_EXEC_COMMAND_TOO_LONG),
                  "命令过长") != NULL);
    assert(strstr(i18n_text(UI_LANG_EN, I18N_EXEC_UNKNOWN_COMMAND_FORMAT),
                  "Unknown command") != NULL);
    assert(strstr(i18n_text(UI_LANG_ZH, I18N_EXEC_UNKNOWN_COMMAND_FORMAT),
                  "未知命令") != NULL);
    assert(strcmp(i18n_ui_lang_code(UI_LANG_EN), "en") == 0);
    assert(strcmp(i18n_ui_lang_code(UI_LANG_ZH), "zh") == 0);
    assert(strcmp(i18n_ui_lang_code((ui_lang_t)99), "en") == 0);
    assert(i18n_next_ui_lang(UI_LANG_EN) == UI_LANG_ZH);
    assert(i18n_next_ui_lang(UI_LANG_ZH) == UI_LANG_EN);
    assert(i18n_next_ui_lang((ui_lang_t)99) == UI_LANG_EN);
}

TEST(text_catalog_is_complete) {
    for (int id = 0; id < I18N_TEXT_COUNT; id++) {
        assert(i18n_text(UI_LANG_EN, (i18n_text_id_t)id)[0] != '\0');
        assert(i18n_text(UI_LANG_ZH, (i18n_text_id_t)id)[0] != '\0');
        assert_ascii_angle_placeholders(
            i18n_text(UI_LANG_ZH, (i18n_text_id_t)id));
    }

    assert(strcmp(i18n_text(UI_LANG_EN,
                            (i18n_text_id_t)I18N_TEXT_COUNT), "") == 0);
    assert(strcmp(i18n_text(UI_LANG_ZH,
                            (i18n_text_id_t)I18N_TEXT_COUNT), "") == 0);
}

int main(void) {
    printf("Running i18n unit tests...\n\n");

    RUN_TEST(parse_explicit_languages);
    RUN_TEST(parse_unknown_uses_fallback);
    RUN_TEST(parse_ignores_surrounding_whitespace);
    RUN_TEST(default_prefers_tnt_lang);
    RUN_TEST(default_uses_locale_when_no_tnt_lang);
    RUN_TEST(text_lookup_matches_language);
    RUN_TEST(text_catalog_is_complete);

    printf("\n✓ All %d tests passed!\n", tests_passed);
    return 0;
}