/*
 * Copyright © 2015 Canonical Limited
 *
 * 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 of the licence, 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/>.
 *
 * Author: Ryan Lortie <desrt@desrt.ca>
 */

#include "config.h"

#include "gcontextspecificgroup.h"

#include <glib-object.h>
#include <gobject/gvaluecollector.h>
#include "glib-private.h"

#include <string.h>

typedef struct
{
  gint   ref_count;
  guint  signal_id;
  GQuark detail;
  guint  n_params;
  GValue params[1];
} GContextSpecificEmission;

static GContextSpecificEmission *
g_context_specific_emission_newv (guint   signal_id,
                                  GQuark  detail,
                                  va_list ap)
{
  GContextSpecificEmission *emission;
  GSignalQuery query;
  guint i;

  g_signal_query (signal_id, &query);

  if (query.n_params == 0 && detail == 0)
    return GUINT_TO_POINTER (signal_id * 2 + 1);

  emission = g_malloc (G_STRUCT_OFFSET (GContextSpecificEmission, params[query.n_params]));
  emission->ref_count = 1;
  emission->signal_id = signal_id;
  emission->detail = detail;
  emission->n_params = query.n_params;

  for (i = 0; i < query.n_params; i++)
    {
      gchar *error;

      G_VALUE_COLLECT_INIT (&emission->params[i], query.param_types[i], ap, 0, &error);
      if (error)
        g_error ("g_contect_specific_emission_newv: %s", error);
    }

  return emission;
}

static GContextSpecificEmission *
g_context_specific_emission_ref (GContextSpecificEmission *emission)
{
  if (~GPOINTER_TO_UINT (emission) & 1)
    g_atomic_int_inc (&emission->ref_count);

  return emission;
}

static void
g_context_specific_emission_unref (GContextSpecificEmission *emission)
{
  if (~GPOINTER_TO_UINT (emission) & 1)
    {
      if (g_atomic_int_dec_and_test (&emission->ref_count))
        {
          guint i;

          for (i = 0; i < emission->n_params; i++)
            g_value_reset (&emission->params[i]);

          g_free (emission);
        }
    }
}

static void
g_context_specific_emission_emit_on_instance (GContextSpecificEmission *emission,
                                              gpointer                  instance)
{
  if (~GPOINTER_TO_UINT (emission) & 1)
    {
      GValue *instance_and_params;

      instance_and_params = g_newa (GValue, 1 + emission->n_params);
      instance_and_params[0].g_type = G_TYPE_FROM_INSTANCE (instance);
      instance_and_params[0].data[0].v_pointer = instance;
      memcpy (instance_and_params + 1, emission->params, sizeof (GValue) * emission->n_params);

      g_signal_emitv (instance_and_params, emission->signal_id, emission->detail, NULL);
    }
  else
    g_signal_emit (instance, GPOINTER_TO_UINT (emission) / 2, 0);
}

struct _GContextSpecificSource
{
  GSource   source;

  GMutex    lock;
  gpointer  instance;
  GQueue    pending;
};

static gboolean
g_context_specific_source_dispatch (GSource     *source,
                                    GSourceFunc  callback,
                                    gpointer     user_data)
{
  GContextSpecificSource *css = (GContextSpecificSource *) source;
  GContextSpecificEmission *emission;

  g_mutex_lock (&css->lock);

  g_assert (!g_queue_is_empty (&css->pending));
  emission = g_queue_pop_head (&css->pending);

  if (g_queue_is_empty (&css->pending))
    g_source_set_ready_time (source, -1);

  g_mutex_unlock (&css->lock);

  g_context_specific_emission_emit_on_instance (emission, css->instance);
  g_context_specific_emission_unref (emission);

  return TRUE;
}

static void
g_context_specific_source_finalize (GSource *source)
{
  GContextSpecificSource *css = (GContextSpecificSource *) source;

  while (!g_queue_is_empty (&css->pending))
    g_context_specific_emission_unref (g_queue_pop_head (&css->pending));

  g_queue_clear (&css->pending);
  g_mutex_clear (&css->lock);
}

gpointer
g_context_specific_source_new (const gchar *name,
                               gpointer     instance,
                               gsize        user_data_size)
{
  static GSourceFuncs source_funcs = {
    NULL,
    NULL,
    g_context_specific_source_dispatch,
    g_context_specific_source_finalize
  };
  GContextSpecificSource *css;
  GSource *source;

  /* generally, simple addition would be unsafe, but in this case we
   * know that the GContextSpecificSource struct will be at least as
   * aligned as anything that we will put in the user_data.  In theory
   * we should change that if this ever becomes a public API.
   */
  source = g_source_new (&source_funcs, sizeof (GContextSpecificSource) + user_data_size);
  css = (GContextSpecificSource *) source;

  g_source_set_name (source, name);

  g_mutex_init (&css->lock);
  g_queue_init (&css->pending);
  css->instance = instance;

  return css;
}

