munyal-client/config.py

import os
import shutil
import sys
import pathlib
import pickle

from hashlib import sha256
from tkinter import Button, Entry, Label, StringVar, Tk, filedialog

from uuid import uuid4

from PIL import Image, ImageTk

from misc import alert, get_os


def __get_config_folder():
    _os = get_os()
    if _os == "linux":
        folder = os.path.join(os.getenv("HOME"), ".munyal")
    else:
        folder = os.path.join(os.path.expandvars("%APPDATA%"), "Munyal")
    return folder

def __get_files_folder():
    folder = os.path.join(os.path.expanduser("~"), "Munyal")
    return folder

def get_config():
    for _ in range(3):
        config = __get_config()
        if not config:
            gui_config()
        else:
            return config


def __get_config():
    folder = __get_config_folder()
    config_file = os.path.join(folder, "config")
    if os.path.exists(config_file):
        config_bytes = open(config_file, "rb")
        config = pickle.load(config_bytes)
        config_bytes.close()
        return config


def __validate_values(user, password):
    return True


def _set_config(root, user, password, folder):
    passwd_hash = sha256(password.get().encode("utf-8")).hexdigest()
    if not user:
        alert(root, "Introduce tu usuario de la red Munyal")
    elif not password:
        alert(root, "Introduce tu contraseña")
    elif not folder:
        alert(root, "Elige la carpeta que quieres sincronizar")
    elif not __validate_values(user, passwd_hash):
        alert(
            root,
            "Tu usuario o contraseña son incorrectos, favor de reintentarlo")
    else:
        password.set("")
        folder_config = __get_config_folder()
        path = pathlib.Path(folder_config)
        path.mkdir(parents=True, exist_ok=True)
        with open(os.path.join(folder_config, "config"), "wb") as config_file:
            config_file.write(
                pickle.dumps({
                    "login": {
                        "user": user,
                        "password": passwd_hash
                    },
                    "folder": folder,
                    "uuid": str(uuid4())
                }))
        root.destroy()


def gui_config():
    root = Tk()
    root.geometry("250x420")

    img_logo = Image.open("img/logo.png")
    img_tk = ImageTk.PhotoImage(img_logo.resize((200, 150), Image.ANTIALIAS))

    host = StringVar()
    host.set("localhost")
    host_field = Entry(root, textvariable=host)

    passwd = StringVar()
    passwd_field = Entry(root, textvariable=passwd, show="*")

    folder = StringVar()
    folder.set(__get_files_folder())
    folder_field = Entry(root, textvariable=folder)
    btn_folder = Button(root,
                        text="Examinar",
                        command=lambda: search_folder(folder))

    connect = Button(
        root,
        text="Conectar",
        command=lambda: _set_config(root, host.get(), passwd, folder.get()))

    Label(root, text="MUNYAL").pack()
    Label(root, image=img_tk).pack()
    Label(root, text="").pack()
    Label(root, text="Ruta del servidor").pack()
    host_field.pack()
    Label(root, text="").pack()
    Label(root, text="Contraseña").pack()
    passwd_field.pack()
    Label(root, text="").pack()
    Label(root, text="Carpeta a sincronizar").pack()
    folder_field.pack()
    btn_folder.pack()
    Label(root, text="").pack()
    connect.pack()

    root.mainloop()


def search_folder(field):
    path = filedialog.askdirectory()
    field.set(path)


if __name__ == "__main__":
    if "--restart" in sys.argv:
        if __get_config():
            shutil.rmtree(__get_config_folder())
    print(get_config())