Added custom languages

Author: johnklipi

PolyMod.csproj

@@ -11,7 +11,7 @@
         </RestoreAdditionalProjectSources>
         <Configurations>IL2CPP</Configurations>
         <RootNamespace>PolyMod</RootNamespace>
-        <Version>1.2.11</Version>
+        <Version>1.2.11-pre</Version>
         <PolytopiaVersion>2.16.4.15698</PolytopiaVersion>
         <Authors>PolyModdingTeam</Authors>
         <Description>The Battle of Polytopia's mod loader.</Description>

src/Loader.cs

@@ -51,11 +51,6 @@ internal record TypeMapping(Type type, bool shouldCreateCache = true);
 	/// </summary>
 	internal static List<GameModeButtonsInformation> gamemodes = new();
 
-	/// <summary>
-	/// A dictionary of skin types of tribes, which can flood tiles, keyed by custom flood tile effect.
-	/// </summary>
-	internal static Dictionary<TileData.EffectType, SkinType> customFloodingSkins = new();
-
 	/// <summary>
 	/// Handlers for processing specific data types during mod loading.
 	/// </summary>
@@ -105,7 +100,7 @@ out TribeAbility.Type tribeAbilityType
 											string tileEffectId = skinId.ToLowerInvariant() + "_flood";
 											EnumCache<TileData.EffectType>.AddMapping(tileEffectId, (TileData.EffectType)Registry.autoidx);
 											EnumCache<TileData.EffectType>.AddMapping(tileEffectId, (TileData.EffectType)Registry.autoidx);
-                                            customFloodingSkins.Add((TileData.EffectType)Registry.autoidx, skinValue);
+                                            Visual.customFloodingSkins.Add((TileData.EffectType)Registry.autoidx, skinValue);
 											Plugin.logger.LogInfo("Created mapping for tileEffect with id " + tileEffectId + " and index " + Registry.autoidx);
 											Registry.autoidx++;
 										}
@@ -407,6 +402,17 @@ internal static void LoadMods(Dictionary<string, Mod> mods, out bool dependencyC
 				{
 					LoadSpriteInfoFile(mod, file);
 				}
+				Match languageMatch = Regex.Match(Path.GetFileName(file.name), @"^language(?:_(.*))?\.json$");
+				if (languageMatch.Success)
+				{
+					string languageName = languageMatch.Groups[1].Value;
+					LoadLanguageFile(
+						mod,
+						file,
+						languageName
+					);
+					continue;
+				}
 			}
 			if (!mod.client && id != "polytopia")
 			{
@@ -589,6 +595,23 @@ public static void LoadLocalizationFile(Mod mod, Mod.File file)
 		}
 	}
 
+	public static void LoadLanguageFile(Mod mod, Mod.File file, string languageName)
+	{
+		try
+		{
+			var dict = JsonSerializer.Deserialize<Dictionary<string, string>>(file.bytes);
+			if(dict != null)
+			{
+				Loc.languagesToAdd.Add(languageName, dict);
+				Plugin.logger.LogInfo($"Registered language from {mod.id} mod");
+			}
+		}
+		catch (Exception e)
+		{
+			Plugin.logger.LogError($"Error on loading language file from {mod.id} mod: {e.StackTrace}");
+		}
+	}
+
 	/// <summary>
 	/// Loads a sprite file from a mod.
 	/// </summary>

src/Managers/Loc.cs

@@ -11,6 +11,7 @@ namespace PolyMod.Managers;
 /// </summary>
 public static class Loc
 {
+	internal static Dictionary<string, Dictionary<string, string>> languagesToAdd = new();
 	/// <summary>
 	/// Patches the localization getter to handle custom enum values.
 	/// </summary>
@@ -89,6 +90,48 @@ public static void BuildAndLoadLocalization(Dictionary<string, Dictionary<string
 		}
 	}
 
