/* -*- mode: C; c-file-style: "gnu" -*- */

/* dbus-binding-tool-glib.c: Output C glue

 *

 * Copyright (C) 2003, 2004, 2005 Red Hat, Inc.

 * Copyright (C) 2005 Nokia

 * Portion Copyright © 2008 Nokia Corporation and/or its subsidiary(-ies). All rights reserved.

 * Licensed under the Academic Free License version 2.1

 *

 * 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 */

#ifndef __SYMBIAN32__

#include <config.h>

#else

#include "config.h"

#endif //__SYMBIAN32__

#include "dbus/dbus-glib.h"

#include "dbus-gidl.h"

#include "dbus-gparser.h"

#include "dbus-gutils.h"

#include "dbus-gtype-specialized.h"

#include "dbus-gsignature.h"

#include "dbus-gvalue-utils.h"

#include "dbus-glib-tool.h"

#include "dbus-binding-tool-glib.h"

#ifndef __SYMBIAN32__

#include <glib/gi18n.h>

#include <libintl.h>

#define _(x) dgettext (GETTEXT_PACKAGE, x)

#define N_(x) x

#else

#define _(x) x

#define N_(x) x

#endif

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <unistd.h>

#define MARSHAL_PREFIX "dbus_glib_marshal_"

typedef struct

{

  gboolean ignore_unsupported;

  const char* prefix;

  GIOChannel *channel;

  GError **error;

  GHashTable *generated;

  GString *blob;

  GString *signal_blob;

  GString *property_blob;

  guint count;

} DBusBindingToolCData;

static gboolean gather_marshallers (BaseInfo *base, DBusBindingToolCData *data, GError **error);

static gboolean generate_glue (BaseInfo *base, DBusBindingToolCData *data, GError **error);

static gboolean generate_client_glue (BaseInfo *base, DBusBindingToolCData *data, GError **error);

static const char *

dbus_g_type_get_marshal_name (GType gtype)

{

  switch (G_TYPE_FUNDAMENTAL (gtype))

    {

    case G_TYPE_NONE:

      return "NONE";

    case G_TYPE_BOOLEAN:

      return "BOOLEAN";

    case G_TYPE_UCHAR:

      return "UCHAR";

    case G_TYPE_INT:

      return "INT";

    case G_TYPE_UINT:

      return "UINT";

    case G_TYPE_INT64:

      return "INT64";

    case G_TYPE_UINT64:

      return "UINT64";

    case G_TYPE_DOUBLE:

      return "DOUBLE";

    case G_TYPE_STRING:

      return "STRING";

    case G_TYPE_POINTER:

      return "POINTER";

    case G_TYPE_BOXED:

      return "BOXED";

    case G_TYPE_OBJECT:

      return "OBJECT";

    default:

      return NULL;

    }

}

/* This entire function is kind of...ugh. */

static const char *

dbus_g_type_get_c_name (GType gtype)

{

  GType subtype;

  if (dbus_g_type_is_struct (gtype))

    {

      return "GValueArray";

    }

  if (dbus_g_type_is_collection (gtype))

    {

      subtype = dbus_g_type_get_collection_specialization(gtype);

      if (_dbus_g_type_is_fixed (subtype))

        return "GArray";

      else

        return "GPtrArray";

    }

  if (dbus_g_type_is_map (gtype))

    return "GHashTable";

  if (g_type_is_a (gtype, G_TYPE_STRING))

    return "char *";

  /* This one is even more hacky...we get an extra *

   * because G_TYPE_STRV is a G_TYPE_BOXED

   */

  if (g_type_is_a (gtype, G_TYPE_STRV))

    return "char *";

  if (g_type_is_a (gtype, DBUS_TYPE_G_OBJECT_PATH))

    return "char";

  return g_type_name (gtype);

}

static gboolean

compute_gsignature (MethodInfo *method, GType *rettype, GArray **params, GError **error)

