#include "global.h"

// The font index is a value specific to font_cjk.ttc
i32_map_t *translator_cjk_font_indices = NULL;
char      *translator_last_locale      = "en";
char      *_translator_load_translations_cjk_font_path;
char      *_translator_load_translations_cjk_font_disk_path;
bool       _translator_init_font_cjk;
char      *_translator_init_font_font_path;
f32        _translator_init_font_font_scale;

// Mark strings as localizable in order to be parsed by the extract_locale script
// The string will not be translated to the currently selected locale though
char *_tr(char *s) {
	return s;
}

// Localizes a string with the given placeholders replaced (format is "{placeholder_name}")
// If the string isn't available in the translation, this method will return the source English string
char *vtr(char *id, any_map_t *vars) {
	char *translation = id;

	// English is the source language
	if (!string_equals(g_config->locale, "en")) {
		if (string_index_of(id, "\n") > -1) {
			id = string_replace_all_tmp(id, "\n", "\\n");
		}
		char *s = any_map_get(translator_translations, id);
		if (s != NULL) {
			translation = string_index_of(s, "\\n") > -1 ? string_replace_all_tmp(s, "\\n", "\n") : s;
		}
	}

	if (vars != NULL) {
		string_array_t *keys = vars->keys;
		for (i32 i = 0; i < keys->capacity; ++i) {
			char *k = keys->buffer[i];
			if (k == NULL) {
				continue;
			}
			translation = string_replace_all_tmp(translation, string_tmp("{%s}", k), any_map_get(vars, k));
		}
	}

	return translation;
}

char *tr(char *id) {
	return vtr(id, NULL);
}

void translator_init_font_on_next_frame(void *_) {
	bool  cjk        = _translator_init_font_cjk;
	char *font_path  = _translator_init_font_font_path;
	f32   font_scale = _translator_init_font_font_scale;

	draw_font_t *f = data_get_font(font_path);
	if (cjk) {
		i32 font_index    = i32_map_get(translator_cjk_font_indices, g_config->locale) != -1 ? i32_map_get(translator_cjk_font_indices, g_config->locale) : 0;
		f->index          = font_index;
		f->glyphs_version = 0;
		draw_font_init(f);
	}
	g_font = f;

	// Scale up the font size and elements width a bit
	g_theme->FONT_SIZE = math_floor(base_default_font_size * font_scale);
	g_theme->ELEMENT_W = math_floor(base_default_element_w * (!string_equals(g_config->locale, "en") ? 1.4 : 1.0));

	ui_set_font(g_ui, f);
	ui_set_scale(UI_SCALE());
}

void translator_init_font(bool cjk, char *font_path, f32 font_scale) {
	_translator_init_font_cjk        = cjk;
	_translator_init_font_font_path  = string_copy(font_path);
	_translator_init_font_font_scale = font_scale;

	// Load and assign font with cjk characters
	sys_notify_on_next_frame(&translator_init_font_on_next_frame, NULL);
}

void translator_extended_glyphs() {
	// Basic Latin + Latin-1 Supplement + Latin Extended-A
	draw_font_init_glyphs(32, 383);
	// + Greek
	for (i32 i = 880; i < 1023; ++i) {
		draw_font_add_glyph(i);
	}
	// + Cyrillic
	for (i32 i = 1024; i < 1119; ++i) {
		draw_font_add_glyph(i);
	}
}

void translator_load_translations_on_cjk_downloaded(char *url) {
	if (!iron_file_exists(_translator_load_translations_cjk_font_disk_path)) {
		// Fall back to English
		g_config->locale = "en";
		translator_extended_glyphs();
		translator_translations = any_map_create();
		translator_init_font(false, "font.ttf", 1.0);
	}
	else {
		translator_init_font(true, _translator_load_translations_cjk_font_path, 1.4);
	}
}