+	[HarmonyPrefix]
+	[HarmonyPatch(typeof(Localization), nameof(Localization.Init))]
+	public static bool Localization_Init()
+	{
+		if (Localization.initialized)
+			return true;
+
+		if (LocalizationManager.Sources.Count == 0)
+			LocalizationManager.UpdateSources();
+
+		foreach(var name in languagesToAdd.Keys)
+		{
+			Dictionary<string, string> terms = languagesToAdd[name];
+
+			LanguageSourceData source = LocalizationManager.Sources[0];
+			int languageIndex = source.GetLanguageIndex(name);
+			string languageName = terms["language"];
+			if (languageIndex == -1)
+			{
+				source.AddLanguage(languageName, name);
+				languageIndex = source.GetLanguageIndex(languageName);
+			}
+
+			foreach (var kvp in terms)
+			{
+				TermData term = source.GetTermData(kvp.Key);
+				if (term == null)
+				{
+					source.AddTerm(kvp.Key);
+					term = source.GetTermData(kvp.Key);
+				}
+
+				term.Languages[languageIndex] = kvp.Value;
+			}
+
+			LocalizationManager.UpdateSources();
+
+			Plugin.logger.LogInfo($"{name} language added and loaded!");
+		}
+		return true;
+	}
+
 	/// <summary>
 	/// Initializes the Loc manager by patching the necessary methods.
 	/// </summary>

src/Managers/Main.cs

@@ -388,12 +388,14 @@ internal static void Load(GameLogicData gameLogicData, JObject json)
 			if (mod.status != Mod.Status.Success) continue;
 			foreach (var file in mod.files)
 			{
+				if (mod.status != Mod.Status.Success) break;
 				if (Path.GetFileName(file.name) == "localization.json")
 				{
 					Loader.LoadLocalizationFile(mod, file);
 					continue;
 				}
-				if (Regex.IsMatch(Path.GetFileName(file.name), @"^patch(_.*)?\.json$"))
+				Match patchMatch = Regex.Match(Path.GetFileName(file.name), @"^patch(_.*)?\.json$");
+				if (patchMatch.Success)
 				{
 					var patchText = new StreamReader(new MemoryStream(file.bytes)).ReadToEnd();
 					var template = new Api.GldConfigTemplate(patchText, mod.id);
@@ -410,7 +412,8 @@ internal static void Load(GameLogicData gameLogicData, JObject json)
 					);
 					continue;
 				}
-				if (Regex.IsMatch(Path.GetFileName(file.name), @"^prefab(_.*)?\.json$"))
+				Match prefabMatch = Regex.Match(Path.GetFileName(file.name), @"^prefab(?:_(.*))?\.json$");
+				if (prefabMatch.Success)
 				{
 					Loader.LoadPrefabInfoFile(
 						mod,

src/Managers/Visual.cs

@@ -70,6 +70,10 @@ public UnitPrefabInfo(string type, string tribe, string skin)
 	private static bool firstTimeOpeningPreview = true;
 	private static UnitData.Type currentUnitTypeUI = UnitData.Type.None;
 	private static TribeType attackerTribe = TribeType.None;
+	/// <summary>
+	/// A dictionary of skin types of tribes, which can flood tiles, keyed by custom flood tile effect.
+	/// </summary>
+	internal static Dictionary<TileData.EffectType, SkinType> customFloodingSkins = new();
 
 	/// <summary>The type of a custom prefab.</summary>
 	public enum PrefabType
@@ -312,9 +316,9 @@ private static void TerrainRenderer_UpdateGraphics(TerrainRenderer __instance, T
 		{
 			foreach (var effect in tile.data.effects)
 			{
-				if(Loader.customFloodingSkins.ContainsKey(effect))
+				if(customFloodingSkins.ContainsKey(effect))
 				{
-					skinType = Loader.customFloodingSkins[effect];
+					skinType = customFloodingSkins[effect];
 					break;
 				}
 			}
@@ -355,9 +359,9 @@ private static void TerrainRenderer_UpdateGraphics(TerrainRenderer __instance, T
 	[HarmonyPatch(typeof(TileData), nameof(TileData.Flood))]
 	private static void TileData_Flood(TileData __instance, PlayerState playerState)
 	{
-		if(Loader.customFloodingSkins.ContainsValue(playerState.skinType))
+		if(customFloodingSkins.ContainsValue(playerState.skinType))
 		{
-			TileData.EffectType effectType = Loader.customFloodingSkins.FirstOrDefault(x => x.Value == playerState.skinType).Key;
+			TileData.EffectType effectType = customFloodingSkins.FirstOrDefault(x => x.Value == playerState.skinType).Key;
 			__instance.AddEffect(effectType);	
 		}
 	}