/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
 * 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:
 *
 * Copyright (C) 2011 Aleksander Morgado <aleksander@gnu.org>
 */

#include <string.h>

#include "mm-errors-types.h"
#include "mm-common-helpers.h"
#include "mm-common-connect-properties.h"

G_DEFINE_TYPE (MMCommonConnectProperties, mm_common_connect_properties, G_TYPE_OBJECT);

#define PROPERTY_PIN             "pin"
#define PROPERTY_OPERATOR_ID     "operator-id"
#define PROPERTY_ALLOWED_BANDS   "allowed-bands"
#define PROPERTY_ALLOWED_MODES   "allowed-modes"
#define PROPERTY_PREFERRED_MODE  "preferred-mode"

struct _MMCommonConnectPropertiesPrivate {
    /* PIN */
    gchar *pin;
    /* Operator ID */
    gchar *operator_id;
    /* Bands */
    MMModemBand *allowed_bands;
    guint n_allowed_bands;
    /* Modes */
    gboolean allowed_modes_set;
    MMModemMode allowed_modes;
    MMModemMode preferred_mode;
    /* Bearer properties */
    MMCommonBearerProperties *bearer_properties;
};

/*****************************************************************************/

void
mm_common_connect_properties_set_pin (MMCommonConnectProperties *self,
                                      const gchar *pin)
{
    g_free (self->priv->pin);
    self->priv->pin = g_strdup (pin);
}

void
mm_common_connect_properties_set_operator_id (MMCommonConnectProperties *self,
                                              const gchar *operator_id)
{
    g_free (self->priv->operator_id);
    self->priv->operator_id = g_strdup (operator_id);
}

void
mm_common_connect_properties_set_allowed_bands (MMCommonConnectProperties *self,
                                                const MMModemBand *bands,
                                                guint n_bands)
{
    g_free (self->priv->allowed_bands);
    self->priv->n_allowed_bands = n_bands;
    self->priv->allowed_bands = g_new (MMModemBand, self->priv->n_allowed_bands);
    memcpy (self->priv->allowed_bands,
            bands,
            sizeof (MMModemBand) * self->priv->n_allowed_bands);
}

void
mm_common_connect_properties_set_allowed_modes (MMCommonConnectProperties *self,
                                                MMModemMode allowed,
                                                MMModemMode preferred)
{
    self->priv->allowed_modes = allowed;
    self->priv->preferred_mode = preferred;
    self->priv->allowed_modes_set = TRUE;
}

void
mm_common_connect_properties_set_apn (MMCommonConnectProperties *self,
                                      const gchar *apn)
{
    mm_common_bearer_properties_set_apn (self->priv->bearer_properties,
                                         apn);
}

void
mm_common_connect_properties_set_user (MMCommonConnectProperties *self,
                                       const gchar *user)
{
    mm_common_bearer_properties_set_user (self->priv->bearer_properties,
                                          user);
}

void
mm_common_connect_properties_set_password (MMCommonConnectProperties *self,
                                           const gchar *password)
{
    mm_common_bearer_properties_set_password (self->priv->bearer_properties,
                                              password);
}

void
mm_common_connect_properties_set_ip_type (MMCommonConnectProperties *self,
                                          const gchar *ip_type)
{
    mm_common_bearer_properties_set_ip_type (self->priv->bearer_properties,
                                             ip_type);
}

void
mm_common_connect_properties_set_allow_roaming (MMCommonConnectProperties *self,
                                                gboolean allow_roaming)
{
    mm_common_bearer_properties_set_allow_roaming (self->priv->bearer_properties,
                                                   allow_roaming);
}

void
mm_common_connect_properties_set_number (MMCommonConnectProperties *self,
                                         const gchar *number)
{
    mm_common_bearer_properties_set_number (self->priv->bearer_properties,
                                            number);
}

/*****************************************************************************/

MMCommonBearerProperties *
mm_common_connect_properties_get_bearer_properties (MMCommonConnectProperties *self)
{
    return g_object_ref (self->priv->bearer_properties);
}

