app.py

import os, stat, re, sys, platform, subprocess, string, shutil, configparser, tkinter as tk
from tkinter import ttk, simpledialog
from tkinter.messagebox import askyesno
from tkinter import filedialog
from tkinter import messagebox
from pathlib import Path
from ftplib import FTP
from send2trash import send2trash

from tkinter import *
import tkinter.ttk as ttk
from functools import partial

from back.basic_instruction import *
from back.git_init import *

from back.git_status import *
from back.git_add import *
from back.git_rm import *
from back.git_rm_cached import *
from back.git_restore import *
from back.git_restore_staged import *
from back.git_mv import *
from back.git_commit import *

from back.git_history import *
from back.git_b_create import *
from back.git_b_delete import *
from back.git_b_rename import *
from back.git_b_checkout import *
from back.git_clone import *
from back.git_merge import *

# Interface

def sort_name_reverse():
    global sort
    global reverse
    sort = "name"
    if reverse == False:
        reverse = True
        tree.heading("#0", text="   Name ↓")
    elif reverse == True:
        reverse = False
        tree.heading("#0", text="   Name ↑")
    tree.heading("#1", text="Size")
    update_files(entry.get())


def sort_size_reverse():
    global sort
    global reverse
    sort = "size"
    if reverse == False:
        reverse = True
        tree.heading("#1", text="Size ↓")
    elif reverse == True:
        reverse = False
        tree.heading("#1", text="Size ↑")
    tree.heading("#0", text="   Name")
    update_files(entry.get())


def move_up():
    global last_lower_folder
    up_path = entry.get().rsplit("/" if ftp != None else slash, 1)
    last_lower_folder = up_path[1]
    update_files(up_path[0])


def show_hide():
    global hidden
    if hidden_menu.get() == 1:
        hidden = True
    elif hidden_menu.get() == 0:
        hidden = False
    update_files(entry.get())


def open_right_menu(event):
    try:
        right_menu.tk_popup(event.x_root, event.y_root)
    finally:
        right_menu.grab_release()


def scan_disks_add_buttons():
    if sys.platform == "win32":
        letters = string.ascii_uppercase
        letter_c = 0
        column = 2
        for _ in range(26):
            disk = letters[letter_c]
            if os.path.exists(f"{disk}:{slash}"):
                tk.Button(frame_b, text=disk.lower(), font=("Arial", 14), relief="flat", bg="white", fg="black",
                    command=lambda disk=disk:update_files(f"{disk}:{slash}")).grid(column=column, row=1)
                column += 1
            letter_c += 1
    if sys.platform == "linux":
        column = 2
        os_user = home_path.rsplit(slash, 1)[1]
        if os.path.exists(f"/media/{os_user}/"):
            for l_disk in os.listdir(f"/media/{os_user}/"):
                tk.Button(frame_b, text=l_disk[0].lower(), font=("Arial", 14), relief="flat", bg="white", fg="black",
                    command=lambda l_disk=l_disk:update_files(f"/media/{os_user}/{l_disk}")).grid(column=column, row=1)
                column += 1


def up_down_focus():
    """Focus item on start"""
    if tree.focus() == "":
        try:
            item = tree.get_children()[0]
            tree.selection_set(item)
            tree.focus(item)
            tree.see(item)
        except:pass


def select():
    """Enable some menu items"""
    if len(tree.selection()) > 0:
        right_menu.entryconfig("Open", state="normal")
        if ftp == None:
            right_menu.entryconfig("Copy", state="normal")
            right_menu.entryconfig("Rename", state="normal")
            right_menu.entryconfig("Delete in trash", state="normal")
        if ftp != None:
            right_menu.entryconfig("Copy to folder", state="normal")


def remove_selection(menu_only=False):
    """Remove selection + Disable some menu items"""
    if menu_only == False:
        tree.selection_remove(tree.focus())
    right_menu.entryconfig("Open", state="disabled")
    right_menu.entryconfig("Copy", state="disabled")
    right_menu.entryconfig("Rename", state="disabled")
    right_menu.entryconfig("Delete in trash", state="disabled")
    if ftp != None:
        right_menu.entryconfig("Copy to folder", state="disabled")


def click():
    stop = False
    for i in tree.selection():
        path = tree.item(i)["values"][1]
        if tree.item(i)["values"][0] == "dir":
            if stop == False:
                update_files(path)
                stop = True
        else:
            if ftp == None:
                if sys.platform == "win32":
                    os.startfile(path)
                else:
                    opener = "open" if sys.platform == "darwin" else "xdg-open"
                    subprocess.call([opener, path])

# Operations

def new(goal: str):
    try:
        test = False
        cancel = False
        if goal == "dir":
            info_text = "Enter catalog name"
        elif goal == "file":
            info_text = "Enter file name"
        while test == False and cancel == False:
            name_dir = simpledialog.askstring(title="Files", prompt=info_text, parent=tree_frame)
            if name_dir is not None:
                test = True
                for f in os.listdir(last_path):
                    if f.lower() == name_dir.lower():
                        test = False
                        info_text = "Name is taken"
            else:
                cancel = True
        if test == True:
            path = os.path.join(last_path, name_dir)
            if goal == "dir":
                os.mkdir(path)
            elif goal == "file":
                Path(path).touch()
            update_files(last_path)

    except Exception as e:
        tk.messagebox.showerror(title="Error", message=str(e))


def copy():
    list_i = []
    for i in tree.selection():
        path = tree.item(i)["values"][1]
        if "\\" in path:
            path = path.replace("\\", "/")
        list_i.append(f"'{path}'")

    if len(list_i) > 1:
        items = ",".join(list_i)
    else:
        items = list_i[0]

    if sys.platform == "win32":
        os.system(f"powershell.exe Set-Clipboard -path {items}")
        right_menu.entryconfig("Paste", state="normal")
    else:
        global non_win_clipboard
        non_win_clipboard = items
        right_menu.entryconfig("Paste", state="normal")


