#include <gmyth/gmyth_backendinfo.h>

#include <gmyth/gmyth_epg.h>

#include "common.h"

static gboolean

test_epg_connection (GMythBackendInfo *backend_info)

{

    GMythEPG *epg = gmyth_epg_new ();

    gboolean res = FALSE;

    res = gmyth_epg_connect (epg, backend_info);

    gmyth_epg_disconnect (epg);

    if ( epg != NULL )

    	g_object_unref (epg);

    return res; 

}

static gboolean

test_epg_get_channels (GMythBackendInfo *backend_info)

{

    GMythEPG *epg = gmyth_epg_new ();

    GList *clist;

    gint i, length;

    if (!gmyth_epg_connect (epg, backend_info)) {

        return FALSE;

    }

    length = gmyth_epg_get_channel_list (epg, &clist);

    g_debug ("==== %d channels found in the EPG ====\n", length);

    for (i=0; i<length; i++) {

        GMythChannelInfo *channel_info = (GMythChannelInfo*) g_list_nth_data (clist, i);

        gmyth_channel_info_print(channel_info);

    }

    g_list_free (clist);

    gmyth_epg_disconnect (epg);

    if ( epg != NULL )

    	g_object_unref (epg);

}

static gboolean

test_epg_get_channel_icon (GMythBackendInfo *backend_info)

{

    GMythEPG *epg = gmyth_epg_new ();

    GList *clist;

    gint i, length;

    if (!gmyth_epg_connect (epg, backend_info)) {

        return FALSE;

    }

    length = gmyth_epg_get_channel_list (epg, &clist);

    g_debug ("==== %d channels found in the EPG ====\n", length);

    for (i=0; i<length; i++) {

        GMythChannelInfo *channel_info = (GMythChannelInfo*) g_list_nth_data (clist, i);

        if (gmyth_epg_channel_has_icon (epg, channel_info)) {

            gchar *icon_name = g_strdup_printf ("%s.jpg", channel_info->channel_name->str);

            guint8 *icon_data = NULL;

            guint icon_length;

            g_debug ("Channel %s has icon %s\n", channel_info->channel_name->str, channel_info->channel_icon->str);

            if (gmyth_epg_channel_get_icon (epg, channel_info, &icon_data, &icon_length)) {

                FILE *outfile = fopen (icon_name, "w+");

                if (fwrite (icon_data, icon_length, 1, outfile) == icon_length)

                    g_debug ("\tIcon saved as %s", icon_name);

                else

                    g_debug ("\tError while downloading the file or writing it");

                g_free (icon_data);

            }

            g_free (icon_name);

        } else {

            g_debug ("Channel %s does not have icon\n", channel_info->channel_name->str);

        }

        gmyth_channel_info_print(channel_info);

    }

    g_list_free (clist);

    gmyth_epg_disconnect (epg);

    g_object_unref (epg);

    return TRUE;

}

int

main (int args, const char **argv)

{

    GMythBackendInfo *backend_info;

    g_type_init ();

    g_thread_init (NULL);

    if (args < 2) {

        g_debug ("Type %s myth://hostname:port/?mythconverg\n", argv[0]);

        return -1;

    }

    backend_info = gmyth_backend_info_new_with_uri (argv[1]);

    fprintf(stdout, SYNC_STRING);

    fflush(NULL);

    getchar();

    test_epg_connection (backend_info);

    test_epg_get_channels (backend_info);

    test_epg_get_channel_icon (backend_info);

    if ( backend_info != NULL )

    	g_object_unref( backend_info );

    return(0);

}