const gchar *
mm_common_connect_properties_get_pin (MMCommonConnectProperties *self)
{
    return self->priv->pin;
}

const gchar *
mm_common_connect_properties_get_operator_id (MMCommonConnectProperties *self)
{
    return self->priv->operator_id;
}

void
mm_common_connect_properties_get_allowed_bands (MMCommonConnectProperties *self,
                                                const MMModemBand **bands,
                                                guint *n_bands)
{
    *bands = self->priv->allowed_bands;
    *n_bands = self->priv->n_allowed_bands;
}

void
mm_common_connect_properties_get_allowed_modes (MMCommonConnectProperties *self,
                                                MMModemMode *allowed,
                                                MMModemMode *preferred)
{
    *allowed = self->priv->allowed_modes;
    *preferred = self->priv->preferred_mode;
}

const gchar *
mm_common_connect_properties_get_apn (MMCommonConnectProperties *self)
{
    return mm_common_bearer_properties_get_apn (self->priv->bearer_properties);
}

const gchar *
mm_common_connect_properties_get_user (MMCommonConnectProperties *self)
{
    return mm_common_bearer_properties_get_user (self->priv->bearer_properties);
}

const gchar *
mm_common_connect_properties_get_password (MMCommonConnectProperties *self)
{
    return mm_common_bearer_properties_get_password (self->priv->bearer_properties);
}

const gchar *
mm_common_connect_properties_get_ip_type (MMCommonConnectProperties *self)
{
    return mm_common_bearer_properties_get_ip_type (self->priv->bearer_properties);
}

gboolean
mm_common_connect_properties_get_allow_roaming (MMCommonConnectProperties *self)
{
    return mm_common_bearer_properties_get_allow_roaming (self->priv->bearer_properties);
}

const gchar *
mm_common_connect_properties_get_number (MMCommonConnectProperties *self)
{
    return mm_common_bearer_properties_get_number (self->priv->bearer_properties);
}

/*****************************************************************************/

GVariant *
mm_common_connect_properties_get_dictionary (MMCommonConnectProperties *self)
{
    GVariantBuilder builder;
    GVariantIter iter;
    gchar *key;
    GVariant *value;
    GVariant *bearer_properties_dictionary;

    /* We do allow NULL */
    if (!self)
        return NULL;

    g_variant_builder_init (&builder, G_VARIANT_TYPE ("a{sv}"));

    if (self->priv->pin)
        g_variant_builder_add (&builder,
                               "{sv}",
                               PROPERTY_PIN,
                               g_variant_new_string (self->priv->pin));

    if (self->priv->operator_id)
        g_variant_builder_add (&builder,
                               "{sv}",
                               PROPERTY_OPERATOR_ID,
                               g_variant_new_string (self->priv->operator_id));

    if (self->priv->allowed_bands)
        g_variant_builder_add (&builder,
                               "{sv}",
                               PROPERTY_ALLOWED_BANDS,
                               mm_common_bands_array_to_variant (self->priv->allowed_bands,
                                                                 self->priv->n_allowed_bands));

    if (self->priv->allowed_modes_set) {
        g_variant_builder_add (&builder,
                               "{sv}",
                               PROPERTY_ALLOWED_MODES,
                               g_variant_new_uint32 (self->priv->allowed_modes));
        g_variant_builder_add (&builder,
                               "{sv}",
                               PROPERTY_PREFERRED_MODE,
                               g_variant_new_uint32 (self->priv->preferred_mode));
    }

    /* Merge dictionaries */
    bearer_properties_dictionary = mm_common_bearer_properties_get_dictionary (self->priv->bearer_properties);
    g_variant_iter_init (&iter, bearer_properties_dictionary);
    while (g_variant_iter_next (&iter, "{sv}", &key, &value)) {
        g_variant_builder_add (&builder,
                               "{sv}",
                               key,
                               value);
        g_variant_unref (value);
        g_free (key);
    }
    g_variant_unref (bearer_properties_dictionary);

    return g_variant_ref_sink (g_variant_builder_end (&builder));
}

/*****************************************************************************/