def paste():
    if sys.platform == "win32":
        clipboard = window.selection_get(selection="CLIPBOARD").replace("/", slash).split("\n")
    else:
        clipboard = non_win_clipboard.replace("'", "").split(",")

    for source in clipboard:
        edit = source.rsplit(slash, 1)
        # Search copies, create destination path
        file_copies = 1
        for f in os.listdir(entry.get()):
            if f.lower() == edit[1].lower():
                file_copies += 1
        if file_copies > 1:
            while True:
                for f in os.listdir(entry.get()):
                    if f.lower() == f"({file_copies}){edit[1]}".lower():
                        file_copies += 1
                        continue
                break
            destination = entry.get() + slash + f"({file_copies})" + edit[1]
        else:
            destination = entry.get() + slash + edit[1]
        # Paste
        if os.path.isdir(source):
            shutil.copytree(source, destination)
        else:
            shutil.copy2(source, destination)

    update_files(entry.get())


def delete():
    try:
        del_list = []
        for i in tree.selection():
            del_path = tree.item(i)["values"][1]
            if os.path.exists(del_path):
                del_list.append(del_path)
        answer = askyesno(title="Files", message=f"Delete {len(tree.selection())} objects in trash?")
        if answer:
            for di in del_list:
                send2trash(di)
    except:pass
    update_files(entry.get())


def rename():
    for i in tree.selection():
        r_path = tree.item(i)["values"][1]
        e_path = r_path.rsplit(slash, 1)
        if os.path.exists(r_path):
            info_text = f"Rename '{e_path[1]}'"
            test = False
            while test == False:
                test = True
                new_name = simpledialog.askstring(title="Files", prompt=info_text, parent=tree_frame, initialvalue=e_path[1])
                if new_name is not None:
                    for f in os.listdir(entry.get()):
                        if f.lower() == new_name.lower() and new_name.lower() != e_path[1].lower():
                            test = False
                            info_text = "Name is taken"
            if new_name is not None and new_name.lower() != e_path[1].lower():
                n_path = e_path[0] + slash + new_name
                os.rename(r_path, n_path)
    update_files(entry.get())