// (Re)loads translations for the specified locale
void translator_load_translations(char *new_locale) {

	if (translator_cjk_font_indices == NULL) {
		translator_cjk_font_indices = i32_map_create();
		i32_map_set(translator_cjk_font_indices, "ja", 0);
		i32_map_set(translator_cjk_font_indices, "ko", 1);
		i32_map_set(translator_cjk_font_indices, "zh_cn", 2);
		i32_map_set(translator_cjk_font_indices, "zh_tw", 3);
		i32_map_set(translator_cjk_font_indices, "zh_tw.big5", 4);
	}

	if (string_equals(new_locale, "system")) {
		g_config->locale = string_copy(iron_language());
	}

	// Check whether the requested or detected locale is available
	if (!string_equals(g_config->locale, "en") && string_array_index_of(translator_get_supported_locales(), g_config->locale) == -1) {
		// Fall back to English
		g_config->locale = "en";
	}

	// No translations to load, as source strings are in English
	// Clear existing translations if switching languages at runtime
	translator_translations = any_map_create();

	if (string_equals(g_config->locale, "en") && string_equals(translator_last_locale, "en")) {
		// No need to generate extended font atlas for English locale
		return;
	}

	translator_last_locale = string_copy(g_config->locale);

	if (!string_equals(g_config->locale, "en")) {
		// Load the translation file
		char *translation_json  = sys_buffer_to_string(iron_load_blob(string("data/locale/%s.json", g_config->locale)));
		translator_translations = json_parse_to_map(translation_json);
	}

	// Generate extended font atlas
	translator_extended_glyphs();

	// Push additional char codes contained in translation file
	bool            cjk  = false;
	string_array_t *keys = map_keys(translator_translations);
	for (i32 i = 0; i < keys->length; ++i) {
		char *s = any_map_get(translator_translations, keys->buffer[i]);

		for (i32 i = 0; char_code_at(s, i) != 0;) {
			i32 l         = 0;
			i32 codepoint = string_utf8_decode((s) + i, &l);
			i += l;

			// Assume cjk in the > 1119 range
			if (codepoint > 1119 && !draw_font_has_glyph(codepoint)) {
				cjk = true;
				draw_font_add_glyph(codepoint);
			}
		}
	}
	array_free(keys);
	free(keys);

	if (cjk) {
		if (path_is_protected()) {
			_translator_load_translations_cjk_font_path      = string_copy(iron_internal_save_path());
			_translator_load_translations_cjk_font_disk_path = string_copy(iron_internal_save_path());
		}
		else {
			_translator_load_translations_cjk_font_path      = "";
			_translator_load_translations_cjk_font_disk_path = string("%s%s", path_data(), PATH_SEP);
		}
		_translator_load_translations_cjk_font_path      = string("%sfont_cjk.ttc", _translator_load_translations_cjk_font_path);
		_translator_load_translations_cjk_font_disk_path = string("%sfont_cjk.ttc", _translator_load_translations_cjk_font_disk_path);

		if (!iron_file_exists(_translator_load_translations_cjk_font_disk_path)) {
			file_download_to("https://github.com/armory3d/armorbase/raw/main/Assets/common/extra/font_cjk.ttc",
			                 _translator_load_translations_cjk_font_disk_path, &translator_load_translations_on_cjk_downloaded, 20332392);
		}
		else {
			translator_init_font(true, _translator_load_translations_cjk_font_path, 1.4);
		}
	}
	else {
		translator_init_font(false, "font.ttf", 1.0);
	}
}

// Returns a list of supported locales (plus English and the automatically detected system locale)
string_array_t *translator_get_supported_locales() {
	string_array_t *locales = any_array_create_from_raw(
	    (void *[]){
	        "system",
	        "en",
	    },
	    2);
	string_array_t *files = file_read_directory(string("%s%slocale", path_data(), PATH_SEP));
	for (i32 i = 0; i < files->length; ++i) {
		char *locale_filename = files->buffer[i];
		// Trim the ".json" file extension from file names
		any_array_push(locales, substring(locale_filename, 0, string_length(locale_filename) - 5));
	}
	return locales;
}