Continued

2026-06-04 00:06:18 -07:00
parent 8e9b4412ce
commit 5c11ff4fad
42 changed files with 1287 additions and 29 deletions
@@ -0,0 +1,92 @@
+using ProjectM.Simulation;
+using Unity.Burst;
+using Unity.Collections;
+using Unity.Entities;
+using Unity.NetCode;
+
+namespace ProjectM.Server
+{
+    /// <summary>
+    /// Server-authoritative ability-damage upgrade (handles <see cref="AbilityUpgradeRequest"/> RPCs). Resolves
+    /// the sender's player (SourceConnection -&gt; NetworkId -&gt; GhostOwner) and, if the global ledger affords the
+    /// Aether cost, withdraws it IN-PLACE and grows a single damage <see cref="StatModifier"/> on the player
+    /// (replace-by-SourceId so the [InternalBufferCapacity(8)] buffer stays bounded — repeated upgrades grow one
+    /// row's percent rather than appending). StatRecomputeSystem folds it into EffectiveAbilityStats.Damage on
+    /// both worlds. Plain server SimulationSystemGroup (not predicted → applied once).
+    /// </summary>
+    [BurstCompile]
+    [WorldSystemFilter(WorldSystemFilterFlags.ServerSimulation)]
+    public partial struct AbilityUpgradeSystem : ISystem
+    {
+        const uint UpgradeSourceId = 0x00A0E711u; // distinct sentinel so the upgrade modifier is found + grown
+        const float TierStep = 0.25f;             // +25% damage per tier
+        const int CostAmount = 20;                // Aether per tier
+
+        [BurstCompile]
+        public void OnCreate(ref SystemState state)
+        {
+            state.RequireForUpdate<ResourceLedger>();
+            var builder = new EntityQueryBuilder(Allocator.Temp)
+                .WithAll<AbilityUpgradeRequest, ReceiveRpcCommandRequest>();
+            state.RequireForUpdate(state.GetEntityQuery(builder));
+        }
+
+        [BurstCompile]
+        public void OnUpdate(ref SystemState state)
+        {
+            var ledger = SystemAPI.GetBuffer<StorageEntry>(SystemAPI.GetSingletonEntity<ResourceLedger>());
+
+            var playerByConn = new NativeHashMap<int, Entity>(8, Allocator.Temp);
+            foreach (var (owner, entity) in
+                     SystemAPI.Query<RefRO<GhostOwner>>().WithAll<PlayerTag, StatModifier>().WithEntityAccess())
+                playerByConn[owner.ValueRO.NetworkId] = entity;
+
+            var ecb = new EntityCommandBuffer(Allocator.Temp);
+
+            foreach (var (receive, requestEntity) in
+                     SystemAPI.Query<RefRO<ReceiveRpcCommandRequest>>().WithAll<AbilityUpgradeRequest>().WithEntityAccess())
+            {
+                var conn = receive.ValueRO.SourceConnection;
+                if (SystemAPI.HasComponent<NetworkId>(conn)
+                    && playerByConn.TryGetValue(SystemAPI.GetComponent<NetworkId>(conn).Value, out var player))
+                {
+                    int have = 0;
+                    for (int i = 0; i < ledger.Length; i++)
+                        if (ledger[i].ItemId == ResourceId.Aether) { have = ledger[i].Count; break; }
+
+                    if (have >= CostAmount)
+                    {
+                        StorageMath.Withdraw(ledger, ResourceId.Aether, CostAmount);
+
+                        var mods = SystemAPI.GetBuffer<StatModifier>(player);
+                        bool grown = false;
+                        for (int i = 0; i < mods.Length; i++)
+                        {
+                            if (mods[i].SourceId == UpgradeSourceId && mods[i].Target == (byte)StatTarget.Damage)
+                            {
+                                var m = mods[i];
+                                m.Value += TierStep;
+                                mods[i] = m;
+                                grown = true;
+                                break;
+                            }
+                        }
+                        if (!grown)
+                            mods.Add(new StatModifier
+                            {
+                                Target = (byte)StatTarget.Damage,
+                                Op = (byte)ModOp.PercentAdd,
+                                Value = TierStep,
+                                SourceId = UpgradeSourceId,
+                            });
+                    }
+                }
+
+                ecb.DestroyEntity(requestEntity);
+            }
+
+            ecb.Playback(state.EntityManager);
+            playerByConn.Dispose();
+        }
+    }
+}