moved from Mate-Extra

1 files changed, 209 insertions, 0 deletions

diff --git a/src/gpm-disks.c b/src/gpm-disks.c
new file mode 100644
index 0000000..eec3358
--- /dev/null
+++ b/src/gpm-disks.c
@@ -0,0 +1,209 @@
+/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*-
+ *
+ * Copyright (C) 2009 Richard Hughes <[email protected]>
+ *
+ * Licensed under the GNU General Public License Version 2
+ *
+ * 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 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.
+ */
+
+#include "config.h"
+
+#include <glib.h>
+#include <dbus/dbus-glib.h>
+
+#include "egg-debug.h"
+#include "gpm-disks.h"
+
+static void     gpm_disks_finalize   (GObject	  *object);
+
+#define GPM_DISKS_GET_PRIVATE(o) (G_TYPE_INSTANCE_GET_PRIVATE ((o), GPM_TYPE_DISKS, GpmDisksPrivate))
+
+struct GpmDisksPrivate
+{
+	DBusGProxy		*proxy;
+	gchar			*cookie;
+};
+
+static gpointer gpm_disks_object = NULL;
+
+G_DEFINE_TYPE (GpmDisks, gpm_disks, G_TYPE_OBJECT)
+
+/**
+ * gpm_disks_unregister:
+ **/
+static gboolean
+gpm_disks_unregister (GpmDisks *disks)
+{
+	gboolean ret = FALSE;
+	GError *error = NULL;
+
+	/* no UDisks */
+	if (disks->priv->proxy == NULL) {
+		egg_warning ("no UDisks");
+		goto out;
+	}
+
+	/* clear spindown timeouts */
+	ret = dbus_g_proxy_call (disks->priv->proxy, "DriveUnsetAllSpindownTimeouts", &error,
+				 G_TYPE_STRING, disks->priv->cookie,
+				 G_TYPE_INVALID,
+				 G_TYPE_INVALID);
+	if (!ret) {
+		egg_warning ("failed to clear spindown timeout: %s", error->message);
+		g_error_free (error);
+		goto out;
+	}
+out:
+	/* reset */
+	g_free (disks->priv->cookie);
+	disks->priv->cookie = NULL;
+
+	return ret;
+}
+
+/**
+ * gpm_disks_register:
+ **/
+static gboolean
+gpm_disks_register (GpmDisks *disks, gint timeout)
+{
+	gboolean ret = FALSE;
+	GError *error = NULL;
+	const gchar **options = {NULL};
+
+	/* no UDisks */
+	if (disks->priv->proxy == NULL) {
+		egg_warning ("no UDisks");
+		goto out;
+	}
+
+	/* set spindown timeouts */
+	ret = dbus_g_proxy_call (disks->priv->proxy, "DriveSetAllSpindownTimeouts", &error,
+				 G_TYPE_INT, timeout,
+				 G_TYPE_STRV, options,
+				 G_TYPE_INVALID,
+				 G_TYPE_STRING, &disks->priv->cookie,
+				 G_TYPE_INVALID);
+	if (!ret) {
+		egg_warning ("failed to set spindown timeout: %s", error->message);
+		g_error_free (error);
+		goto out;
+	}
+out:
+	return ret;
+}
+
+/**
+ * gpm_disks_set_spindown_timeout:
+ **/
+gboolean
+gpm_disks_set_spindown_timeout (GpmDisks *disks, gint timeout)
+{
+	gboolean ret = TRUE;
+
+	/* get rid of last request */
+	if (disks->priv->cookie != NULL) {
+		egg_debug ("unregistering %s", disks->priv->cookie);
+		gpm_disks_unregister (disks);
+	}
+
+	/* is not enabled? */
+	if (timeout == 0) {
+		egg_debug ("disk spindown disabled");
+		goto out;
+	}
+
+	/* register */
+	ret = gpm_disks_register (disks, timeout);
+	if (ret)
+		egg_debug ("registered %s (%i)", disks->priv->cookie, timeout);
+out:
+	return ret;
+}
+
+/**
+ * gpm_disks_init:
+ */
+static void
+gpm_disks_init (GpmDisks *disks)
+{
+	GError *error = NULL;
+	DBusGConnection *connection;
+
+	disks->priv = GPM_DISKS_GET_PRIVATE (disks);
+
+	disks->priv->cookie = NULL;
+
+	/* get proxy to interface */
+	connection = dbus_g_bus_get (DBUS_BUS_SYSTEM, NULL);
+	disks->priv->proxy = dbus_g_proxy_new_for_name_owner (connection,
+							      "org.freedesktop.UDisks",
+							      "/org/freedesktop/UDisks",
+							      "org.freedesktop.UDisks", &error);
+	if (disks->priv->proxy == NULL) {
+		egg_warning ("DBUS error: %s", error->message);
+		g_error_free (error);
+	}
+}
+
+/**
+ * gpm_disks_coldplug:
+ *
+ * @object: This disks instance
+ */
+static void
+gpm_disks_finalize (GObject *object)
+{
+	GpmDisks *disks;
+	g_return_if_fail (object != NULL);
+	g_return_if_fail (GPM_IS_DISKS (object));
+	disks = GPM_DISKS (object);
+	g_return_if_fail (disks->priv != NULL);
+
+	g_free (disks->priv->cookie);
+	g_object_unref (disks->priv->proxy);
+
+	G_OBJECT_CLASS (gpm_disks_parent_class)->finalize (object);
+}
+
+/**
+ * gpm_disks_class_init:
+ * @klass: This class instance
+ **/
+static void
+gpm_disks_class_init (GpmDisksClass *klass)
+{
+	GObjectClass *object_class = G_OBJECT_CLASS (klass);
+	object_class->finalize = gpm_disks_finalize;
+	g_type_class_add_private (klass, sizeof (GpmDisksPrivate));
+}
+
+/**
+ * gpm_disks_new:
+ * Return value: new #GpmDisks instance.
+ **/
+GpmDisks *
+gpm_disks_new (void)
+{
+	if (gpm_disks_object != NULL) {
+		g_object_ref (gpm_disks_object);
+	} else {
+		gpm_disks_object = g_object_new (GPM_TYPE_DISKS, NULL);
+		g_object_add_weak_pointer (gpm_disks_object, &gpm_disks_object);
+	}
+	return GPM_DISKS (gpm_disks_object);
+}
+