initial commit

1 files changed, 414 insertions, 0 deletions

diff --git a/src/river.c b/src/river.c
new file mode 100644
index 0000000..44225a6
--- /dev/null
+++ b/src/river.c
@@ -0,0 +1,414 @@
+
+
+#include "river-private.h"
+#include <gio/gio.h>
+#include <wayland-client-protocol.h>
+#include <wayland-client.h>
+#include "river-status-unstable-v1-client.h"
+#include "wayland-source.h"
+#include <json-glib/json-glib.h>
+
+
+struct _AstalRiverRiver {
+    GObject parent_instance;
+    GList *outputs;
+    gchar *focused_output;
+    gchar *focused_view;
+    gchar *mode;
+};
+
+typedef struct {
+    GHashTable *signal_ids;
+
+    struct wl_registry *wl_registry;
+    struct wl_seat *seat;
+    struct wl_display *display;
+    WLSource *wl_source;
+    struct zriver_status_manager_v1 *river_status_manager;
+    struct zriver_seat_status_v1 *river_seat_status;
+} AstalRiverRiverPrivate;
+
+static JsonSerializableIface *serializable_iface = NULL;
+static void astal_river_river_initable_iface_init (GInitableIface *iface);
+static void astal_river_river_json_serializable_iface_init(JsonSerializableIface *g_iface);
+
+G_DEFINE_TYPE_WITH_CODE (AstalRiverRiver, astal_river_river, G_TYPE_OBJECT,
+                         G_ADD_PRIVATE (AstalRiverRiver)
+                         G_IMPLEMENT_INTERFACE (G_TYPE_INITABLE, astal_river_river_initable_iface_init)
+                         G_IMPLEMENT_INTERFACE (JSON_TYPE_SERIALIZABLE, astal_river_river_json_serializable_iface_init))
+
+typedef enum {
+    ASTAL_RIVER_RIVER_PROP_FOCUSED_OUTPUT = 1,
+    ASTAL_RIVER_RIVER_PROP_FOCUSED_VIEW,
+    ASTAL_RIVER_RIVER_PROP_MODE,
+    ASTAL_RIVER_RIVER_PROP_OUTPUTS,
+    ASTAL_RIVER_RIVER_N_PROPERTIES
+} AstalRiverOutputProperties;
+
+typedef enum {
+    ASTAL_RIVER_RIVER_SIGNAL_CHANGED,
+    ASTAL_RIVER_RIVER_SIGNAL_OUTPUT_ADDED,
+    ASTAL_RIVER_RIVER_SIGNAL_OUTPUT_REMOVED,
+    ASTAL_RIVER_RIVER_N_SIGNALS
+} AstalRiverOutputSignals;
+
+static guint astal_river_river_signals[ASTAL_RIVER_RIVER_N_SIGNALS] = {0,};
+static GParamSpec *astal_river_river_properties[ASTAL_RIVER_RIVER_N_PROPERTIES] = {NULL,};
+
+static void reemit_changed(AstalRiverOutput *output, AstalRiverRiver  *self) {
+    g_signal_emit (self, astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_CHANGED], 0);
+}
+
+static AstalRiverOutput *find_output_by_id (AstalRiverRiver *self, uint32_t id) {
+    GList *output = self->outputs;
+    while(output != NULL) {
+        AstalRiverOutput *river_output = output->data;
+        if(astal_river_output_get_id (river_output) == id) return river_output;
+        output = output->next;
+    }
+    return NULL;
+}
+
+static AstalRiverOutput *find_output_by_output (AstalRiverRiver *self, struct wl_output *wl_output) {
+    GList *output = self->outputs;
+    while(output != NULL) {
+        AstalRiverOutput *river_output = output->data;
+        if(astal_river_output_get_wl_output (river_output) == wl_output) return river_output;
+        output = output->next;
+    }
+    return NULL;
+}
+
+static AstalRiverOutput *find_output_by_name (AstalRiverRiver *self, gchar* name) {
+    GList *output = self->outputs;
+    while(output != NULL) {
+        AstalRiverOutput *river_output = output->data;
+        if(strcmp(astal_river_output_get_name (river_output), name) == 0) return river_output;
+        output = output->next;
+    }
+    return NULL;
+}
+
+GList* astal_river_river_get_outputs(AstalRiverRiver *self){
+  return self->outputs;
+}
+
+AstalRiverOutput* astal_river_river_get_output(AstalRiverRiver *self, gchar* name) {
+  return find_output_by_name(self, name);
+}
+
+gchar* astal_river_river_get_focused_view(AstalRiverRiver *self) {
+  return self->focused_view;
+}
+
+gchar* astal_river_river_get_focused_output(AstalRiverRiver *self) {
+  return self->focused_output;
+}
+
+gchar* astal_river_river_get_mode(AstalRiverRiver *self) {
+  return self->mode;
+}
+
+static void astal_river_river_get_property(GObject *object, guint property_id, GValue *value, GParamSpec *pspec) {
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER(object);
+
+    switch (property_id) {
+        case ASTAL_RIVER_RIVER_PROP_MODE:
+            g_value_set_string (value, self->mode);
+            break;
+        case ASTAL_RIVER_RIVER_PROP_FOCUSED_VIEW:
+            g_value_set_string (value, self->focused_view);
+            break;
+        case ASTAL_RIVER_RIVER_PROP_FOCUSED_OUTPUT:
+            g_value_set_string (value, self->focused_output);
+            break;
+        case ASTAL_RIVER_RIVER_PROP_OUTPUTS:
+            g_value_set_pointer(value, self->outputs);
+            break;
+        default:
+            G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
+            break;
+    }
+}
+
+static JsonNode* astal_river_river_serialize_property(JsonSerializable *serializable,
+                                                      const gchar *name,
+                                                      const GValue *value,
+                                                      GParamSpec *pspec) {
+    JsonNode *retval = NULL;
+    if (strcmp (name, "outputs") == 0) {
+        JsonArray *outputs = json_array_new ();
+        retval = json_node_new (JSON_NODE_ARRAY);
+        GList *output = g_value_get_pointer (value);
+        while(output != NULL) {
+            AstalRiverOutput *river_output = output->data;
+            json_array_add_element (outputs, json_gobject_serialize (G_OBJECT(river_output)));
+            output = output->next;
+        }
+        json_node_take_array (retval, outputs);
+    }
+    else
+        retval = serializable_iface->serialize_property(serializable, name, value, pspec);
+
+    return retval;
+}
+
+static void noop(){}
+
+static void river_seat_status_handle_focused_output(void* data,
+                                                    struct zriver_seat_status_v1* seat_status,
+                                                    struct wl_output* focused_output) {
+
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER(data);
+    g_free(self->focused_output);
+    AstalRiverOutput *output = find_output_by_output(self, focused_output);
+    if(output == NULL) return;
+    self->focused_output = g_strdup(astal_river_output_get_name(output));
+    g_object_notify(G_OBJECT(self), "focused-output");
+    g_signal_emit (self, astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_CHANGED], 0);
+}
+
+static void river_seat_status_handle_focused_view(void* data,
+                                                  struct zriver_seat_status_v1* seat_status,
+                                                  const char* focused_view) {
+
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER(data);
+    g_free(self->focused_view);
+    self->focused_view = g_strdup(focused_view);
+    AstalRiverOutput *output = find_output_by_name(self, self->focused_output);
+    astal_river_output_set_focused_view (output, focused_view);
+    g_object_notify(G_OBJECT(self), "focused-view");
+    g_signal_emit (self, astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_CHANGED], 0);
+}
+
+static void river_seat_status_handle_mode(void* data,
+                                          struct zriver_seat_status_v1* seat_status,
+                                          const char* mode) {
+
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER(data);
+    g_free(self->mode);
+    self->mode = g_strdup(mode);
+    g_object_notify(G_OBJECT(self), "mode");
+    g_signal_emit (self, astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_CHANGED], 0);
+}
+
+static const struct zriver_seat_status_v1_listener river_seat_status_listener = {
+    .focused_output = river_seat_status_handle_focused_output,
+    .unfocused_output = noop,
+    .focused_view = river_seat_status_handle_focused_view,
+    .mode = river_seat_status_handle_mode,
+};
+
+static void global_registry_handler(void* data,
+                                    struct wl_registry* registry,
+                                    uint32_t id,
+                                    const char* interface,
+                                    uint32_t version) {
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER(data);
+    AstalRiverRiverPrivate *priv = astal_river_river_get_instance_private (self);
+    if (strcmp(interface, wl_output_interface.name) == 0) {
+        struct wl_output *wl_out = wl_registry_bind(registry, id, &wl_output_interface, 4);
+        AstalRiverOutput *output = astal_river_output_new(id, wl_out, priv->river_status_manager, priv->display);
+
+        self->outputs = g_list_append (self->outputs, output);
+        g_object_notify (G_OBJECT(self), "outputs");
+        g_signal_emit(G_OBJECT(self),
+                      astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_OUTPUT_ADDED], 0,
+                      astal_river_output_get_name(output));
+        g_signal_emit (self, astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_CHANGED], 0);
+        guint signal_id = g_signal_connect(output, "changed", G_CALLBACK(reemit_changed), self);
+        g_hash_table_insert (priv->signal_ids, GUINT_TO_POINTER (id), GUINT_TO_POINTER (signal_id));
+    } else if (strcmp(interface, wl_seat_interface.name) == 0) {
+        priv->seat = wl_registry_bind(registry, id, &wl_seat_interface, 4);
+    } else if (strcmp(interface, zriver_status_manager_v1_interface.name) == 0) {
+        priv->river_status_manager = wl_registry_bind(
+          registry, id, &zriver_status_manager_v1_interface, 4);
+    }
+}
+
+static void global_registry_remover(void* data, struct wl_registry* registry, uint32_t id) {
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER (data);
+    AstalRiverRiverPrivate *priv = astal_river_river_get_instance_private (self);
+    AstalRiverOutput *output = find_output_by_id(self, id);
+	if ( output != NULL ) {
+        guint signal_id = GPOINTER_TO_UINT (g_hash_table_lookup (priv->signal_ids, GUINT_TO_POINTER (id)));
+        g_hash_table_remove (priv->signal_ids, GUINT_TO_POINTER(id));
+        g_signal_handler_disconnect (output, signal_id);
+        g_signal_emit(G_OBJECT(self),
+                      astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_OUTPUT_ADDED], 0,
+                      astal_river_output_get_name(output));
+        self->outputs = g_list_remove (self->outputs, output);
+        g_object_notify (G_OBJECT(self), "outputs");
+        g_object_unref(output);
+		return;
+	}
+    g_signal_emit (self, astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_CHANGED], 0);
+}
+
+
+static const struct wl_registry_listener registry_listener = {
+    global_registry_handler,
+    global_registry_remover
+};
+
+static void astal_river_river_json_serializable_iface_init(JsonSerializableIface *iface){
+    iface->serialize_property = astal_river_river_serialize_property;
+    serializable_iface = g_type_default_interface_peek (JSON_TYPE_SERIALIZABLE);
+}
+
+static gboolean astal_river_river_initable_init (GInitable *initable, GCancellable *cancellable, GError **error) {
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER (initable);
+    AstalRiverRiverPrivate *priv = astal_river_river_get_instance_private (self);
+
+    priv->seat = NULL;
+    priv->display = NULL;
+    priv->river_status_manager = NULL;
+    priv->signal_ids = g_hash_table_new (g_direct_hash, g_direct_equal);
+
+    priv->wl_source = wl_source_new(NULL, NULL);
+
+    if (priv->wl_source == NULL) {
+        g_set_error_literal(error, G_IO_ERROR, G_IO_ERROR_FAILED, "Can not connect to wayland display");
+        return FALSE;
+    }
+
+    priv->display = wl_source_get_display(priv->wl_source);
+
+    priv->wl_registry = wl_display_get_registry(priv->display);
+    wl_registry_add_listener(priv->wl_registry, &registry_listener, self);
+
+    wl_display_roundtrip(priv->display);
+
+    if (priv->river_status_manager == NULL) {
+        g_set_error_literal(error, G_IO_ERROR, G_IO_ERROR_FAILED, "Can not connect river status protocol");
+        return FALSE;
+    }
+
+    priv->river_seat_status = zriver_status_manager_v1_get_river_seat_status(
+      priv->river_status_manager, priv->seat);
+    zriver_seat_status_v1_add_listener(
+      priv->river_seat_status, &river_seat_status_listener, self);
+
+    wl_display_roundtrip(priv->display);
+
+    return TRUE;
+}
+
+static void astal_river_river_initable_iface_init (GInitableIface *iface) {
+  iface->init = astal_river_river_initable_init;
+}
+
+static void astal_river_river_init(AstalRiverRiver *self) {
+    self->outputs = NULL;
+}
+
+AstalRiverRiver *astal_river_river_new() {
+  return g_initable_new(ASTAL_RIVER_TYPE_RIVER, NULL, NULL, NULL);
+}
+
+static void disconnect_signal (gpointer key, gpointer value, gpointer user_data) {
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER(user_data);
+    AstalRiverRiverPrivate *priv = astal_river_river_get_instance_private (self);
+
+    AstalRiverOutput *output = find_output_by_id (self, GPOINTER_TO_UINT(key));
+    g_signal_handler_disconnect (output, GPOINTER_TO_UINT(value));
+}
+
+static void astal_river_river_finalize(GObject *object) {
+
+    AstalRiverRiver *self = ASTAL_RIVER_RIVER(object);
+    AstalRiverRiverPrivate *priv = astal_river_river_get_instance_private(self);
+
+    g_hash_table_foreach (priv->signal_ids, disconnect_signal, self);
+    g_hash_table_destroy (priv->signal_ids);
+
+    wl_display_roundtrip (priv->display);
+
+    g_clear_list(&self->outputs, g_object_unref);
+    self->outputs = NULL;
+
+    wl_registry_destroy(priv->wl_registry);
+    zriver_status_manager_v1_destroy(priv->river_status_manager);
+    zriver_seat_status_v1_destroy(priv->river_seat_status);
+    wl_seat_destroy(priv->seat);
+    wl_display_flush(priv->display);
+
+    wl_source_free(priv->wl_source);
+
+    g_free(self->focused_view);
+    g_free(self->focused_output);
+    g_free(self->mode);
+
+
+    G_OBJECT_CLASS (astal_river_river_parent_class)->finalize(object);
+}
+
+static void astal_river_river_class_init(AstalRiverRiverClass *class) {
+
+    GObjectClass *object_class = G_OBJECT_CLASS (class);
+    object_class->get_property = astal_river_river_get_property;
+    object_class->finalize = astal_river_river_finalize;
+
+    astal_river_river_properties[ASTAL_RIVER_RIVER_PROP_MODE] =
+        g_param_spec_string("mode",
+                            "mode",
+                            "currently active mode",
+                            NULL,
+                            G_PARAM_READABLE);
+    astal_river_river_properties[ASTAL_RIVER_RIVER_PROP_FOCUSED_VIEW] =
+        g_param_spec_string("focused-view",
+                            "focused-view",
+                            "currently focused view",
+                            NULL,
+                            G_PARAM_READABLE);
+    astal_river_river_properties[ASTAL_RIVER_RIVER_PROP_FOCUSED_OUTPUT] =
+        g_param_spec_string("focused-output",
+                            "focused-output",
+                            "currently focused-output",
+                            NULL,
+                            G_PARAM_READABLE);
+    astal_river_river_properties[ASTAL_RIVER_RIVER_PROP_OUTPUTS] =
+        g_param_spec_pointer("outputs",
+                             "outputs",
+                             "a list of all outputs",
+                             G_PARAM_READABLE);
+
+    g_object_class_install_properties(object_class,
+                                      ASTAL_RIVER_RIVER_N_PROPERTIES,
+                                      astal_river_river_properties);
+
+
+    astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_OUTPUT_ADDED] =
+      g_signal_new("output-added",
+                 G_TYPE_FROM_CLASS(class),
+                 G_SIGNAL_RUN_FIRST,
+                 0,
+                 NULL,
+                 NULL,
+                 NULL,
+                 G_TYPE_NONE,
+                 1,
+                 G_TYPE_STRING);
+
+    astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_OUTPUT_REMOVED] =
+      g_signal_new("output-removed",
+                 G_TYPE_FROM_CLASS(class),
+                 G_SIGNAL_RUN_FIRST,
+                 0,
+                 NULL,
+                 NULL,
+                 NULL,
+                 G_TYPE_NONE,
+                 1,
+                 G_TYPE_STRING);
+    astal_river_river_signals[ASTAL_RIVER_RIVER_SIGNAL_CHANGED] =
+        g_signal_new("changed",
+                     G_TYPE_FROM_CLASS(class),
+                     G_SIGNAL_RUN_FIRST,
+                     0,
+                     NULL,
+                     NULL,
+                     NULL,
+                     G_TYPE_NONE,
+                     0);
+}