/*
 *  xfdesktop - xfce4's desktop manager
 *
 *  Copyright (c) 2006,2024 Brian Tarricone, <brian@tarricone.org>
 *
 *  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 2 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 Library General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software Foundation,
 *  Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <glib-object.h>

#include <libxfce4util/libxfce4util.h>
#include <libxfce4ui/libxfce4ui.h>
#include <xfconf/xfconf.h>

#include "common/xfdesktop-common.h"
#include "common/xfdesktop-keyboard-shortcuts.h"
#include "xfce-desktop.h"
#include "xfdesktop-backdrop-manager.h"
#include "xfdesktop-icon-view-manager.h"

#define XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager) ((XfdesktopIconViewManagerPrivate *)xfdesktop_icon_view_manager_get_instance_private(XFDESKTOP_ICON_VIEW_MANAGER(manager)))

typedef struct _XfdesktopIconViewManagerPrivate
{
    XfwScreen *screen;
    GList *desktops;
    XfconfChannel *channel;
    XfdesktopBackdropManager *backdrop_manager;
    GtkAccelGroup *accel_group;

    gboolean icons_on_primary;
    gboolean confirm_sorting;
} XfdesktopIconViewManagerPrivate;

enum {
    PROP0 = 0,
    PROP_SCREEN,
    PROP_DESKTOPS,
    PROP_CHANNEL,
    PROP_BACKDROP_MANAGER,
    PROP_ACCEL_GROUP,
    PROP_ICON_ON_PRIMARY,
    PROP_CONFIRM_SORTING,
};

static void xfdesktop_icon_view_manager_constructed(GObject *obj);
static void xfdesktop_icon_view_manager_set_property(GObject *obj,
                                                     guint prop_id,
                                                     const GValue *value,
                                                     GParamSpec *pspec);
static void xfdesktop_icon_view_manager_get_property(GObject *obj,
                                                     guint prop_id,
                                                     GValue *value,
                                                     GParamSpec *pspec);
static void xfdesktop_icon_view_manager_finalize(GObject *obj);

static void xfdesktop_icon_view_manager_set_show_icons_on_primary(XfdesktopIconViewManager *manager,
                                                                  gboolean icons_on_primary);

static void icon_view_action_fixup(XfceGtkActionEntry *entry);
static void accel_map_changed(XfdesktopIconViewManager *manager);

static const struct {
    const gchar *setting;
    GType setting_type;
    const gchar *property;
} setting_bindings[] = {
    { DESKTOP_ICONS_ON_PRIMARY_PROP, G_TYPE_BOOLEAN, "icons-on-primary" },
    { DESKTOP_ICONS_CONFIRM_SORTING_PROP, G_TYPE_BOOLEAN, "confirm-sorting" },
};


G_DEFINE_ABSTRACT_TYPE_WITH_PRIVATE(XfdesktopIconViewManager, xfdesktop_icon_view_manager, G_TYPE_OBJECT)


static void
xfdesktop_icon_view_manager_class_init(XfdesktopIconViewManagerClass *klass)
{
    GObjectClass *gobject_class = G_OBJECT_CLASS(klass);

    gobject_class->constructed = xfdesktop_icon_view_manager_constructed;
    gobject_class->set_property = xfdesktop_icon_view_manager_set_property;
    gobject_class->get_property = xfdesktop_icon_view_manager_get_property;
    gobject_class->finalize = xfdesktop_icon_view_manager_finalize;

#define PARAM_FLAGS  (G_PARAM_READWRITE \
                      | G_PARAM_STATIC_NAME \
                      | G_PARAM_STATIC_NICK \
                      | G_PARAM_STATIC_BLURB)

    g_object_class_install_property(gobject_class,
                                    PROP_SCREEN,
                                    g_param_spec_object("screen",
                                                        "screen",
                                                        "XfwScreen",
                                                        XFW_TYPE_SCREEN,
                                                        G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));

    g_object_class_install_property(gobject_class, PROP_DESKTOPS,
                                    g_param_spec_pointer("desktops",
                                                         "desktops",
                                                         "All known XfceDesktop instances",
                                                         PARAM_FLAGS | G_PARAM_CONSTRUCT_ONLY));

    g_object_class_install_property(gobject_class, PROP_CHANNEL,
                                    g_param_spec_object("channel",
                                                        "channel",
                                                        "xfconf channel",
                                                        XFCONF_TYPE_CHANNEL,
                                                        PARAM_FLAGS | G_PARAM_CONSTRUCT_ONLY));

    g_object_class_install_property(gobject_class,
                                    PROP_BACKDROP_MANAGER,
                                    g_param_spec_object("backdrop-manager",
                                                        "backdrop-manager",
                                                        "backdrop manager",
                                                        XFDESKTOP_TYPE_BACKDROP_MANAGER,
                                                        G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));

    g_object_class_install_property(gobject_class,
                                    PROP_ACCEL_GROUP,
                                    g_param_spec_object("accel-group",
                                                        "accel-group",
                                                        "GtkAccelGroup",
                                                        GTK_TYPE_ACCEL_GROUP,
                                                        G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS));

    g_object_class_install_property(gobject_class, PROP_ICON_ON_PRIMARY,
                                    g_param_spec_boolean("icons-on-primary",
                                                         "icons on primary",
                                                         "show icons on primary desktop",
                                                         DEFAULT_ICONS_ON_PRIMARY,
                                                         PARAM_FLAGS));

    g_object_class_install_property(gobject_class, PROP_CONFIRM_SORTING,
                                    g_param_spec_boolean("confirm-sorting",
                                                         "confirm-sorting",
                                                         "confirm-sorting",
                                                         TRUE,
                                                         PARAM_FLAGS));

#undef PARAM_FLAGS
}

static void
xfdesktop_icon_view_manager_init(XfdesktopIconViewManager *manager)
{
    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);
    priv->icons_on_primary = DEFAULT_ICONS_ON_PRIMARY;
    priv->confirm_sorting = TRUE;
}

static void
xfdesktop_icon_view_manager_constructed(GObject *obj)
{
    XfdesktopIconViewManager *manager = XFDESKTOP_ICON_VIEW_MANAGER(obj);
    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);

    G_OBJECT_CLASS(xfdesktop_icon_view_manager_parent_class)->constructed(obj);

    accel_map_changed(manager);
    g_signal_connect_data(gtk_accel_map_get(),
                          "changed",
                          G_CALLBACK(accel_map_changed),
                          manager,
                          NULL,
                          G_CONNECT_SWAPPED | G_CONNECT_AFTER);

    for (gsize i = 0; i < G_N_ELEMENTS(setting_bindings); ++i) {
        xfconf_g_property_bind(priv->channel,
                               setting_bindings[i].setting,
                               setting_bindings[i].setting_type,
                               manager,
                               setting_bindings[i].property);
    }
}

static void
xfdesktop_icon_view_manager_set_property(GObject *obj,
                                         guint prop_id,
                                         const GValue *value,
                                         GParamSpec *pspec)
{
    XfdesktopIconViewManager *manager = XFDESKTOP_ICON_VIEW_MANAGER(obj);
    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);

    switch (prop_id) {
        case PROP_SCREEN:
            priv->screen = g_value_dup_object(value);
            break;

        case PROP_DESKTOPS:
            priv->desktops = g_list_copy(g_value_get_pointer(value));
            break;

        case PROP_CHANNEL:
            priv->channel = g_value_dup_object(value);
            break;

        case PROP_BACKDROP_MANAGER:
            priv->backdrop_manager = g_value_dup_object(value);
            break;

        case PROP_ACCEL_GROUP:
            priv->accel_group = g_value_dup_object(value);
            break;

        case PROP_ICON_ON_PRIMARY:
            xfdesktop_icon_view_manager_set_show_icons_on_primary(manager,
                                                                  g_value_get_boolean(value));
            break;

        case PROP_CONFIRM_SORTING:
            priv->confirm_sorting = g_value_get_boolean(value);
            break;

        default:
            G_OBJECT_WARN_INVALID_PROPERTY_ID(obj, prop_id, pspec);
            break;
    }
}

static void
xfdesktop_icon_view_manager_get_property(GObject *obj,
                                         guint prop_id,
                                         GValue *value,
                                         GParamSpec *pspec)
{
    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(obj);

    switch (prop_id) {
        case PROP_SCREEN:
            g_value_set_object(value, priv->screen);
            break;

        case PROP_DESKTOPS:
            g_value_set_pointer(value, priv->desktops);
            break;

        case PROP_CHANNEL:
            g_value_set_object(value, priv->channel);
            break;

        case PROP_BACKDROP_MANAGER:
            g_value_set_object(value, priv->backdrop_manager);
            break;

        case PROP_ACCEL_GROUP:
            g_value_set_object(value, priv->accel_group);
            break;

        case PROP_ICON_ON_PRIMARY:
            g_value_set_boolean(value, priv->icons_on_primary);
            break;

        case PROP_CONFIRM_SORTING:
            g_value_set_boolean(value, priv->confirm_sorting);
            break;

        default:
            G_OBJECT_WARN_INVALID_PROPERTY_ID(obj, prop_id, pspec);
            break;
    }
}

static void
xfdesktop_icon_view_manager_finalize(GObject *obj)
{
    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(obj);

    g_signal_handlers_disconnect_by_func(gtk_accel_map_get(), accel_map_changed, obj);

    gsize n_actions;
    XfceGtkActionEntry *actions = xfdesktop_get_icon_view_actions(icon_view_action_fixup, &n_actions);
    xfce_gtk_accel_group_disconnect_action_entries(priv->accel_group, actions, n_actions);
    g_object_unref(priv->accel_group);

    g_object_unref(priv->backdrop_manager);
    g_object_unref(priv->channel);
    g_object_unref(priv->screen);
    g_list_free(priv->desktops);

    G_OBJECT_CLASS(xfdesktop_icon_view_manager_parent_class)->finalize(obj);
}

static void
xfdesktop_icon_view_manager_set_show_icons_on_primary(XfdesktopIconViewManager *manager,
                                                      gboolean icons_on_primary)
{
    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);

    if (priv->icons_on_primary != icons_on_primary) {
        priv->icons_on_primary = icons_on_primary;
        g_object_notify(G_OBJECT(manager), "icons-on-primary");
    }
}

static void
icon_view_action_activate(XfdesktopIconViewManager *manager) {
    XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager)->activate_icons(manager);
}

static void
icon_view_action_toggle_cursor(XfdesktopIconViewManager *manager) {
    XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager)->toggle_cursor_icon(manager);
}

static void
icon_view_action_select_all(XfdesktopIconViewManager *manager) {
    XfdesktopIconViewManagerClass *klass = XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager);
    if (klass->select_all_icons != NULL) {
        klass->select_all_icons(manager);
    }
}

static void
icon_view_action_unselect_all(XfdesktopIconViewManager *manager) {
    XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager)->unselect_all_icons(manager);
}

static void
icon_view_action_arrange_icons(XfdesktopIconViewManager *manager) {
    XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager)->sort_icons(manager,
                                                               GTK_SORT_ASCENDING,
                                                               XFDESKTOP_ICON_VIEW_MANAGER_SORT_NONE);
}

static void
icon_view_action_fixup(XfceGtkActionEntry *entry) {
    switch (entry->id) {
        case XFDESKTOP_ICON_VIEW_ACTION_ACTIVATE:
        case XFDESKTOP_ICON_VIEW_ACTION_ACTIVATE_ALT_1:
        case XFDESKTOP_ICON_VIEW_ACTION_ACTIVATE_ALT_2:
        case XFDESKTOP_ICON_VIEW_ACTION_ACTIVATE_ALT_3:
        case XFDESKTOP_ICON_VIEW_ACTION_ACTIVATE_ALT_4:
            entry->callback = G_CALLBACK(icon_view_action_activate);
            break;
        case XFDESKTOP_ICON_VIEW_ACTION_TOGGLE_CURSOR:
        case XFDESKTOP_ICON_VIEW_ACTION_TOGGLE_CURSOR_ALT_1:
            entry->callback = G_CALLBACK(icon_view_action_toggle_cursor);
            break;
        case XFDESKTOP_ICON_VIEW_ACTION_SELECT_ALL:
            entry->callback = G_CALLBACK(icon_view_action_select_all);
            break;
        case XFDESKTOP_ICON_VIEW_ACTION_UNSELECT_ALL:
            entry->callback = G_CALLBACK(icon_view_action_unselect_all);
            break;
        case XFDESKTOP_ICON_VIEW_ACTION_ARRANGE_ICONS:
            entry->callback = G_CALLBACK(icon_view_action_arrange_icons);
            break;
        default:
            g_assert_not_reached();
            break;
    }
}

static void
accel_map_changed(XfdesktopIconViewManager *manager) {
    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);

    gsize n_actions;
    XfceGtkActionEntry *actions = xfdesktop_get_icon_view_actions(icon_view_action_fixup, &n_actions);

    xfce_gtk_accel_group_disconnect_action_entries(priv->accel_group, actions, n_actions);
    xfce_gtk_accel_group_connect_action_entries(priv->accel_group, actions, n_actions, manager);
}

XfwScreen *
xfdesktop_icon_view_manager_get_screen(XfdesktopIconViewManager *manager) {
    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), NULL);
    return XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager)->screen;
}

GList *
xfdesktop_icon_view_manager_get_desktops(XfdesktopIconViewManager *manager)
{
    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), NULL);
    return XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager)->desktops;
}

XfconfChannel *
xfdesktop_icon_view_manager_get_channel(XfdesktopIconViewManager *manager)
{
    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), NULL);
    return XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager)->channel;
}

XfdesktopBackdropManager *
xfdesktop_icon_view_manager_get_backdrop_manager(XfdesktopIconViewManager *manager) {
    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), NULL);
    return XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager)->backdrop_manager;
}

GtkAccelGroup *
xfdesktop_icon_view_manager_get_accel_group(XfdesktopIconViewManager *manager) {
    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), NULL);
    return XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager)->accel_group;
}

gboolean
xfdesktop_icon_view_manager_get_show_icons_on_primary(XfdesktopIconViewManager *manager)
{
    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), DEFAULT_ICONS_ON_PRIMARY);
    return XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager)->icons_on_primary;
}

void
xfdesktop_icon_view_manager_desktop_added(XfdesktopIconViewManager *manager, XfceDesktop *desktop) {
    g_return_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager));
    g_return_if_fail(XFCE_IS_DESKTOP(desktop));

    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);
    priv->desktops = g_list_append(priv->desktops, desktop);

    XfdesktopIconViewManagerClass *klass = XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager);
    klass->desktop_added(manager, desktop);

    g_object_notify(G_OBJECT(manager), "desktops");
}

void
xfdesktop_icon_view_manager_desktop_removed(XfdesktopIconViewManager *manager, XfceDesktop *desktop) {
    g_return_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager));
    g_return_if_fail(XFCE_IS_DESKTOP(desktop));

    XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);
    priv->desktops = g_list_remove(priv->desktops, desktop);

    XfdesktopIconViewManagerClass *klass = XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager);
    klass->desktop_removed(manager, desktop);

    g_object_notify(G_OBJECT(manager), "desktops");
}

XfceDesktop *
xfdesktop_icon_view_manager_get_focused_desktop(XfdesktopIconViewManager *manager) {
    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), NULL);
    return XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager)->get_focused_desktop(manager);
}

GtkMenu *
xfdesktop_icon_view_manager_get_context_menu(XfdesktopIconViewManager *manager,
                                             XfceDesktop *desktop,
                                             gint popup_x,
                                             gint popup_y)
{
    XfdesktopIconViewManagerClass *klass;

    g_return_val_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager), NULL);

    klass = XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager);
    if (klass->get_context_menu != NULL) {
        return klass->get_context_menu(manager, desktop, popup_x, popup_y);
    } else {
        return NULL;
    }
}

void
xfdesktop_icon_view_manager_sort_icons(XfdesktopIconViewManager *manager,
                                       GtkSortType sort_type,
                                       XfdesktopIconViewManagerSortFlags flags)
{
    XfdesktopIconViewManagerClass *klass;

    g_return_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager));

    klass = XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager);
    if (klass->sort_icons != NULL) {
        XfdesktopIconViewManagerPrivate *priv = XFDESKTOP_ICON_VIEW_MANAGER_GET_PRIVATE(manager);
        if (priv->confirm_sorting) {
            GtkWidget *dialog = gtk_dialog_new_with_buttons(_("Arrange Icons"),
                                                            NULL,
                                                            GTK_DIALOG_DESTROY_WITH_PARENT,
                                                            _("_Cancel"), GTK_RESPONSE_CANCEL,
                                                            _("_OK"), GTK_RESPONSE_ACCEPT,
                                                            NULL);

            GtkWidget *hbox = gtk_box_new(GTK_ORIENTATION_HORIZONTAL, 6);
            gtk_container_add(GTK_CONTAINER(gtk_dialog_get_content_area(GTK_DIALOG(dialog))), hbox);

            GIcon *icon = g_themed_icon_new("dialog-question");
            GtkWidget *image = gtk_image_new_from_gicon(icon, GTK_ICON_SIZE_DIALOG);
            g_object_set(image,
                         "valign", GTK_ALIGN_START,
                         NULL);
            gtk_box_pack_start(GTK_BOX(hbox), image, FALSE, FALSE, 0);
            g_object_unref(icon);

            GtkWidget *vbox = gtk_box_new(GTK_ORIENTATION_VERTICAL, 6);
            gtk_box_pack_start(GTK_BOX(hbox), vbox, TRUE, TRUE, 0);

            const gchar *text = _("This will reorder all desktop items and place them on different screen positions.\nAre you sure?");
            GtkWidget *label = gtk_label_new(text);
            gtk_box_pack_start(GTK_BOX(vbox), label, FALSE, FALSE, 0);

            GtkWidget *checkbox = gtk_check_button_new_with_mnemonic(_("Do _not ask me again"));
            gtk_box_pack_start(GTK_BOX(vbox), checkbox, FALSE, FALSE, 0);

            gtk_window_set_icon_name(GTK_WINDOW(dialog), "dialog-question");
            gtk_window_set_resizable(GTK_WINDOW(dialog), FALSE);
            gtk_dialog_set_default_response(GTK_DIALOG(dialog), GTK_RESPONSE_ACCEPT);
            
            /* remove focus outline on checkbox widget visible when sorting via command line and putting the cursor over the checkbox */
            GtkWidget *button = gtk_dialog_get_widget_for_response(GTK_DIALOG(dialog), GTK_RESPONSE_ACCEPT);
            gtk_widget_grab_focus(button);

            gtk_widget_show_all(dialog);

            if (gtk_dialog_run(GTK_DIALOG(dialog)) == GTK_RESPONSE_ACCEPT) {
                priv->confirm_sorting = !gtk_toggle_button_get_active(GTK_TOGGLE_BUTTON(checkbox));
                if (!priv->confirm_sorting) {
                    xfconf_channel_set_bool(priv->channel, DESKTOP_ICONS_CONFIRM_SORTING_PROP, FALSE);
                }
                klass->sort_icons(manager, sort_type, flags);
            }

            gtk_widget_destroy(dialog);
        } else {
            klass->sort_icons(manager, sort_type, flags);
        }
    }
}

void
xfdesktop_icon_view_manager_reload(XfdesktopIconViewManager *manager) {
    g_return_if_fail(XFDESKTOP_IS_ICON_VIEW_MANAGER(manager));
    XfdesktopIconViewManagerClass *klass = XFDESKTOP_ICON_VIEW_MANAGER_GET_CLASS(manager);
    if (klass->reload != NULL) {
        klass->reload(manager);
    }
}