/* Pinos
 * Copyright (C) 2015 Wim Taymans <wim.taymans@gmail.com>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include <string.h>
#include <sys/socket.h>

#include <gio/gunixfdlist.h>

#include "pinos/client/enumtypes.h"

#include "pinos/server/daemon.h"
#include "pinos/server/source-output.h"

#include "pinos/dbus/org-pinos.h"

struct _PinosSourceOutputPrivate
{
  PinosDaemon *daemon;
  PinosSourceOutput1 *iface;

  gchar *object_path;
  gchar *client_path;
  gchar *source_path;

  GBytes *possible_formats;
  PinosProperties *properties;
  GBytes *requested_format;
  PinosSourceOutputState state;
  GBytes *format;

  GSocket *socket;
};

#define PINOS_SOURCE_OUTPUT_GET_PRIVATE(obj)  \
   (G_TYPE_INSTANCE_GET_PRIVATE ((obj), PINOS_TYPE_SOURCE_OUTPUT, PinosSourceOutputPrivate))

G_DEFINE_TYPE (PinosSourceOutput, pinos_source_output, G_TYPE_OBJECT);

enum
{
  PROP_0,
  PROP_DAEMON,
  PROP_OBJECT_PATH,
  PROP_CLIENT_PATH,
  PROP_SOURCE_PATH,
  PROP_POSSIBLE_FORMATS,
  PROP_PROPERTIES,
  PROP_REQUESTED_FORMAT,
  PROP_FORMAT,
  PROP_SOCKET,
  PROP_STATE,
};

enum
{
  SIGNAL_REMOVE,
  LAST_SIGNAL
};

static guint signals[LAST_SIGNAL] = { 0 };

static void
pinos_source_output_get_property (GObject    *_object,
                                  guint       prop_id,
                                  GValue     *value,
                                  GParamSpec *pspec)
{
  PinosSourceOutput *output = PINOS_SOURCE_OUTPUT (_object);
  PinosSourceOutputPrivate *priv = output->priv;

  switch (prop_id) {
    case PROP_DAEMON:
      g_value_set_object (value, priv->daemon);
      break;

    case PROP_OBJECT_PATH:
      g_value_set_string (value, priv->object_path);
      break;

    case PROP_CLIENT_PATH:
      g_value_set_string (value, priv->client_path);
      break;

    case PROP_SOURCE_PATH:
      g_value_set_string (value, priv->source_path);
      break;

    case PROP_POSSIBLE_FORMATS:
      g_value_set_boxed (value, priv->possible_formats);
      break;

    case PROP_PROPERTIES:
      g_value_set_boxed (value, priv->properties);
      break;

    case PROP_REQUESTED_FORMAT:
      g_value_set_boxed (value, priv->requested_format);
      break;

    case PROP_FORMAT:
      g_value_set_boxed (value, priv->format);
      break;

    case PROP_SOCKET:
      g_value_set_object (value, priv->socket);
      break;

    case PROP_STATE:
      g_value_set_uint (value, priv->state);
      break;

    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (output, prop_id, pspec);
      break;
  }
}

static void
pinos_source_output_set_property (GObject      *_object,
                                  guint         prop_id,
                                  const GValue *value,
                                  GParamSpec   *pspec)
{
  PinosSourceOutput *output = PINOS_SOURCE_OUTPUT (_object);
  PinosSourceOutputPrivate *priv = output->priv;

  switch (prop_id) {
    case PROP_DAEMON:
      priv->daemon = g_value_dup_object (value);
      break;

    case PROP_OBJECT_PATH:
      priv->object_path = g_value_dup_string (value);
      break;

    case PROP_CLIENT_PATH:
      priv->client_path = g_value_dup_string (value);
      g_object_set (priv->iface, "client", priv->client_path, NULL);
      break;

    case PROP_SOURCE_PATH:
      priv->source_path = g_value_dup_string (value);
      g_object_set (priv->iface, "source", priv->source_path, NULL);
      break;

    case PROP_POSSIBLE_FORMATS:
      if (priv->possible_formats)
        g_bytes_unref (priv->possible_formats);
      priv->possible_formats = g_value_dup_boxed (value);
      g_object_set (priv->iface, "possible-formats",
          g_bytes_get_data (priv->possible_formats, NULL), NULL);
      break;

    case PROP_PROPERTIES:
      if (priv->properties)
        pinos_properties_free (priv->properties);
      priv->properties = g_value_dup_boxed (value);
      g_object_set (priv->iface, "properties",
          pinos_properties_to_variant (priv->properties), NULL);
      break;

    case PROP_FORMAT:
      if (priv->format)
        g_bytes_unref (priv->format);
      priv->format = g_value_dup_boxed (value);
      break;

    default:
      G_OBJECT_WARN_INVALID_PROPERTY_ID (output, prop_id, pspec);
      break;
  }
}

static void
clear_formats (PinosSourceOutput *output)
{
  PinosSourceOutputPrivate *priv = output->priv;

  g_clear_pointer (&priv->requested_format, g_bytes_unref);
  g_clear_pointer (&priv->format, g_bytes_unref);
}

static void
stop_transfer (PinosSourceOutput *output)
{
  PinosSourceOutputPrivate *priv = output->priv;

  if (priv->socket) {
    g_clear_object (&priv->socket);
    g_object_notify (G_OBJECT (output), "socket");
  }
  clear_formats (output);
  priv->state = PINOS_SOURCE_OUTPUT_STATE_IDLE;
  g_object_set (priv->iface,
               "state", priv->state,
                NULL);
}

static gboolean
handle_start (PinosSourceOutput1     *interface,
              GDBusMethodInvocation  *invocation,
              const gchar            *arg_requested_format,
              gpointer                user_data)
{
  PinosSourceOutput *output = user_data;
  PinosSourceOutputPrivate *priv = output->priv;
  GUnixFDList *fdlist;
  gint fd[2];

  priv->state = PINOS_SOURCE_OUTPUT_STATE_STARTING;

  priv->requested_format = g_bytes_new (arg_requested_format,
                                        strlen (arg_requested_format) + 1);

  socketpair (AF_UNIX, SOCK_STREAM, 0, fd);

  g_clear_object (&priv->socket);
  priv->socket = g_socket_new_from_fd (fd[0], NULL);
  g_object_set_data (priv->socket, "pinos-client-path", priv->client_path);
  g_object_notify (G_OBJECT (output), "socket");

  /* the notify of the socket above should configure the format */
  if (priv->format == NULL)
    goto no_format;

  priv->state = PINOS_SOURCE_OUTPUT_STATE_STREAMING;

  fdlist = g_unix_fd_list_new ();
  g_unix_fd_list_append (fdlist, fd[1], NULL);

  g_dbus_method_invocation_return_value_with_unix_fd_list (invocation,
           g_variant_new ("(hs@a{sv})",
                          0,
                          g_bytes_get_data (priv->format, NULL),
                          pinos_properties_to_variant (priv->properties)),
           fdlist);
  g_object_unref (fdlist);
  close (fd[1]);

  g_object_set (priv->iface,
               "format", g_bytes_get_data (priv->format, NULL),
               "state", priv->state,
                NULL);

  return TRUE;

  /* error */
