from tkinter import *
from tkinter import ttk
from tkinter import filedialog as fd
from tkinter import messagebox
import get_char
import count_char
import global_var
import ttc_get
import ntpath
import os
#drag and drop
import sys
import argparse
first_time = True
parser = argparse.ArgumentParser(
                    prog='CJK Character Count',
                    description='A program that parse a font file and list CJK characters. Currently only CJK Ideographs are supported.',)
parser.add_argument('filename', # positional argument
                    nargs='?', # may be optional
                    default=None)
parser.add_argument('-r', '--report',
                    action='store_true', # on/off flag
                    default =False,
                    help='Output a .txt file with CSV structure under cjk_report folder.')
args = parser.parse_args()

#start window
main = Tk()
#set title
main.title("CJK Character Count")
#window size
main.geometry('')
#prepare font
title_font=("Segoe UI", 12, "bold underline")
text_font=("Segoe UI", 12)
text_sum_font=("Segoe UI", 12, "bold")
#add frame
frame = Frame(main)
frame.pack(padx=(25,40), pady=(0,30))
#set module language
current_lang="en"

#export text file checkbox
# initialize the boolean state from the argument parser
bool_exportfile = BooleanVar(value=args.report)
checkbox_exportfile = Checkbutton(frame, text='Text report', variable=bool_exportfile, onvalue=True, offvalue=False)
checkbox_exportfile.grid(column=0, row=0, sticky=W)

#text label + button
input_lbl = Label(frame, text="Choose font file:　　　　", font=text_font)
input_lbl.grid(column=0, row=0, sticky=E)
btn = Button(frame, text ='Open', font=text_font, command = lambda:open_file(None) ) #just send None as no drag and drop file
btn.grid(column=0, row=0, sticky=E)

#show file name
font_name = StringVar(main)
font_name_lbl = Label(frame, textvariable=font_name, justify="left", font=text_font)
font_name_lbl.grid(column=3, row=0, sticky=W, columnspan=2)
font_name.set("No input file")

font_txt_lbl = Label(frame, text="File name: ", justify="right", font=text_font)
font_txt_lbl.grid(column=1, row=0, sticky=E, padx=(5,0), columnspan=2)


#section 1: chinese encoding
#prepare listing for string var
cjk_lbl={}
cjk_empty={}
cjk_count={}
cjk_var={}

cjk_jian_list = global_var.cjk_jian_list_en
cjk_jian_fan_list = global_var.cjk_jian_fan_list_en
cjk_fan_list = global_var.cjk_fan_list_en
unicode_list = global_var.unicode_list
titles = global_var.titles_en

cjk_jian_start=2
cjk_enc_lbl = Label(frame, text=titles["simp"], font=title_font, anchor="w")
cjk_enc_lbl.grid(column=0, row=cjk_jian_start, sticky=W, padx=5)
#loop label for cjk jian
for i, (cjk_enc, cjk_enc_name) in enumerate(cjk_jian_list.items()):
    cjk_var[cjk_enc] = StringVar(main)
    cjk_lbl[cjk_enc] = Label(frame, text=cjk_enc_name+"：", justify="left", font=text_font).grid(column=0, row=i+cjk_jian_start+1, sticky=W)
    cjk_empty[cjk_enc] = Label(frame, textvariable=cjk_var[cjk_enc], justify="left", font=text_font).grid(column=1, row=i+cjk_jian_start+1)
    cjk_count[cjk_enc] = Label(frame, text="/"+str(global_var.cjk_count[cjk_enc]), justify="left", font=text_font).grid(column=2, row=i+cjk_jian_start+1, sticky=W)
    cjk_jian_fan_start=cjk_jian_start+i+1

cjk_jian_fan_start+=2
cjk_enc_lbl = Label(frame, text=titles["simptrad"], font=title_font, anchor="w")
cjk_enc_lbl.grid(column=0, row=cjk_jian_fan_start-1, rowspan=2, sticky="SW", padx=5)
#loop label for cjk jian-fan
for i, (cjk_enc, cjk_enc_name) in enumerate(cjk_jian_fan_list.items()):
    cjk_var[cjk_enc] = StringVar(main)
    cjk_lbl[cjk_enc] = Label(frame, text=cjk_enc_name+"：", justify="left", font=text_font).grid(column=0, row=i+cjk_jian_fan_start+1, sticky=W)
    cjk_empty[cjk_enc] = Label(frame, textvariable=cjk_var[cjk_enc], justify="left", font=text_font).grid(column=1, row=i+cjk_jian_fan_start+1)
    cjk_count[cjk_enc] = Label(frame, text="/"+str(global_var.cjk_count[cjk_enc]), justify="left", font=text_font).grid(column=2, row=i+cjk_jian_fan_start+1, sticky=W, padx=(0,16))
    cjk_fan_start=cjk_jian_fan_start+i+1

