Move meta swap functionality to command process

gui/fitCommands/__init__.py

@@ -18,4 +18,5 @@ from .guiRemoveProjected import GuiRemoveProjectedCommand
 from .guiCargoToModule import GuiCargoToModuleCommand
 from .guiModuleToCargo import GuiModuleToCargoCommand
 from .guiAddFighter import GuiAddFighterCommand
-from .guiRemoveFighter import GuiRemoveFighterCommand
+from .guiRemoveFighter import GuiRemoveFighterCommand
+from .guiMetaSwap import GuiMetaSwapCommand

gui/fitCommands/calc/fitAddFighter.py

@@ -14,7 +14,7 @@ class FitAddFighterCommand(wx.Command):
     """"
     from sFit.addFighter
     """
-    def __init__(self, fitID, itemID, amount=1, replace=False):
+    def __init__(self, fitID, itemID):
         wx.Command.__init__(self, True, "Cargo add")
         self.fitID = fitID
         self.itemID = itemID

gui/fitCommands/guiMetaSwap.py

@@ -0,0 +1,64 @@
+import wx
+from service.fit import Fit
+
+import gui.mainFrame
+from gui import globalEvents as GE
+from .calc.fitRemoveImplant import FitRemoveImplantCommand
+from .calc.fitAddImplant import FitAddImplantCommand
+from .calc.fitRemoveBooster import FitRemoveBoosterCommand
+from .calc.fitAddBooster import FitAddBoosterCommand
+from .calc.fitRemoveCargo import FitRemoveCargoCommand
+from .calc.fitAddCargo import FitAddCargoCommand
+from .calc.fitReplaceModule import FitReplaceModuleCommand
+from .calc.fitAddFighter import FitAddFighterCommand
+from .calc.fitRemoveFighter import FitRemoveFighterCommand
+
+class GuiMetaSwapCommand(wx.Command):
+    def __init__(self, fitID, context, itemID, selection: list):
+        wx.Command.__init__(self, True, "Meta Swap")
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.sFit = Fit.getInstance()
+        self.internal_history = wx.CommandProcessor()
+        self.fitID = fitID
+        self.itemID = itemID
+        self.context = context
+        self.data = []
+        fit = self.sFit.getFit(fitID)
+
+        if context == 'fittingModule':
+            for x in selection:
+                self.data.append(((FitReplaceModuleCommand, fitID, fit.modules.index(x), itemID),),)
+        elif context == 'implantItem':
+            for x in selection:
+                idx = fit.implants.index(x)
+                self.data.append(((FitRemoveImplantCommand, fitID, idx), (FitAddImplantCommand, fitID, itemID)))
+        elif context == 'boosterItem':
+            for x in selection:
+                idx = fit.boosters.index(x)
+                self.data.append(((FitRemoveBoosterCommand, fitID, idx), (FitAddBoosterCommand, fitID, itemID)))
+        elif context == 'cargoItem':
+            for x in selection:
+                self.data.append(((FitRemoveCargoCommand, fitID, x.itemID, 1, True), (FitAddCargoCommand, fitID, itemID, x.amount)))
+        elif context == 'fighterItem':
+            for x in selection:
+                self.data.append(((FitRemoveFighterCommand, fitID, fit.fighters.index(x)), (FitAddFighterCommand, fitID, itemID)))
+        elif context == 'droneItem':
+            raise NotImplementedError()
+
+
+    def Do(self):
+        for cmds in self.data:
+            for cmd in cmds:
+                self.internal_history.Submit(cmd[0](*cmd[1:]))
+
+        self.sFit.recalc(self.fitID)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+        return True
+
+    def Undo(self):
+        for x in self.internal_history.Commands:
+            self.internal_history.Undo()
+        self.sFit.recalc(self.fitID)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+        return True
+