from database import Database
from scrapper import get_files, get_uri, get_url
import utils
from utils import get
import platform
import urllib
import colorama
from terminal_menu.Menu import Menu


uri = ""
preview_site = {}
preview_database = ""

def get_user_choice_by_menu(options: list, title = None, preview_command=None, preview_size: float = 0.3, show_search_hint: bool = False, skip_empty_entries = False, cursor_pos: int = 0):
    menu = Menu(options, title, preview_body_function=preview_command, preview_ratio=preview_size, skip_empty_option=skip_empty_entries, cursor_pos=cursor_pos
            )
    return (menu.show());

def add_site(database: Database) -> dict:
    print("add a site:")
    site = {}
    site.update({"url": input("url without protocol (ip:port):")})
    site.update({"user": input("user(leave blank):")})
    site.update({"password": input("password(leave blank):")})
    site.update({"id": database.get_sites_table_len()})
    name = input(f"name[{str(database.get_sites_table_len())}]")
    if (name == ""):
        name = database.get_sites_table_len()
    site.update({"name": name})
    return (site)

def files_preview(filename: str) -> str:
    if (not filename.endswith("/")):
        return (None)
    files = get_files(preview_site, uri + filename)
    return ("\n".join(files))

def get_files_formated_by_viewing_data(database: Database, files: list, site: dict, path: str):
    out = []
    current_url = utils.get_base_url(site) + urllib.parse.quote(path);
    for file in files:
        if file == "..":
            out.append(file);
            continue
        url = current_url + urllib.parse.quote(file)
        viewing_data = database.get_viewing_data(url);
        if (viewing_data == None) or (viewing_data.get("finished") == False):
            out.append(colorama.Style.BRIGHT + file)
        else:
            out.append(colorama.Fore.LIGHTBLACK_EX + file);
    return (out);

def files(site:dict, path:str, database:Database, pos:int):
    global uri
    global preview_site

    uri = path
    preview_site = site
    current_url = get_url(site, path);
    files = get_files(site, path)
    files_displayed = get_files_formated_by_viewing_data(database, files, site, path)
    choose = get_user_choice_by_menu(files_displayed,
                                 preview_command=files_preview,
                                 preview_size=0.3,
                                 show_search_hint=True,
                                 title=f"Index of {path}",
                                 cursor_pos = pos)
    if (choose == None):
        database.set_last_cursor_pos(0)
        return (None)
    database.set_last_cursor_pos(choose)
    return (files[choose])

def config_preview(site_name:str):
    id = int(site_name.split(": ")[0])
    site = preview_database.get_site_by_id(id)
    if (site == None):
        return
    str = f"""
    url: {site.get('url')}
    user: {site.get('user')}
    password: {site.get('password')}
    """
    return (str)

def site_deleter(sites, database: Database):
    global preview_database

    preview_database = database
    choose = get_user_choice_by_menu(sites,
                                 show_search_hint=True,
                                 preview_title="delete",
                                 preview_command=config_preview)
    return (choose)

def param_editor(lst: list):
    choose = get_user_choice_by_menu(lst)
    return (choose)

def site_editor(lst: list, database: Database):
    global preview_database

    preview_database = database
    choose = get_user_choice_by_menu(lst + ["", "quit"],
                                 skip_empty_entries=True,
                                 show_search_hint=True,
                                 title="Edit")
    if (choose == len(lst) + 1):
        return (None)
    return (choose)

def sites(sites: list):
    nb_site = len(sites)
    choose = get_user_choice_by_menu(get(sites, "name") + ["", "add", "edit", "delete"],
                                 skip_empty_entries=True,
                                 show_search_hint=True)
    return (choose)