cjk_fan_start+=2
cjk_enc_lbl = Label(frame, text=titles["trad"], font=title_font, anchor="w")
cjk_enc_lbl.grid(column=0, row=cjk_fan_start-1, rowspan=2, sticky="SW", padx=5)
#loop label for cjk fan
for i, (cjk_enc, cjk_enc_name) in enumerate(cjk_fan_list.items()):
    cjk_var[cjk_enc] = StringVar(main)
    cjk_lbl[cjk_enc] = Label(frame, text=cjk_enc_name+"：", justify="left", font=text_font).grid(column=0, row=i+cjk_fan_start+1, sticky=W)
    cjk_empty[cjk_enc] = Label(frame, textvariable=cjk_var[cjk_enc], justify="left", font=text_font).grid(column=1, row=i+cjk_fan_start+1)
    cjk_count[cjk_enc] = Label(frame, text="/"+str(global_var.cjk_count[cjk_enc]), justify="left", font=text_font).grid(column=2, row=i+cjk_fan_start+1, sticky=W)

#section 2: unicode section
unicode_lbl = Label(frame, text=titles["uni"], font=title_font)
unicode_lbl.grid(column=3, row=2, sticky=W, padx=5)
unicode_lbl={}
unicode_empty={}
unicode_count={}
unicode_var={}
unicode_last_tag=0
#listing with unicode area
for i, (unicode_enc, unicode_enc_name) in enumerate(unicode_list.items()):
    unicode_row=i+3
    unicode_var[unicode_enc] = StringVar(main)
    #only bold total
    if unicode_enc == "total":
        unicode_text_font = text_sum_font
    else:
        unicode_text_font = text_font
    unicode_lbl[unicode_enc] = Label(frame, text=unicode_enc_name+"：", justify="left", font=unicode_text_font).grid(column=3, row=unicode_row, sticky=W)
    unicode_empty[unicode_enc] = Label(frame, textvariable=unicode_var[unicode_enc], justify="left", font=unicode_text_font).grid(column=4, row=unicode_row)
    unicode_count[unicode_enc] = Label(frame, text="/"+str(global_var.unicode_count[unicode_enc]), justify="left", font=unicode_text_font).grid(column=5, row=unicode_row, sticky=W)

#open file react function
def open_file(filename_arg):
    #check if file is dragged onto here
    if filename_arg:
        #set filename to font
        filename = filename_arg
    else:
        #get file with open file dialog
        filename = fd.askopenfilename(initialdir = 'shell:Desktop', title="Select font file",
                                    filetypes = [("All suppported font format","*.ttf *.otf *.woff *.woff2 *.otc *.ttc"),
                                                ("Single font file format","*.ttf *.otf *.woff *.woff2"),
                                                ("TrueType font","*.ttf"),
                                                ("OpenType font","*.otf"),
                                                ("Web Open Font Format (WOFF)","*.woff *.woff2"),
                                                ("OpenType collection font","*.otc *.ttc"),
                                                ("All files","*.*")
                                                ]
                                    )

    #no file selected, fd.askopenfilename() will return empty string
    if filename == '' or filename is None:
        font_name.set("No input file")
        messagebox.showwarning(title="No file", message="No font selected.")
        return
    
    #test if the file is a valid font file
    is_a_font = False
    if (filename.lower().endswith((".otf",".ttf",".woff",".woff2",".otc",".ttc"))):
        # external module: get_char
        is_a_font = get_char.is_font(filename)
    
    #default value for non-TTCollection
    font_id = -1
    #for TTCollections, get which font to check
    if filename.lower().endswith((".otc",".ttc")):
        #get which font to use, if no choose will return None - external module: ttc_get
        font_id = ttc_get.TTC_popup(main, filename, lang=current_lang).show()
        #if user not select
        if font_id is None:
            filename = "" #trigger as no font selected
            is_a_font = False #prevent font to process
            
    if not is_a_font:
        #not a font, change filename back to null n give error message
        font_name.set("No input file")
        messagebox.showwarning(title="Not a font", message="This is not a valid font file.")
        return

    #show filename
    font_name.set(path_leaf(filename))
    #reset 0
    for cjk_enc in global_var.cjk_list:
        cjk_var[cjk_enc].set(0)
    for unicode_enc in global_var.unicode_list:
        unicode_var[unicode_enc].set(0)
    #get character list in font - external module: get_char
    char_list = get_char.font_import(filename, font_id, lang=current_lang)
    #get list count - external module: count_char
    output = count_char.count_char(char_list)
    cjk_char_count = output[0]
    unicode_char_count = output[1]
    #update list count
    for cjk_enc in global_var.cjk_list:
        cjk_var[cjk_enc].set(cjk_char_count[cjk_enc])
    for unicode_enc in global_var.unicode_list:
        unicode_var[unicode_enc].set(unicode_char_count[unicode_enc])

    if bool_exportfile.get():
        import write_csv
        write_csv.write(path_leaf(filename), cjk_char_count, unicode_char_count, current_lang)


#get filename only
def path_leaf(path):
    head, tail = ntpath.split(path)
    return tail or ntpath.basename(head)

#add icon
pic_path = os.path.join(global_var.main_directory, 'appicon.png')
main.tk.call('wm', 'iconphoto', main._w, PhotoImage(file=pic_path))

#if dragged file onto exe, received file path as parameter 1 and directly start counting, icon loaded alrdy by this
if first_time:
    #reset to finish counting first time
    first_time = False
    try:
        if args.filename is None:
            raise FileNotFoundError
        font_filename_arg = args.filename
        if font_filename_arg.lower().endswith((".otf",".ttf",".woff",".woff2",".otc",".ttc")):
            open_file(font_filename_arg)
    except: 
        pass

#UI stay appear
main.mainloop()