#!/usr/bin/env python
import csv
from bitFields import bitField
import sys
from table import MultiListbox
from Tkinter import *
import tkFileDialog
import tkMessageBox
import os


#data1 = "00235b00020100e04c681a7e080045000130b8d1f0f08011df8ec0a81001c0a8100b08340834011cbadb465449880005002001040000001000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000"
#data = "00235b00020100e04c681a7e080045000130b8d100aa8011df8ec0a81001c0a8100b08340834011cbadb46544988"
data = ""
#format_filename="C:\Python27\UDPHeader.csv"
format_filename=""
data_filename=""
fields = []

class FormatFields:
    def __init__(self,row):
        self.label,self.length,self.format = row[0:3]
        if self.format == "Bin":
            if len(row) > 3:
                self.type = row[3:]
                #print self.type
                self.bitFields = bitField(self.type)
                #print self.bitFields.bits
        else:
            self.bitFields = []
        self.length = int(self.length)
        self.data = []
        self.formatted = []
    def get_f(self):
        return "%-25s|%-5s|%-7s|%30s"  %(self.label,self.length,self.format,self.formatted)
    def get(self):
        return (self.label,self.length,self.format,self.type,self.data,self.formatted)
    def format_data(self):
        if (self.format == "IP"):
            self.formatted = formatIP(self.data)
        elif (self.format == "MAC"):
            self.formatted = formatMAC(self.data)
        elif (self.format == "Bin"):
            self.formatted = formatBin(self.data)
            self.bitFields.add_data(self.data)
        elif (self.format == "Int"):
            self.formatted = formatInt(self.data)
        elif (self.format == "Hex"):
            self.formatted = formatHex(self.data)
        else:
            self.formatted = self.data

def splitToList(data,lengths):
    result = []
    start = 0
    for x in lengths:
        res = data[start:start+(x*2)]
        result.append(res)
        start += (x*2)
    return result

def splitString(data,length):

    split = data[:length*2]
    data = data[length*2+1:]

    return split

def formatList(s,count):
    return [''.join(x) for x in zip(*[list(s[z::count]) for z in range(count)])]

def formatSep(lst,sep):
    return sep.join(lst)

def formatBin(s):
    """Return Grab the test results for range testing this
    >>>formatBin('0xFF')
    ['1','1','1','1','1','1','1','1']
    """
    #print s
    binary = bin(int(s, 16))[2:]
    size = len(binary)
    if (len(binary) % 8):
        size = size + (8 - size % 8)
        binary = ('0' * (size - len(binary))) + binary
        
    return binary

def formatInt(s):
    return int(s,16)

def formatIP(s):
    lst = splitToList(s,[1,1,1,1])
    for z in range(len(lst)):
        lst[z] = int(lst[z],16)
        lst[z] = str(lst[z])
    return formatSep(lst,".")

def formatMAC(s):
    lst = splitToList(s,[1,1,1,1,1,1])
    return formatSep(lst,":")

def formatHex(s):
    return "0x" + s

def open_format():
    global format_filename
    filename = tkFileDialog.askopenfilename(filetypes=[('CSV','*.csv')])
    #print filename
    format_filename = filename
    name = "Format: " + os.path.basename(filename)
    format_file.config(text=name)
    process_format()

def open_data():
    global data_filename
    
    data_filename = tkFileDialog.askopenfilename(filetypes=[('Data','*.dat'),('All Files','*.*')])
    #print data_filename
    if data_filename!="":
        name = "Data: " + os.path.basename(data_filename)
        data_file.config(text=name)
        data_read()

def data_read():
    global data_filename
    global data

    #print data_filename

    if data_filename=="":
        tkMessageBox.showwarning("Data file","No Data has been selected\nPlease select one")
        return
    else:
        #print "Success Opening: " + data_filename
        pass
    
    try:
        file = open(data_filename)
    except IOError:
        tkMessageBox.showwarning("Open file","Cannot open this file\n(%s)" % data_filename)
        return

    data = file.read()
    process()

def export():
    exportfilename = tkFileDialog.asksaveasfilename(filetypes=[('Text','*.txt')])
    #print exportfilename

    try:
        exportfile = open(exportfilename,'w')
    except IOError:
        tkMessageBox.showwarning("Open file","Cannot open this file\n(%s)" % exportfilename)
        return
    exportfile.write("     Data Field Name     |Size |Format |              Data            \n")
    exportfile.write("-------------------------|-----|-------|------------------------------\n")
    for field in fields:
        exportfile.write(field.get_f())
        exportfile.write("\n")
    exportfile.close()

def donothing():
    return

def process_format():
    global fields
    global format_filename

    mlb.delete(0,last='end')

    if format_filename=="":
        tkMessageBox.showwarning("Process file","No Format File has been selected\nPlease select one")
        return
    
    try:
        file = open(format_filename)
    except IOError:
        tkMessageBox.showwarning("Open file","Cannot open this file\n(%s)" % filename)
        return

    reader = csv.reader(file)
    fields = list(map(FormatFields, reader))

    for field in fields:
        mlb.insert(END, (field.label,field.format, field.formatted))

def process():
    global format_filename
    global data
    global fields

    if data=="":
        tkMessageBox.showwarning("No Data","No data file has been selected.\nPlease Open one.")
        return

    size_list = []
    
    for field in fields:
        size_list.append(field.length)

    data_list = splitToList(data,size_list)
    i = 0

    #Add the data from the data list to the datafields,
    #format the data based on its format
    for field in fields:
        field.data = data_list[i]
        #print field.label +" "+ field.data
        field.format_data()
        i += 1
        if i >= len(data_list):
            break
    #print "Size: " + str(mlb.size())
 
    mlb.delete(0,last='end')

    #Finally print out the table
    #for field in fields:
    #    field.get_f()
    for field in fields:
        mlb.insert(END, (field.label,field.format, field.formatted))
        # Binary Formatted Commented out until it works better
        if field.format=="Bin":
            #print len(field.bitFields.bits)
            for i in range(len(field.bitFields.bits)):
                #print "Index: " + str(i)
                if i>= len(field.bitFields.data):
                    break
                mlb.insert(END,("|_",field.bitFields.bits[i],field.bitFields.data[i]))
                       
if __name__ == '__main__':
    tk = Tk()
    tk.title("Data Formatter")
    tk.minsize(width=400,height=400)
    
    menubar = Menu(tk)
    
    filemenu = Menu(menubar, tearoff=0)
    filemenu.add_command(label="Open Format", command=open_format)
    filemenu.add_command(label="Open Data", command=open_data)
    filemenu.add_command(label="Export", command=export)
    filemenu.add_separator()
    filemenu.add_command(label="Exit", command=tk.destroy)
    
    menubar.add_cascade(label="File",menu=filemenu)
    #menubar.add_command(label="Process",command=process)
    tk.config(menu=menubar)
    
    mlb = MultiListbox(tk, (('Field Name', 40),('Type', 20), ('Data', 20)))
    mlb.pack(expand=YES,fill=BOTH)

    format_file = Label(tk,text="Format:",relief="ridge",justify='left')
    format_file.pack(side="left",fill=X,expand=1)
    
    data_file = Label(tk,text="Data:",relief="ridge",justify='left')
    data_file.pack(side="right",fill=X,expand=1)

    formatname = "Format: " + os.path.basename(format_filename)
    format_file.config(text=formatname)
    if format_filename:
        process_format()

    dataname = "Data: " + os.path.basename(data_filename)
    data_file.config(text=dataname)
    
    tk.mainloop()