# Python imports
import threading, subprocess, traceback
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 program in programs_list:
                        fixed_programs_list.append({'name': program})

                    self.favorites = self.call_method("set_favorites_menu", [fixed_programs_list])["set_faves"]
                    self.settings.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"]

                if entry not in base_options:
                    self.execute_program(self.flat_menu_data, entry)
            except Exception as e:
                self.logger.debug(f"Traceback:  {traceback.print_exc()}")
                self.logger.debug(f"Exception:  {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:
            self.list_and_update_desktop_iles(path, menu_objects);

        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()
                        main_exec = xdg_object.getExec()
                        try_exec  = 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"

                        chunk_data = {
                                        "groups": groups, "comment": comment,
                                        "exec": main_exec, "try_exec": try_exec,
                                        "fileName": f
                                    }

                        menu_objects[group][title] = chunk_data
                        self.flat_menu_data[title] = chunk_data
        except Exception as e:
            self.logger.debug(e)


    def get_sub_group(self, group, query = ""):
        desktop_objects = []
        if "Search..." in group:
            for key in self.flat_menu_data.keys():
                option = self.flat_menu_data[key]
                keys   = option.keys()

                if "comment" in keys and len(option["comment"]) > 0:
                    if query.lower() in option["comment"].lower():
                        desktop_objects.append( f"{key} || {option['comment']}" )
                elif query.lower() in key.lower() or query.lower() in option["fileName"].lower():
                        desktop_objects.append( f"{key} || {option['fileName'].replace('.desktop', '')}" )
        elif "Favorites" in group:
            desktop_objects = self.favorites
        else:
            for option in self.menu_data[group]:
                keys = option.keys()
                if "comment" in keys and len(option["comment"]) > 0 :
                    desktop_objects.append( f"{option['title']} || {option['comment']}" )
                else:
                    desktop_objects.append( f"{option['title']} || {option['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()