gpointer
g_context_specific_source_get_user_data (GContextSpecificSource *source)
{
  return G_STRUCT_MEMBER_P (source, sizeof (GContextSpecificSource));
}

void
g_context_specific_source_emit (GContextSpecificSource *css,
                                guint                   signal_id,
                                GQuark                  detail,
                                ...)
{
  GContextSpecificEmission *emission;
  va_list ap;

  va_start (ap, detail);
  emission = g_context_specific_emission_newv (signal_id, detail, ap);
  va_end (ap);

  g_mutex_lock (&css->lock);
  /* this will find repeat instances of void handlers */
  if (!g_queue_find (&css->pending, emission))
    g_queue_push_tail (&css->pending, emission);
  g_mutex_unlock (&css->lock);
}

gpointer
g_context_specific_group_get (GContextSpecificGroup *group,
                              GType                  type,
                              goffset                context_offset,
                              GSourceFunc            start_func)
{
  GContextSpecificSource *css;
  GMainContext *context;

  context = g_main_context_get_thread_default ();
  if (!context)
    context = g_main_context_default ();

  if (!g_main_context_ensure_is_owner (context))
    g_critical ("Object %s is being set up in the wrong thread (or you forgot "
                "to set the thread-default main context).", g_type_name (type));

  g_mutex_lock (&group->lock);

  if (!group->table)
    group->table = g_hash_table_new (NULL, NULL);

  /* already started */
  if (start_func && g_hash_table_size (group->table) > 0)
    start_func = 0;

  css = g_hash_table_lookup (group->table, context);

  g_mutex_unlock (&group->lock);

  if (!css)
    {
      gpointer instance;

      instance = g_object_new (type, NULL);
      css = g_context_specific_source_new (g_type_name (type), instance, 0);
      G_STRUCT_MEMBER (GMainContext *, instance, context_offset) = context;
      g_source_attach ((GSource *) css, context);

      g_mutex_lock (&group->lock);
      /* We know that nobody else could have added an item for 'context'
       * since we are the owner and there can only be one owner.
       */
      g_hash_table_insert (group->table, context, css);
      g_mutex_unlock (&group->lock);
    }

  if (start_func)
    g_main_context_invoke (GLIB_PRIVATE_CALL(g_get_worker_context) (), start_func, NULL);

  return css->instance;
}

void
g_context_specific_group_remove (GContextSpecificGroup *group,
                                 GMainContext          *context,
                                 gpointer               instance,
                                 GSourceFunc            stop_func)
{
  GContextSpecificSource *css;

  if (!context)
    {
      g_critical ("Removing %s with NULL context.  This object was probably directly constructed from a "
                  "dynamic language.  This is not a valid use of the API.", G_OBJECT_TYPE_NAME (instance));
      return;
    }

  if (!g_main_context_ensure_is_owner (context))
    g_critical ("Object %s is being torn down in the wrong thread.", G_OBJECT_TYPE_NAME (instance));

  g_mutex_lock (&group->lock);
  css = g_hash_table_lookup (group->table, context);
  g_hash_table_remove (group->table, context);
  g_assert (css);

  /* don't stop if there are others */
  if (stop_func && g_hash_table_size (group->table) > 0)
    stop_func = NULL;

  g_mutex_unlock (&group->lock);

  g_assert (css->instance == instance);

  g_source_unref ((GSource *) css);

  if (stop_func)
    g_main_context_invoke (GLIB_PRIVATE_CALL(g_get_worker_context) (), stop_func, NULL);
}

void
g_context_specific_group_emit (GContextSpecificGroup *group,
                               guint                  signal_id,
                               GQuark                 detail,
                               ...)
{
  GContextSpecificEmission *emission;
  va_list ap;

  va_start (ap, detail);
  emission = g_context_specific_emission_newv (signal_id, detail, ap);
  va_end (ap);

  g_mutex_lock (&group->lock);

  if (group->table)
    {
      GHashTableIter iter;
      gpointer value;

      g_hash_table_iter_init (&iter, group->table);
      while (g_hash_table_iter_next (&iter, NULL, &value))
        {
          GContextSpecificSource *css = value;

          g_mutex_lock (&css->lock);

          /* this will find repeat instances of void handlers */
          if (!g_queue_find (&css->pending, emission))
            g_queue_push_tail (&css->pending, g_context_specific_emission_ref (emission));

          g_source_set_ready_time ((GSource *) css, 0);

          g_mutex_unlock (&css->lock);
        }
    }

  g_mutex_unlock (&group->lock);

  g_context_specific_emission_unref (emission);
}