Feature/issue 26107

Content.Client/Store/Ui/StoreBoundUserInterface.cs

@@ -2,6 +2,7 @@
 using JetBrains.Annotations;
 using System.Linq;
 using Content.Shared.Store.Components;
+using Content.Shared.StoreDiscount.Components;
 using Robust.Shared.Prototypes;
 
 namespace Content.Client.Store.Ui;
@@ -20,6 +21,9 @@ public sealed class StoreBoundUserInterface : BoundUserInterface
     [ViewVariables]
     private HashSet<ListingData> _listings = new();
 
+    [ViewVariables]
+    private StoreDiscountData[] _discounts = default!;
+
     public StoreBoundUserInterface(EntityUid owner, Enum uiKey) : base(owner, uiKey)
     {
     }
@@ -68,8 +72,10 @@ protected override void UpdateState(BoundUserInterfaceState state)
         {
             case StoreUpdateState msg:
                 _listings = msg.Listings;
+                _discounts = msg.Discounts;
 
                 _menu?.UpdateBalance(msg.Balance);
+
                 UpdateListingsWithSearchFilter();
                 _menu?.SetFooterVisibility(msg.ShowFooter);
                 _menu?.UpdateRefund(msg.AllowRefund);
@@ -97,7 +103,7 @@ private void UpdateListingsWithSearchFilter()
             filteredListings.RemoveWhere(listingData => !ListingLocalisationHelpers.GetLocalisedNameOrEntityName(listingData, _prototypeManager).Trim().ToLowerInvariant().Contains(_search) &&
                                                         !ListingLocalisationHelpers.GetLocalisedDescriptionOrEntityDescription(listingData, _prototypeManager).Trim().ToLowerInvariant().Contains(_search));
         }
-        _menu.PopulateStoreCategoryButtons(filteredListings);
-        _menu.UpdateListing(filteredListings.ToList());
+        _menu.PopulateStoreCategoryButtons(filteredListings, _discounts);
+        _menu.UpdateListing(filteredListings.ToList(), _discounts);
     }
 }

Content.Client/Store/Ui/StoreListingControl.xaml

@@ -2,6 +2,8 @@
     <BoxContainer Margin="8,8,8,8" Orientation="Vertical">
         <BoxContainer Orientation="Horizontal">
             <Label Name="StoreItemName" HorizontalExpand="True" />
+            <Label Name="DiscountSubText"
+                   HorizontalAlignment="Right"/>
             <Button
                 Name="StoreItemBuyButton"
                 MinWidth="64"

Content.Client/Store/Ui/StoreListingControl.xaml.cs

@@ -21,7 +21,8 @@ public sealed partial class StoreListingControl : Control
 
     private readonly bool _hasBalance;
     private readonly string _price;
-    public StoreListingControl(ListingData data, string price, bool hasBalance, Texture? texture = null)
+    private readonly string _discount;
+    public StoreListingControl(ListingData data, string price, string discount, bool hasBalance, Texture? texture = null)
     {
         IoCManager.InjectDependencies(this);
         RobustXamlLoader.Load(this);
@@ -31,6 +32,7 @@ public StoreListingControl(ListingData data, string price, bool hasBalance, Text
         _data = data;
         _hasBalance = hasBalance;
         _price = price;
+        _discount = discount;
 
         StoreItemName.Text = ListingLocalisationHelpers.GetLocalisedNameOrEntityName(_data, _prototype);
         StoreItemDescription.SetMessage(ListingLocalisationHelpers.GetLocalisedDescriptionOrEntityDescription(_data, _prototype));
@@ -63,6 +65,7 @@ private void UpdateBuyButtonText()
         }
         else
         {
+            DiscountSubText.Text = _discount;
             StoreItemBuyButton.Text = _price;
         }
     }

Content.Client/Store/Ui/StoreMenu.xaml.cs

@@ -3,6 +3,7 @@
 using Content.Client.Message;
 using Content.Shared.FixedPoint;
 using Content.Shared.Store;
+using Content.Shared.StoreDiscount.Components;
 using Robust.Client.AutoGenerated;
 using Robust.Client.GameObjects;
 using Robust.Client.Graphics;
