# Python imports
import threading, subprocess
from os.path import isfile
from os import listdir


# Gtk imports
from xdg.DesktopEntry import DesktopEntry

# Application imports
from .mixins import *
from . import Menu, Controller_Data



def threaded(fn):
    def wrapper(*args, **kwargs):
        threading.Thread(target=fn, args=args, kwargs=kwargs).start()

    return wrapper


class Controller(ProcessorMixin, Menu, Controller_Data):
    def __init__(self, _settings, args, unknownargs):
        super().__init__(_settings, args)

        self.setup_controller_data(_settings)

        base_options   = ["[  TO MAIN MENU  ]", "Favorites"]
        self.menu_data = self.get_desktop_files_info(self.app_paths)
        query          = ""

        while True:
            try:
                self.clear_console()
                group = self.call_method("main_menu")["group"]
                self.clear_console()

                if "Search..." in group:
                    query = self.call_method("search_menu")["query"]
                if "[ Set Favorites ]" in group:
                    programs_list       = self.get_sub_group("Search...", "")
                    fixed_programs_list = []

                    for prog in programs_list:
                        fixed_programs_list.append({'name': prog})

                    self.favorites = self.call_method("set_favorites_menu", [fixed_programs_list])["setFaves"]
                    self.save_faves(self.favorites)
                    continue
                if "[ Exit ]" in group:
                    break

                programs_list = ["[  TO MAIN MENU  ]"]
                programs_list += self.get_sub_group(group, query)
                entry     = self.call_method("sub_menu", [group, programs_list])["prog"]

                self.logger.debug(entry)
                if entry not in base_options:
                    self.logger.info(f"[Executing Program] Group: {group} Entry: {entry}")
                    self.execute_program(group, entry)
            except Exception as e:
                self.logger.error(e)



    def get_desktop_files_info(self, paths):
        menu_objects = {
            "Accessories": [],
            "Multimedia":  [],
            "Graphics":    [],
            "Game":        [],
            "Office":      [],
            "Development": [],
            "Internet":    [],
            "Settings":    [],
            "System":      [],
            "Wine":        [],
            "Other":       []
        }

        for path in paths:
            if not "/opt/" in path:
                self.list_and_update_desktop_iles(path, menu_objects);
            else:
                for folder in listdir(path):
                    try:
                        full_path = f"{path}{folder}/"
                        self.list_and_update_desktop_iles(full_path, menu_objects);
                    except Exception as e:
                        self.logger.debug(e)

        return menu_objects

    def list_and_update_desktop_iles(self, path, menu_objects):
        try:
            for f in listdir(path):
                full_path = f"{path}{f}"
                if isfile(full_path) and f.endswith(".desktop"):
                    xdg_object = DesktopEntry(full_path)
                    hidden     = xdg_object.getHidden()
                    nodisplay  = xdg_object.getNoDisplay()
                    type       = xdg_object.getType()
                    groups     = xdg_object.getCategories()
                    # Do not show those marked as hidden or not to display
                    if hidden or nodisplay:
                        continue

                    if type == "Application" and groups != "":
                        title    = xdg_object.getName()
                        comment  = xdg_object.getComment()
                        # icon     = xdg_object.getIcon()
                        mainExec = xdg_object.getExec()
                        tryExec  = xdg_object.getTryExec()

                        group    = ""
                        if "Accessories" in groups or "Utility" in groups:
                            group = "Accessories"
                        elif "Multimedia" in groups or "Video" in groups or "Audio" in groups:
                            group = "Multimedia"
                        elif "Development" in groups:
                            group = "Development"
                        elif "Game" in groups:
                            group = "Game"
                        elif "Internet" in groups or "Network" in groups:
                            group = "Internet"
                        elif "Graphics" in groups:
                            group = "Graphics"
                        elif "Office" in groups:
                            group = "Office"
                        elif "System" in groups:
                            group = "System"
                        elif "Settings" in groups:
                            group = "Settings"
                        elif "Wine" in groups:
                            group = "Wine"
                        else:
                            group = "Other"

                        menu_objects[group].append( {"title":  title,   "groups": groups,
                                                    "comment": comment, "exec": mainExec,
                                                    "tryExec": tryExec, "fileName": f
                                                })
        except Exception as e:
            self.logger.debug(e)


    def get_sub_group(self, group, query = ""):
        desktop_objects = []
        if "Search..." in group:
            group_keys = self.menu_data.keys()
            for group_key in group_keys:
                for opt in self.menu_data[group_key]:
                    keys = opt.keys()
                    if "comment" in keys and len(opt["comment"]) > 0 :
                        if query.lower() in opt["comment"].lower():
                            desktop_objects.append( opt["title"] + " || " + opt["comment"] )
                    if query.lower() in opt["title"].lower() or query.lower() in opt["fileName"].lower():
                            desktop_objects.append( opt["title"] + " || " + opt["fileName"].replace(".desktop", "") )
        elif "Favorites" in group:
            desktop_objects = self.favorites
        else:
            for opt in self.menu_data[group]:
                keys = opt.keys()
                if "comment" in keys and len(opt["comment"]) > 0 :
                    desktop_objects.append( opt["title"] + " || " + opt["comment"] )
                else:
                    desktop_objects.append( opt["title"] + " || " + opt["fileName"].replace(".desktop", "") )

        return desktop_objects




    def tear_down(self, widget=None, eve=None):
        quit()

    def get_clipboard_data(self):
        proc    = subprocess.Popen(['xclip','-selection', 'clipboard', '-o'], stdout=subprocess.PIPE)
        retcode = proc.wait()
        data    = proc.stdout.read()
        return data.decode("utf-8").strip()

    def set_clipboard_data(self, data):
        proc = subprocess.Popen(['xclip','-selection','clipboard'], stdin=subprocess.PIPE)
        proc.stdin.write(data)
        proc.stdin.close()
        retcode = proc.wait()