def update_files(orig_dirname: str):
    cd_dir(orig_dirname)
    def convert_size(var) -> tuple:
        if type(var) == type(1):
            byte_size = var
        else:
            byte_size = os.stat(var).st_size
        #
        if byte_size >= 1000000000000:
            size = str(round(byte_size/1000000000000, 2)) + " TB"
        elif byte_size >= 1000000000:
            size = str(round(byte_size/1000000000, 2)) + " GB"
        elif byte_size >= 1000000:
            size = str(round(byte_size/1000000, 2)) + " MB"
        elif byte_size >= 1000:
            size = str(round(byte_size/1000, 2)) + " KB"
        elif byte_size < 1000:
            size = str(byte_size) + " B"
        return size, byte_size

    try:
        global last_path
        global ftp
        global url_ftp
        dirname = orig_dirname
        # FTP
        if "ftp://" in dirname:
            x = dirname.split("//", 1)
            if "/" in x[1]:
                dirname = "/" + x[1].split("/", 1)[1]
            else:
                dirname = "/"
            #
            if ftp == None:
                ftp = FTP("")
                try:
                    if ":" in x[1]:
                        ftp_split = x[1].split(":", 1)
                        ftp.connect(ftp_split[0],int(ftp_split[1]))
                        url_ftp = f"ftp://{ftp_split[0]}:{ftp_split[1]}"
                    else:
                        ftp.connect(x[1])
                        url_ftp = f"ftp://{x[1]}"
                    ftp.login()
                    #
                    right_menu.add_command(label="Copy to folder", command=copy_from_ftp, state="disable")
                    right_menu.entryconfig("Show hidden files", state="disable")
                    right_menu.entryconfig("New file", state="disable")
                    right_menu.entryconfig("New catalog", state="disable")
                except:
                    ftp = None
                    url_ftp = None
        else:
            if ftp != None:
                ftp.quit()
                ftp = None
                url_ftp = None
                #
                right_menu.delete("Copy to folder")
                right_menu.entryconfig("Show hidden files", state="normal")
                right_menu.entryconfig("New file", state="normal")
                right_menu.entryconfig("New catalog", state="normal")
        # Check path
        if ftp == None and op_slash in dirname:
            dirname = dirname.replace(op_slash, slash)
        elif ftp != None and "\\" in dirname:
            dirname = dirname.replace("\\", "/")
        if re.match(r".+\\$", dirname) or re.match(r".+/$", dirname):
            dirname = dirname[0:-1]
        if re.match(r"\w:$", dirname) or dirname == "":
            if ftp == None:
                dirname = dirname + slash
            else:
                dirname = "/"
        # Scan
        files_list, dirs_list = [], []
        if ftp == None:
            files = os.scandir(dirname)
            for f in files:
                f_stat = f.stat()
                size = convert_size(f_stat.st_size)
                if f.is_dir():
                    if hidden == False:
                        if sys.platform == "win32":
                            if not f.is_symlink() and not bool(f_stat.st_file_attributes & stat.FILE_ATTRIBUTE_HIDDEN):
                                dirs_list.append([f.name, "dir", f.path, folder_icon])
                        else:
                            if not f.name.startswith("."):
                                dirs_list.append([f.name, "dir", f.path, folder_icon])
                    else: 
                        if sys.platform == "win32":
                            if bool(f_stat.st_file_attributes & stat.FILE_ATTRIBUTE_HIDDEN):
                                dirs_list.append([f.name, "dir", f.path, folder_hidden_icon])
                            else:
                                if f.is_symlink():
                                    dirs_list.append([f.name, "dir", f.path, folder_hidden_icon])
                                else:
                                    dirs_list.append([f.name, "dir", f.path, folder_icon])
                        else:
                            if f.name.startswith("."):
                                dirs_list.append([f.name, "dir", f.path, folder_hidden_icon])
                            else:
                                dirs_list.append([f.name, "dir", f.path, folder_icon])
                if f.is_file():
                    if hidden == False:
                        if sys.platform == "win32":
                            if not bool(f_stat.st_file_attributes & stat.FILE_ATTRIBUTE_HIDDEN):
                                files_list.append([f.name, size[0], f.path, file_icon, size[1]])
                        else:
                            if not f.name.startswith("."):
                                files_list.append([f.name, size[0], f.path, file_icon, size[1]])
                    else:
                        if sys.platform == "win32":
                            if bool(f_stat.st_file_attributes & stat.FILE_ATTRIBUTE_HIDDEN):
                                files_list.append([f.name, size[0], f.path, file_hidden_icon, size[1]])
                            else:
                                files_list.append([f.name, size[0], f.path, file_icon, size[1]])
                        else:
                            if f.name.startswith("."):
                                files_list.append([f.name, size[0], f.path, file_hidden_icon, size[1]])
                            else:
                                files_list.append([f.name, size[0], f.path, file_icon, size[1]])
        # FTP
        else:
            ftp.cwd(dirname)
            try:
                for f in ftp.mlsd():
                    if f[1]["type"] == "dir":
                        dirs_list.append([f[0], "dir", f"{orig_dirname}/{f[0]}", folder_icon])
                    elif f[1]["type"] == "file":
                        size = convert_size(int(f[1]["size"]))
                        files_list.append([f[0], size[0], f"{orig_dirname}/{f[0]}", file_icon, size[1]])
            except:
                ftp.voidcmd('TYPE I')
                for f in ftp.nlst():
                    try:
                        if "." in f and not f.startswith("."):
                            size = convert_size(ftp.size(f))
                            files_list.append([f, size[0], f"{orig_dirname}/{f}", file_icon, size[1]])    
                        else:
                            dirs_list.append([f, "dir", f"{orig_dirname}/{f}", folder_icon])
                    except:
                        dirs_list.append([f, "dir", f"{orig_dirname}/{f}", folder_icon])
        # Sorting
        if sort == "size":
            if reverse == False:
                dirs_list.sort(key=lambda s: s[0])
                files_list.sort(key=lambda s: s[4])
            if reverse == True:
                dirs_list.sort(key=lambda s: s[0], reverse=True)
                files_list.sort(key=lambda s: s[4], reverse=True)
        elif sort == "name":
            if reverse == False:
                dirs_list.sort(key=lambda f: f[0])
                files_list.sort(key=lambda f: f[0])
            elif reverse == True:
                dirs_list.sort(key=lambda f: f[0], reverse=True)
                files_list.sort(key=lambda f: f[0], reverse=True)
        # Clean old data
        for item in tree.get_children():
            tree.delete(item)
        entry.delete(0, "end")
        # Add new data
        ################state icon 추가부분################
        count = 0

        result = git_status()
        #print(result)

        if result == None: # .git 파일이 없는 디렉토리인 경우
            print("no exist .git")
            for i in dirs_list:
                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=i[3])
                count += 1
            for i in files_list:
                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=i[3])       
                count += 1
        else: # .git 파일이 있는 디렉토리인 경우
             # directory insert
            inserted_folder_list = []
            for i in dirs_list:
               folder_inserted = False
               for j in range(len(result)):
                    for k in range(len(result[str(j)])):
                        folder_name = result[str(j)][k].rsplit('/', maxsplit=1)[0] # 파일명 제외한 폴더 경로 추출
                        folder_name2 = result[str(j)][k].split('/', maxsplit=1)[0] # 최상위 폴더명만 추출
                        git_path = os.popen('git rev-parse --show-toplevel').read().strip() + '/'  
                        #print("1 " + i[2].replace('\\', '/').replace(git_path, ''))
                        #print("2 " + folder_name)
                        if ((i[2].replace('\\', '/').replace(git_path, '') == folder_name and ((folder_name in inserted_folder_list) == False))) or ((i[2].replace('\\', '/').replace(git_path, '') == folder_name2) and ((folder_name2 in inserted_folder_list) == False)):
                            if j == 0:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=untracked_folder_icon)
                                if(i[2].replace('\\', '/').replace(git_path, '') == folder_name):
                                    inserted_folder_list.append(folder_name)
                                else:
                                    inserted_folder_list.append(folder_name2)
                                folder_inserted = True
                                break
                            elif j == 1:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=modified_folder_icon)
                                if(i[2].replace('\\', '/').replace(git_path, '') == folder_name):
                                    inserted_folder_list.append(folder_name)
                                else:
                                    inserted_folder_list.append(folder_name2)
                                folder_inserted = True
                                break
                            elif j == 2:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=staged_folder_icon)
                                if(i[2].replace('\\', '/').replace(git_path, '') == folder_name):
                                    inserted_folder_list.append(folder_name)
                                else:
                                    inserted_folder_list.append(folder_name2)
                                folder_inserted = True
                                break
                            elif j == 3:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=commited_folder_icon)
                                if(i[2].replace('\\', '/').replace(git_path, '') == folder_name):
                                    inserted_folder_list.append(folder_name)
                                else:
                                    inserted_folder_list.append(folder_name2) 
                                folder_inserted = True
               if folder_inserted == False: # 숨김 처리된 디렉토리 표시하기 위해(윈도우만 가능)
                   tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=i[3])                                       
               count += 1

            # file insert
            inserted_file_list = []
            for i in files_list:
                file_inserted = False                    
                for j in range(len(result)):
                    for k in range(len(result[str(j)])):
                        git_path = os.popen('git rev-parse --show-toplevel').read().strip() + '/'
                        if (i[2].replace('\\', '/').replace(git_path, '') == result[str(j)][k] and i[0] == result[str(j)][k].split('/')[-1]) and ((i[0] in inserted_file_list) == False):
                            if j == 0:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=untracked_file_icon)
                                inserted_file_list.append(i[0])
                                file_inserted = True
                                break
                            elif j == 1:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=modified_file_icon)
                                inserted_file_list.append(i[0])
                                file_inserted = True
                                break
                            elif j == 2:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=staged_file_icon)
                                inserted_file_list.append(i[0])
                                file_inserted = True
                                break
                            elif j == 3:
                                tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=commited_file_icon)
                                inserted_file_list.append(i[0])
                                file_inserted = True
                                break
                if file_inserted == False: # 숨김 처리된 파일 표시하기 위해(윈도우만 가능)
                    tree.insert("", tk.END, text=i[0], values=[f"{i[1]}", i[2]], open=False, image=i[3])             
                count += 1


        print_curr_branch()
        draw_tree()
                
            
    #################################################################
        if ftp == None:
            last_path = dirname
            entry.insert("end", dirname)
        else:
            last_path = ftp.pwd()
            entry.insert("end", f"{url_ftp}{ftp.pwd()}")
        #
        label["text"]=f"   {str(count)} objects"
        # Set title = folder name
        if ftp == None:
            if re.match(r"\w:\\$", dirname):
                window.title(f"Disk ({dirname[0:2]})")
            elif dirname == slash:
                window.title("Files")
            else:
                window.title(dirname.rsplit(slash, 1)[1])
        else:
            window.title(url_ftp)
        #
        remove_selection(menu_only=True)
        # If clipboard on windows has a path - change button
        if sys.platform == "win32" and ftp == None:
            try:
                if re.match(r"\w:", window.selection_get(selection="CLIPBOARD")):
                    right_menu.entryconfig("Paste", state="normal")
                else:
                    right_menu.entryconfig("Paste", state="disabled")
            except:
                right_menu.entryconfig("Paste", state="disabled")
        # Focus on the folder from which you returned
        if tree.focus() == "" and last_lower_folder != None:
            for item in tree.get_children():
                if tree.item(item)["text"] == last_lower_folder:
                    tree.selection_set(item)
                    tree.focus(item)
                    tree.see(item)
                    break

    except Exception as e:
        tk.messagebox.showerror(title="Error", message=str(e))


