import wx
from logbook import Logger

import eos.db
import gui.mainFrame
from service.fit import Fit

pyfalog = Logger(__name__)


class FitSetChargeCommand(wx.Command):
    def __init__(self, fitID, positions, chargeID=None, projected=False):
        # todo: determine if this command really should be used with a group of modules, or a simple per module basis
        wx.Command.__init__(self, True, "Module Charge Add")
        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
        self.sFit = Fit.getInstance()
        self.fitID = fitID
        self.chargeID = chargeID
        self.positions = positions
        self.projected = projected
        self.cache = None

    def Do(self):
        return self.__setAmmo(self.positions, self.chargeID)

    def Undo(self):
        for position, chargeID in self.cache.items():
            self.__setAmmo([position], chargeID)
        return True

    def __setAmmo(self, positions, chargeID):
        fit = eos.db.getFit(self.fitID)
        source = fit.modules if not self.projected else fit.projectedModules
        self.cache = {source[i].modPosition: source[i].chargeID for i in positions}
        ammo = eos.db.getItem(chargeID) if chargeID else None

        if ammo is not None and not ammo.isCharge:
            return False
        result = False

        for pos in positions:
            mod = source[pos]
            if not mod.isEmpty and mod.isValidCharge(ammo):
                pyfalog.debug("Set ammo {} for {} on fit {}", ammo, mod, self.fitID)
                result = True
                mod.charge = ammo
        eos.db.commit()
        return result