#===============================================================================
# Copyright (C) 2010 Diego Duclos
#
# This file is part of pyfa.
#
# pyfa is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# pyfa is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
#===============================================================================

import wx
import service
import config
import gui.mainFrame
import os
import gui.shipBrowser

class ImportDialog(wx.Dialog):
    def __init__(self, parent):
        wx.Dialog.__init__ (self, parent, id=wx.ID_ANY, title=u"Import fitting from ...", pos=wx.DefaultPosition, size=wx.Size(500, 300), style=wx.DEFAULT_DIALOG_STYLE)
        self._toggleEdit = -1
        self.SetSizeHintsSz(wx.DefaultSize, wx.DefaultSize)
        self.SetMinSize((500,300))
        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
        self._fitsFromFile = None
        self._fitsFromEdit = None

        mainSizer = wx.BoxSizer(wx.VERTICAL)

        headerSizer = wx.BoxSizer(wx.HORIZONTAL)

        self.cFilePicker = wx.FilePickerCtrl(self, wx.ID_ANY, wx.EmptyString, u"Select a fit file", u"*.*", style=wx.FLP_DEFAULT_STYLE | wx.FLP_FILE_MUST_EXIST | wx.FLP_CHANGE_DIR)
        headerSizer.Add(self.cFilePicker, 1, wx.ALL | wx.ALIGN_CENTER_VERTICAL, 5)

        self.tbtnEdit = wx.ToggleButton( self, wx.ID_ANY, u"Text Edit", wx.DefaultPosition, wx.DefaultSize, 0 )
        headerSizer.Add( self.tbtnEdit, 0, wx.ALIGN_CENTER_VERTICAL|wx.RIGHT, 5 )


        mainSizer.Add(headerSizer, 0, wx.EXPAND, 5)
        self.m_staticline2 = wx.StaticLine( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL )
        self.m_staticline2.SetMinSize( wx.Size( 480,1 ) )
        mainSizer.Add( self.m_staticline2, 0, wx.EXPAND, 5 )
        contentSizer = wx.BoxSizer(wx.VERTICAL)

        self.tcEdit = wx.TextCtrl(self, wx.ID_ANY, u"", style=wx.TE_MULTILINE)

        contentSizer.Add(self.tcEdit, 1, wx.EXPAND, 5)

        mainSizer.Add(contentSizer, 1, wx.EXPAND, 5)

        footerSizer = wx.BoxSizer(wx.HORIZONTAL)


        self.stStatus = wx.StaticText( self, wx.ID_ANY, u"Status: File mode.", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.stStatus.Wrap( -1 )
        footerSizer.Add( self.stStatus, 1, wx.ALIGN_CENTER_VERTICAL|wx.LEFT, 5 )

        self.btnImport = wx.Button( self, wx.ID_ANY, u"Import", wx.DefaultPosition, wx.DefaultSize, 0 )
        footerSizer.Add( self.btnImport, 0, wx.ALIGN_CENTER_VERTICAL|wx.TOP|wx.BOTTOM|wx.RIGHT, 5 )

        self.btnOK = wx.Button(self, wx.ID_OK, u"OK", wx.DefaultPosition, wx.DefaultSize, 0)
        footerSizer.Add(self.btnOK, 0, wx.TOP|wx.BOTTOM|wx.RIGHT|wx.ALIGN_CENTER_VERTICAL, 5)

        mainSizer.Add(footerSizer, 0, wx.EXPAND, 5)

        self.SetSizer(mainSizer)

        self.tbtnEdit.SetValue( False )
        self.tcEdit.Enable(False)
        self.btnImport.Enable(False)

        self.Layout()


        self.Bind( wx.EVT_CLOSE, self.CloseDlg )
        self.btnOK.Bind( wx.EVT_BUTTON, self.CloseDlg )
        self.cFilePicker.Bind( wx.EVT_FILEPICKER_CHANGED, self.prepareFileFits )
        self.tbtnEdit.Bind( wx.EVT_TOGGLEBUTTON, self.SwitchEditCtrl )
        self.btnImport.Bind( wx.EVT_BUTTON, self.ImportFittings )
        self.Centre(wx.BOTH)


    def ImportFittings( self, event ):
        sFit = service.Fit.getInstance()
        if self._toggleEdit == -1:
            if self._fitsFromFile:
                try:
                    IDs = sFit.saveImportedFits(self._fitsFromFile)
                    self.stStatus.SetLabel("Status: %d fit(s) imported" % len(self._fitsFromFile))
                    self._openAfterImport(len(self._fitsFromFile), IDs)
                except:
                    self.stStatus.SetLabel("Status: Error importing from file!")
                self._fitsFromFile = None
            else:
                self.stStatus.SetLabel("Status: No fits were specified. Use Browse button.")
        else:
            buffer = self.tcEdit.GetValue()
            if len(buffer) != 0:
                try:
                    self._fitsFromEdit = sFit.importFitFromBuffer(buffer)
                    IDs = sFit.saveImportedFits(self._fitsFromEdit)
                    self.stStatus.SetLabel("Status: %d fit(s) imported" % len(self._fitsFromEdit))
                    self._openAfterImport(len(self._fitsFromEdit), IDs)
                except:
                    self.stStatus.SetLabel("Status: Error importing from text editor!")
            else:
                self.stStatus.SetLabel("Status: Nothing specified.")
        event.Skip()

    def _openAfterImport(self, importCount, fitIDs):
        if importCount == 1:
            if self.mainFrame.getActiveFit() != fitIDs[0]:
                wx.PostEvent(self.mainFrame, gui.shipBrowser.FitSelected(fitID=fitIDs[0]))

    def prepareFileFits(self, event):
        sFit = service.Fit.getInstance()
        try:
            self._fitsFromFile = sFit.importFit(event.Path)
            self.stStatus.SetLabel("Status: Found %d fit(s)." % len(self._fitsFromFile))
            self.btnImport.Enable(True)
        except:
            self.stStatus.SetLabel("Status: Invalid fitting file!")

    def CloseDlg( self, event ):
        event.Skip()

    def SwitchEditCtrl( self, event ):
        self._toggleEdit *= -1
        if self._toggleEdit == -1:
            self.tcEdit.Enable(False)
            self.cFilePicker.Enable( True )
            self.stStatus.SetLabel("Status: File mode.")
            self.btnImport.Enable(False)
        else:
            self.tcEdit.Enable(True)
            self.cFilePicker.Enable( False )
            self.stStatus.SetLabel("Status: Text edit mode.")
            self.btnImport.Enable(True)

        event.Skip()

class ExportDialog ( wx.Dialog ):

    def __init__( self, parent ):
        wx.Dialog.__init__ ( self, parent, id = wx.ID_ANY, title = u"Export fit as ...", pos = wx.DefaultPosition, size = wx.Size( -1,-1 ), style = wx.DEFAULT_DIALOG_STYLE )

        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
        self.srvFit = service.fit.Fit.getInstance()

        self.SetSizeHintsSz( wx.DefaultSize, wx.DefaultSize )

        mainSizer = wx.BoxSizer( wx.VERTICAL )

        fileSizer = wx.BoxSizer( wx.HORIZONTAL )

        self.cFileSavePicker = wx.FilePickerCtrl( self, wx.ID_ANY, wx.EmptyString, u"Select a file", u"XML files (*.xml)|*.xml|TXT files (*.txt)|*.txt", wx.DefaultPosition, wx.DefaultSize, wx.FLP_CHANGE_DIR|wx.FLP_OVERWRITE_PROMPT|wx.FLP_SAVE|wx.FLP_USE_TEXTCTRL )
        fileSizer.Add( self.cFileSavePicker, 1, wx.ALL, 5 )

        mainSizer.Add( fileSizer, 0, wx.EXPAND, 5 )

        self.m_staticline2 = wx.StaticLine( self, wx.ID_ANY, wx.DefaultPosition, wx.DefaultSize, wx.LI_HORIZONTAL )
        self.m_staticline2.SetMinSize( wx.Size( 400,-1 ) )

        mainSizer.Add( self.m_staticline2, 0, wx.EXPAND, 5 )

        choiceSizer = wx.BoxSizer( wx.VERTICAL )

        chCtrlChoices = [ u"XML file", u"EFT file", u"XML && EFT file" ]
        self.chCtrl = wx.RadioBox( self, wx.ID_ANY, u"Export type", wx.DefaultPosition, wx.DefaultSize, chCtrlChoices, 3, wx.RA_SPECIFY_COLS )
        self.chCtrl.SetSelection( 0 )
        choiceSizer.Add( self.chCtrl, 0, wx.EXPAND|wx.ALL, 5 )

        mainSizer.Add( choiceSizer, 1, wx.EXPAND, 5 )

        footerSizer = wx.BoxSizer( wx.HORIZONTAL )

        self.stStatus = wx.StaticText( self, wx.ID_ANY, u"", wx.DefaultPosition, wx.DefaultSize, 0 )
        self.stStatus.Wrap( -1 )
        footerSizer.Add( self.stStatus, 1, wx.ALIGN_CENTER_VERTICAL|wx.TOP|wx.BOTTOM|wx.LEFT, 5 )

        self.btnExport = wx.Button( self, wx.ID_ANY, u"Export", wx.DefaultPosition, wx.DefaultSize, 0 )
        footerSizer.Add( self.btnExport, 0, wx.ALIGN_CENTER_VERTICAL|wx.TOP|wx.BOTTOM, 5 )

        self.btnOK = wx.Button( self, wx.ID_OK, u"OK", wx.DefaultPosition, wx.DefaultSize, 0 )
        footerSizer.Add( self.btnOK, 0, wx.TOP|wx.BOTTOM|wx.ALIGN_CENTER_VERTICAL, 5 )

        mainSizer.Add( footerSizer, 0, wx.EXPAND, 5 )

        self.SetSizer( mainSizer )
        self.Layout()

        self.fitID = self.mainFrame.getActiveFit()
        self.fit = self.srvFit.getFit(self.fitID)
        if self.fit is None:
            self.stStatus.SetLabel("Error: please select a fit before trying to export!")
            self.cFileSavePicker.Enable( False )
            self.chCtrl.Enable( False )
            self.btnExport.Enable( False )
        else:
            self.stStatus.SetLabel("Selected fit: %s" % self.fit.name)
            self.filePath = os.path.join(config.staticPath,self.fit.name)
            self.cFileSavePicker.SetPath(self.filePath)
        mainSizer.Fit( self )

        self.Centre( wx.BOTH )

        # Connect Events
        self.cFileSavePicker.Bind( wx.EVT_FILEPICKER_CHANGED, self.OnFileChoose )
        self.btnOK.Bind( wx.EVT_BUTTON, self.Close )
        self.btnExport.Bind( wx.EVT_BUTTON, self.ExportFit )
        self.chCtrl.Bind( wx.EVT_RADIOBOX, self.OnExtSelect )


    # Virtual event handlers, overide them in your derived class
    def OnExtSelect( self, event ):
#        selection = self.chCtrl.GetSelection()
#        if selection == 0:
#            ext = "XML files (*.xml)|*.xml"
#        elif selection == 1:
#            ext = "TXT files (*.txt)|*.txt"
#        else:
#            ext = "XML files (*.xml)|*.xml|TXT files (*.txt)|*.txt"
#        self.cFileSavePicker.SetWildcard(ext)
        event.Skip()

    def OnFileChoose( self, event ):
        self.filePath,ext = os.path.splitext(event.Path)
        self.cFileSavePicker.SetPath(self.filePath)
        if ext.lower() == ".xml":
            self.chCtrl.SetSelection(0)
        else:
            self.chCtrl.SetSelection(1)
        event.Skip()

    def ExportFit( self, event ):
        extensions = ["xml", "txt"]
        sFit = service.Fit.getInstance()
        selection  = self.chCtrl.GetSelection()
        if selection <2:
            filename = "%s.%s" %(self.filePath,extensions[selection])
            if selection == 0:
                output = sFit.exportXml(self.fitID)
            else:
                output = sFit.exportFit(self.fitID)

            self.WriteFile(filename,output)
        else:
            xmlfname = "%s.%s" %(self.filePath,extensions[0])
            txtfname = "%s.%s" %(self.filePath,extensions[1])
            xmlbuff = sFit.exportXml(self.fitID)
            txtbuff = sFit.exportFit(self.fitID)
            self.WriteFile(xmlfname, xmlbuff)
            self.WriteFile(txtfname, txtbuff)
        self.stStatus.SetLabel("%s exported." % self.fit.name)
        self.chCtrl.Enable(False)
        self.cFileSavePicker.Enable(False)
        self.btnExport.Enable(False)
        event.Skip()

    def Close( self, event ):

        event.Skip()

    def WriteFile(self,filename,buffer):
        ofile = file(filename,"w")
        ofile.write(buffer)
        ofile.close()