import tkinter as tk
import tkinter.ttk as ttk
from tkinter.filedialog import (askopenfilename,
    askopenfilenames, askdirectory, asksaveasfilename)
from tkinter.colorchooser import askcolor

class DialogButton(ttk.Frame):
    def __init__(self, master, 
        height, widthL, widthR, logtype, label=None, text=None, 
        frmDct={}, btnDct={}, enyDct={}, logDct={}):
        w = widthL + widthR
        super().__init__(master, 
            height=height, width = w, **frmDct)
        self.pack(fill=tk.X)

        self.text = tk.StringVar() if not text else text
        ttk.Entry(self, width=widthL, textvariable=self.text, 
            **enyDct).pack(side=tk.LEFT, fill = tk.X, expand=True)
        
        ttk.Button(self, width=widthR, 
            text=self.setLabel(logtype, label),
            command = self.Click, **btnDct).pack(side=tk.RIGHT, padx=5)
        self.logtype = logtype
        self.logDct = logDct

    def setLabel(self, key, label=None):
        if label:
            return label
        labelDct = {
            "文件"   : "選擇文件",
            "文件夾" : "選擇路徑",
            "多文件" : "選擇多個(gè)文件",
            "保存" : "存儲(chǔ)路徑",
            "顏色"   : "選擇顏色",
        }
        return labelDct[key]

    def Click(self):
        typeDct = {
            "文件"  : askopenfilename,
            "文件夾": askdirectory,
            "多文件": askopenfilenames,
            "保存"  : asksaveasfilename,
            "顏色"  : askcolor,
        }
        text = typeDct[self.logtype](**self.logDct)
        if self.logtype == "顏色":
            text = text[1]
        self.text.set(text)

    def get(self):
        return self.text.get()

    def set(self, txt):
        self.text.set(txt)

class ExFolder(ttk.Frame):
    def __init__(self, master,
        folder=None, ex=None, **options):
        super().__init__(master, **options)
        self.pack()

        self.folder = tk.StringVar()
        if folder : self.setFolder(folder)
        self.ex = tk.StringVar()
        if ex : self.setEx(ex)

        self.initWidgets()

    def initWidgets(self):
        ttk.Entry(self, textvariable=self.folder,
            width=10).pack(side=tk.LEFT, padx=2)
        ttk.Entry(self, textvariable=self.ex
            ).pack(side=tk.LEFT, fill=tk.X, expand=True)

    def getFolder(self):
        return self.folder.get()

    def setFolder(self, f):
        self.folder.set(f)

    def getEx(self):
        exs = self.ex.get().split(",")
        return [ex.strip() for ex in exs]

    def setEx(self, exs):
        if type(exs)==list:
            exs = ', '.join([ex.strip() for ex in exs])
        self.ex.set(exs)

class FolderSplit(ttk.Frame):
    def __init__(self, master, **options):
        super().__init__(master, **options)
        self.pack()
        self.fs = FileSystem()

        self.initWidgets()

    def initWidgets(self):
        self.exForders = []
        pDct = dict(side=tk.TOP, expand=True, fill=tk.X)
        self.srcFolder = DialogButton(self, 5, 25, 8,
                                      "文件夾", label="源路徑")
        self.srcFolder.pack(**pDct)
        self.dstFolder = DialogButton(self, 5, 25, 8,
                                      "文件夾", label="目標(biāo)路徑")
        self.dstFolder.pack(**pDct)

        btns = ttk.Frame(self)
        btns.pack(**pDct)
        ttk.Button(btns, text="? ",
            command=self.btnAddFolder).grid(row=0,column=0)
        ttk.Button(btns, text="?",
            command=self.btnDelFolder).grid(row=0,column=1)
        ttk.Button(btns, text="移動(dòng)",
            command=self.btnMove).grid(row=0,column=2)
        ttk.Button(btns, text="撤銷",
            command=self.btnReMove).grid(row=0,column=3)

        self.exFrm = ttk.LabelFrame(self, text="文件夾映射表")
        self.exFrm.pack(**pDct)

        for k, v in self.fs.fileDct.items():
            self.addFolders(self.exFrm, k, v)

    def btnAddFolder(self):
        self.addFolders(self.exFrm, None, None)

    def addFolders(self, frm, folder, exs):
        ef = ExFolder(frm, folder, exs)
        ef.pack(side=tk.TOP, expand=True, fill=tk.X, padx=2, pady=1)
        self.exForders.append(ef)

    def btnDelFolder(self):
        if len(self.exForders) == 0: return
        self.exForders[-1].pack_forget()
        del self.exForders[-1]

    def setExDct(self):
        fileDct = {ef.getFolder() : ef.getEx()
                        for ef in self.exForders}
        self.fs.update(fileDct)

    def btnMove(self):
        src = self.srcFolder.get()
        dst = self.dstFolder.get()
        self.setExDct()
        if dst=="": dst = src
        self.fs.start(src, dst)

    def btnReMove(self):
        self.fs.restore()