#!/usr/bin/env python3

import gi
import os

gi.require_version("Gtk", "3.0")
from gi.repository import Gtk, Gio


css = b'''
.button {
    min-height: 38px;
    font-family: 'Ubuntu Condensed', sans-serif;
    font-weight: 600;
    font-size: 18px;
    color: #000;
    text-shadow: none;
    outline-style: none;
    border-radius: 25px;
    border-width: 0;
    box-shadow: none;
    padding: 2px 20px;
    margin: 0;
}
.button--f90 {
    background-color: #f90;
    background: #f90; /* for Ubuntu */
}
.button--c9c {
    background-color: #c9c;
    background: #c9c; /* for Ubuntu */
}
.button--99c {
    background-color: #99c;
    background: #99c; /* for Ubuntu */
}
.button--c66 {
    background-color: #c66;
    background: #c66; /* for Ubuntu */
}
.button--fc9 {
    background-color: #fc9;
    background: #fc9; /* for Ubuntu */
}
.button--99f {
    background-color: #99f;
    background: #99f; /* for Ubuntu */
}
.button--f96 {
    background-color: #f96;
    background: #f96; /* for Ubuntu */
}
.button--c69 {
    background-color: #f96;
    background: #f96; /* for Ubuntu */
}
.category {
    font-family: 'Ubuntu Condensed', sans-serif;
    font-weight: 600;
    font-size: 24px;
    color: #f90;
}
.line-end {
    min-width: 20px;
    background-color: #99F;
    background: #99F; /* for Ubuntu */
    outline-style: none;
    border-width: 0;
    box-shadow: none;
    padding: 0;
    margin: 0;
}
.line-end--left {
    border-radius: 20px 0 0 20px;
}
.line-end--right {
    border-radius: 0 20px 20px 0;
}
.window {
    background-color: #000;
}
'''


def sort_dict_by_key(data):
    new_dict = {}
    for key in sorted(data.keys()):
        new_dict[key] = data[key]
    return new_dict


def adjust_name(name):
    if name != name[:18]:
        return name[:18] + "…"
    return name


class CategoryLabel(Gtk.Box):
    def __init__(self, label, css_provider):
        Gtk.Box.__init__(self, orientation=Gtk.Orientation.HORIZONTAL, spacing=8)

        line_end_left = Gtk.Label()
        line_end_left.get_style_context().add_class("line-end")
        line_end_left.get_style_context().add_class("line-end--left")
        line_end_left.get_style_context().add_provider(css_provider, Gtk.STYLE_PROVIDER_PRIORITY_USER)
        self.add(line_end_left)

        category_label = Gtk.Label(label=label)
        category_label.get_style_context().add_class("category")
        category_label.get_style_context().add_provider(css_provider, Gtk.STYLE_PROVIDER_PRIORITY_USER)
        self.add(category_label)

        line_end_right = Gtk.Label()
        line_end_right.get_style_context().add_class("line-end")
        line_end_right.get_style_context().add_class("line-end--right")
        line_end_right.get_style_context().add_provider(css_provider, Gtk.STYLE_PROVIDER_PRIORITY_USER)
        self.add(line_end_right)


class AppButton(Gtk.Button):
    COLORS = [
        "c9c",
        "99c",
        "c66",
        "99f",
        "f96",
        "c69"]

    def __init__(self, label, css_provider, file_path):
        Gtk.Button.__init__(self, label=adjust_name(label))
        self.file_path = file_path

        color_index = sum([ord(c) for c in label]) % len(self.COLORS)
        color = self.COLORS[color_index]

        self.set_alignment(1, 1)
        self.connect("clicked", self.start_application)
        self.get_style_context().add_class("button")
        self.get_style_context().add_class("button--{}".format(color))
        self.get_style_context().add_provider(css_provider, Gtk.STYLE_PROVIDER_PRIORITY_USER)

    def start_application(self, widget):
        launcher = Gio.DesktopAppInfo.new_from_filename(self.file_path)
        launcher.launch([], None)
        Gtk.main_quit()


class LcarsdeApplicationStarter(Gtk.Window):
    """
    Application selector main window
    """
    PREFERRED_CATEGORIES = ["System", "Game", "Network", "Office", "Settings", "AudioVideo", "Development", "Graphics",
                            "Utility"]

    def __init__(self):
        Gtk.Window.__init__(self, title="Application Selector")

        self.applications = {}
        self.load_system_applications()
        self.load_flatpak_applications()
        self.load_user_applications()

        self.css_provider = Gtk.CssProvider()
        self.css_provider.load_from_data(css)

        self.scroll_container = Gtk.ScrolledWindow()
        self.scroll_container.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.AUTOMATIC)

        self.app_container = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=8)
        self.display_apps()

        self.scroll_container.add(self.app_container)
        self.add(self.scroll_container)
        self.connect('size-allocate', self.view_changed)

        self.get_style_context().add_class("window")
        self.get_style_context().add_provider(self.css_provider, Gtk.STYLE_PROVIDER_PRIORITY_USER)

    def view_changed(self, widget, event, data=None):
        adj = self.scroll_container.get_vadjustment()
        adj.set_value(0)

    def add_application(self, file_path, app_data):
        data_lines = app_data.splitlines()
        name = None
        categories = None
        no_display = False
        for line in data_lines:
            if line.startswith("Name="):
                name = line[5:]
            elif line.startswith("Categories="):
                categories = line[11:]
            elif line.startswith("NoDisplay="):
                no_display = line[10:].lower() == "true"

            if name is not None and categories is not None:
                break

        if not name or no_display:
            return

        if not categories:
            categories = "Utility"

        categories = list(categories.split(";"))
        category = self.get_category(categories).strip()
        if category == "":
            category = "Utility"

        if category not in self.applications.keys():
            self.applications[category] = list()

        self.applications[category].append((name, file_path))

    def get_category(self, categories):
        for preferred_category in self.PREFERRED_CATEGORIES:
            if preferred_category in categories:
                return preferred_category
        return categories[0]

    def load_applications(self, app_directory):
        try:
            app_files = [os.path.join(app_directory, f)
                         for f in os.listdir(app_directory)
                         if os.path.isfile(os.path.join(app_directory, f))]
            app_data = set()
            for file_path in app_files:
                try:
                    with open(file_path, 'r') as file:
                        app_data.add((file_path, file.read()))
                except FileNotFoundError:
                    print("Unable to load {0}".format(file_path))

            for (file_path, data) in app_data:
                self.add_application(file_path, data)
        except FileNotFoundError:
            print("directory not found:", app_directory)

    def load_system_applications(self):
        self.load_applications("/usr/share/applications")
        
    def load_flatpak_applications(self):
        self.load_applications("/var/lib/flatpak/exports/share/applications")

    def load_user_applications(self):
        path = "{0}/.local/share/applications".format(os.environ.get('HOME'))
        self.load_applications(path)

    def display_apps(self):
        for category, apps in sort_dict_by_key(self.applications).items():
            category_label = CategoryLabel(category, self.css_provider)
            self.app_container.add(category_label)

            apps.sort(key=lambda e: e[0])
            flow_box = Gtk.FlowBox(homogeneous=True)
            for app in apps:
                flow_box.add(AppButton(app[0], self.css_provider, app[1]))

            self.app_container.add(flow_box)


if __name__ == "__main__":
    win = LcarsdeApplicationStarter()
    win.connect("destroy", Gtk.main_quit)
    win.show_all()
    Gtk.main()