Добавьте файлы проекта.

2025-10-26 17:02:04 +03:00
parent 842a1a2c70
commit caa830ba06
12 changed files with 1136 additions and 0 deletions
--- a/Source/AIImages/PawnGizmoPatch.cs
+++ b/Source/AIImages/PawnGizmoPatch.cs
@@ -0,0 +1,71 @@
+using System.Collections.Generic;
+using System.Linq;
+using HarmonyLib;
+using RimWorld;
+using UnityEngine;
+using Verse;
+
+#pragma warning disable IDE1006 // Naming Styles
+
+namespace AIImages
+{
+    /// <summary>
+    /// Harmony patch to add a gizmo (button) to all colonist pawns
+    /// </summary>
+    [HarmonyPatch(typeof(Pawn), "GetGizmos")]
+    [System.Diagnostics.CodeAnalysis.SuppressMessage(
+        "Style",
+        "IDE1006:Naming Styles",
+        Justification = "RimWorld Harmony patch naming convention"
+    )]
+    [System.Diagnostics.CodeAnalysis.SuppressMessage(
+        "Minor Code Smell",
+        "S101:Types should be named in PascalCase",
+        Justification = "RimWorld Harmony patch naming convention"
+    )]
+    public static class Pawn_GetGizmos_Patch
+    {
+        [HarmonyPostfix]
+        public static IEnumerable<Gizmo> Postfix(IEnumerable<Gizmo> __result, Pawn __instance)
+        {
+            // First, return all original gizmos
+            foreach (Gizmo gizmo in __result)
+            {
+                yield return gizmo;
+            }
+
+            // Only add button to colonist pawns that are spawned
+            if (
+                __instance.IsColonist
+                && __instance.Spawned
+                && __instance.Faction == Faction.OfPlayer
+            )
+            {
+                yield return new Command_Action
+                {
+                    defaultLabel = "AI Image",
+                    defaultDesc = "Open AI Image window",
+                    icon = ContentFinder<Texture2D>.Get("UI/Commands/AttackMelee", true),
+                    action = delegate()
+                    {
+                        // Проверяем, открыто ли уже окно AI Image
+                        Window_AIImage existingWindow = Find
+                            .WindowStack.Windows.OfType<Window_AIImage>()
+                            .FirstOrDefault();
+
+                        if (existingWindow != null)
+                        {
+                            // Если окно открыто, обновляем пешку
+                            existingWindow.UpdatePawn(__instance);
+                        }
+                        else
+                        {
+                            // Если окна нет, создаём новое
+                            Find.WindowStack.Add(new Window_AIImage(__instance));
+                        }
+                    },
+                };
+            }
+        }
+    }
+}