Merge pull request #789 from The-International-Screeps-Bot/feature/ad…

…d-etrd-stat-tracking Feature add etrd stat tracking
The-International-Screeps-Bot · Apr 7, 2024 · 84a2aa0 · 84a2aa0
2 parents 54b0ded + 2863e47
commit 84a2aa0
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/src/constants/stats.ts b/src/constants/stats.ts
@@ -22,6 +22,7 @@ export enum RoomStatsKeys {
   MinHaulerCost = 'mhc',
   EnergyOutputTransactionCosts = 'eotc',
   EnergyTerminalSentDomestic = 'etsd',
+  EnergyTerminalReceivedDomestic = 'etrd',
   EnergyTerminalSentOther = 'etso',
   CpuUsed = 'cpu',
 
@@ -61,6 +62,7 @@ export const averageStatNames: Set<keyof CommuneStats | keyof RoomStats> = new S
   RoomStatsKeys.RemoteEnergyOutputBuild,
   RoomStatsKeys.EnergyOutputTransactionCosts,
   RoomStatsKeys.EnergyTerminalSentDomestic,
+  RoomStatsKeys.EnergyTerminalReceivedDomestic,
   RoomStatsKeys.EnergyTerminalSentOther,
   RoomStatsKeys.CpuUsed,
 ])
diff --git a/src/international/stats.ts b/src/international/stats.ts
@@ -43,6 +43,7 @@ export class StatsManager {
         [RoomStatsKeys.MinHaulerCost]: 0,
         [RoomStatsKeys.EnergyOutputTransactionCosts]: 0,
         [RoomStatsKeys.EnergyTerminalSentDomestic]: 0,
+        [RoomStatsKeys.EnergyTerminalReceivedDomestic]: 0,
         [RoomStatsKeys.EnergyTerminalSentOther]: 0,
         [RoomStatsKeys.CpuUsed]: 0,
       })

diff --git a/src/international/transactions.ts b/src/international/transactions.ts
@@ -90,6 +90,11 @@ export class TransactionsManager {
         RoomStatsKeys.EnergyTerminalSentDomestic,
         transaction.amount,
       )
+      StatsManager.updateCommuneStat(
+        transaction.to,
+        RoomStatsKeys.EnergyTerminalReceivedDomestic,
+        transaction.amount,
+      )
     }
     // Not a domestic trade
     else {

diff --git a/src/types/stats.ts b/src/types/stats.ts
@@ -29,6 +29,7 @@ export interface CommuneStats extends RoomStats {
     [RoomStatsKeys.MinHaulerCost]: number
     [RoomStatsKeys.EnergyOutputTransactionCosts]: number
     [RoomStatsKeys.EnergyTerminalSentDomestic]: number
+    [RoomStatsKeys.EnergyTerminalReceivedDomestic]: number
     [RoomStatsKeys.EnergyTerminalSentOther]: number
     [RoomStatsKeys.BatteriesStoredTimes10]: number
     [RoomStatsKeys.CpuUsed]: number