#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);

}