aboutsummaryrefslogtreecommitdiff
path: root/extension/src/openvic-extension/LoadLocalisation.cpp
blob: 418f01ca42f9097b90a5a7a0d1ddd342e2dfd08b (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#include "LoadLocalisation.hpp"

#include <godot_cpp/classes/dir_access.hpp>
#include <godot_cpp/classes/file_access.hpp>
#include <godot_cpp/classes/translation_server.hpp>
#include <godot_cpp/variant/utility_functions.hpp>

#include "openvic-extension/Utilities.hpp"

using namespace godot;
using namespace OpenVic;

LoadLocalisation* LoadLocalisation::singleton = nullptr;

void LoadLocalisation::_bind_methods() {
   ClassDB::bind_method(D_METHOD("load_file", "file_path", "locale"), &LoadLocalisation::load_file);
   ClassDB::bind_method(D_METHOD("load_locale_dir", "dir_path", "locale"), &LoadLocalisation::load_locale_dir);
   ClassDB::bind_method(D_METHOD("load_localisation_dir", "dir_path"), &LoadLocalisation::load_localisation_dir);
}

LoadLocalisation* LoadLocalisation::get_singleton() {
   return singleton;
}

LoadLocalisation::LoadLocalisation() {
   ERR_FAIL_COND(singleton != nullptr);
   singleton = this;
}

LoadLocalisation::~LoadLocalisation() {
   ERR_FAIL_COND(singleton != this);
   singleton = nullptr;
}

Error LoadLocalisation::_load_file_into_translation(String const& file_path, Ref<Translation> translation) const {
   const Ref<FileAccess> file = FileAccess::open(file_path, FileAccess::ModeFlags::READ);
   Error err = FileAccess::get_open_error();
   if (err != OK || file.is_null()) {
      UtilityFunctions::push_error("Failed to load localisation file: ", file_path);
      return err == OK ? FAILED : err;
   }
   int line_number = 0;
   while (!file->eof_reached()) {
      static const String delimeter = ";";
      const PackedStringArray line = file->get_csv_line(delimeter);
      line_number++;
      if (line.size() < 2 || line[0].is_empty() || line[1].is_empty()) {
         if (!line[0].is_empty()) {
            UtilityFunctions::push_warning("Key \"", line[0], "\" missing value on line ", line_number, " in file: ", file_path);
            err = FAILED;
         } else if (line.size() >= 2 && !line[1].is_empty()) {
            UtilityFunctions::push_warning("Value \"", line[1], "\" missing key on line ", line_number, " in file: ", file_path);
            err = FAILED;
         }
         continue;
      }
      translation->add_message(line[0], line[1].c_unescape());
   }
   return err;
}

Ref<Translation> LoadLocalisation::_get_translation(String const& locale) const {
   TranslationServer* server = TranslationServer::get_singleton();
   if (server == nullptr) {
      UtilityFunctions::push_error("Failed to get TranslationServer singleton");
      return nullptr;
   }
   Ref<Translation> translation = server->get_translation_object(locale);
   if (translation.is_null() || translation->get_locale() != locale) {
      translation.instantiate();
      translation->set_locale(locale);
      server->add_translation(translation);
   }
   return translation;
}

Error LoadLocalisation::load_file(String const& file_path, String const& locale) const {
   return _load_file_into_translation(file_path, _get_translation(locale));
}

/* REQUIREMENTS
 * FS-18, FS-24, FS-25
 */
Error LoadLocalisation::load_locale_dir(String const& dir_path, String const& locale) const {
   if (!DirAccess::dir_exists_absolute(dir_path)) {
      UtilityFunctions::push_error("Locale directory does not exist: ", dir_path);
      return FAILED;
   }
   /* This will add the locale to the list of loaded locales even if it has no
    * localisation files - this is useful for testing other aspects of localisation
    * such as number formatting and text direction. To disable this behaviour and
    * only show non-empty localisations, move the `_get_translation` call to after
    * the `files.size()` check.
    */
   const Ref<Translation> translation = _get_translation(locale);
   const PackedStringArray files = DirAccess::get_files_at(dir_path);
   if (files.size() < 1) {
      UtilityFunctions::push_error("Locale directory does not contain any files: ", dir_path);
      return FAILED;
   }
   Error err = OK;
   for (String const& file_name : files) {
      if (file_name.get_extension().to_lower() == "csv") {
         if (_load_file_into_translation(dir_path.path_join(file_name), translation) != OK)
            err = FAILED;
      }
   }
   return err;
}

/* REQUIREMENTS
 * FS-23
 */
Error LoadLocalisation::load_localisation_dir(String const& dir_path) const {
   if (!DirAccess::dir_exists_absolute(dir_path)) {
      UtilityFunctions::push_error("Localisation directory does not exist: ", dir_path);
      return FAILED;
   }
   PackedStringArray const dirs = DirAccess::get_directories_at(dir_path);
   if (dirs.size() < 1) {
      UtilityFunctions::push_error("Localisation directory does not contain any sub-directories: ", dir_path);
      return FAILED;
   }
   TranslationServer* server = TranslationServer::get_singleton();
   if (server == nullptr) {
      UtilityFunctions::push_error("Failed to get TranslationServer singleton");
      return FAILED;
   }
   Error err = OK;
   for (String const& locale_name : dirs) {
      if (locale_name != server->standardize_locale(locale_name))
         UtilityFunctions::push_error("Invalid locale directory name: ", locale_name);
      else if (load_locale_dir(dir_path.path_join(locale_name), locale_name) == OK)
         continue;
      err = FAILED;
   }
   return err;
}
bool LoadLocalisation::add_message(std::string_view key, Dataloader::locale_t locale, std::string_view localisation) {
   static Ref<Translation> translations[Dataloader::_LocaleCount] = { nullptr };
   Ref<Translation>& translation = translations[locale];
   if (translation.is_null()) {
      translation = singleton->_get_translation(Dataloader::locale_names[locale]);
      if (translation.is_null()) {
         UtilityFunctions::push_error("Failed to get translation object: ", Dataloader::locale_names[locale]);
         return false;
      }
   }
   const StringName godot_key = Utilities::std_view_to_godot_string(key);
   const StringName godot_localisation = Utilities::std_view_to_godot_string(localisation);
   if (0) {
      const StringName old_localisation = translation->get_message(godot_key);
      if (!old_localisation.is_empty()) {
         UtilityFunctions::push_warning("Changing translation ", godot_key, " (", Dataloader::locale_names[locale], ") from \"", old_localisation, "\" to \"", godot_localisation, "\"");
      }
   }
   translation->add_message(godot_key, godot_localisation);
   return true;
}