diff options
Diffstat (limited to 'drivemount/drive-list.c')
| -rw-r--r-- | drivemount/drive-list.c | 461 |
1 files changed, 461 insertions, 0 deletions
diff --git a/drivemount/drive-list.c b/drivemount/drive-list.c new file mode 100644 index 00000000..1013a886 --- /dev/null +++ b/drivemount/drive-list.c @@ -0,0 +1,461 @@ +/* -*- mode: C; c-basic-offset: 4 -*- + * Drive Mount Applet + * Copyright (c) 2004 Canonical Ltd + * Copyright 2008 Pierre Ossman + * + * This library 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 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + * + * Author: + * James Henstridge <[email protected]> + */ + +#ifdef HAVE_CONFIG_H +# include <config.h> +#endif + +#include <gio/gio.h> +#include "drive-list.h" +#include "drive-button.h" +#include <glib/gi18n.h> + +G_DEFINE_TYPE (DriveList, drive_list, GTK_TYPE_TABLE); + +static GVolumeMonitor *volume_monitor = NULL; + +static void drive_list_finalize (GObject *object); +static void drive_list_destroy (GtkObject *object); +static void drive_list_add (GtkContainer *container, GtkWidget *child); +static void drive_list_remove (GtkContainer *container, GtkWidget *child); + +static void mount_added (GVolumeMonitor *monitor, + GMount *mount, + DriveList *self); +static void mount_changed (GVolumeMonitor *monitor, + GMount *mount, + DriveList *self); +static void mount_removed (GVolumeMonitor *monitor, + GMount *mount, + DriveList *self); +static void volume_added (GVolumeMonitor *monitor, + GVolume *volume, + DriveList *self); +static void volume_changed (GVolumeMonitor *monitor, + GVolume *volume, + DriveList *self); +static void volume_removed (GVolumeMonitor *monitor, + GVolume *volume, + DriveList *self); +static void add_volume (DriveList *self, + GVolume *volume); +static void remove_volume (DriveList *self, + GVolume *volume); +static void add_mount (DriveList *self, + GMount *mount); +static void remove_mount (DriveList *self, + GMount *mount); + +static void +drive_list_class_init (DriveListClass *class) +{ + G_OBJECT_CLASS (class)->finalize = drive_list_finalize; + GTK_OBJECT_CLASS (class)->destroy = drive_list_destroy; + GTK_CONTAINER_CLASS (class)->add = drive_list_add; + GTK_CONTAINER_CLASS (class)->remove = drive_list_remove; +} + +static void +drive_list_init (DriveList *self) +{ + GList *volumes, *mounts, *tmp; + + gtk_table_set_homogeneous (GTK_TABLE (self), TRUE); + + self->volumes = g_hash_table_new (NULL, NULL); + self->mounts = g_hash_table_new (NULL, NULL); + self->orientation = GTK_ORIENTATION_HORIZONTAL; + self->layout_tag = 0; + self->icon_size = 24; + self->relief = GTK_RELIEF_NORMAL; + + /* listen for drive connects/disconnects, and add + * currently connected drives. */ + if (!volume_monitor) + volume_monitor = g_volume_monitor_get (); + + g_signal_connect_object (volume_monitor, "mount_added", + G_CALLBACK (mount_added), self, 0); + g_signal_connect_object (volume_monitor, "mount_changed", + G_CALLBACK (mount_changed), self, 0); + g_signal_connect_object (volume_monitor, "mount_removed", + G_CALLBACK (mount_removed), self, 0); + g_signal_connect_object (volume_monitor, "volume_added", + G_CALLBACK (volume_added), self, 0); + g_signal_connect_object (volume_monitor, "volume_changed", + G_CALLBACK (volume_changed), self, 0); + g_signal_connect_object (volume_monitor, "volume_removed", + G_CALLBACK (volume_removed), self, 0); + volumes = g_volume_monitor_get_volumes (volume_monitor); + for (tmp = volumes; tmp != NULL; tmp = tmp->next) { + GVolume *volume = tmp->data; + + add_volume (self, volume); + g_object_unref (volume); + } + g_list_free (volumes); + + mounts = g_volume_monitor_get_mounts (volume_monitor); + for (tmp = mounts; tmp != NULL; tmp = tmp->next) { + GMount *mount = tmp->data; + + add_mount (self, mount); + g_object_unref (mount); + } + g_list_free (mounts); +} + +GtkWidget * +drive_list_new (void) +{ + return g_object_new (DRIVE_TYPE_LIST, NULL); +} + +static void +drive_list_finalize (GObject *object) +{ + DriveList *self = DRIVE_LIST (object); + + g_hash_table_destroy (self->volumes); + g_hash_table_destroy (self->mounts); + + if (G_OBJECT_CLASS (drive_list_parent_class)->finalize) + (* G_OBJECT_CLASS (drive_list_parent_class)->finalize) (object); +} + +static void +drive_list_destroy (GtkObject *object) +{ + DriveList *self = DRIVE_LIST (object); + + g_signal_handlers_disconnect_by_func (volume_monitor, + G_CALLBACK (mount_added), self); + g_signal_handlers_disconnect_by_func (volume_monitor, + G_CALLBACK (mount_changed), self); + g_signal_handlers_disconnect_by_func (volume_monitor, + G_CALLBACK (mount_removed), self); + g_signal_handlers_disconnect_by_func (volume_monitor, + G_CALLBACK (volume_added), self); + g_signal_handlers_disconnect_by_func (volume_monitor, + G_CALLBACK (volume_changed), self); + g_signal_handlers_disconnect_by_func (volume_monitor, + G_CALLBACK (volume_removed), self); + + if (self->layout_tag) + g_source_remove (self->layout_tag); + self->layout_tag = 0; + + if (GTK_OBJECT_CLASS (drive_list_parent_class)->destroy) + (* GTK_OBJECT_CLASS (drive_list_parent_class)->destroy) (object); +} + +static void +drive_list_add (GtkContainer *container, GtkWidget *child) +{ + DriveList *self; + DriveButton *button; + + g_return_if_fail (DRIVE_IS_LIST (container)); + g_return_if_fail (DRIVE_IS_BUTTON (child)); + + if (GTK_CONTAINER_CLASS (drive_list_parent_class)->add) + (* GTK_CONTAINER_CLASS (drive_list_parent_class)->add) (container, + child); + + self = DRIVE_LIST (container); + button = DRIVE_BUTTON (child); + if (button->volume) + g_hash_table_insert (self->volumes, button->volume, button); + else + g_hash_table_insert (self->mounts, button->mount, button); +} + +static void +drive_list_remove (GtkContainer *container, GtkWidget *child) +{ + DriveList *self; + DriveButton *button; + + g_return_if_fail (DRIVE_IS_LIST (container)); + g_return_if_fail (DRIVE_IS_BUTTON (child)); + + self = DRIVE_LIST (container); + button = DRIVE_BUTTON (child); + if (button->volume) + g_hash_table_remove (self->volumes, button->volume); + else + g_hash_table_remove (self->mounts, button->mount); + + if (GTK_CONTAINER_CLASS (drive_list_parent_class)->remove) + (* GTK_CONTAINER_CLASS (drive_list_parent_class)->remove) (container, + child); +} + +static void +list_buttons (gpointer key, gpointer value, gpointer user_data) +{ + GtkWidget *button = value; + GList **sorted_buttons = user_data; + + *sorted_buttons = g_list_insert_sorted (*sorted_buttons, button, + (GCompareFunc)drive_button_compare); +} + +static gboolean +relayout_buttons (gpointer data) +{ + DriveList *self = DRIVE_LIST (data); + GList *sorted_buttons = NULL, *tmp; + int i; + + + self->layout_tag = 0; + g_hash_table_foreach (self->volumes, list_buttons, &sorted_buttons); + g_hash_table_foreach (self->mounts, list_buttons, &sorted_buttons); + + /* position buttons in the table according to their sorted order */ + for (tmp = sorted_buttons, i = 0; tmp != NULL; tmp = tmp->next, i++) { + GtkWidget *button = tmp->data; + + if (self->orientation == GTK_ORIENTATION_HORIZONTAL) { + gtk_container_child_set (GTK_CONTAINER (self), button, + "left_attach", i, "right_attach", i+1, + "top_attach", 0, "bottom_attach", 1, + "x_options", GTK_FILL, + "y_options", GTK_FILL, + NULL); + } else { + gtk_container_child_set (GTK_CONTAINER (self), button, + "left_attach", 0, "right_attach", 1, + "top_attach", i, "bottom_attach", i+1, + "x_options", GTK_FILL, + "y_options", GTK_FILL, + NULL); + } + } + /* shrink wrap the table */ + gtk_table_resize (GTK_TABLE (self), 1, 1); + + return FALSE; +} + +static void +queue_relayout (DriveList *self) +{ + if (!self->layout_tag) { + self->layout_tag = g_idle_add (relayout_buttons, self); + } +} + +static void +mount_added (GVolumeMonitor *monitor, + GMount *mount, + DriveList *self) +{ + add_mount (self, mount); + + mount_changed (monitor, mount, self); +} + +static void +mount_changed (GVolumeMonitor *monitor, + GMount *mount, + DriveList *self) +{ + GVolume *volume; + DriveButton *button = NULL;; + + volume = g_mount_get_volume (mount); + if (volume) { + button = g_hash_table_lookup (self->volumes, volume); + g_object_unref (volume); + } else { + button = g_hash_table_lookup (self->mounts, mount); + } + if (button) + drive_button_queue_update (button); +} + +static void +mount_removed (GVolumeMonitor *monitor, + GMount *mount, + DriveList *self) +{ + remove_mount (self, mount); + + mount_changed (monitor, mount, self); +} + +static void +volume_added (GVolumeMonitor *monitor, + GVolume *volume, + DriveList *self) +{ + add_volume (self, volume); +} + +static void +volume_changed (GVolumeMonitor *monitor, + GVolume *volume, + DriveList *self) +{ + DriveButton *button = NULL;; + + button = g_hash_table_lookup (self->volumes, volume); + if (button) + drive_button_queue_update (button); +} + +static void +volume_removed (GVolumeMonitor *monitor, + GVolume *volume, + DriveList *self) +{ + remove_volume (self, volume); +} + +static void +add_volume (DriveList *self, GVolume *volume) +{ + GtkWidget *button; + + /* if the volume has already been added, return */ + if (g_hash_table_lookup (self->volumes, volume) != NULL) + return; + + button = drive_button_new (volume); + gtk_button_set_relief (GTK_BUTTON (button), self->relief); + drive_button_set_size (DRIVE_BUTTON (button), self->icon_size); + gtk_container_add (GTK_CONTAINER (self), button); + gtk_widget_show (button); + queue_relayout (self); +} + +static void +remove_volume (DriveList *self, GVolume *volume) +{ + GtkWidget *button; + + /* if the volume has already been added, return */ + button = g_hash_table_lookup (self->volumes, volume); + if (button) { + gtk_container_remove (GTK_CONTAINER (self), button); + queue_relayout (self); + } +} + +static void +add_mount (DriveList *self, GMount *mount) +{ + GtkWidget *button; + GVolume *volume; + + /* ignore mounts attached to a volume */ + volume = g_mount_get_volume (mount); + if (volume) { + g_object_unref (volume); + return; + } + + /* if the mount has already been added, return */ + if (g_hash_table_lookup (self->mounts, mount) != NULL) + return; + + button = drive_button_new_from_mount (mount); + gtk_button_set_relief (GTK_BUTTON (button), self->relief); + drive_button_set_size (DRIVE_BUTTON (button), self->icon_size); + gtk_container_add (GTK_CONTAINER (self), button); + gtk_widget_show (button); + queue_relayout (self); +} + +static void +remove_mount (DriveList *self, GMount *mount) +{ + GtkWidget *button; + + /* if the mount has already been added, return */ + button = g_hash_table_lookup (self->mounts, mount); + if (button) { + gtk_container_remove (GTK_CONTAINER (self), button); + queue_relayout (self); + } +} + +void +drive_list_set_orientation (DriveList *self, + GtkOrientation orientation) +{ + g_return_if_fail (DRIVE_IS_LIST (self)); + + if (orientation != self->orientation) { + self->orientation = orientation; + queue_relayout (self); + } +} + +static void +set_icon_size (gpointer key, gpointer value, gpointer user_data) +{ + DriveButton *button = value; + DriveList *self = user_data; + + drive_button_set_size (button, self->icon_size); +} + + +void +drive_list_set_panel_size (DriveList *self, int panel_size) +{ + g_return_if_fail (DRIVE_IS_LIST (self)); + + if (self->icon_size != panel_size) { + self->icon_size = panel_size; + g_hash_table_foreach (self->volumes, set_icon_size, self); + g_hash_table_foreach (self->mounts, set_icon_size, self); + } +} + +static void +set_button_relief (gpointer key, gpointer value, gpointer user_data) +{ + GtkButton *button = value; + DriveList *self = user_data; + + gtk_button_set_relief (button, self->relief); +} + +void +drive_list_set_transparent (DriveList *self, gboolean transparent) +{ + GtkReliefStyle relief; + + relief = transparent ? GTK_RELIEF_NONE : GTK_RELIEF_NORMAL; + + if (relief == self->relief) + return; + + self->relief = relief; + g_hash_table_foreach (self->volumes, set_button_relief, self); + g_hash_table_foreach (self->mounts, set_button_relief, self); +} |