def copy_from_ftp():
    path = filedialog.askdirectory()
    print(path)
    for i in tree.selection():
        text = tree.item(i)["text"]
        with open(f"{path}/{text}", "wb") as file:
            ftp.retrbinary(f"RETR {text}", file.write)
            file.close()


# Fix graphic on Win 10
if sys.platform == "win32"and platform.release() == "10":
    from ctypes import windll
    windll.shcore.SetProcessDpiAwareness(1)

# Ini config home path, showing hidden files + other variables
config = configparser.ConfigParser()
config.read("files.ini")
home_path = str(Path.home()) if config["USER SETTINGS"]["home_path"] == "" else config["USER SETTINGS"]["home_path"]
hidden = True if config["USER SETTINGS"]["show_hidden_files"] == "True" else False
sort = "size" if config["USER SETTINGS"]["sort"] == "size" else "name"
reverse = False
last_lower_folder = None
last_path = None
non_win_clipboard = None
slash = "\\" if sys.platform == "win32" else "/"
op_slash = "/" if sys.platform == "win32" else "\\"
ftp = None
url_ftp = None

# Window
window = tk.Tk()
window.resizable(True, True)
window.iconphoto(True, tk.PhotoImage(file="data/icon.png"))
window.minsize(width=800, height=500)

#tab 추가
tab = ttk.Notebook(window)
tab.pack(expand=1, fill="both")
frame_workspace = tk.Frame(tab)
frame_history = tk.Frame(tab)
frame_branch = tk.Frame(tab)
tab.add(frame_workspace, text="Version control")
tab.add(frame_branch, text="Branch management")
tab.add(frame_history, text="Commit History")


frame_up = tk.Frame(frame_workspace, border=1, bg="white")
frame_up.pack(fill="x", side="top")

frame_vc_up = tk.Frame(frame_up, border=1, bg="white")
frame_vc_up.pack(fill="x", side="top")
frame_vc_command = tk.Frame(frame_vc_up, border=2, relief="groove", bg="white")
frame_vc_command.pack(fill = "x", side="left", expand=1)
# frame_clone = tk.Frame(frame_vc_up, border=2, relief="groove", bg="white")
# frame_clone.pack(side="left")

frame_branch_up = tk.Frame(frame_branch, border=1, bg="white")
frame_branch_up.pack(fill="x", side="top")
# label_brnch_command =tk.Label(frame_branch_up, text="Branch management", width =15, font=("Arial", 10),  fg = "grey", bg="white")
# label_brnch_command.pack(side="left")
# frame_branch_command = tk.Frame(frame_branch_up, border=2, relief="groove", bg="white")
# frame_branch_command.pack(fill = "x", side="left", expand=1)

# # current branch
# frame_curr_branch = tk.Frame(frame_branch_up, border=2, relief="groove", bg="white")
# frame_curr_branch.pack(fill = "both", side="right")


frame_b = tk.Frame(frame_up, border=2, relief="groove", bg="white")
frame_b.pack(side="left")


##############version control 영역#############
# Top of window
folder_icon = tk.PhotoImage(file="data/untracked_folder.png")
file_icon = tk.PhotoImage(file="data/untracked_file.png")
folder_hidden_icon = tk.PhotoImage(file="data/icon_folder_hidden.png")
file_hidden_icon = tk.PhotoImage(file="data/icon_file_hidden.png")
home_icon = tk.PhotoImage(file="data/icon_home.png")
up_icon = tk.PhotoImage(file="data/icon_up.png")

# file state icon
modified_file_icon = tk.PhotoImage(file="data/modified_file.png")
untracked_file_icon = tk.PhotoImage(file="data/untracked_file.png")
staged_file_icon = tk.PhotoImage(file="data/staged_file.png")
commited_file_icon = tk.PhotoImage(file="data/commited_file.png")

# directory state icon
modified_folder_icon = tk.PhotoImage(file="data/modified_folder.png")
untracked_folder_icon = tk.PhotoImage(file="data/untracked_folder.png")
staged_folder_icon = tk.PhotoImage(file="data/staged_folder.png")
commited_folder_icon = tk.PhotoImage(file="data/commited_folder.png")