MMCommonConnectProperties *
mm_common_connect_properties_new_from_string (const gchar *str,
                                              GError **error)
{
    GError *inner_error = NULL;
    MMCommonConnectProperties *properties;
    gchar **words;
    gchar *key;
    gchar *value;
    guint i;
    const gchar *allowed_modes_str = NULL;
    const gchar *preferred_mode_str = NULL;

    properties = mm_common_connect_properties_new ();

    /* Expecting input as:
     *   key1=string,key2=true,key3=false...
     * */

    words = g_strsplit_set (str, ",= ", -1);
    if (!words)
        return properties;

    i = 0;
    key = words[i];
    while (key) {
        value = words[++i];

        if (!value) {
            inner_error = g_error_new (MM_CORE_ERROR,
                                       MM_CORE_ERROR_INVALID_ARGS,
                                       "Invalid properties string, no value for key '%s'",
                                       key);
            break;
        }

        /* First, check if we can consume this as bearer properties */
        if (!mm_common_bearer_properties_consume_string (properties->priv->bearer_properties,
                                                         key, value,
                                                         NULL)) {
            if (g_str_equal (key, PROPERTY_PIN))
                mm_common_connect_properties_set_pin (properties, value);
            else if (g_str_equal (key, PROPERTY_OPERATOR_ID))
                mm_common_connect_properties_set_operator_id (properties, value);
            else if (g_str_equal (key, PROPERTY_ALLOWED_BANDS)) {
                MMModemBand *bands = NULL;
                guint n_bands = 0;

                mm_common_get_bands_from_string (value, &bands, &n_bands, &inner_error);
                if (!inner_error)
                    mm_common_connect_properties_set_allowed_bands (properties, bands, n_bands);
                g_free (bands);
            } else if (g_str_equal (key, PROPERTY_ALLOWED_MODES)) {
                allowed_modes_str = value;
            } else if (g_str_equal (key, PROPERTY_PREFERRED_MODE)) {
                preferred_mode_str = value;
            } else {
                inner_error = g_error_new (MM_CORE_ERROR,
                                           MM_CORE_ERROR_INVALID_ARGS,
                                           "Invalid properties string, unexpected key '%s'",
                                           key);
                break;
            }
        }

        key = words[++i];
    }

    /* If error, destroy the object */
    if (inner_error) {
        g_propagate_error (error, inner_error);
        g_object_unref (properties);
        properties = NULL;
    }
    else if (allowed_modes_str || preferred_mode_str) {
        MMModemMode allowed_modes;
        MMModemMode preferred_mode;

        allowed_modes = (allowed_modes_str ?
                         mm_common_get_modes_from_string (allowed_modes_str,
                                                          &inner_error) :
                         MM_MODEM_MODE_ANY);
        if (!inner_error) {
            preferred_mode = (preferred_mode_str ?
                              mm_common_get_modes_from_string (preferred_mode_str,
                                                               &inner_error) :
                              MM_MODEM_MODE_NONE);
        }

        if (inner_error) {
            g_propagate_error (error, inner_error);
            g_object_unref (properties);
            properties = NULL;
        } else {
            mm_common_connect_properties_set_allowed_modes (
                properties,
                allowed_modes,
                preferred_mode);
        }
    }

    g_strfreev (words);
    return properties;
}

/*****************************************************************************/

