code restructure

2024-07-02 00:54:27 -07:00
parent 799fc6fa94
commit 89656fa468
6 changed files with 0 additions and 0 deletions
--- a/ContextMenus/InsuranceInteractions.cs
+++ b/ContextMenus/InsuranceInteractions.cs
@@ -0,0 +1,105 @@
+using Comfort.Common;
+using EFT.InventoryLogic;
+using EFT.UI;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace UIFixes
+{
+    public class InsuranceInteractions(IEnumerable<Item> items, ItemUiContext uiContext, int playerRubles) : ItemInfoInteractionsAbstractClass<InsuranceInteractions.EInsurers>(uiContext)
+    {
+        private readonly InsuranceCompanyClass insurance = uiContext.Session.InsuranceCompany;
+        private readonly List<Item> items = items.ToList();
+        private readonly int playerRubles = playerRubles;
+        private List<ItemClass> insurableItems;
+        private readonly Dictionary<string, int> prices = [];
+
+        public InsuranceInteractions(Item item, ItemUiContext uiContext, int playerRubles) : this([item], uiContext, playerRubles) { }
+
+        public void LoadAsync(Action callback)
+        {
+            IEnumerable<ItemClass> itemClasses = items.Select(ItemClass.FindOrCreate);
+            insurableItems = itemClasses.SelectMany(insurance.GetItemChildren)
+                .Flatten(insurance.GetItemChildren)
+                .Concat(itemClasses)
+                .Where(i => insurance.ItemTypeAvailableForInsurance(i) && !insurance.InsuredItems.Contains(i))
+                .ToList();
+
+            insurance.GetInsurePriceAsync(insurableItems, _ =>
+            {
+                foreach (var insurer in insurance.Insurers)
+                {
+                    int price = this.insurableItems.Select(i => insurance.InsureSummary[insurer.Id][i]).Where(s => s.Loaded).Sum(s => s.Amount);
+                    prices[insurer.Id] = price;
+
+                    string priceColor = price > playerRubles ? "#FF0000" : "#ADB8BC";
+
+                    string text = string.Format("<b><color=#C6C4B2>{0}</color> <color={1}>({2} ₽)</color></b>", insurer.LocalizedName, priceColor, price);
+
+                    base.method_2(MakeInteractionId(insurer.Id), text, () => this.Insure(insurer.Id));
+                }
+
+                callback();
+            });
+        }
+
+        private void Insure(string insurerId)
+        {
+            insurance.SelectedInsurerId = insurerId;
+            insurance.InsureItems(this.insurableItems, result => { });
+        }
+
+        public IResult GetButtonInteraction(string interactionId)
+        {
+            string traderId = interactionId.Split(':')[1];
+            if (prices[traderId] > playerRubles)
+            {
+                return new FailedResult("ragfair/Not enough money", 0);
+            }
+
+            return SuccessfulResult.New;
+        }
+
+        public override void ExecuteInteractionInternal(EInsurers interaction)
+        {
+        }
+
+        public override bool IsActive(EInsurers button)
+        {
+            return button == EInsurers.None && !this.insurance.Insurers.Any();
+        }
+
+        public override IResult IsInteractive(EInsurers button)
+        {
+            return new FailedResult("No insurers?", 0);
+        }
+
+        public override bool HasIcons
+        {
+            get { return false; }
+        }
+
+        public enum EInsurers
+        {
+            None
+        }
+        private static string MakeInteractionId(string traderId)
+        {
+            return "UIFixesInsurerId:" + traderId;
+        }
+
+        public static bool IsInsuranceInteractionId(string id)
+        {
+            return id.StartsWith("UIFixesInsurerId:");
+        }
+    }
+
+    public static class InsuranceExtensions
+    {
+        public static bool IsInsuranceInteraction(this DynamicInteractionClass interaction)
+        {
+            return interaction.Id.StartsWith("UIFixesInsurerId:");
+        }
+    }
+}