import wx
from service.fit import Fit

import gui.mainFrame
from gui import globalEvents as GE
from eos.saveddata.module import Module
from .calc.fitRemoveProjectedModule import FitRemoveProjectedModuleCommand
from .calc.fitRemoveProjectedEnv import FitRemoveProjectedEnvCommand
from .calc.fitRemoveProjectedFit import FitRemoveProjectedFitCommand
from logbook import Logger
import eos.db
pyfalog = Logger(__name__)

from eos.saveddata.drone import Drone
from eos.saveddata.module import Module
from eos.saveddata.fighter import Fighter


class GuiRemoveProjectedCommand(wx.Command):
    def __init__(self, fitID, thing):
        wx.Command.__init__(self, True, "Projected Add")
        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
        self.sFit = Fit.getInstance()
        self.internal_history = wx.CommandProcessor()
        self.fitID = fitID
        fit = self.sFit.getFit(fitID)

        if isinstance(thing, Drone):
            self.data = fit.projectedDrones.index(thing)
            self.type = 'drone'
        elif isinstance(thing, Module):
            # todo: projected stuff should be wrapped in a projected class wrapper for easier maintainence
            if thing.item.group.name in Module.SYSTEM_GROUPS:
                self.type = 'env'
                self.data = thing.itemID
            else:
                self.type = 'module'
                self.data = fit.projectedModules.index(thing)
        elif isinstance(thing, Fighter):
            self.data = fit.projectedFighters.index(thing)
            self.type = 'fighter'
        else:
            # todo: fix!
            self.data = thing.ID
            self.type = 'fit'

    def Do(self):
        result = False
        # since we can project various types, we need to switch of the fit command. We can't do this switch easily in
        # the fit command since each type might have a different kind of undo, easier to split it out
        if self.type == 'module':
            result = self.internal_history.Submit(FitRemoveProjectedModuleCommand(self.fitID, self.data))
        elif self.type == 'env':
            result = self.internal_history.Submit(FitRemoveProjectedEnvCommand(self.fitID, self.data))
        elif self.type == 'fit':
            result = self.internal_history.Submit(FitRemoveProjectedFitCommand(self.fitID, self.data))

        # if item.category.name == "Drone":
        #     pyfalog.warn("DRONE REMOVE PROJECTION NOT IMPLEMENTED")
        # elif item.category.name == "Fighter":
        #     pyfalog.warn("FIGHTER REMOVE PROJECTION NOT IMPLEMENTED")
        # elif item.group.name in Module.SYSTEM_GROUPS:
        #     result = self.internal_history.Submit(FitRemoveProjectedEnvCommand(self.fitID, self.id))
        # else:
        #     # attempt a regular module projection
        #
        # elif self.type == 'fit':
        #     pyfalog.warn("FIT REMOVE PROJECTION NOT IMPLEMENTED")

        if result:
            self.sFit.recalc(self.fitID)
            wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
            return True
        return False

    def Undo(self):
        for _ 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