From 73b3e347fb6160e9b6ef339ce49dbf04e3626d7b Mon Sep 17 00:00:00 2001
From: DarkPhoenix <phoenix@mail.ru>
Date: Sat, 24 May 2025 11:28:56 +0200
Subject: [PATCH] Hardcode sov upgrade buffs onto their carrying items

---
 db_update.py   | 52 +++++++++++++++++++++++++++++++++++++++++++++-----
 eos/effects.py | 17 +++++++++++++++--
 2 files changed, 62 insertions(+), 7 deletions(-)

diff --git a/db_update.py b/db_update.py
index 55cb4f317..acb51dddf 100644
--- a/db_update.py
+++ b/db_update.py
@@ -643,13 +643,17 @@ def update_db():
             else:
                 attr.value = value
 
-    def _hardcodeEffects(typeID, effectMap):
+    def _hardcodeEffects(typeID, effectMap, clearEffects=True):
         item = eos.db.gamedata_session.query(eos.gamedata.Item).filter(eos.gamedata.Item.ID == typeID).one()
-        item.effects.clear()
+        if clearEffects:
+            item.effects.clear()
         for effectID, effectName in effectMap.items():
-            effect = eos.gamedata.Effect()
-            effect.effectID = effectID
-            effect.effectName = effectName
+            try:
+                effect = eos.db.gamedata_session.query(eos.gamedata.Effect).filter(eos.gamedata.Effect.ID == effectID).one()
+            except sqlalchemy.orm.exc.NoResultFound:
+                effect = eos.gamedata.Effect()
+                effect.effectID = effectID
+                effect.effectName = effectName
             item.effects[effectName] = effect
 
     def hardcodeSuppressionTackleRange():
@@ -660,6 +664,43 @@ def update_db():
         effectMap = {100000: 'pyfaCustomSuppressionTackleRange'}
         _hardcodeAttribs(beaconTypeID, attrMap)
         _hardcodeEffects(beaconTypeID, effectMap)
+        eos.db.gamedata_session.flush()
+
+    def hardcodeSovUpgradeBuffs():
+        typeBuffMap = {
+            # Gamma
+            87815: {
+                'warfareBuff1ID': 2433,
+                'warfareBuff1Value': 5,
+                'warfareBuff2ID': 2434,
+                'warfareBuff2Value': 10,
+                'warfareBuff3ID': 2441,
+                'warfareBuff3Value': 5},
+            # Plasma
+            87949: {
+                'warfareBuff1ID': 2442,
+                'warfareBuff1Value': 5,
+                'warfareBuff2ID': 2435,
+                'warfareBuff2Value': 5,
+                'warfareBuff3ID': 2436,
+                'warfareBuff3Value': 10},
+            # Electric
+            87950: {
+                'warfareBuff1ID': 2437,
+                'warfareBuff1Value': -25,
+                'warfareBuff2ID': 2438,
+                'warfareBuff2Value': 25},
+            # Exotic
+            87951: {
+                'warfareBuff1ID': 2440,
+                'warfareBuff1Value': 2,
+                'warfareBuff2ID': 2439,
+                'warfareBuff2Value': 25}}
+        effectMap = {100001: 'pyfaCustomSovUpgradeBuffEffect'}
+        for typeID, attrMap in typeBuffMap.items():
+            _hardcodeAttribs(typeID, attrMap)
+            _hardcodeEffects(typeID, effectMap, clearEffects=False)
+            eos.db.gamedata_session.flush()
 
 
     def hardcodeShapash():
@@ -806,6 +847,7 @@ def update_db():
         _hardcodeEffects(cybeleTypeID, effectMap)
 
     hardcodeSuppressionTackleRange()
+    hardcodeSovUpgradeBuffs()
 
     eos.db.gamedata_session.commit()
     eos.db.gamedata_engine.execute('VACUUM')
diff --git a/eos/effects.py b/eos/effects.py
index dc1a9e19d..46a123dd0 100644
--- a/eos/effects.py
+++ b/eos/effects.py
@@ -50,13 +50,26 @@ class Effect100000(BaseEffect):
 
     @staticmethod
     def handler(fit, beacon, context, projectionRange, **kwargs):
-        if beacon.getModifiedItemAttr('warfareBuff1ID'):
+        if id := beacon.getModifiedItemAttr('warfareBuff1ID'):
             value = beacon.getModifiedItemAttr('warfareBuff1Value')
-            id = beacon.getModifiedItemAttr('warfareBuff1ID')
             if id:
                 fit.addCommandBonus(id, value, beacon, kwargs['effect'], 'early')
 
 
+class Effect100001(BaseEffect):
+
+    runTime = 'early'
+    type = ('projected', 'passive', 'gang')
+
+    @staticmethod
+    def handler(fit, beacon, context, projectionRange, **kwargs):
+        for i in (1, 4):
+            if id := beacon.getModifiedItemAttr(f'warfareBuff{i}ID'):
+                value = beacon.getModifiedItemAttr(f'warfareBuff{i}Value')
+                if id:
+                    fit.addCommandBonus(id, value, beacon, kwargs['effect'], 'early')
+
+
 class Effect4(BaseEffect):
     """
     shieldBoosting