static void Postfix(Pawn_DraftController __instance, ref IEnumerable <Gizmo> __result) { try { Pawn pawn = __instance.pawn; if (pawn.Faction == Faction.OfPlayer) { List <Gizmo> l = null; if (Settings.ShowWeaponStorageButtonForPawns && WorldComp.HasStorages()) { l = new List <Gizmo>() { new Command_Action { icon = AssignUI.weaponStorageTexture, defaultLabel = "WeaponStorage.UseWeaponStorage".Translate(), activateSound = SoundDef.Named("Click"), action = delegate { Find.WindowStack.Add(new AssignUI(null, pawn)); } } }; l.AddRange(__result); } if (WorldComp.TryGetAssignedWeapons(__instance.pawn, out AssignedWeaponContainer weapons)) { if (l == null) { l = new List <Gizmo>(); l.AddRange(__result); } //if (pawn.equipment.Primary != null) // l.Add(CreateUnequipGizmo(pawn, weapons)); foreach (ThingWithComps weapon in weapons.Weapons) { bool isTool = Settings.IsTool(weapon); bool show = false; if (pawn.Drafted) { show = true; } else // Not drafted { if (isTool || Settings.ShowWeaponsWhenNotDrafted) { show = true; } } if (show) { show = pawn.equipment.Primary != weapon; } if (show) { l.Add(CreateEquipWeaponGizmo(weapon.def, pawn, delegate { HarmonyPatchUtil.EquipWeapon(weapon, pawn, weapons); weapons.SetLastThingUsed(pawn, weapon, false); })); } } } foreach (SharedWeaponFilter f in WorldComp.SharedWeaponFilter) { if (l == null) { l = new List <Gizmo>(); l.AddRange(__result); } f.UpdateFoundDefCache(); if (f.AssignedPawns.Contains(pawn)) { if (l == null) { l = new List <Gizmo>(); if (__result != null) { l.AddRange(__result); } } foreach (ThingDef d in f.AllowedDefs) { if (d != pawn.equipment.Primary?.def && f.FoundDefCacheContains(d)) { l.Add(CreateEquipWeaponGizmo(d, pawn, delegate { if (WorldComp.TryRemoveWeapon(d, f, false, out ThingWithComps weapon)) { HarmonyPatchUtil.EquipWeapon(weapon, pawn); f.UpdateDefCache(d); } }, "WeaponStorage.EquipShared")); }