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);
+}