#include <glib-object.h>

#include "gmyth_backendinfo.h"

#include "gmyth_remote_util.h"

#include "gmyth_query.h"

#include "gmyth_epg.h"

#include "gmyth_common.h"

static gboolean

test_recorder_availability (GMythBackendInfo *backend_info)

{

    GMythRecorder* recorder;

    GMythSocket *socket = gmyth_socket_new ();

    if (gmyth_socket_connect_to_backend (socket, 

	    gmyth_backend_info_get_hostname (backend_info),

	    gmyth_backend_info_get_port (backend_info), TRUE) == FALSE) {

	g_debug ("Test recorder failed: Connection failed");

	return FALSE;

    }

    recorder = remote_request_next_free_recorder (socket, -1);

    gmyth_socket_close_connection (socket);

    if (recorder == NULL) {

	g_debug ("Recorder not available\n");

	return FALSE;

    }

    g_debug ("Recorder found (num): %d", recorder->recorder_num);

    return TRUE;

}

static gboolean

test_recorder_setup (GMythBackendInfo *backend_info)

{

    GMythQuery *query = gmyth_query_new ();

    if (gmyth_query_connect_with_timeout (query, backend_info, 3) == TRUE) {

	g_debug ("Mysql connection success");

	return TRUE;

    } else {

	g_debug ("Mysql connection failed");

	return FALSE;

    }

}

static gboolean

test_recorder_check_channels (GMythBackendInfo *backend_info)

{

    GMythRecorder* recorder;

    GMythSocket *socket = gmyth_socket_new ();

    GMythEPG *epg = gmyth_epg_new ();

    GList *clist;

    gint i, length;

    // Gets the free recorder

    if (gmyth_socket_connect_to_backend (socket, 

	    gmyth_backend_info_get_hostname (backend_info),

	    gmyth_backend_info_get_port (backend_info), TRUE) == FALSE) {

	g_debug ("Test recorder failed: Connection failed");

	return FALSE;

    }

    recorder = remote_request_next_free_recorder (socket, -1);

    gmyth_socket_close_connection (socket);

    if (recorder == NULL) {

	g_debug ("[%s] Recorder not available", __FUNCTION__);

	return FALSE;

    }

    // Connects the recorder socket

    gmyth_recorder_setup (recorder);

    // Gets the list of channels

    if (!gmyth_epg_connect (epg, backend_info)) {

        g_debug ("%s: Not connected\n", __FUNCTION__);

        return FALSE;

    }

    length = gmyth_epg_get_channel_list (epg, &clist);

    gmyth_epg_disconnect (epg);

    g_object_unref (epg);

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

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

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

	gboolean res;

	// Checks the channels

        res = gmyth_recorder_check_channel (recorder, channel_info->channel_ID);

	g_debug ("Channel %d %s", channel_info->channel_ID, res ? "Found" : "Not found");

    }

    g_list_free (clist);

}

int

main (int args, const char **argv)

{

    const char* uri = argv[1];

    GMythBackendInfo *backend_info;

    g_type_init ();

    g_thread_init (NULL);

    backend_info = gmyth_backend_info_new_with_uri (argv[1]);

    printf ("******** Testing recorder availability ***********\n");

    test_recorder_availability (backend_info);

    printf ("******** Testing recorder check channels function ***********\n");

    test_recorder_check_channels (backend_info);

}