{

  GSList *elt;

  GType retval_type;

  GArray *ret;

  gboolean is_async;

  const char *arg_type;

  gboolean retval_signals_error;

  is_async = method_info_get_annotation (method, DBUS_GLIB_ANNOTATION_ASYNC) != NULL;

  retval_signals_error = FALSE;

  ret = g_array_new (TRUE, TRUE, sizeof (GType));

  if (is_async)

    retval_type = G_TYPE_NONE;

  else

    {

      gboolean found_retval;

      /* Look for return value */

      found_retval = FALSE;

      for (elt = method_info_get_args (method); elt; elt = elt->next)

	{

	  ArgInfo *arg = elt->data;

	  const char *returnval_annotation;

	  returnval_annotation = arg_info_get_annotation (arg, DBUS_GLIB_ANNOTATION_RETURNVAL);

	  if (returnval_annotation != NULL)

	    {

	      arg_type = arg_info_get_type (arg);

	      retval_type = _dbus_gtype_from_signature (arg_type, FALSE);

	      if (retval_type == G_TYPE_INVALID)

		goto invalid_type;

	      found_retval = TRUE;

	      if (!strcmp (returnval_annotation, "error"))

		retval_signals_error = TRUE;

	      break;

	    }

	}

      if (!found_retval)

	{

	  retval_type = G_TYPE_BOOLEAN;

	  retval_signals_error = TRUE;

	}

    }

  *rettype = retval_type;

  /* Handle all input arguments */

  for (elt = method_info_get_args (method); elt; elt = elt->next)

    {

      ArgInfo *arg = elt->data;

      if (arg_info_get_direction (arg) == ARG_IN)

	{

	  GType gtype;

	  arg_type = arg_info_get_type (arg);

	  gtype = _dbus_gtype_from_signature (arg_type, FALSE);

	  if (gtype == G_TYPE_INVALID)

	    goto invalid_type;

	  g_array_append_val (ret, gtype);

	}

    }

  if (!is_async)

    {

      /* Append pointer for each out arg storage */

      for (elt = method_info_get_args (method); elt; elt = elt->next)

	{

	  ArgInfo *arg = elt->data;

	  /* Skip return value */

	  if (arg_info_get_annotation (arg, DBUS_GLIB_ANNOTATION_RETURNVAL) != NULL)

	    continue;

	  if (arg_info_get_direction (arg) == ARG_OUT)

	    {

	      GType gtype;

	      arg_type = arg_info_get_type (arg);

	      gtype = _dbus_gtype_from_signature (arg_type, FALSE);

	      if (gtype == G_TYPE_INVALID)

		goto invalid_type;

	      /* We actually just need a pointer for the return value

		 storage */

	      gtype = G_TYPE_POINTER;

	      g_array_append_val (ret, gtype);

	    }

	}

      if (retval_signals_error)

	{

	  /* Final GError parameter */

	  GType gtype = G_TYPE_POINTER;

	  g_array_append_val (ret, gtype);

	}

    }

  else

    {

      /* Context pointer */

      GType gtype = G_TYPE_POINTER;

      g_array_append_val (ret, gtype);

    }

  *params = ret;

  return TRUE;

 invalid_type:

  g_set_error (error,

	       DBUS_BINDING_TOOL_ERROR,

	       DBUS_BINDING_TOOL_ERROR_UNSUPPORTED_CONVERSION,

	       _("Unsupported conversion from D-BUS type %s to glib-genmarshal type"),

	       arg_type);

  return FALSE;

}

static char *

compute_marshaller (MethodInfo *method, GError **error)

{

  GArray *signature;

  GType rettype;

  const char *marshal_name;

  GString *ret;

  guint i;

  if (!compute_gsignature (method, &rettype, &signature, error))

    return NULL;

  ret = g_string_new ("");

  marshal_name = dbus_g_type_get_marshal_name (rettype);

  g_assert (marshal_name != NULL);

  g_string_append (ret, marshal_name);

  g_string_append_c (ret, ':');

  for (i = 0; i < signature->len; i++)

    {

      marshal_name = dbus_g_type_get_marshal_name (g_array_index (signature, GType, i));

      g_assert (marshal_name != NULL);

      g_string_append (ret, marshal_name);

      if (i < signature->len - 1)

	g_string_append_c (ret, ',');

    }

  if (signature->len == 0)

    {

      marshal_name = dbus_g_type_get_marshal_name (G_TYPE_NONE);

      g_assert (marshal_name != NULL);

      g_string_append (ret, marshal_name);

    }

  g_array_free (signature, TRUE);

  return g_string_free (ret, FALSE);

}

static char *

compute_marshaller_name (MethodInfo *method, const char *prefix, GError **error)

{

  GString *ret;

  GArray *signature;

  GType rettype;

  const char *marshal_name;

  guint i;

  if (!compute_gsignature (method, &rettype, &signature, error))

    return NULL;

  ret = g_string_new (MARSHAL_PREFIX);

  g_string_append (ret, prefix);

  g_string_append_c (ret, '_');

  marshal_name = dbus_g_type_get_marshal_name (rettype);

  g_assert (marshal_name != NULL);

  g_string_append (ret, marshal_name);

  g_string_append (ret, "__");

  for (i = 0; i < signature->len; i++)

    {

      marshal_name = dbus_g_type_get_marshal_name (g_array_index (signature, GType, i));

      g_assert (marshal_name != NULL);

      g_string_append (ret, marshal_name);

      if (i < signature->len - 1)

	g_string_append_c (ret, '_');

    }

  if (signature->len == 0)

    {

      marshal_name = dbus_g_type_get_marshal_name (G_TYPE_NONE);

      g_assert (marshal_name != NULL);

      g_string_append (ret, marshal_name);

    }

  g_array_free (signature, TRUE);

  return g_string_free (ret, FALSE);

}

static gboolean

gather_marshallers_list (GSList *list, DBusBindingToolCData *data, GError **error)

{

  GSList *tmp;

  tmp = list;

  while (tmp != NULL)

    {

      if (!gather_marshallers (tmp->data, data, error))

	return FALSE;

      tmp = tmp->next;

    }

  return TRUE;

}

static gboolean

gather_marshallers (BaseInfo *base, DBusBindingToolCData *data, GError **error)

