mirror of
https://github.com/Gurge44/EndlessHostRoles.git
synced 2026-10-20 01:46:25 +00:00
301 lines
12 KiB
C#
301 lines
12 KiB
C#
using HarmonyLib;
|
||
using Il2CppInterop.Runtime.InteropTypes.Arrays;
|
||
using System;
|
||
using System.Collections.Generic;
|
||
using System.Globalization;
|
||
using System.IO;
|
||
using System.Linq;
|
||
using System.Reflection;
|
||
using System.Text;
|
||
using System.Text.Json;
|
||
using System.Text.RegularExpressions;
|
||
using Object = Il2CppSystem.Object;
|
||
|
||
namespace TOHE;
|
||
|
||
public static class Translator
|
||
{
|
||
public static Dictionary<string, Dictionary<int, string>> translateMaps;
|
||
public const string LANGUAGE_FOLDER_NAME = "Language";
|
||
|
||
public static void Init()
|
||
{
|
||
Logger.Info("Loading Custom Translations...", "Translator");
|
||
LoadLangs();
|
||
Logger.Info("Loaded Custom Translations", "Translator");
|
||
}
|
||
|
||
public static void LoadLangs()
|
||
{
|
||
try
|
||
{
|
||
// Get the directory containing the JSON files (e.g., TOHE.Resources.Lang)
|
||
string jsonDirectory = "TOHE.Resources.Lang";
|
||
// Get the assembly containing the resources
|
||
var assembly = Assembly.GetExecutingAssembly();
|
||
string[] jsonFileNames = GetJsonFileNames(assembly, jsonDirectory);
|
||
|
||
translateMaps = [];
|
||
|
||
if (jsonFileNames.Length == 0)
|
||
{
|
||
Logger.Warn("Json Translation files does not exist.", "Translator");
|
||
return;
|
||
}
|
||
|
||
foreach (string jsonFileName in jsonFileNames)
|
||
{
|
||
// Read the JSON file content
|
||
using Stream resourceStream = assembly.GetManifestResourceStream(jsonFileName);
|
||
if (resourceStream != null)
|
||
{
|
||
using StreamReader reader = new(resourceStream);
|
||
string jsonContent = reader.ReadToEnd();
|
||
|
||
// Deserialize the JSON into a dictionary
|
||
Dictionary<string, string> jsonDictionary = JsonSerializer.Deserialize<Dictionary<string, string>>(jsonContent);
|
||
if (jsonDictionary.TryGetValue("LanguageID", out string languageIdObj) && int.TryParse(languageIdObj, out int languageId))
|
||
{
|
||
// Remove the "LanguageID" entry
|
||
jsonDictionary.Remove("LanguageID");
|
||
|
||
// Handle the rest of the data and merge it into the resulting translation map
|
||
MergeJsonIntoTranslationMap(translateMaps, languageId, jsonDictionary);
|
||
}
|
||
else
|
||
{
|
||
//Logger.Warn(jsonDictionary["HostText"], "Translator");
|
||
Logger.Warn($"Invalid JSON format in {jsonFileName}: Missing or invalid 'LanguageID' field.", "Translator");
|
||
}
|
||
}
|
||
}
|
||
|
||
// Convert the resulting translation map to JSON
|
||
string mergedJson = JsonSerializer.Serialize(translateMaps, new JsonSerializerOptions
|
||
{
|
||
WriteIndented = true
|
||
});
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
Logger.Error($"Error: {ex}", "Translator");
|
||
}
|
||
|
||
// Loading custom translation files
|
||
if (!Directory.Exists(LANGUAGE_FOLDER_NAME)) Directory.CreateDirectory(LANGUAGE_FOLDER_NAME);
|
||
|
||
// Creating a translation template
|
||
CreateTemplateFile();
|
||
foreach (var lang in EnumHelper.GetAllValues<SupportedLangs>())
|
||
{
|
||
if (File.Exists(@$"./{LANGUAGE_FOLDER_NAME}/{lang}.dat"))
|
||
{
|
||
UpdateCustomTranslation($"{lang}.dat" /*, lang*/);
|
||
LoadCustomTranslation($"{lang}.dat", lang);
|
||
}
|
||
}
|
||
}
|
||
|
||
static void MergeJsonIntoTranslationMap(Dictionary<string, Dictionary<int, string>> translationMaps, int languageId, Dictionary<string, string> jsonDictionary)
|
||
{
|
||
foreach (var kvp in jsonDictionary)
|
||
{
|
||
string textString = kvp.Key;
|
||
if (kvp.Value is string translation)
|
||
{
|
||
// If the textString is not already in the translation map, add it
|
||
if (!translationMaps.ContainsKey(textString))
|
||
{
|
||
translationMaps[textString] = [];
|
||
}
|
||
|
||
// Add or update the translation for the current id and textString
|
||
translationMaps[textString][languageId] = translation.Replace("\\n", "\n").Replace("\\r", "\r");
|
||
}
|
||
}
|
||
}
|
||
|
||
// Function to get a list of JSON file names in a directory
|
||
static string[] GetJsonFileNames(Assembly assembly, string directoryName)
|
||
{
|
||
string[] resourceNames = assembly.GetManifestResourceNames();
|
||
return resourceNames.Where(resourceName => resourceName.StartsWith(directoryName) && resourceName.EndsWith(".json")).ToArray();
|
||
}
|
||
|
||
public static string GetString(string s, Dictionary<string, string> replacementDic = null, bool console = false)
|
||
{
|
||
var langId = TranslationController.InstanceExists ? TranslationController.Instance.currentLanguage.languageID : SupportedLangs.English;
|
||
if (console) langId = SupportedLangs.English;
|
||
if (Main.ForceOwnLanguage.Value) langId = GetUserTrueLang();
|
||
string str = GetString(s, langId);
|
||
if (replacementDic != null)
|
||
foreach (var rd in replacementDic)
|
||
{
|
||
str = str.Replace(rd.Key, rd.Value);
|
||
}
|
||
|
||
return str;
|
||
}
|
||
|
||
public static string GetString(string str, SupportedLangs langId)
|
||
{
|
||
var res = $"<INVALID:{str}>";
|
||
try
|
||
{
|
||
if (translateMaps.TryGetValue(str, out var dic) && (!dic.TryGetValue((int)langId, out res) || res == "" || (langId is not SupportedLangs.SChinese and not SupportedLangs.TChinese && Regex.IsMatch(res, @"[\u4e00-\u9fa5]") && res == GetString(str, SupportedLangs.SChinese)))) //strに該当する&無効なlangIdかresが空
|
||
{
|
||
res = langId == SupportedLangs.English ? $"*{str}" : GetString(str, SupportedLangs.English);
|
||
}
|
||
|
||
if (!translateMaps.ContainsKey(str)) //translateMapsにない場合、StringNamesにあれば取得する
|
||
{
|
||
var stringNames = EnumHelper.GetAllValues<StringNames>().Where(x => x.ToString() == str).ToArray();
|
||
if (stringNames.Length > 0)
|
||
res = GetString(stringNames.FirstOrDefault());
|
||
}
|
||
}
|
||
catch (Exception Ex)
|
||
{
|
||
Logger.Fatal($"Error oucured at [{str}] in String.csv", "Translator");
|
||
Logger.Error("Here was the error:\n" + Ex, "Translator");
|
||
}
|
||
|
||
return res;
|
||
}
|
||
|
||
public static string GetString(StringNames stringName)
|
||
=> DestroyableSingleton<TranslationController>.Instance.GetString(stringName, new Il2CppReferenceArray<Object>(0));
|
||
|
||
public static string GetRoleString(string str, bool forUser = true)
|
||
{
|
||
var CurrentLanguage = TranslationController.Instance.currentLanguage.languageID;
|
||
var lang = forUser ? CurrentLanguage : SupportedLangs.English;
|
||
if (Main.ForceOwnLanguageRoleName.Value)
|
||
lang = GetUserTrueLang();
|
||
|
||
return GetString(str, lang);
|
||
}
|
||
|
||
public static SupportedLangs GetUserTrueLang()
|
||
{
|
||
try
|
||
{
|
||
var name = CultureInfo.CurrentUICulture.Name;
|
||
if (name.StartsWith("en")) return SupportedLangs.English;
|
||
if (name.StartsWith("zh_CHT")) return SupportedLangs.TChinese;
|
||
if (name.StartsWith("zh")) return SupportedLangs.SChinese;
|
||
if (name.StartsWith("ru")) return SupportedLangs.Russian;
|
||
return TranslationController.Instance.currentLanguage.languageID;
|
||
}
|
||
catch
|
||
{
|
||
return SupportedLangs.English;
|
||
}
|
||
}
|
||
|
||
static void UpdateCustomTranslation(string filename /*, SupportedLangs lang*/)
|
||
{
|
||
string path = @$"./{LANGUAGE_FOLDER_NAME}/{filename}";
|
||
if (File.Exists(path))
|
||
{
|
||
Logger.Info("Updating Custom Translations", "UpdateCustomTranslation");
|
||
try
|
||
{
|
||
List<string> textStrings = [];
|
||
using (StreamReader reader = new(path, Encoding.GetEncoding("UTF-8")))
|
||
{
|
||
string line;
|
||
while ((line = reader.ReadLine()) != null)
|
||
{
|
||
// Split the line by ':' to get the first part
|
||
string[] parts = line.Split(':');
|
||
|
||
// Check if there is at least one part before ':'
|
||
if (parts.Length >= 1)
|
||
{
|
||
// Trim any leading or trailing spaces and add it to the list
|
||
string textString = parts[0].Trim();
|
||
textStrings.Add(textString);
|
||
}
|
||
}
|
||
}
|
||
|
||
var sb = new StringBuilder();
|
||
foreach (var templateString in translateMaps.Keys)
|
||
{
|
||
if (!textStrings.Contains(templateString)) sb.Append($"{templateString}:\n");
|
||
}
|
||
|
||
using FileStream fileStream = new(path, FileMode.Append, FileAccess.Write);
|
||
using StreamWriter writer = new(fileStream);
|
||
writer.WriteLine(sb.ToString());
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Logger.Error("An error occurred: " + e.Message, "Translator");
|
||
}
|
||
}
|
||
}
|
||
|
||
public static void LoadCustomTranslation(string filename, SupportedLangs lang)
|
||
{
|
||
string path = @$"./{LANGUAGE_FOLDER_NAME}/{filename}";
|
||
if (File.Exists(path))
|
||
{
|
||
Logger.Info($"Loading Custom Translation File：{filename}", "LoadCustomTranslation");
|
||
try
|
||
{
|
||
using StreamReader sr = new(path, Encoding.GetEncoding("UTF-8"));
|
||
string text;
|
||
string[] tmp = [];
|
||
while ((text = sr.ReadLine()) != null)
|
||
{
|
||
tmp = text.Split(":");
|
||
if (tmp.Length > 1 && tmp[1] != "")
|
||
{
|
||
try
|
||
{
|
||
translateMaps[tmp[0]][(int)lang] = tmp.Skip(1).Join(delimiter: ":").Replace("\\n", "\n").Replace("\\r", "\r");
|
||
}
|
||
catch (KeyNotFoundException)
|
||
{
|
||
Logger.Warn($"Invalid Key：{tmp[0]}", "LoadCustomTranslation");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
catch (ObjectDisposedException)
|
||
{
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Logger.Error(e.ToString(), "Translator.LoadCustomTranslation");
|
||
}
|
||
}
|
||
else
|
||
{
|
||
Logger.Error($"Custom Translation File Not Found：{filename}", "LoadCustomTranslation");
|
||
}
|
||
}
|
||
|
||
private static void CreateTemplateFile()
|
||
{
|
||
var sb = new StringBuilder();
|
||
foreach (var title in translateMaps) sb.Append($"{title.Key}:\n");
|
||
File.WriteAllText(@$"./{LANGUAGE_FOLDER_NAME}/template.dat", sb.ToString());
|
||
}
|
||
|
||
public static void ExportCustomTranslation()
|
||
{
|
||
LoadLangs();
|
||
var sb = new StringBuilder();
|
||
var lang = TranslationController.Instance.currentLanguage.languageID;
|
||
foreach (var title in translateMaps)
|
||
{
|
||
var text = title.Value.GetValueOrDefault((int)lang, "");
|
||
sb.Append($"{title.Key}:{text.Replace("\n", "\\n").Replace("\r", "\\r")}\n");
|
||
}
|
||
|
||
File.WriteAllText(@$"./{LANGUAGE_FOLDER_NAME}/export_{lang}.dat", sb.ToString());
|
||
}
|
||
} |