crystall-punk-14/Content.Server/Traitor/Uplink/SurplusBundle/SurplusBundleSystem.cs

using System.Linq;
using Content.Server.Storage.EntitySystems;
using Content.Server.Store.Systems;
using Content.Shared.FixedPoint;
using Content.Shared.Store;
using Robust.Shared.Prototypes;
using Robust.Shared.Random;

namespace Content.Server.Traitor.Uplink.SurplusBundle;

public sealed class SurplusBundleSystem : EntitySystem
{
    [Dependency] private readonly IPrototypeManager _prototypeManager = default!;
    [Dependency] private readonly IRobustRandom _random = default!;
    [Dependency] private readonly EntityStorageSystem _entityStorage = default!;
    [Dependency] private readonly StoreSystem _store = default!;

    private ListingData[] _listings = default!;

    public override void Initialize()
    {
        base.Initialize();
        SubscribeLocalEvent<SurplusBundleComponent, MapInitEvent>(OnMapInit);

        SubscribeLocalEvent<SurplusBundleComponent, ComponentInit>(OnInit);
    }

    private void OnInit(EntityUid uid, SurplusBundleComponent component, ComponentInit args)
    {
        var storePreset = _prototypeManager.Index<StorePresetPrototype>(component.StorePreset);

        _listings = _store.GetAvailableListings(uid, null, storePreset.Categories).ToArray();

        Array.Sort(_listings, (a, b) => (int) (b.Cost.Values.Sum() - a.Cost.Values.Sum())); //this might get weird with multicurrency but don't think about it
    }

    private void OnMapInit(EntityUid uid, SurplusBundleComponent component, MapInitEvent args)
    {
        FillStorage(uid, component);
    }

    private void FillStorage(EntityUid uid, SurplusBundleComponent? component = null)
    {
        if (!Resolve(uid, ref component))
            return;

        var cords = Transform(uid).Coordinates;

        var content = GetRandomContent(component.TotalPrice);
        foreach (var item in content)
        {
            var ent = EntityManager.SpawnEntity(item.ProductEntity, cords);
            _entityStorage.Insert(ent, uid);
        }
    }

    // wow, is this leetcode reference?
    private List<ListingData> GetRandomContent(FixedPoint2 targetCost)
    {
        var ret = new List<ListingData>();
        if (_listings.Length == 0)
            return ret;

        var totalCost = FixedPoint2.Zero;
        var index = 0;
        while (totalCost < targetCost)
        {
            // All data is sorted in price descending order
            // Find new item with the lowest acceptable price
            // All expansive items will be before index, all acceptable after
            var remainingBudget = targetCost - totalCost;
            while (_listings[index].Cost.Values.Sum() > remainingBudget)
            {
                index++;
                if (index >= _listings.Length)
                {
                    // Looks like no cheap items left
                    // It shouldn't be case for ss14 content
                    // Because there are 1 TC items
                    return ret;
                }
            }

            // Select random listing and add into crate
            var randomIndex = _random.Next(index, _listings.Length);
            var randomItem = _listings[randomIndex];
            ret.Add(randomItem);
            totalCost += randomItem.Cost.Values.Sum();
        }

        return ret;
    }
}