no_format:
  {
    g_dbus_method_invocation_return_dbus_error (invocation,
        "org.pinos.Error", "No format");
    close (fd[0]);
    close (fd[1]);
    g_clear_pointer (&priv->requested_format, g_bytes_unref);
    g_clear_object (&priv->socket);
    return TRUE;
  }
}

static gboolean
handle_stop (PinosSourceOutput1     *interface,
             GDBusMethodInvocation  *invocation,
             gpointer                user_data)
{
  PinosSourceOutput *output = user_data;

  stop_transfer (output);

  g_dbus_method_invocation_return_value (invocation, NULL);

  return TRUE;
}

static gboolean
handle_remove (PinosSourceOutput1     *interface,
               GDBusMethodInvocation  *invocation,
               gpointer                user_data)
{
  PinosSourceOutput *output = user_data;

  stop_transfer (output);

  g_signal_emit (output, signals[SIGNAL_REMOVE], 0, NULL);

  g_dbus_method_invocation_return_value (invocation, NULL);

  return TRUE;
}

static void
output_register_object (PinosSourceOutput *output,
                        const gchar       *prefix)
{
  PinosSourceOutputPrivate *priv = output->priv;
  PinosObjectSkeleton *skel;
  gchar *name;

  name = g_strdup_printf ("%s/output", prefix);
  skel = pinos_object_skeleton_new (name);
  g_free (name);

  pinos_object_skeleton_set_source_output1 (skel, priv->iface);

  g_free (priv->object_path);
  priv->object_path = pinos_daemon_export_uniquely (priv->daemon, G_DBUS_OBJECT_SKELETON (skel));
}

static void
output_unregister_object (PinosSourceOutput *output)
{
  PinosSourceOutputPrivate *priv = output->priv;

  pinos_daemon_unexport (priv->daemon, priv->object_path);
}

static void
pinos_source_output_dispose (GObject * object)
{
  PinosSourceOutput *output = PINOS_SOURCE_OUTPUT (object);
  PinosSourceOutputPrivate *priv = output->priv;

  clear_formats (output);
  g_clear_object (&priv->socket);
  output_unregister_object (output);

  G_OBJECT_CLASS (pinos_source_output_parent_class)->dispose (object);
}

static void
pinos_source_output_finalize (GObject * object)
{
  PinosSourceOutput *output = PINOS_SOURCE_OUTPUT (object);
  PinosSourceOutputPrivate *priv = output->priv;

  if (priv->possible_formats)
    g_bytes_unref (priv->possible_formats);
  if (priv->properties)
    pinos_properties_free (priv->properties);
  g_clear_object (&priv->daemon);
  g_clear_object (&priv->iface);
  g_free (priv->client_path);
  g_free (priv->object_path);
  g_free (priv->source_path);

  G_OBJECT_CLASS (pinos_source_output_parent_class)->finalize (object);
}