branch_icon = tk.PhotoImage(file="data/branch_icon.png")

#init
def init_bttn_clicked():
    success, message = git_init()
    if success == False:
        show_message(message)
    else: 
        update_files(entry.get())
        show_message(message)
#add
def add_bttn_clicked():
    for i in tree.selection():
        r_path = tree.item(i)["values"][1]
        e_path = r_path.rsplit(slash, 1)
    success, message = git_add(e_path[1])
    if success:
        update_files(entry.get())
    show_message(message)

#restore
def restore_bttn_clicked():
    for i in tree.selection():
        r_path = tree.item(i)["values"][1]
        e_path = r_path.rsplit(slash, 1)
    success, message = git_restore(e_path[1])
    if success:
        update_files(entry.get())
    show_message(message)


#unstage
def unstage_bttn_clicked():
    for i in tree.selection():
        r_path = tree.item(i)["values"][1]
        e_path = r_path.rsplit(slash, 1)
    success, message = git_restore_staged(e_path[1])
    if success:
        update_files(entry.get())
    show_message(message)

#remove
def rm_bttn_clicked():
    for i in tree.selection():
        r_path = tree.item(i)["values"][1]
        e_path = r_path.rsplit(slash, 1)
    success, message = git_rm(r_path)
    if success:
        update_files(entry.get())
    show_message(message)

#rm_cached
def rm_cached_bttn_clicked():
    for i in tree.selection():
        r_path = tree.item(i)["values"][1]
        e_path = r_path.rsplit(slash, 1)
    success, message = git_rm_cached(e_path[1])
    if success:
        update_files(entry.get())
    show_message(message)

# mv 
def mv_bttn_clicked(input):
    new_name=input.get() # 입력받은 새 이름 new_name에 저장
    for i in tree.selection():
        r_path = tree.item(i)["values"][1]
        e_path = r_path.rsplit(slash, 1)
    success, message = git_mv(e_path[1], new_name)
    if success:
        update_files(entry.get())
    show_message(message)
    mv_new_win.destroy() # 새 창 닫기

def show_message(message):
    message_box = messagebox.showinfo("알림", message)

def mv_new_window():
    global mv_new_win
    mv_new_win = Toplevel()
    mv_new_win.title("mv")
    label=tk.Label(mv_new_win, text="새 파일명을 입력하세요", bg="white")
    label.pack()
    input = Entry(mv_new_win, width=30)
    input.pack()
    cnfrm_button = Button(mv_new_win, text="move", relief="flat", bg="white", command=partial(mv_bttn_clicked, input))
    cnfrm_button.pack()

# commit
def commit_bttn_clicked(input):
    commit_message=input.get() # 입력받은 commit message commit_message에 저장
    success, message = git_commit(commit_message)
    if success:
        update_files(entry.get())
    show_message(message)
    cmmt_new_win.destroy()

def commit_new_window():
    global cmmt_new_win
    cmmt_new_win = Toplevel()
    cmmt_new_win.title("commit")

    # staged 파일 목록 영역
    frame_added_files = Frame(cmmt_new_win, border=2, relief="groove", bg="white")
    frame_added_files.pack(side="top", fill="both", expand=True)
    
    tree_frame = tk.Frame(frame_added_files, border=1, relief="flat", bg="white")
    tree_frame.pack(expand=1, fill="both")
    treeview = ttk.Treeview(tree_frame, selectmode="extended", show="tree headings", style="mystyle.Treeview")
    treeview.pack(side="left", expand=1, fill="both")
    treeview.heading("#0", text="added files")
    style = ttk.Style()
    style.configure("Treeview", rowheight=30, font=("Arial", 12))
    style.configure("Treeview.Heading", font=("Arial", 12), foreground="grey")
    style.layout("mystyle.Treeview", [("mystyle.Treeview.treearea", {"sticky":"nswe"})])
    scrollbar = ttk.Scrollbar(tree_frame, orient="vertical", command=treeview.yview)
    treeview.configure(yscroll=scrollbar.set)
    scrollbar.pack(side="right",fill="y")

    status = git_status()
    for i in range(len(status['2'])):
        treeview.insert("", tk.END, text=status['2'][i], values="", open=False, image=staged_file_icon)
        

    # commit message 입력 영역
    frame_commit_message = Frame(cmmt_new_win, border=2, relief="groove", bg="white")
    frame_commit_message.pack(side="bottom", fill="both", expand=True)
     
    label=tk.Label(frame_commit_message, text="commit message를 입력하세요", bg="white")
    label.pack()
    input = tk.Entry(frame_commit_message, bg="white", width=50)
    input.pack()
    cnfrm_button = tk.Button(frame_commit_message, text="commit", relief="groove", bg="white", border=1, command=partial(commit_bttn_clicked, input))
    cnfrm_button.pack()


# version control 관련 버튼
init_bttn = tk.Button(frame_vc_command, text="init", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 7, command=init_bttn_clicked)
add_bttn = tk.Button(frame_vc_command, text="add", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 7, command = add_bttn_clicked)
restore_bttn = tk.Button(frame_vc_command, text="restore", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 7, command=restore_bttn_clicked)
unstage_bttn = tk.Button(frame_vc_command, text="restore --staged", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 15, command=unstage_bttn_clicked)
rm_bttn = tk.Button(frame_vc_command, text="remove", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 7, command=rm_bttn_clicked)
rm_cached_bttn = tk.Button(frame_vc_command, text="rm --cached", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 10, command=rm_cached_bttn_clicked)
mv_bttn = tk.Button(frame_vc_command, text="move", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 7, command=mv_new_window)
commit_bttn = tk.Button(frame_vc_command, text="commit", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 7, command=commit_new_window)
init_bttn.pack(side="left", expand=1)
add_bttn.pack(side="left", expand=1)
restore_bttn.pack(side="left", expand=1)
unstage_bttn.pack(side="left", expand=1)
rm_bttn.pack(side="left", expand=1)
rm_cached_bttn.pack(side="left", expand=1)
mv_bttn.pack(side="left", expand=1)
commit_bttn.pack(side="left", expand=1)

