/*

 * Copyright © 2013 Lars Uebernickel

 *

 * This library is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2.1 of the License, or (at your option) any later version.

 *

 * This library 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

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General

 * Public License along with this library; if not, see <http://www.gnu.org/licenses/>.

 *

 * Authors: Lars Uebernickel <lars@uebernic.de>

 */

#include "gnotification-server.h"

#include <gio/gio.h>

typedef GObjectClass GNotificationServerClass;

struct _GNotificationServer

{

  GObject parent;

  GDBusConnection *connection;

  guint name_owner_id;

  guint object_id;

  guint is_running;

  /* app_ids -> hashtables of notification ids -> a{sv} */

  GHashTable *applications;

};

G_DEFINE_TYPE (GNotificationServer, g_notification_server, G_TYPE_OBJECT)

enum

{

  PROP_0,

  PROP_IS_RUNNING

};

static GDBusInterfaceInfo *

org_gtk_Notifications_get_interface (void)

{

  static GDBusInterfaceInfo *iface_info;

  if (iface_info == NULL)

    {

      GDBusNodeInfo *info;

      GError *error = NULL;

      info = g_dbus_node_info_new_for_xml (

        "<node>"

        "  <interface name='org.gtk.Notifications'>"

        "    <method name='AddNotification'>"

        "      <arg type='s' direction='in' />"

        "      <arg type='s' direction='in' />"

        "      <arg type='a{sv}' direction='in' />"

        "    </method>"

        "    <method name='RemoveNotification'>"

        "      <arg type='s' direction='in' />"

        "      <arg type='s' direction='in' />"

        "    </method>"

        "  </interface>"

        "</node>", &error);

      if (info == NULL)

        g_error ("%s", error->message);

      iface_info = g_dbus_node_info_lookup_interface (info, "org.gtk.Notifications");

      g_assert (iface_info);

      g_dbus_interface_info_ref (iface_info);

      g_dbus_node_info_unref (info);

    }

  return iface_info;

}

static void

g_notification_server_notification_added (GNotificationServer *server,

                                          const gchar         *app_id,

                                          const gchar         *notification_id,

                                          GVariant            *notification)

{

  GHashTable *notifications;

  notifications = g_hash_table_lookup (server->applications, app_id);

  if (notifications == NULL)

    {

      notifications = g_hash_table_new_full (g_str_hash, g_str_equal,

                                             g_free, (GDestroyNotify) g_variant_unref);

      g_hash_table_insert (server->applications, g_strdup (app_id), notifications);

    }

  g_hash_table_replace (notifications, g_strdup (notification_id), g_variant_ref (notification));

  g_signal_emit_by_name (server, "notification-received", app_id, notification_id, notification);

}

static void

g_notification_server_notification_removed (GNotificationServer *server,

                                            const gchar         *app_id,

                                            const gchar         *notification_id)

{

  GHashTable *notifications;

  notifications = g_hash_table_lookup (server->applications, app_id);

  if (notifications)

    {

      g_hash_table_remove (notifications, notification_id);

      if (g_hash_table_size (notifications) == 0)

        g_hash_table_remove (server->applications, app_id);

    }

  g_signal_emit_by_name (server, "notification-removed", app_id, notification_id);

}

static void

org_gtk_Notifications_method_call (GDBusConnection       *connection,

                                   const gchar           *sender,

                                   const gchar           *object_path,

                                   const gchar           *interface_name,

                                   const gchar           *method_name,

                                   GVariant              *parameters,

                                   GDBusMethodInvocation *invocation,

                                   gpointer               user_data)

{

  GNotificationServer *server = user_data;

  if (g_str_equal (method_name, "AddNotification"))

    {

      const gchar *app_id;

      const gchar *notification_id;

      GVariant *notification;

      g_variant_get (parameters, "(&s&s@a{sv})", &app_id, &notification_id, &notification);

      g_notification_server_notification_added (server, app_id, notification_id, notification);

      g_dbus_method_invocation_return_value (invocation, NULL);

      g_variant_unref (notification);

    }

  else if (g_str_equal (method_name, "RemoveNotification"))

    {

      const gchar *app_id;

      const gchar *notification_id;

      g_variant_get (parameters, "(&s&s)", &app_id, &notification_id);

      g_notification_server_notification_removed (server, app_id, notification_id);

      g_dbus_method_invocation_return_value (invocation, NULL);

    }

  else

    {

      g_dbus_method_invocation_return_dbus_error (invocation, "UnknownMethod", "No such method");

    }

}

static void

g_notification_server_dispose (GObject *object)

{

  GNotificationServer *server = G_NOTIFICATION_SERVER (object);

  g_notification_server_stop (server);

  g_clear_pointer (&server->applications, g_hash_table_unref);

  g_clear_object (&server->connection);

  G_OBJECT_CLASS (g_notification_server_parent_class)->dispose (object);

}