{

  if (base_info_get_type (base) == INFO_TYPE_NODE)

    {

      if (!gather_marshallers_list (node_info_get_nodes ((NodeInfo *) base),

				    data, error))

	return FALSE;

      if (!gather_marshallers_list (node_info_get_interfaces ((NodeInfo *) base),

				    data, error))

	return FALSE;

    }

  else

    {

      InterfaceInfo *interface;

      GSList *methods;

      GSList *tmp;

      const char *interface_c_name;

      interface = (InterfaceInfo *) base;

      interface_c_name = interface_info_get_annotation (interface, DBUS_GLIB_ANNOTATION_C_SYMBOL);

      if (interface_c_name == NULL)

        {

	  if (!data->prefix)

	    return TRUE;

        }

      methods = interface_info_get_methods (interface);

      /* Generate the necessary marshallers for the methods. */

      for (tmp = methods; tmp != NULL; tmp = g_slist_next (tmp))

        {

          MethodInfo *method;

          char *marshaller_name;

          method = (MethodInfo *) tmp->data;

          marshaller_name = compute_marshaller (method, error);

	  if (!marshaller_name)

	    return FALSE;

	  if (g_hash_table_lookup (data->generated, marshaller_name))

	    {

	      g_free (marshaller_name);

	      continue;

	    }

	  g_hash_table_insert (data->generated, marshaller_name, NULL);

        }

    }

  return TRUE;

}

static gboolean

generate_glue_list (GSList *list, DBusBindingToolCData *data, GError **error)

{

  GSList *tmp;

  tmp = list;

  while (tmp != NULL)

    {

      if (!generate_glue (tmp->data, data, error))

	return FALSE;

      tmp = tmp->next;

    }

  return TRUE;

}

#define WRITE_OR_LOSE(x) do { gsize bytes_written; if (!g_io_channel_write_chars (channel, x, -1, &bytes_written, error)) goto io_lose; } while (0)

static gboolean

write_printf_to_iochannel (const char *fmt, GIOChannel *channel, GError **error, ...)

{

  char *str;

  va_list args;

  GIOStatus status;

  gsize written;

  gboolean ret;

  va_start (args, error);

  str = g_strdup_vprintf (fmt, args);

  if ((status = g_io_channel_write_chars (channel, str, -1, &written, error)) == G_IO_STATUS_NORMAL)

    ret = TRUE;

  else

    ret = FALSE;

  g_free (str);

  va_end (args);

  return ret;

}

static gboolean

write_quoted_string (GIOChannel *channel, GString *string, GError **error)

{

  guint i;

  WRITE_OR_LOSE ("\"");

  for (i = 0; i < string->len; i++)

    {

      if (string->str[i] != '\0')

	{

	  if (!g_io_channel_write_chars (channel, string->str + i, 1, NULL, error))

	    return FALSE;

	}

      else

	{

	  if (!g_io_channel_write_chars (channel, "\\0", -1, NULL, error))

	    return FALSE;

	}

    }

  WRITE_OR_LOSE ("\\0\"");

  return TRUE;

 io_lose:

  return FALSE;

}

static gboolean

generate_glue (BaseInfo *base, DBusBindingToolCData *data, GError **error)

{

  if (base_info_get_type (base) == INFO_TYPE_NODE)

    {

      GString *object_introspection_data_blob;

      GIOChannel *channel;

      channel = data->channel;

      object_introspection_data_blob = g_string_new_len ("", 0);

      data->blob = object_introspection_data_blob;

      data->count = 0;

      data->signal_blob = g_string_new_len ("", 0);

      data->property_blob = g_string_new_len ("", 0);

      if (!write_printf_to_iochannel ("static const DBusGMethodInfo dbus_glib_%s_methods[] = {\n", channel, error, data->prefix))

	goto io_lose;

      if (!generate_glue_list (node_info_get_nodes ((NodeInfo *) base),

			       data, error))

	return FALSE;

      if (!generate_glue_list (node_info_get_interfaces ((NodeInfo *) base),

			       data, error))

	return FALSE;

      WRITE_OR_LOSE ("};\n\n");

      /* Information about the object. */

      if (!write_printf_to_iochannel ("const DBusGObjectInfo dbus_glib_%s_object_info = {\n",

				      channel, error, data->prefix))

	goto io_lose;

      WRITE_OR_LOSE ("  0,\n");

      if (!write_printf_to_iochannel ("  dbus_glib_%s_methods,\n", channel, error, data->prefix))

	goto io_lose;

      if (!write_printf_to_iochannel ("  %d,\n", channel, error, data->count))

	goto io_lose;

      if (!write_quoted_string (channel, object_introspection_data_blob, error))

	goto io_lose;

      WRITE_OR_LOSE (",\n");

      if (!write_quoted_string (channel, data->signal_blob, error))

	goto io_lose;

      WRITE_OR_LOSE (",\n");

      if (!write_quoted_string (channel, data->property_blob, error))

	goto io_lose;

      WRITE_OR_LOSE ("\n};\n\n");

      g_string_free (object_introspection_data_blob, TRUE);

      g_string_free (data->signal_blob, TRUE);

      g_string_free (data->property_blob, TRUE);