tk.Button(frame_b, image=up_icon, width=25, height=32, relief="flat", bg="white", fg="black", command=move_up).grid(column=0, row=1)
tk.Button(frame_b, image=home_icon, width=25, height=32, relief="flat", bg="white", fg="black", command=lambda:update_files(home_path)).grid(column=1, row=1)
entry = tk.Entry(frame_up, font=("Arial", 12), justify="left", highlightcolor="white", highlightthickness=0, relief="groove", border=2)
entry.pack(side="right",fill="both", expand=1)
label = tk.Label(frame_workspace, font=("Arial", 12), anchor="w", bg="white", foreground="grey", border=2)
label.pack(side="bottom",fill="both")

# Tree view
tree_frame = tk.Frame(frame_workspace, border=1, relief="flat", bg="white")
tree_frame.pack(expand=1, fill="both")
tree = ttk.Treeview(tree_frame, columns=(["#1"]), selectmode="extended", show="tree headings", style="mystyle.Treeview")
tree.heading("#0", text="   Name ↑", anchor="w", command=sort_name_reverse) 
tree.heading("#1", text="Size", anchor="w", command=sort_size_reverse)
tree.column("#0", anchor="w")
tree.column("#1", anchor="e", stretch=False, width=120)
tree.pack(side="left", expand=1, fill="both")
style = ttk.Style()
style.configure("Treeview", rowheight=40, font=("Arial", 12))
style.configure("Treeview.Heading", font=("Arial", 12), foreground="grey")
style.layout("mystyle.Treeview", [("mystyle.Treeview.treearea", {"sticky":"nswe"})])
scrollbar = ttk.Scrollbar(tree_frame, orient="vertical", command=tree.yview)
tree.configure(yscroll=scrollbar.set)
scrollbar.pack(side="right",fill="y") 

# On/off hidden files checkbutton on start
hidden_menu = tk.IntVar() 
if hidden == False:
    hidden_menu.set(0)
if hidden == True:
    hidden_menu.set(1)


################## branch 영역 #################
frame_branch_up = tk.Frame(frame_branch, border=1, bg="white")
frame_branch_up.pack(fill="x", side="top")
frame_branch_command = tk.Frame(frame_branch_up, border=2, relief="groove", bg="white")
frame_branch_command.pack(fill = "x", side="left", expand=1)
frame_curr_branch = tk.Frame(frame_branch_up, border=2, relief="groove", bg="white")
frame_curr_branch.pack(fill = "both", side="right")

frame_branch_list = Frame(frame_branch, border=2, relief="groove", bg="white")
frame_branch_list.pack(side="top", fill="both", expand=True)
global treeview
treeview = ttk.Treeview(frame_branch_list, selectmode="extended", show="tree headings", style="mystyle.Treeview")
treeview.pack(side="left", expand=1, fill="both")
treeview.heading("#0", text="branch list")
style = ttk.Style()
style.configure("Treeview", rowheight=30, font=("Arial", 12))
style.configure("Treeview.Heading", font=("Arial", 12), foreground="grey")
style.layout("mystyle.Treeview", [("mystyle.Treeview.treearea", {"sticky":"nswe"})])
scrollbar = ttk.Scrollbar(frame_branch_list, orient="vertical", command=treeview.yview)
treeview.configure(yscroll=scrollbar.set)
scrollbar.pack(side="right",fill="y")

def select_branch(event): 
    selected_item = treeview.focus()
    get_text = treeview.item(selected_item).get('text')
    return get_text
    
def print_curr_branch(): # 현재 브랜치 출력 함수
    for widgets in frame_curr_branch.winfo_children():
      widgets.destroy()
    success, curr_branch = get_current_branch()
    if success == True:
        label_curr_branch=tk.Label(frame_curr_branch, border=2, padx = 5, font=("Arial", 12), fg = "blue", bg="white")
        label_curr_branch.config(text="Current branch: " + curr_branch)
        label_curr_branch.pack()
        
    elif success == False:
        label_curr_branch=tk.Label(frame_curr_branch, border=2, padx = 5, font=("Arial", 12), fg = "blue", bg="white")
        label_curr_branch.config(text="Current branch: not a git repo")
        label_curr_branch.pack()

def draw_tree(): # branch list 출력
    success, curr_branch = get_current_branch()
    if success == True:
        treeview.delete(*treeview.get_children())
        branch_list = get_branches()
        for i in range(len(branch_list[1])):
            treeview.insert("", tk.END, text=branch_list[1][i], values= "", open=False, image=branch_icon)
    elif success == False:
        treeview.delete(*treeview.get_children())
 
# create
def create_bttn_clicked(input):
    branch_name=input.get() 
    success, message = git_b_create(branch_name)
    crt_new_win.destroy()
    show_message(message)
    draw_tree()

def create_new_window():
    success, message = get_current_branch()
    if success == True:
        global crt_new_win
        crt_new_win = Toplevel()
        crt_new_win.attributes("-topmost", True)
        crt_new_win.title("Create")
        
        label=tk.Label(crt_new_win, text="Enter new branch name.", bg="white")
        label.pack()
        input = Entry(crt_new_win, width=30)
        input.pack()
        cnfrm_button = Button(crt_new_win, text="confirm", relief="flat", bg="white", command=partial(create_bttn_clicked, input))
        cnfrm_button.pack()

    elif success == False:
        show_message(message)
        
    
# delete
    
def delete_bttn_clicked():
    selected_brnch = select_branch("<ButtonRelease-1>")
    success, message = git_b_delete(selected_brnch)
    show_message(message)
    if success:
        print_curr_branch()
        draw_tree()
    
       
