#!/usr/bin/python
# -*- coding: utf-8 -*-

from Tkinter import *
from tkFileDialog import askopenfilename
import tkFont
import re, os

class App:
    def __init__(self,master):
        
        frame = Frame(master, width=400, height=300)
        frame.pack()

        self.getFile = Entry(frame, width=30)
        self.getFile.grid(row=0,column=0)

        self.browse = Button(frame, text="Browse", command=self.browseFiles)
        self.browse.grid(row=0,column=1)

        self.next = Button(frame, text="Next >", command=self.chooseColors)
        self.next.grid(row=1,column=1, pady=5)

        Letters = tkFont.Font(family='Gulim',size=10)

        self.nickScroll = Scrollbar(frame)
        self.nickList = Listbox(frame, selectmode=SINGLE, font=Letters, \
                                yscrollcommand=self.nickScroll.set)

        self.nickScroll.config(command=self.nickList.yview)

        self.nickList.bind("<Button-1>", self.activateIndex)

        self.colorScroll = Scrollbar(frame)
        self.colorChoices = Listbox(frame,selectmode=SINGLE, font=Letters)
        self.colorScroll.config(command=self.colorChoices.yview)

        #colors = ['검정','빨강','파랑','초록','보라','잿빛','갈색','군청','진적색','청록','분홍']
        colors = ['Black','Red','Blue','Green','Purple','Gray','Brown','Navy','Maroon','Teal','Pink']
        
        for color in colors:
            self.colorChoices.insert(END,color)

        self.colorChoices.bind("<Button-1>", self.saveToFile)

        self.getHTML = Button(frame, text="Apply Style >", command=self.genHTML)

        self.abortStyle = Button(frame, text="Abort", command=frame.destroy)

        self.previous = Button(frame, text="< Back", command=self.previous)

        self.doneMsg = Label(frame,text="You have generated an HTML file of your log.")

        self.finish = Button(frame, text="Done", command=frame.destroy)

    def saveToFile(self,event):
        selected = self.colorChoices.nearest(event.y)
        selectedColor = self.colorChoices.get(selected)
        currnick = self.nickList.get(ANCHOR)
        currindex = self.nickList.index(ANCHOR)
        leftBracket = currnick.rfind('(')
        if leftBracket != -1:
            currnick = currnick[0:leftBracket]
        currnick = currnick.strip()
        ucurrnick = currnick.encode('utf-8')
        self.nickList.delete(currindex)
        self.nickList.insert(currindex,currnick+' ('+selectedColor+')')
        self.nickList.select_anchor(currindex)
        uselectedColor = selectedColor.encode('utf-8')
        config = self.readINI()
        nickcolor = self.findINI(config,ucurrnick)
        if nickcolor:
            newconfig = config.replace(nickcolor,"%s=%s"%\
                                       (ucurrnick,uselectedColor))
            self.writeINI(newconfig)
        else:
            self.appendINI("%s=%s"%(ucurrnick,uselectedColor))

    def activateIndex(self,event):
        selected = self.nickList.nearest(event.y)
        self.nickList.select_anchor(selected)

    def browseFiles(self):

        contents = self.readINI()

        try:
            prevfileline = self.findINI(contents,'file')
        except:
            prevfileline = None

        if prevfileline:
            prevfile = prevfileline.replace('file=','')
            prevfile = prevfile.strip()
            lastdash = prevfile.rfind('/')
            startdir = prevfile[0:lastdash]
        else:
            startdir = None

        filename = askopenfilename(initialdir=startdir)
        filename = filename.strip()
        self.getFile.delete(0,END)
        self.getFile.insert(0,filename)
        
        if contents:
            fileinfo = self.findINI(contents,'file')

            if fileinfo:
                contents = contents.replace(fileinfo,'file='+filename,1)
                self.writeINI(contents)

            else:
                self.appendINI('file='+filename)

        else:
            self.appendINI('file='+filename)
        

    def chooseColors(self):
        #This function handles the main HTML file-writing work
        #Maybe a name change is in order, to prevent confusion?

        contents = self.readINI()

        if contents:
            fileline = self.findINI(contents,'file')
        else:
            fileline = None

        if fileline:
            #get the file...
            origfile = fileline.replace('file=','')
            origfile = origfile.strip()

            lastdot = origfile.rfind('.')
            #if there is a . in the file name
            if lastdot != -1:
                origfileName = origfile[0:lastdot]
                origex = origfile[lastdot:]
            #if no . in the file name
            else:
                origfileName = origfile
                origex = ''

            #debug
            #print origex

            if origex != '.log' and origex !='.txt':
                import tkMessageBox
                tkMessageBox.showwarning('Extension error',\
                                         'File must be a .log or .txt file.')
                return
            else:
                #process head part of file
                head = self.readINI('filehead.html')
                lastdash = origfileName.rfind('/')
                justfileName = origfileName[lastdash + 1:]
                puncpattern = re.compile('[^&#\w]')
                logtitle = "<title>%s</title>"%puncpattern.sub(' ',justfileName)
                head = head.replace('<title></title>',logtitle,1)

                newfile = origfileName + '.html'

                #open in write mode and write HTML head
                newfilewrite = open(newfile,'w')
                newfilewrite.write(head)
                newfilewrite.close()

                #get the contents of original logfile...
                origlog = '\n' + self.readINI(origfile)

                #erase all time stamps, if any
                timestamp = re.compile('\n\[.*?\][ ]?')
                origlog = timestamp.sub('\n',origlog)

                #erase all color codes
                colorcode = re.compile('[0-9]{1,2}')
                origlog = colorcode.sub('',origlog)

                #split into list of lines
                lines = origlog.split("\n")
                lines = [line.strip() for line in lines]

                nicks = []
                cleanNicks = []

                newfileprocess = open(newfile,'a')

                #start and end chat characters
                startchars = ['(','[']
                endchars = [')',']']

                #process by line
                lineno = 0

                while lineno < len(lines):
                    line = lines[lineno]
                    prevchat = False

                    if lineno > 0:
                        prevline = lines[lineno - 1]
                        prevchat = self.wasChat(prevline,startchars,endchars)
                    
                    #spoken line
                    rightBracket = line.find('>')
                    if line.startswith('<') and rightBracket !=-1:
                        #extract nick
                        nick = line[1:rightBracket]
                        #clean up nickname
                        cleanup = re.compile('[@]')
                        nick = cleanup.sub('',nick)
                        spacedNick = nick.replace('_',' ')
                        cleanNick = spacedNick.strip()
                        #append to lists if not already there
                        try:
                            x = nicks.index(nick)
                        except ValueError:
                            nicks.append(nick)
                            cleanNicks.append(cleanNick)
                        #extract spoken line
                        spoken = line[rightBracket+1:]
                        spoken = spoken.strip()
                        currchat = self.isChat(spoken,startchars,endchars)
                        if prevchat and not currchat:
                            #end of chatter
                            div = '</div>'
                        elif not prevchat and currchat:
                            #start of chatter
                            div = "<div class='chatter'>"
                        else:
                            div = ''
                            
                        #create new line
                        newline0="</p>%s<p class='%s'><span class='nick'>%s:</span>"\
                                   %(div,nick,cleanNick)
                        newline1=" %s\n" % (spoken,)
                        newline = newline0 + newline1
                    else:
                        #append to new file
                        if prevchat:
                            div = ''
                        else:
                            div = "<div class='chatter'>"
                        newline = "</p>%s<p class='note'>%s\n" % (div,line)


                    #write to html log file
                    try:
                        newline = unicode(newline,"cp949").encode("utf-8")
                        newfileprocess.write(newline)
                    except UnicodeDecodeError:
                        pass
                    lineno = lineno + 1

                #add closing tags
                newfileprocess.write("\n</p></div></body></html>")
                #close out file
                newfileprocess.close()
                #write nicks to log.ini
                nickline = ','.join(nicks)
                cleanNickline = ','.join(cleanNicks)
                self.appendINI("nicks(%s)=%s"%(newfile,nickline))
                self.appendINI("cleanNicks(%s)=%s"%(newfile,cleanNickline))
        
                self.getFile.grid_forget()
                self.browse.grid_forget()
                self.next.grid_forget()

                nickslist = self.getNickLists()['nicks']
                orignickslist = self.getNickLists()['origs']

            for nick in nickslist:
                nick = unicode(nick,"euc-kr")
                unick = nick.encode("utf-8")
                config = self.readINI()
                nickcolor = self.findINI(config,unick)
                if nickcolor:
                    try:
                        color = nickcolor.split('=')[1].strip()
                    except IndexError:
                        color = None
                    if color:
                        kocolor = unicode(color,'euc-kr')
                        nick = nick + ' (' + kocolor + ')'

                self.nickList.insert(END,nick)
                
            self.nickList.grid(row=0,column=0)
            self.nickList.select_anchor(0)
            self.nickScroll.grid(row=0,column=1,padx=0)
            self.colorChoices.grid(row=0,column=2)
            self.colorScroll.grid(row=0,column=3,padx=0)
            self.previous.grid(row=1,column=0,pady=5)
            self.getHTML.grid(row=1,column=2,pady=5)

    def genHTML(self):
        #This function fills out the stylesheet
        #Doesn't work very well yet

        nicks = self.getNickLists()['nicks']
        origs = self.getNickLists()['origs']

        #debug
        #for nick in nicks:
            #print unicode(nick,'cp949').encode('utf-8')


        filename = self.getNewFileName()
        logfile = self.readINI(filename)

        #fill out the stylesheet
        i=0
        while i<len(nicks):

            nick = nicks[i]
            nick = unicode(nick,'cp949')
            nick = nick.encode('utf-8')

            orig = origs[i]
            orig = unicode(orig,'cp949')
            orig = orig.encode('utf-8')

            config = self.readINI()
            colorConfigLine = self.findINI(config,nick)

            if colorConfigLine:
                #debug
                #print 'found it!'
                colorConfig = colorConfigLine.split('=')
                try:
                    color = colorConfig[1]
                except IndexError:
                    color = 'Black'

                classDeclare = '.'+color
                nickClass = classDeclare+', .'+orig
                #debug
                #print nickClass

                classLine = self.findINI(logfile,classDeclare)

                if classLine:
                    #debug
                    #print classLine
                    #print classDeclare
                    logfile = logfile.replace(classDeclare,nickClass)

            i = i+1

        self.writeINI(logfile,filename)
        self.eraseNickList()

        self.nickList.delete(0,END)        
        self.nickList.grid_forget()
        self.nickScroll.grid_forget()
        self.colorChoices.grid_forget()
        self.colorScroll.grid_forget()
        self.previous.grid_forget()
        self.getHTML.grid_forget()
        #completion message
        self.doneMsg.grid(row=0,column=0,columnspan=2)
        #'Back' button
        self.previous.grid(row=1,column=0)
        self.finish.grid(row=1,column=1)

    def previous (self):
        self.doneMsg.grid_forget()
        self.previous.grid_forget()
        self.finish.grid_forget()
        self.nickList.delete(0,END)        
        self.nickList.grid_forget()
        self.colorChoices.grid_forget()
        self.nickScroll.grid_forget()
        self.colorScroll.grid_forget()
        self.getHTML.grid_forget()
        self.getFile.grid(row=0,column=0)
        self.browse.grid(row=0,column=1)
        self.next.grid(row=1,column=1, pady=5)

        self.eraseNickList()

    def eraseNickList(self):
        config = self.readINI()
        fileline = self.findINI(config,'file')
        filename = fileline.replace('file=','')
        lastdot = filename.rfind('.')
        filenameOnly = filename[0:lastdot]
        #debug
        #print filenameOnly
        lines = config.split("\n")
        newlines = []
        for line in lines:
            Nicks = line.find('nicks(' + filenameOnly)
            CleanNicks = line.find('cleanNicks(' + filenameOnly)
            #debug
            #print CleanNicks
            if Nicks == -1 and CleanNicks == -1 and not line.startswith('='):
                newlines.append(line)
        newconfig = "\n".join(newlines)
        self.writeINI(newconfig)

    def getFileName(self,filename='log.ini'):
        config = self.readINI()
        fileline = self.findINI(config,'file')
        if fileline:
            filename = fileline.replace('file=','')
        else:
            filename = None
        return filename

    def getNewFileName(self,filename='log.ini'):
        filename = self.getFileName()
        lastdot = filename.rfind('.')
        if lastdot != -1:
            filenameOnly = filename[0:lastdot]
        else:
            filenameOnly = filename

        newfileName = filenameOnly + '.html'
        return newfileName

    def readINI(self,filename='log.ini'):
        try:
            readfile = open(filename,'r')
            contents = readfile.read()
            readfile.close()
        except IOError:
            contents = None
        return contents

    def findINI(self,contents,lookfor):
        foundit = False
        lines = contents.split("\n")
        for line in lines:
            if line.startswith(lookfor):
                found = line
                foundit = True
                break

        if foundit:
            return found
        else:
            return None
        
    def writeINI(self,contents,filename='log.ini'):
        writefile = open(filename,'w')
        writefile.write(contents)
        writefile.close()

    def appendINI(self,line,filename='log.ini'):
        appendfile = open(filename,'a')
        appendfile.write("\n"+line)
        appendfile.close()

    def wasChat(self,line,startchars,endchars):
        chatter = False

        rightBracket = line.find('>')
        if line.startswith('<') and rightBracket != -1:
            #is spoken line
            #extract spoken part
            spoken = line[rightBracket+1:]
            spoken = spoken.strip()
            ischatter = self.isChat(spoken,startchars,endchars)
            if ischatter:
                chatter = True
            else:
                chatter = False            
        else:
            #no nick, no dialogue
            chatter = True
                
        return chatter

    def isChat(self,line,startchars,endchars):
        chatter = False

        startchat = False
        for startchar in startchars:
           if line.startswith(startchar):
               startchat = True
               break

        endchat = False
        for endchar in endchars:
            if line.endswith(endchar):
                endchat = True
                break

        if startchat and endchat:
            chatter = True

        return chatter

    def getNickLists(self):
        config = self.readINI()
        filename = self.getNewFileName()
        nicksline = self.findINI(config,'cleanNicks('+filename)
        orignicksline = self.findINI(config,'nicks('+filename)

        nickslinesplit = nicksline.split(')=')
        orignickslinesplit = orignicksline.split(')=')

        try:
            nickslist = nickslinesplit[1].split(',')
            orignickslist = orignickslinesplit[1].split(',')
        except IndexError:
            nickslist = []
            orignickslist = []

        return {'nicks':nickslist,'origs':orignickslist}

root = Tk()
application = App(root)
root.mainloop()