MMCommonConnectProperties *
mm_common_connect_properties_new_from_dictionary (GVariant *dictionary,
                                                  GError **error)
{
    GError *inner_error = NULL;
    GVariantIter iter;
    gchar *key;
    GVariant *value;
    MMCommonConnectProperties *properties;
    GVariant *allowed_modes_variant = NULL;
    GVariant *preferred_mode_variant = NULL;

    properties = mm_common_connect_properties_new ();
    if (!dictionary)
        return properties;

    g_variant_iter_init (&iter, dictionary);
    while (!inner_error &&
           g_variant_iter_next (&iter, "{sv}", &key, &value)) {

        /* First, check if we can consume this as bearer properties */
        if (!mm_common_bearer_properties_consume_variant (properties->priv->bearer_properties,
                                                          key, value,
                                                          NULL)) {
            if (g_str_equal (key, PROPERTY_PIN))
                mm_common_connect_properties_set_pin (
                    properties,
                    g_variant_get_string (value, NULL));
            else if (g_str_equal (key, PROPERTY_OPERATOR_ID))
                mm_common_connect_properties_set_operator_id (
                    properties,
                    g_variant_get_string (value, NULL));
            else if (g_str_equal (key, PROPERTY_ALLOWED_BANDS)) {
                GArray *array;

                array = mm_common_bands_variant_to_garray (value);
                mm_common_connect_properties_set_allowed_bands (
                    properties,
                    (MMModemBand *)array->data,
                    array->len);
                g_array_unref (array);
            } else if (g_str_equal (key, PROPERTY_ALLOWED_MODES))
                allowed_modes_variant = g_variant_ref (value);
            else if (g_str_equal (key, PROPERTY_PREFERRED_MODE))
                preferred_mode_variant = g_variant_ref (value);
            else {
                /* Set inner error, will stop the loop */
                inner_error = g_error_new (MM_CORE_ERROR,
                                           MM_CORE_ERROR_INVALID_ARGS,
                                           "Invalid properties dictionary, unexpected key '%s'",
                                           key);
            }
        }

        g_free (key);
        g_variant_unref (value);
    }

    /* If error, destroy the object */
    if (inner_error) {
        g_propagate_error (error, inner_error);
        g_object_unref (properties);
        properties = NULL;
    }
    /* If we got allowed modes variant, check if we got preferred mode */
    else if (allowed_modes_variant) {
        mm_common_connect_properties_set_allowed_modes (
            properties,
            g_variant_get_uint32 (allowed_modes_variant),
            (preferred_mode_variant ?
             g_variant_get_uint32 (preferred_mode_variant) :
             MM_MODEM_MODE_NONE));
    }
    /* If we only got preferred mode, assume allowed is ANY */
    else if (preferred_mode_variant) {
        mm_common_connect_properties_set_allowed_modes (
            properties,
            MM_MODEM_MODE_ANY,
            g_variant_get_uint32 (preferred_mode_variant));
    }

    /* Cleanup last things before exiting */
    if (allowed_modes_variant)
        g_variant_unref (allowed_modes_variant);
    if (preferred_mode_variant)
        g_variant_unref (preferred_mode_variant);

    return properties;
}

/*****************************************************************************/

MMCommonConnectProperties *
mm_common_connect_properties_new (void)
{
    return (MM_COMMON_CONNECT_PROPERTIES (
                g_object_new (MM_TYPE_COMMON_CONNECT_PROPERTIES, NULL)));
}

static void
mm_common_connect_properties_init (MMCommonConnectProperties *self)
{
    self->priv = G_TYPE_INSTANCE_GET_PRIVATE ((self),
                                              MM_TYPE_COMMON_CONNECT_PROPERTIES,
                                              MMCommonConnectPropertiesPrivate);

    /* Some defaults */
    self->priv->bearer_properties = mm_common_bearer_properties_new ();
    self->priv->allowed_modes = MM_MODEM_MODE_ANY;
    self->priv->preferred_mode = MM_MODEM_MODE_NONE;
    self->priv->allowed_bands = g_new (MMModemBand, 1);
    self->priv->allowed_bands[0] = MM_MODEM_BAND_ANY;
    self->priv->n_allowed_bands = 1;
}

static void
finalize (GObject *object)
{
    MMCommonConnectProperties *self = MM_COMMON_CONNECT_PROPERTIES (object);

    g_free (self->priv->pin);
    g_free (self->priv->operator_id);
    g_free (self->priv->allowed_bands);
    g_object_unref (self->priv->bearer_properties);

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

static void
mm_common_connect_properties_class_init (MMCommonConnectPropertiesClass *klass)
{
    GObjectClass *object_class = G_OBJECT_CLASS (klass);

    g_type_class_add_private (object_class, sizeof (MMCommonConnectPropertiesPrivate));

    object_class->finalize = finalize;
}