static void
pinos_source_output_constructed (GObject * object)
{
  PinosSourceOutput *output = PINOS_SOURCE_OUTPUT (object);
  PinosSourceOutputPrivate *priv = output->priv;

  output_register_object (output, priv->object_path);

  G_OBJECT_CLASS (pinos_source_output_parent_class)->constructed (object);
}

static void
pinos_source_output_class_init (PinosSourceOutputClass * klass)
{
  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);

  g_type_class_add_private (klass, sizeof (PinosSourceOutputPrivate));

  gobject_class->constructed = pinos_source_output_constructed;
  gobject_class->dispose = pinos_source_output_dispose;
  gobject_class->finalize = pinos_source_output_finalize;
  gobject_class->set_property = pinos_source_output_set_property;
  gobject_class->get_property = pinos_source_output_get_property;

  g_object_class_install_property (gobject_class,
                                   PROP_DAEMON,
                                   g_param_spec_object ("daemon",
                                                        "Daemon",
                                                        "The Daemon",
                                                        PINOS_TYPE_DAEMON,
                                                        G_PARAM_READWRITE |
                                                        G_PARAM_CONSTRUCT_ONLY |
                                                        G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_OBJECT_PATH,
                                   g_param_spec_string ("object-path",
                                                        "Object Path",
                                                        "The object path",
                                                        NULL,
                                                        G_PARAM_READWRITE |
                                                        G_PARAM_CONSTRUCT_ONLY |
                                                        G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_CLIENT_PATH,
                                   g_param_spec_string ("client-path",
                                                        "Client Path",
                                                        "The client object path",
                                                        NULL,
                                                        G_PARAM_READWRITE |
                                                        G_PARAM_CONSTRUCT_ONLY |
                                                        G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_SOURCE_PATH,
                                   g_param_spec_string ("source-path",
                                                        "Source Path",
                                                        "The source object path",
                                                        NULL,
                                                        G_PARAM_READWRITE |
                                                        G_PARAM_CONSTRUCT_ONLY |
                                                        G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_POSSIBLE_FORMATS,
                                   g_param_spec_boxed ("possible-formats",
                                                       "Possible Formats",
                                                       "The possbile formats of the stream",
                                                       G_TYPE_BYTES,
                                                       G_PARAM_READWRITE |
                                                       G_PARAM_CONSTRUCT |
                                                       G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_PROPERTIES,
                                   g_param_spec_boxed ("properties",
                                                       "Properties",
                                                       "Extra properties of the stream",
                                                       PINOS_TYPE_PROPERTIES,
                                                       G_PARAM_READWRITE |
                                                       G_PARAM_CONSTRUCT |
                                                       G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_REQUESTED_FORMAT,
                                   g_param_spec_boxed ("requested-format",
                                                       "Requested Format",
                                                       "The requested format of the stream",
                                                       G_TYPE_BYTES,
                                                       G_PARAM_READABLE |
                                                       G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_FORMAT,
                                   g_param_spec_boxed ("format",
                                                       "Format",
                                                       "The format of the stream",
                                                       G_TYPE_BYTES,
                                                       G_PARAM_READWRITE |
                                                       G_PARAM_STATIC_STRINGS));

  g_object_class_install_property (gobject_class,
                                   PROP_SOCKET,
                                   g_param_spec_object ("socket",
                                                        "Socket",
                                                        "The socket with data",
                                                        G_TYPE_SOCKET,
                                                        G_PARAM_READABLE |
                                                        G_PARAM_STATIC_STRINGS));

  signals[SIGNAL_REMOVE] = g_signal_new ("remove",
                                         G_TYPE_FROM_CLASS (klass),
                                         G_SIGNAL_RUN_LAST,
                                         0,
                                         NULL,
                                         NULL,
                                         g_cclosure_marshal_generic,
                                         G_TYPE_NONE,
                                         0,
                                         G_TYPE_NONE);
}

static void
pinos_source_output_init (PinosSourceOutput * output)
{
  PinosSourceOutputPrivate *priv = output->priv = PINOS_SOURCE_OUTPUT_GET_PRIVATE (output);

  priv->iface = pinos_source_output1_skeleton_new ();
  g_signal_connect (priv->iface, "handle-start", (GCallback) handle_start, output);
  g_signal_connect (priv->iface, "handle-stop", (GCallback) handle_stop, output);
  g_signal_connect (priv->iface, "handle-remove", (GCallback) handle_remove, output);

  priv->state = PINOS_SOURCE_OUTPUT_STATE_IDLE;
  g_object_set (priv->iface, "state", priv->state, NULL);
}

void
pinos_source_output_remove (PinosSourceOutput *output)
{
  stop_transfer (output);

  g_signal_emit (output, signals[SIGNAL_REMOVE], 0, NULL);
}

const gchar *
pinos_source_output_get_object_path (PinosSourceOutput *output)
{
  PinosSourceOutputPrivate *priv;

  g_return_val_if_fail (PINOS_IS_SOURCE_OUTPUT (output), NULL);
  priv = output->priv;

  return priv->object_path;
}