@@ -16,6 +17,8 @@ namespace Content.Client.Store.Ui;
 [GenerateTypedNameReferences]
 public sealed partial class StoreMenu : DefaultWindow
 {
+    private const string DiscountedCategoryPrototypeKey = "DiscountedItems";
+
     [Dependency] private readonly IEntityManager _entityManager = default!;
     [Dependency] private readonly IPrototypeManager _prototypeManager = default!;
 
@@ -31,6 +34,7 @@ public sealed partial class StoreMenu : DefaultWindow
     public string CurrentCategory = string.Empty;
 
     private List<ListingData> _cachedListings = new();
+    private StoreDiscountData[] _cachedDiscounts = Array.Empty<StoreDiscountData>();
 
     public StoreMenu()
     {
@@ -68,22 +72,32 @@ public void UpdateBalance(Dictionary<ProtoId<CurrencyPrototype>, FixedPoint2> ba
         WithdrawButton.Disabled = disabled;
     }
 
-    public void UpdateListing(List<ListingData> listings)
+    public void UpdateListing(List<ListingData> listings, StoreDiscountData[] discounts)
     {
         _cachedListings = listings;
+        _cachedDiscounts = discounts;
         UpdateListing();
     }
 
     public void UpdateListing()
     {
-        var sorted = _cachedListings.OrderBy(l => l.Priority).ThenBy(l => l.Cost.Values.Sum());
+        var sorted = _cachedListings.OrderBy(l => l.Priority)
+                                    .ThenBy(l => l.Cost.Values.Sum());
 
         // should probably chunk these out instead. to-do if this clogs the internet tubes.
         // maybe read clients prototypes instead?
         ClearListings();
+        var storeDiscounts = _cachedDiscounts.Where(x => x.Count > 0)
+                                      .ToDictionary(x => x.ListingId);
+
         foreach (var item in sorted)
         {
-            AddListingGui(item);
+            storeDiscounts.TryGetValue(item.ID, out var discountData);
+            if (discountData != null)
+            {
+                item.Categories.Add(DiscountedCategoryPrototypeKey);
+            }
+            AddListingGui(item, discountData);
         }
     }
 
@@ -114,13 +128,13 @@ private void OnRefundButtonDown(BaseButton.ButtonEventArgs args)
         OnRefundAttempt?.Invoke(args);
     }
 
-    private void AddListingGui(ListingData listing)
+    private void AddListingGui(ListingData listing, StoreDiscountData? discountData)
     {
         if (!listing.Categories.Contains(CurrentCategory))
             return;
 
         var listingPrice = listing.Cost;
-        var hasBalance = HasListingPrice(Balance, listingPrice);
+        var hasBalance = CanBuyListing(Balance, listingPrice, discountData);
 
         var spriteSys = _entityManager.EntitySysManager.GetEntitySystem<SpriteSystem>();
 
@@ -143,50 +157,98 @@ private void AddListingGui(ListingData listing)
             }
         }
 
-        var newListing = new StoreListingControl(listing, GetListingPriceString(listing), hasBalance, texture);
+        var (listingInStock, discount) = GetListingPriceString(listing, discountData);
+
+        var newListing = new StoreListingControl(listing, listingInStock, discount, hasBalance, texture);
         newListing.StoreItemBuyButton.OnButtonDown += args
             => OnListingButtonPressed?.Invoke(args, listing);
 
         StoreListingsContainer.AddChild(newListing);
     }
 
-    public bool HasListingPrice(Dictionary<ProtoId<CurrencyPrototype>, FixedPoint2> currency, Dictionary<ProtoId<CurrencyPrototype>, FixedPoint2> price)
+    public bool CanBuyListing(Dictionary<ProtoId<CurrencyPrototype>, FixedPoint2> currentBalance, Dictionary<ProtoId<CurrencyPrototype>, FixedPoint2> price, StoreDiscountData? discountData)
     {
-        foreach (var type in price)
+        foreach (var (currency, value) in price)
         {
-            if (!currency.ContainsKey(type.Key))
+            if (!currentBalance.ContainsKey(currency))
                 return false;
 
-            if (currency[type.Key] < type.Value)
+            var amount = value;
+            if (discountData != null && discountData.DiscountAmountByCurrency.TryGetValue(currency, out var discount))
+            {
+                amount -= discount;
+            }
+
+            if (currentBalance[currency] < amount)
                 return false;
         }
+
         return true;
     }
 
