# window.py
#
# Change the look of Adwaita, with ease
# Copyright (C) 2022  Gradience Team
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.


import os

from gi.repository import Gtk, Adw, Gio, Gdk

from svglib.svglib import svg2rlg
from reportlab.graphics import renderPM
from material_color_utilities_python import *

from .error import GradienceError
from .settings_schema import settings_schema
from .palette_shades import GradiencePaletteShades
from .option import GradienceOption
from .app_type_dialog import GradienceAppTypeDialog
from .custom_css_group import GradienceCustomCSSGroup
from .presets_manager_window import GradiencePresetWindow
from .plugins_list import GradiencePluginsList
from .modules.utils import buglog
from .constants import rootdir, app_id, build_type


@Gtk.Template(resource_path=f"{rootdir}/ui/window.ui")
class GradienceMainWindow(Adw.ApplicationWindow):
    __gtype_name__ = "GradienceMainWindow"

    content = Gtk.Template.Child()
    toast_overlay = Gtk.Template.Child()
    content_monet = Gtk.Template.Child("content_monet")
    content_plugins = Gtk.Template.Child("content_plugins")
    save_preset_button = Gtk.Template.Child("save-preset-button")
    main_menu = Gtk.Template.Child("main-menu")
    errors_button = Gtk.Template.Child("errors-button")
    errors_list = Gtk.Template.Child("errors-list")
    monet_image_file = None

    def __init__(self, **kwargs):
        super().__init__(**kwargs)

        # Set devel style
        if build_type == "debug":
            self.get_style_context().add_class("devel")

        self.setup_monet_page()
        self.setup_plugins_page()
        self.setup_colors_page()

        self.settings = Gio.Settings(app_id)

        self.settings.bind(
            "window-width",
            self,
            "default-width",
            Gio.SettingsBindFlags.DEFAULT)

        self.settings.bind(
            "window-height",
            self,
            "default-height",
            Gio.SettingsBindFlags.DEFAULT)
        self.settings.bind(
            "window-maximized",
            self,
            "maximized",
            Gio.SettingsBindFlags.DEFAULT)

        self.settings.bind(
            "window-fullscreen",
            self,
            "fullscreened",
            Gio.SettingsBindFlags.DEFAULT)

        self.connect("close-request", self.__close_window)
        self.style_manager = self.get_application().style_manager
        self.first_apply = True

        self.get_default_wallpaper()

    # FIXME: This function works only when build using meson, because Flatpak
    # can't access host's dconf with current config/impl
    def get_default_wallpaper(self):
        background_settings = Gio.Settings("org.gnome.desktop.background")
        if self.style_manager.get_dark():
            picture_uri = background_settings.get_string(
                "picture-uri-dark")
        else:
            picture_uri = background_settings.get_string(
                "picture-uri")
        buglog(picture_uri)
        self.monet_image_file = Gio.File.new_for_uri(picture_uri)
        image_basename = self.monet_image_file.get_basename()
        buglog(image_basename)
        self.monet_file_chooser_button.set_label(image_basename)
        self.monet_image_file = self.monet_image_file.get_path()
        self.monet_file_chooser_button.set_tooltip_text(self.monet_image_file)
        buglog(self.monet_image_file)
        # self.on_apply_button() # Comment out for now, because it always shows
        # that annoying toast on startup

    def on_file_picker_button_clicked(self, *args):
        self.monet_file_chooser_dialog.show()

    def __close_window(self, widegt):
        if self.get_application().is_dirty:
            buglog("app is dirty")

    def on_monet_file_chooser_response(self, widget, response):
        if response == Gtk.ResponseType.ACCEPT:
            self.monet_image_file = self.monet_file_chooser_dialog.get_file()
            image_basename = self.monet_image_file.get_basename()
            self.monet_file_chooser_button.set_label(image_basename)
            self.monet_file_chooser_button.set_tooltip_text(
                image_basename)
        self.monet_file_chooser_dialog.hide()

        if response == Gtk.ResponseType.ACCEPT:
            self.monet_image_file = self.monet_image_file.get_path()
            self.on_apply_button()

    def setup_monet_page(self):

        self.monet_pref_group = Adw.PreferencesGroup()
        self.monet_pref_group.set_name("monet")
        self.monet_pref_group.set_title(_("Monet Engine"))
        self.monet_pref_group.set_description(
            _("Monet is an engine that generates a Material Design 3 palette from an image's color."))

        self.apply_button = Gtk.Button()
        self.apply_button.set_label(_("Apply"))
        self.apply_button.set_valign(Gtk.Align.CENTER)
        self.apply_button.connect("clicked", self.on_apply_button)
        self.apply_button.set_css_classes("suggested-action")
        self.monet_pref_group.set_header_suffix(self.apply_button)
        self.monet_file_chooser_row = Adw.ActionRow()
        self.monet_file_chooser_row.set_title(_("Background Image"))

        self.monet_file_chooser_dialog = Gtk.FileChooserNative()
        self.monet_file_chooser_dialog.set_transient_for(self)

        self.monet_file_chooser_button = Gtk.Button()
        self.monet_file_chooser_button.set_valign(Gtk.Align.CENTER)
        #self.monet_file_chooser_button.set_label(_("Choose a file"))
        # self.monet_file_chooser_button.set_icon_name("folder-pictures-symbolic")

        child_button = Gtk.Box()
        label = Gtk.Label()
        label.set_label(_("Choose a file"))
        child_button.append(label)

        icon = Gtk.Image()
        icon.set_from_icon_name("folder-pictures-symbolic")
        child_button.append(icon)
        child_button.set_spacing(5)

        self.monet_file_chooser_button.set_child(child_button)

        # self.monet_file_chooser_button.set_margin_top(5)
        # self.monet_file_chooser_button.set_margin_bottom(5)

        self.monet_file_chooser_button.connect(
            "clicked", self.on_file_picker_button_clicked
        )
        self.monet_file_chooser_dialog.connect(
            "response", self.on_monet_file_chooser_response
        )
        self.monet_file_chooser_row.add_suffix(self.monet_file_chooser_button)
        self.monet_pref_group.add(self.monet_file_chooser_row)

        self.monet_palette_shades = GradiencePaletteShades(
            "monet", _("Monet Palette"), 6
        )
        self.get_application(
        ).pref_palette_shades["monet"] = self.monet_palette_shades
        self.monet_pref_group.add(self.monet_palette_shades)

        self.tone_row = Adw.ComboRow()
        self.tone_row.set_title(_("Tone"))

        store = Gtk.StringList()
        store_values = []
        for i in range(20, 80, 5):
            store_values.append(str(i))
        for v in store_values:
            store.append(v)
        self.tone_row.set_model(store)
        self.monet_pref_group.add(self.tone_row)

        self.monet_theme_row = Adw.ComboRow()
        self.monet_theme_row.set_title(_("Theme"))

        store = Gtk.StringList()
        store.append("Auto")
        store.append("Dark")
        store.append("Light")
        self.monet_theme_row.set_model(store)
        self.monet_pref_group.add(self.monet_theme_row)

        self.content_monet.add(self.monet_pref_group)

    def on_apply_button(self, *_args):
        if self.monet_image_file:
            if self.monet_image_file.endswith(".svg"):
                drawing = svg2rlg(self.monet_image_file)
                self.monet_image_file = os.path.join(
                    os.environ.get("XDG_RUNTIME_DIR"), "gradience_bg.png")
                renderPM.drawToFile(drawing, self.monet_image_file, fmt='PNG')

            if self.monet_image_file.endswith(".xml"):
                buglog("XML WIP")

            try:
                self.monet_img = Image.open(self.monet_image_file)
            except Exception:
                self.toast_overlay.add_toast(
                    Adw.Toast(title=_("Unsupported background type"))
                )
            else:
                basewidth = 64
                wpercent = (basewidth / float(self.monet_img.size[0]))
                hsize = int((float(self.monet_img.size[1]) * float(wpercent)))
                self.monet_img = self.monet_img.resize(
                    (basewidth, hsize), Image.Resampling.LANCZOS)
                self.theme = themeFromImage(self.monet_img)
                self.tone = self.tone_row.get_selected_item()
                self.monet_theme = self.monet_theme_row.get_selected_item()
                self.get_application().update_theme_from_monet(
                    self.theme, self.tone, self.monet_theme
                )
                if not self.first_apply:
                    self.toast_overlay.add_toast(
                        Adw.Toast(title=_("Palette generated"))
                    )
        else:
            self.toast_overlay.add_toast(
                Adw.Toast(title=_("Select a background first"))
            )

    def setup_plugins_page(self):
        self.plugins_list = GradiencePluginsList(self)
        self.plugins_group = self.plugins_list.to_group()

        self.content_plugins.add(self.plugins_group)
        self.get_application().plugins_group = self.plugins_group

        custom_css_group = GradienceCustomCSSGroup()
        for app_type in settings_schema["custom_css_app_types"]:
            self.get_application().custom_css[app_type] = ""
        custom_css_group.load_custom_css(self.get_application().custom_css)
        self.content_plugins.add(custom_css_group)
        self.get_application().custom_css_group = custom_css_group

    def setup_colors_page(self):
        for group in settings_schema["groups"]:
            pref_group = Adw.PreferencesGroup()
            pref_group.set_name(group["name"])
            pref_group.set_title(group["title"])
            pref_group.set_description(group["description"])

            for variable in group["variables"]:
                pref_variable = GradienceOption(
                    variable["name"],
                    variable["title"],
                    variable.get("explanation"),
                    variable["adw_gtk3_support"],
                )
                pref_group.add(pref_variable)
                self.get_application(
                ).pref_variables[variable["name"]] = pref_variable

            self.content.add(pref_group)

        palette_pref_group = Adw.PreferencesGroup()
        palette_pref_group.set_name("palette_colors")
        palette_pref_group.set_title(_("Palette Colors"))
        palette_pref_group.set_description(
            _('Named palette colors used by some applications. Default colors follow the <a href="https://developer.gnome.org/hig/reference/palette.html">GNOME Human Interface Guidelines</a>.'))
        for color in settings_schema["palette"]:
            palette_shades = GradiencePaletteShades(
                color["prefix"], color["title"], color["n_shades"]
            )
            palette_pref_group.add(palette_shades)
            self.get_application(
            ).pref_palette_shades[color["prefix"]] = palette_shades
        self.content.add(palette_pref_group)

    def update_errors(self, errors):
        child = self.errors_list.get_row_at_index(0)
        while child is not None:
            self.errors_list.remove(child)
            child = self.errors_list.get_row_at_index(0)
        self.errors_button.set_visible(len(errors) > 0)
        for error in errors:
            self.errors_list.append(
                GradienceError(
                    error["error"],
                    error["element"],
                    error["line"]))

    @Gtk.Template.Callback()
    def on_presets_button_clicked(self, *args):
        presets = GradiencePresetWindow(self.get_application())
        self.set_transient_for(presets)
        presets.present()