import wx
from logbook import Logger

import eos.db
from service.fit import Fit
from service.market import Market


pyfalog = Logger(__name__)


class FitChangeModuleChargesCommand(wx.Command):

    def __init__(self, fitID, chargeMap, projected=False):
        wx.Command.__init__(self, True, 'Change Module Charges')
        self.fitID = fitID
        self.chargeMap = chargeMap
        self.projected = projected
        self.savedChargeMap = None

    def Do(self):
        pyfalog.debug('Doing change of module charges according to map {} on fit {}'.format(self.chargeMap, self.fitID))
        fit = Fit.getInstance().getFit(self.fitID)
        container = fit.modules if not self.projected else fit.projectedModules
        changes = False
        self.savedChargeMap = {}
        sMkt = Market.getInstance()
        for position, chargeItemID in self.chargeMap.items():
            mod = container[position]
            if mod.isEmpty:
                continue
            if mod.chargeID is None and chargeItemID is None:
                continue
            if mod.chargeID == chargeItemID:
                continue
            chargeItem = sMkt.getItem(chargeItemID) if chargeItemID is not None else None
            if chargeItem is not None and not chargeItem.isCharge:
                continue
            if not mod.isValidCharge(chargeItem):
                pyfalog.warning('Invalid charge {} for {}'.format(chargeItem, mod))
                continue
            pyfalog.debug('Setting charge {} for {} on fit {}'.format(chargeItem, mod, self.fitID))
            self.savedChargeMap[position] = mod.chargeID
            changes = True
            mod.charge = chargeItem
        if changes:
            eos.db.commit()
            return True
        return False

    def Undo(self):
        pyfalog.debug('Undoing change of module charges according to map {} on fit {}'.format(self.chargeMap, self.fitID))
        cmd = FitChangeModuleChargesCommand(fitID=self.fitID, chargeMap=self.savedChargeMap, projected=self.projected)
        return cmd.Do()