static void

g_notification_server_get_property (GObject    *object,

                                    guint       property_id,

                                    GValue     *value,

                                    GParamSpec *pspec)

{

  GNotificationServer *server = G_NOTIFICATION_SERVER (object);

  switch (property_id)

    {

    case PROP_IS_RUNNING:

      g_value_set_boolean (value, server->is_running);

      break;

    default:

      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);

    }

}

static void

g_notification_server_class_init (GNotificationServerClass *class)

{

  GObjectClass *object_class = G_OBJECT_CLASS (class);

  object_class->get_property = g_notification_server_get_property;

  object_class->dispose = g_notification_server_dispose;

  g_object_class_install_property (object_class, PROP_IS_RUNNING,

                                   g_param_spec_boolean ("is-running", "", "", FALSE,

                                                         G_PARAM_READABLE | G_PARAM_STATIC_STRINGS));

  g_signal_new ("notification-received", G_TYPE_NOTIFICATION_SERVER, G_SIGNAL_RUN_FIRST,

                0, NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 3,

                G_TYPE_STRING, G_TYPE_STRING, G_TYPE_VARIANT);

  g_signal_new ("notification-removed", G_TYPE_NOTIFICATION_SERVER, G_SIGNAL_RUN_FIRST,

                0, NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 2,

                G_TYPE_STRING, G_TYPE_STRING);

}

static void

g_notification_server_bus_acquired (GDBusConnection *connection,

                                    const gchar     *name,

                                    gpointer         user_data)

{

  const GDBusInterfaceVTable vtable = {

    org_gtk_Notifications_method_call, NULL, NULL

  };

  GNotificationServer *server = user_data;

  server->object_id = g_dbus_connection_register_object (connection, "/org/gtk/Notifications",

                                                         org_gtk_Notifications_get_interface (),

                                                         &vtable, server, NULL, NULL);

  /* register_object only fails if the same object is exported more than once */

  g_assert (server->object_id > 0);

  server->connection = g_object_ref (connection);

}

static void

g_notification_server_name_acquired (GDBusConnection *connection,

                                     const gchar     *name,

                                     gpointer         user_data)

{

  GNotificationServer *server = user_data;

  server->is_running = TRUE;

  g_object_notify (G_OBJECT (server), "is-running");

}

static void

g_notification_server_name_lost (GDBusConnection *connection,

                                 const gchar     *name,

                                 gpointer         user_data)

{

  GNotificationServer *server = user_data;

  g_notification_server_stop (server);

  if (connection == NULL && server->connection)

    g_clear_object (&server->connection);

}

static void

g_notification_server_init (GNotificationServer *server)

{

  server->applications = g_hash_table_new_full (g_str_hash, g_str_equal,

                                                g_free, (GDestroyNotify) g_hash_table_unref);

  server->name_owner_id = g_bus_own_name (G_BUS_TYPE_SESSION,

                                          "org.gtk.Notifications",

                                          G_BUS_NAME_OWNER_FLAGS_NONE,

                                          g_notification_server_bus_acquired,

                                          g_notification_server_name_acquired,

                                          g_notification_server_name_lost,

                                          server, NULL);

}

GNotificationServer *

g_notification_server_new (void)

{

  return g_object_new (G_TYPE_NOTIFICATION_SERVER, NULL);

}

void

g_notification_server_stop (GNotificationServer *server)

{

  g_return_if_fail (G_IS_NOTIFICATION_SERVER (server));

  if (server->name_owner_id)

    {

      g_bus_unown_name (server->name_owner_id);

      server->name_owner_id = 0;

    }

  if (server->object_id && server->connection)

    {

      g_dbus_connection_unregister_object (server->connection, server->object_id);

      server->object_id = 0;

    }

  if (server->is_running)

    {

      server->is_running = FALSE;

      g_object_notify (G_OBJECT (server), "is-running");

    }

}

gboolean

g_notification_server_get_is_running (GNotificationServer *server)

{

  g_return_val_if_fail (G_IS_NOTIFICATION_SERVER (server), FALSE);

  return server->is_running;

}

gchar **

g_notification_server_list_applications (GNotificationServer *server)

{

  g_return_val_if_fail (G_IS_NOTIFICATION_SERVER (server), NULL);

  return (gchar **) g_hash_table_get_keys_as_array (server->applications, NULL);

}

gchar **

g_notification_server_list_notifications (GNotificationServer *server,

                                          const gchar         *app_id)

{

  GHashTable *notifications;

  g_return_val_if_fail (G_IS_NOTIFICATION_SERVER (server), NULL);

  g_return_val_if_fail (app_id != NULL, NULL);

  notifications = g_hash_table_lookup (server->applications, app_id);

  if (notifications == NULL)

    return NULL;

  return (gchar **) g_hash_table_get_keys_as_array (notifications, NULL);

}