-    public string GetListingPriceString(ListingData listing)
+    private (string Price, string Discount) GetListingPriceString(ListingData listing, StoreDiscountData? discountData)
     {
         var text = string.Empty;
+        var discountMessage = string.Empty;
+
+        var discountMessageList = new List<(FixedPoint2 DiscountPercent, CurrencyPrototype CurrencyPrototype)>();
+
         if (listing.Cost.Count < 1)
             text = Loc.GetString("store-currency-free");
         else
         {
             foreach (var (type, amount) in listing.Cost)
             {
-                var currency = _prototypeManager.Index(type);
-                text += Loc.GetString("store-ui-price-display", ("amount", amount),
-                    ("currency", Loc.GetString(currency.DisplayName, ("amount", amount))));
+                var currency = _prototypeManager.Index<CurrencyPrototype>(type);
+
+                var totalAmount = amount;
+                if (discountData?.DiscountAmountByCurrency.TryGetValue(type, out var discountBy) != null
+                    && discountBy > 0)
+                {
+                    var discountPercent = (float)discountBy.Value / totalAmount.Value;
+                    discountMessageList.Add((discountPercent, currency));
+
+                    totalAmount -= discountBy;
+                }
+
+                text += Loc.GetString(
+                    "store-ui-price-display",
+                    ("amount", totalAmount),
+                    ("currency", Loc.GetString(currency.DisplayName, ("amount", totalAmount)))
+                );
+            }
+
+
+            if (discountMessageList.Count > 1)
+            {
+                var discountMessagesPerCurrency = discountMessageList.Select(
+                    x => Loc.GetString(
+                        "store-ui-discount-display-with-currency",
+                        ("amount", x.DiscountPercent.Value),
+                        ("currency", Loc.GetString(x.CurrencyPrototype.DisplayName))
+                    )
+                );
+                discountMessage = string.Join(',', discountMessagesPerCurrency);
+
+            }
+            else if (discountMessageList.Count == 1)
+            {
+                discountMessage = Loc.GetString(
+                    "store-ui-discount-display",
+                    ("amount", (discountMessageList[0].DiscountPercent.Value))
+                );
             }
         }
 
-        return text.TrimEnd();
+        return (text.TrimEnd(), discountMessage);
     }
 
     private void ClearListings()
     {
         StoreListingsContainer.Children.Clear();
     }
 
-    public void PopulateStoreCategoryButtons(HashSet<ListingData> listings)
+    public void PopulateStoreCategoryButtons(HashSet<ListingData> listings, StoreDiscountData[] discounts)
     {
         var allCategories = new List<StoreCategoryPrototype>();
         foreach (var listing in listings)
@@ -199,6 +261,12 @@ public void PopulateStoreCategoryButtons(HashSet<ListingData> listings)
             }
         }
 
+        if (discounts.Any(x => x.Count > 0))
+        {
+            var proto = _prototypeManager.Index<StoreCategoryPrototype>(DiscountedCategoryPrototypeKey);
+            allCategories.Add(proto);
+        }
+
         allCategories = allCategories.OrderBy(c => c.Priority).ToList();
 
         // This will reset the Current Category selection if nothing matches the search.

Content.Server/GameTicking/Rules/TraitorRuleSystem.cs

@@ -84,7 +84,7 @@ public bool MakeTraitor(EntityUid traitor, TraitorRuleComponent component, bool
             // creadth: we need to create uplink for the antag.
             // PDA should be in place already
             var pda = _uplink.FindUplinkTarget(traitor);
-            if (pda == null || !_uplink.AddUplink(traitor, startingBalance))
+            if (pda == null || !_uplink.AddUplink(traitor, startingBalance, giveDiscounts: true))
                 return false;
 
             // Give traitors their codewords and uplink code to keep in their character info menu