# rename
def rename_bttn_clicked(input):
    new_name=input.get()
    selcted_brnch = select_branch("<ButtonRelease-1>")
    success, message = git_b_rename(selcted_brnch, new_name)
    show_message(message)
    if success:
        print_curr_branch()
        draw_tree() # update branch list
    rn_new_win.destroy() #close new win

def rename_new_window():
    global rn_new_win
    rn_new_win = Toplevel()
    rn_new_win.title("Rename")
    rn_new_win.attributes("-topmost", True)

    label=tk.Label(rn_new_win, text="Select a branch and enter a new name.", bg="white")
    label.pack()
    input = Entry(rn_new_win, width=30)
    input.pack()

    cnfrm_button = Button(rn_new_win, text="rename", relief="flat", bg="white", command=partial(rename_bttn_clicked, input))
    cnfrm_button.pack(side="bottom")
      
# checkout
def checkout_bttn_clicked():
    selected_brnch = select_branch("<ButtonRelease-1>")
    success, message = git_b_checkout(selected_brnch)
    show_message(message)
    if success:
        update_files(entry.get()) 
    

success, curr_branch = get_current_branch()
draw_tree()

create_bttn = tk.Button(frame_branch_command, text="create", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 15, command=create_new_window)
delete_bttn = tk.Button(frame_branch_command, text="delete", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 15, command = delete_bttn_clicked)
rename_bttn = tk.Button(frame_branch_command, text="rename", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 15, command=rename_new_window)
checkout_bttn = tk.Button(frame_branch_command, text="checkout", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 15, command=checkout_bttn_clicked)
create_bttn.pack(side="left", expand=1)
delete_bttn.pack(side="left", expand=1)
rename_bttn.pack(side="left", expand=1)
checkout_bttn.pack(side="left", expand=1)

treeview.bind('<ButtonRelease-1>', select_branch)

#################################

################## merge 영역 ####################
def merge_bttn_clicked():
    selected_brnch = select_branch("<ButtonRelease-1>")
    success, message = git_merge(selected_brnch)
    if success:
        update_files(entry.get())
    show_message(message)

merge_bttn = tk.Button(frame_branch_command, text="merge", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 15, command=merge_bttn_clicked)
merge_bttn.pack(side="left", expand=1)

#################################################

