a lot of refactor & new command getall

SerbiaStrong-220 · Feb 12, 2024 · af56c15 · af56c15
1 parent e71f146
commit af56c15
Showing 1 changed file with 34 additions and 14 deletions.
diff --git a/...er/SS220/aw_cargodolg/CargoDolgCommand.cs → ...20/CargoMoneyCommand/CargoMoneyCommand.cs b/...er/SS220/aw_cargodolg/CargoDolgCommand.cs → ...20/CargoMoneyCommand/CargoMoneyCommand.cs
@@ -7,6 +7,7 @@
 using Content.Server.Cargo.Systems;
 using Content.Server.Cargo.Components;
 using System.Linq;
+using System.Collections.Generic;
 
 namespace Content.Server.Cargo.Commands
 {
@@ -22,11 +23,16 @@ public sealed class CargoMoneyCommand : IConsoleCommand
 
         public void Execute(IConsoleShell shell, string argStr, string[] args)
         {
-            if (args.Length == 2)
+            if (args.Length == 1 && args[0] == "getall")
+            {
+                PrintCargoStationsInfo(shell);
+                return;
+            }
+            if (args.Length == 4)
             {
                 bool bSet = false;
 
-                if (int.TryParse(args[1], out var toAdd))
+                if (int.TryParse(args[2], out var toAdd))
                 {
                     switch (args[0])
                     {
@@ -42,45 +48,59 @@ public void Execute(IConsoleShell shell, string argStr, string[] args)
                             goto invalidArgs;
                     }
 
-                    ProccessMoney(shell, toAdd, bSet);
+                    ProccessMoney(shell, toAdd, bSet, args[1]);
                     return;
                 }
             }
         invalidArgs:
             shell.WriteLine("Expected invalid arguments!");
         }
+        private void PrintCargoStationsInfo(IConsoleShell shell)
+        {
+            var bankQuery = _entityManager.EntityQueryEnumerator<StationBankAccountComponent>();
+
+            while (bankQuery.MoveNext(out var uid, out var bankComp))
+            {
+                shell.WriteLine($"BankEntity: {uid.Id}, Values: {bankComp.Balance}");
+            }
+        }
 
         public CompletionResult GetCompletion(IConsoleShell shell, string[] args)
         {
             CompletionResult res = CompletionResult.Empty;
             switch (args.Length)
             {
                 case 1:
-                    res = CompletionResult.FromHint("set || add || rem");
+                    res = CompletionResult.FromHint("set || add || rem || getall");
                     break;
                 case 2:
+                    res = CompletionResult.FromHint("EntityUid");
+                    break;
+                case 3:
                     res = CompletionResult.FromHint("amount");
                     break;
             }
 
             return res;
         }
 
-        private void ProccessMoney(IConsoleShell shell, int money, bool bSet)
+        private void ProccessMoney(IConsoleShell shell, int money, bool bSet, string station)
         {
-            var cargoSystem = _entitySystemManager.GetEntitySystem<CargoSystem>();
-            var bankQuery = _entityManager.EntityQueryEnumerator<StationBankAccountComponent>();
+            if (EntityUid.TryParse(station, out var bankEnt) && _entityManager.TryGetComponent<StationBankAccountComponent>(bankEnt, out var bankComponent))
+            {
+                var cargoSystem = _entitySystemManager.GetEntitySystem<CargoSystem>();
 
-            bankQuery.MoveNext(out var owner, out var bankComponent);
-            if (!_entityManager.EntityExists(owner) || bankComponent is null)
-                return;
 
-            var currentMoney = bankComponent.Balance;
+                var currentMoney = bankComponent.Balance;
 
-            cargoSystem.UpdateBankAccount(owner, bankComponent, -currentMoney);
-            cargoSystem.UpdateBankAccount(owner, bankComponent, bSet ? money : currentMoney + money);
+                cargoSystem.UpdateBankAccount(bankEnt, bankComponent, -currentMoney);
+                cargoSystem.UpdateBankAccount(bankEnt, bankComponent, bSet ? money : currentMoney + money);
 
-            shell.WriteLine($"Successfully changed cargo's money to {bankComponent.Balance}");
+                shell.WriteLine($"Successfully changed EntityUid {station} cargo's money to {bankComponent.Balance}");
+
+                return;
+            }
+            shell.WriteError("Expected invalid EntityUid!");
         }
     }
 }