################# history 영역 ###################
def _on_mousewheel(event):
    canvas.yview_scroll(-1 * (event.delta // 120), "units")

canvas = Canvas(frame_history, bg="white")
canvas.pack(side=LEFT, expand=1, fill="both")
canvas.bind("<MouseWheel>", _on_mousewheel)

vbar = ttk.Scrollbar(frame_history, orient="vertical")
vbar.pack(side=RIGHT,fill=Y)
vbar.config(command=canvas.yview)
canvas.config(yscrollcommand=vbar.set)

def history_clicked(sum):
    error, result_lst = git_history_detail(sum)
    global mv_new_win
    mv_new_win = Toplevel()
    mv_new_win.title("Commit History Detail")
    for i in range(len(result_lst)):
        label=tk.Label(mv_new_win, text=result_lst[i], wraplength = 700, font=("Arial", 12))
        label.pack(side="top", expand=1)
   
def tab_changed(event):
    selected_tab = event.widget.select()
    tab_text = event.widget.tab(selected_tab, "text")
    if tab_text == "Commit History":
        canvas.delete("all") # canvas 비우기
        return_code, history_list = git_history_list()
        if(return_code == 128):
            print("Error")           
        else:
            pos_y = 15
            for i in range(len(history_list)):
                pos_x = 15              
                graph = history_list[i][0]                   
                for j in graph:
                    if(j == '*'):
                        if(pos_x==15):
                            canvas.create_oval(pos_x-5, pos_y-5, pos_x+5, pos_y+5, fill="blue")
                        else:
                            canvas.create_oval(pos_x-5, pos_y-5, pos_x+5, pos_y+5, fill="black")
                        pos_x += 10
                    elif(j == '|'):
                        if(pos_x==15):
                            canvas.create_line(pos_x, pos_y-10, pos_x, pos_y+10, fill="blue")
                        else:
                            canvas.create_line(pos_x, pos_y-10, pos_x, pos_y+10)
                        pos_x += 10
                    elif(j == '/'):
                        if(pos_x==15):
                            canvas.create_line(pos_x+5, pos_y-10, pos_x-5, pos_y+10, fill="blue")
                        else:
                            canvas.create_line(pos_x+5, pos_y-10, pos_x-5, pos_y+10)
                        pos_x += 10
                    elif(j == '\\'):
                        if(pos_x==15):
                            canvas.create_line(pos_x-5, pos_y-10, pos_x+5, pos_y+10, fill="blue")
                        else:
                            canvas.create_line(pos_x-5, pos_y-10, pos_x+5, pos_y+10)
                        pos_x += 10
                    elif(j == '_'):
                        if(pos_x==15):
                            canvas.create_line(pos_x-4, pos_y-10, pos_x+4, pos_y-10, fill="blue")
                        else:
                            canvas.create_line(pos_x-4, pos_y-10, pos_x+4, pos_y-10)
                        pos_x += 10
                    elif(j == ' '):
                        canvas.create_line(pos_x-4, pos_y-10, pos_x+4, pos_y-10, fill="white")
                        pos_x += 10
                if(len(history_list[i]) != 1): # 그래프만 있는 경우 제외
                    text = canvas.create_text(pos_x, pos_y, text= history_list[i][1], fill="black",anchor="w", font=("Arial", 12), tags = "history" + str(i))
                    canvas.tag_bind("history" + str(i), "<Button-1>", lambda event, sum= history_list[i][2]: history_clicked(sum))
                pos_y += 30
        canvas.update_idletasks()
        canvas.configure(scrollregion=canvas.bbox("all"))

       
tab.bind("<<NotebookTabChanged>>", tab_changed)
################################################


################### clone 영역 ##################

def clone_bttn_clicked(rad_var, addr_entry, id_entry, token_entry):
    rad_selected = rad_var.get()
    addr=addr_entry.get() # 입력받은 commit message commit_message에 저장
    id=id_entry.get()
    token=token_entry.get()
    
    success, message = git_clone(rad_selected, addr, id, token)
    show_message(message)
    if success:
        update_files(entry.get())
    cln_new_win.destroy() # close new window

def clone_new_window():
    global cln_new_win
    cln_new_win = tk.Toplevel()
    cln_new_win.title("Clone")
    rad_var = tk.IntVar(value=0)
    clone_frame = tk.Frame(cln_new_win)
    clone_frame.pack(expand=1, fill="both")

    def print_entry():
        if rad_var.get() == 0:
            for widget in clone_frame.winfo_children():
                widget.destroy()
            button_public = tk.Radiobutton(clone_frame, variable=rad_var, text="Public", value=0, command=print_entry)
            button_private = tk.Radiobutton(clone_frame, variable=rad_var, text="Private", value=1, command=print_entry)
            button_public.pack(side="top", expand=1)
            button_private.pack(side="top", expand=1)
            adddr_label=tk.Label(clone_frame, text="GitHub repo address")
            adddr_label.pack(side="top", fill="x", expand=1)
            addr_entry = Entry(clone_frame, width=50)
            addr_entry.pack(side="top", fill="x", expand=1)
            id_entry = Entry(clone_frame, width=50)
            token_entry = Entry(clone_frame, width=50)
            cnfrm_button = Button(clone_frame, text="clone", bg="white", command=lambda: clone_bttn_clicked(rad_var, addr_entry, id_entry, token_entry))
            cnfrm_button.pack(side="top", expand=1)

        elif rad_var.get() == 1:
            for widget in clone_frame.winfo_children():
                widget.destroy()
            button_public = tk.Radiobutton(clone_frame, variable=rad_var, text="Public", value=0, command=print_entry)
            button_private = tk.Radiobutton(clone_frame, variable=rad_var, text="Private", value=1, command=print_entry)
            button_public.pack(side="top", expand=1)
            button_private.pack(side="top", expand=1)
            adddr_label=tk.Label(clone_frame, text="GitHub repo address")
            adddr_label.pack(side="top", fill="x", expand=1)
            addr_entry = Entry(clone_frame, width=50)
            addr_entry.pack(side="top", fill="x", expand=1)
            id_label=tk.Label(clone_frame, text="ID")
            id_label.pack(side="top", fill="x", expand=1)
            id_entry = Entry(clone_frame, width=50)
            id_entry.pack(side="top", fill="x", expand=1)
            token_label=tk.Label(clone_frame, text="Access token")
            token_label.pack(side="top", fill="x", expand=1)
            token_entry = Entry(clone_frame, width=50)
            token_entry.pack(side="top", fill="x", expand=1)
            cnfrm_button = Button(clone_frame, text="clone", bg="white", command=lambda: clone_bttn_clicked(rad_var, addr_entry, id_entry, token_entry))
            cnfrm_button.pack(side="top", expand=1)
        
    button_public = tk.Radiobutton(clone_frame, variable=rad_var, text="Public", value=0, command=print_entry)
    button_private = tk.Radiobutton(clone_frame, variable=rad_var, text="Private", value=1, command=print_entry)
    button_public.pack(side="top", expand=1)
    button_private.pack(side="top", expand=1)

    adddr_label=tk.Label(clone_frame, text="GitHub repo address")
    adddr_label.pack(side="top", fill="x", expand=1)
    addr_entry = Entry(clone_frame, width=50)
    addr_entry.pack(side="top", fill="x", expand=1)
    id_entry = Entry(clone_frame, width=50)
    token_entry = Entry(clone_frame, width=50)
    cnfrm_button = Button(clone_frame, text="clone", bg="white", command=lambda: clone_bttn_clicked(rad_var, addr_entry, id_entry, token_entry))
    cnfrm_button.pack(side="top", expand=1)

clone_bttn = tk.Button(frame_vc_command, text="clone", font=("Arial", 12), relief="flat", bg="white", fg="black", width = 8, command=clone_new_window)
clone_bttn.pack(side="right", expand=1)   

####################################

# Right click menu
right_menu = tk.Menu(tree_frame, tearoff=0, font=("Arial", 12))
right_menu.add_command(label="Open", command=click, state="disabled")
right_menu.add_command(label="Copy", command=copy, state="disabled")
right_menu.add_command(label="Rename", command=rename, state="disabled")
right_menu.add_command(label="Delete in trash", command=delete, state="disabled")
right_menu.add_separator()
right_menu.add_command(label="Paste", command=paste, state="disabled")
right_menu.add_separator()
right_menu.add_command(label="New file", command=lambda:new("file"), state="normal")
right_menu.add_command(label="New catalog", command=lambda:new("dir"), state="normal")
right_menu.add_separator()
right_menu.add_checkbutton(label="Show hidden files", onvalue=1, offvalue=0, variable=hidden_menu, command=show_hide)
right_menu.bind("<FocusOut>", lambda event:right_menu.unpost())# Click elsewhere - close right click menu
scan_disks_add_buttons()
update_files(home_path)
tree.focus_set()

# Keyboard, mouse buttons
tree.bind("<<TreeviewSelect>>", lambda event:select())
tree.bind("<Double-Button-1>", lambda event:click())
tree.bind("<Button-1>", lambda event:remove_selection())
tree.bind("<Return>", lambda event:click())
tree.bind("<BackSpace>", lambda event:move_up())
tree.bind("<Button-3>", open_right_menu)
tree.bind("<Up>", lambda event:up_down_focus())
tree.bind("<Down>", lambda event:up_down_focus())
tree.bind("<Delete>", lambda event:delete())
tree.bind("<Control-c>", lambda event: copy())
tree.bind("<Control-v>", lambda event: paste() if right_menu.entrycget(index=5, option="state") == "normal" else None)
entry.bind("<Return>", lambda event:update_files(entry.get()))
entry.bind("<KP_Enter>", lambda event:update_files(entry.get()))

window.mainloop()