/**

  * GMyth Library

  *

  * @file gmyth/gmyth_epg.c

  * 

  * @brief <p> GMythEPG class provides access to the program and channel data

  * from the Electronic Program Guide (EPG) of the Mythtv backend.

  *

  * Copyright (C) 2006 INdT - Instituto Nokia de Tecnologia.

  * @author Leonardo Sobral Cunha <leonardo.cunha@indt.org.br>

  * 

  * This program is free software; you can redistribute it and/or modify

  * it under the terms of the GNU Lesser 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 Lesser 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

  */

 #ifdef HAVE_CONFIG_H

 #include "config.h"

 #endif

 #include <mysql/mysql.h>

 #include <stdlib.h>

 #include <string.h>

 #include <assert.h>

 #include "gmyth_epg.h"

 #include "gmyth_programinfo.h"

 #include "gmyth_util.h"

 #include "gmyth_file_transfer.h"

 #include "gmyth_debug.h"

 #define CONNECT_TIMEOUT 6

 static void     gmyth_epg_class_init(GMythEPGClass * klass);

 static void     gmyth_epg_init(GMythEPG * object);

 static void     gmyth_epg_dispose(GObject * object);

 static void     gmyth_epg_finalize(GObject * object);

 G_DEFINE_TYPE(GMythEPG, gmyth_epg, G_TYPE_OBJECT)

     static void     gmyth_epg_class_init(GMythEPGClass * klass)

 {

     GObjectClass   *gobject_class = G_OBJECT_CLASS(klass);

     gobject_class->dispose = gmyth_epg_dispose;

     gobject_class->finalize = gmyth_epg_finalize;

 }

 static void

 gmyth_epg_init(GMythEPG * gmyth_epg)

 {

 }

 static void

 gmyth_epg_dispose(GObject * object)

 {

     GMythEPG       *gmyth_epg = GMYTH_EPG(object);

     if (gmyth_epg->sqlquery != NULL) {

         g_object_unref(gmyth_epg->sqlquery);

         gmyth_epg->sqlquery = NULL;

     }

     G_OBJECT_CLASS(gmyth_epg_parent_class)->dispose(object);

 }

 static void

 gmyth_epg_finalize(GObject * object)

 {

     g_signal_handlers_destroy(object);

     G_OBJECT_CLASS(gmyth_epg_parent_class)->finalize(object);

 }

 /**

  * Creates a new instance of GMythEPG.

  * 

  * @return a new instance of GMythEPG.

  */

 GMythEPG       *

 gmyth_epg_new(void)

 {

     GMythEPG       *epg = GMYTH_EPG(g_object_new(GMYTH_EPG_TYPE, NULL));

     return epg;

 }

 /** Connects to the Mysql database in the backend. The backend address

  * is loaded from the GMythSettings instance.

  *

  * @param gmyth_epg the GMythEPG instance to be connected.

  * @return true if connection was success, false if failed.

  */

 gboolean

 gmyth_epg_connect(GMythEPG * gmyth_epg, GMythBackendInfo * backend_info)

 {

     g_return_val_if_fail(gmyth_epg != NULL, FALSE);

     if (gmyth_epg->sqlquery == NULL) {

         gmyth_debug("[%s] Creating gmyth_query", __FUNCTION__);

         gmyth_epg->sqlquery = gmyth_query_new();

     }

     if (!gmyth_query_connect_with_timeout(gmyth_epg->sqlquery,

                                           backend_info, CONNECT_TIMEOUT)) {

         gmyth_debug("[%s] Error while connecting to db", __FUNCTION__);

         return FALSE;

     }

     gmyth_epg->backend_info = backend_info;

     g_object_ref(backend_info);

     return TRUE;

 }

 gboolean

 gmyth_epg_is_connected (GMythEPG *gmyth_epg)

 {

     g_return_val_if_fail (gmyth_epg != NULL, FALSE);

     return gmyth_query_is_alive (gmyth_epg->sqlquery);

 }

 /** Disconnects from the Mysql database in the backend.

  *

  * @param gmyth_epg the GMythEPG instance to be disconnected

  * @return true if disconnection was success, false if failed.

  */

 gboolean

 gmyth_epg_disconnect(GMythEPG * gmyth_epg)

 {

     g_return_val_if_fail(gmyth_epg != NULL, FALSE);

     if (gmyth_epg->sqlquery != NULL) {

         gmyth_query_disconnect(gmyth_epg->sqlquery);

         g_object_unref(gmyth_epg->sqlquery);

         gmyth_epg->sqlquery = NULL;

     }

     if (gmyth_epg->backend_info != NULL) {

         g_object_unref(gmyth_epg->backend_info);

         gmyth_epg->backend_info = NULL;

     }

     return TRUE;

 }

 /** Retrieves the available list of channels from the backend Mysql database.

  * 

  * @param gmyth_epg the GMythEPG instance.

  * @param glist_ptr the GSList pointer to be filled with the loaded list address.

  * @return The amount of channels retrieved from database,  or -1 if error.

  */

 gint

 gmyth_epg_get_channel_list(GMythEPG * gmyth_epg, GList ** glist_ptr)

 {

     MYSQL_RES      *msql_res;

     g_return_val_if_fail(gmyth_epg != NULL, -1);

     msql_res = gmyth_query_process_statement(gmyth_epg->sqlquery,

                                              "SELECT chanid, channum, name, icon FROM channel;");

     (*glist_ptr) = NULL;

     if (msql_res == NULL) {

         gmyth_debug("[%s] msql query returned NULL MYSQL_RES",

                     __FUNCTION__);

         return -1;

     } else {

         MYSQL_ROW       row;

         GMythChannelInfo *channel_info;

         while ((row = mysql_fetch_row(msql_res)) != NULL) {

             channel_info = g_new0(GMythChannelInfo, 1);

             channel_info->channel_ID =

                 (gint) g_ascii_strtoull(row[0], NULL, 10);

             channel_info->channel_num = g_string_new(row[1]);

             channel_info->channel_name = g_string_new(row[2]);

             channel_info->channel_icon = g_string_new(row[3]);

 #ifdef GMYTH_USE_DEBUG

             gmyth_channel_info_print(channel_info);

 #endif

             (*glist_ptr) = g_list_append((*glist_ptr), channel_info);

         }

     }

     mysql_free_result(msql_res);

     return (!(*glist_ptr)) ? 0 : g_list_length(*glist_ptr);

 }

 GMythChannelInfo *

 gmyth_epg_get_channel_info(GMythEPG * gmyth_epg, gint channel_id)

 {

     GMythChannelInfo *channel_info = NULL;

     MYSQL_RES      *msql_res;

     gchar          *query_str;

     g_return_val_if_fail(gmyth_epg != NULL, NULL);

     query_str =

         g_strdup_printf

         ("SELECT channum, name, icon FROM channel WHERE chanid=%d;",

          channel_id);

     msql_res =

         gmyth_query_process_statement(gmyth_epg->sqlquery, query_str);

     if (msql_res == NULL) {

         gmyth_debug("[%s] msql query returned NULL MYSQL_RES",

                     __FUNCTION__);

         return NULL;

     } else {

         MYSQL_ROW       row;

         if ((row = mysql_fetch_row(msql_res)) != NULL) {

             channel_info = g_new0(GMythChannelInfo, 1);

             channel_info->channel_ID = channel_id;

             channel_info->channel_num = g_string_new(row[0]);

             channel_info->channel_name = g_string_new(row[1]);

             channel_info->channel_icon = g_string_new(row[2]);

 #ifdef GMYTH_USE_DEBUG

             gmyth_channel_info_print(channel_info);

 #endif

         }

     }

     mysql_free_result(msql_res);

     return channel_info;

 }

 /** 

  * Retrieves the available list of channels from the backend Mysql database.

  * 

  * @param gmyth_epg the GMythEPG instance.

  * @param proglist the GSList pointer to be filled with the loaded list.

  * @param chan_num the channel num on which to search for program.

  * @param starttime the start time to search for programs.

  * @param endtime the end time to search for programs.

  * @return The amount of channels retrieved from database, or -1 if error.

  */

 gint

 gmyth_epg_get_program_list(GMythEPG * gmyth_epg, GList ** proglist,

                            const gint chan_num, GTimeVal * starttime,

                            GTimeVal * endtime)

 {

     gchar          *startts =

         gmyth_util_time_to_string_from_time_val(starttime);

     gchar          *endts =

         gmyth_util_time_to_string_from_time_val(endtime);

     MYSQL_ROW       row;

     GString        *querystr;

     assert(gmyth_epg);

     querystr =

         g_string_new

         ("SELECT DISTINCT program.chanid, program.starttime, program.endtime, "

          "    program.title, program.subtitle, program.description, "

          "    program.category, channel.channum, channel.callsign, "

          "    channel.name, program.previouslyshown, channel.commfree, "

          "    channel.outputfilters, program.seriesid, program.programid, "

          "    program.airdate, program.stars, program.originalairdate, "

          "    program.category_type, record.recordid, "

          "    oldrecstatus.rectype, oldrecstatus.recstatus, "

          "    oldrecstatus.findid "

          "FROM program "

          "LEFT JOIN channel ON program.chanid = channel.chanid "

          "LEFT JOIN record ON "

          "    program.chanid = record.chanid AND "

          "    DATE (program.starttime) = record.startdate AND "

          "    TIME (program.starttime) = record.starttime AND "

          "    DATE (program.endtime) = record.enddate AND "

          "    TIME (program.endtime) = record.endtime "

          "LEFT JOIN oldrecorded AS oldrecstatus ON "

          "    program.title = oldrecstatus.title AND "

          "    channel.callsign = oldrecstatus.station AND "

          "    program.starttime = oldrecstatus.starttime ");

     g_string_append_printf(querystr,

                            "WHERE program.chanid = %d "

                            "  AND program.endtime >= '%s' "

                            "  AND program.starttime <= '%s' "

                            "  AND program.manualid = 0 ", chan_num,

                            startts, endts);

     if (!g_strrstr(querystr->str, " GROUP BY "))

         querystr = g_string_append(querystr,

                                    " GROUP BY program.starttime, channel.channum, "

                                    "  channel.callsign, program.title ");

     if (!g_strrstr(querystr->str, " LIMIT "))

         querystr = g_string_append(querystr, " LIMIT 1000 ");

     MYSQL_RES      *res_set =

         gmyth_query_process_statement(gmyth_epg->sqlquery, querystr->str);

     if (res_set == NULL) {

         gmyth_debug("[%s] msql query returned NULL MYSQL_RES",

                     __FUNCTION__);

         return -1;

     }

     *proglist = NULL;

     while ((row = mysql_fetch_row(res_set)) != NULL) {

         GMythProgramInfo *p = gmyth_program_info_new();

         p->channel_id = (int) g_ascii_strtoull (row[0], NULL, 10);

         p->startts = gmyth_util_string_to_time_val(row[1]);

         p->endts = gmyth_util_string_to_time_val(row[2]);

         p->recstartts = g_new0(GTimeVal, 1);

         p->recstartts->tv_sec = p->startts->tv_sec;

         p->recstartts->tv_usec = p->startts->tv_usec;

         p->recendts = g_new0(GTimeVal, 1);

         p->recendts->tv_sec = p->endts->tv_sec;

         p->recendts->tv_usec = p->endts->tv_usec;

         p->lastmodified = g_new0(GTimeVal, 1);

         p->lastmodified->tv_sec = p->startts->tv_sec;

         p->lastmodified->tv_usec = p->startts->tv_usec;

         p->title = g_string_new(row[3]);

         p->subtitle = g_string_new(row[4]);

         p->description = g_string_new(row[5]);

         p->category = g_string_new(row[6]);

         p->chanstr = g_string_new(row[7]);

         p->chansign = g_string_new(row[8]);

         p->channame = g_string_new(row[9]);

         p->repeat = g_ascii_strtoull(row[10], NULL, 10);

         p->chancommfree = g_ascii_strtoull(row[11], NULL, 10);

         p->chanOutputFilters = g_string_new(row[12]);

         p->seriesid = g_string_new(row[13]);

         p->program_id = g_string_new(row[14]);

         p->year = g_string_new(row[15]);

         p->stars = g_ascii_strtod(row[16], NULL);

         if (!row[17] || !strcmp(row[17], "")) {

             p->originalAirDate = 0;

             p->hasAirDate = FALSE;

         } else {

             p->originalAirDate = gmyth_util_string_to_time_val(row[17]);

             p->hasAirDate = TRUE;

         }

         p->catType = g_string_new(row[18]);

         if (row[19] != NULL)

             p->recordid =  g_ascii_strtoull(row[19], NULL, 10);

         *proglist = g_list_append(*proglist, p);

 #ifdef GMYTH_USE_DEBUG

         gmyth_program_info_print(p);

 #endif

     }

     /*

      * deallocate 

      */

     mysql_free_result(res_set);

     g_string_free(querystr, TRUE);

     return g_list_length (*proglist);

 }

 gboolean

 gmyth_epg_channel_has_icon(GMythEPG * gmyth_epg,

                            GMythChannelInfo * channel_info)

 {

     gboolean        res = FALSE;

     g_return_val_if_fail(gmyth_epg != NULL, FALSE);

     g_return_val_if_fail(channel_info != NULL, FALSE);

     if (channel_info->channel_icon != NULL) {

         res = gmyth_util_file_exists(gmyth_epg->backend_info,

                                      channel_info->channel_icon->str);

     }

     return res;

 }

 /**

  * 

  * @param data the data pointer to be filled with icon binary data. It must be freed by the calling function.

  * @return TRUE if success, FALSE if any error happens.

  */

 gboolean

 gmyth_epg_channel_get_icon(GMythEPG * gmyth_epg,

                            GMythChannelInfo * channel_info, guint8 ** data,

                            guint * length)

 {

     gboolean res = FALSE;

     g_return_val_if_fail(gmyth_epg != NULL, FALSE);

     g_return_val_if_fail(channel_info != NULL, FALSE);

     if (gmyth_epg_channel_has_icon(gmyth_epg, channel_info)) {

         GMythFileTransfer *transfer =

             gmyth_file_transfer_new(gmyth_epg->backend_info);

         GMythFileReadResult gmyth_res;

         GByteArray *icon_data;

         guint64 icon_length = 0;

         res = gmyth_file_transfer_open(transfer,

                                        channel_info->channel_icon->str);

         if (!res) {

             gmyth_debug("Channel icon could not be opened");

             return FALSE;

         }

         icon_length = gmyth_file_transfer_get_filesize(transfer);

         if (icon_length <= 0) {

             gmyth_debug("Channel icon file size is zero or negative");

             return FALSE;

         }

         icon_data = g_byte_array_new();

         gmyth_res = gmyth_file_transfer_read(transfer,

                                              icon_data,

                                              icon_length, FALSE);

         if (gmyth_res == GMYTH_FILE_READ_EOF) {

             *length = icon_length;

             *data = icon_data->data;

             g_byte_array_free(icon_data, FALSE);

             res = TRUE;

         } else {

             *length = 0;

             *data = NULL;

             g_byte_array_free(icon_data, TRUE);

         }

     }

     return res;

 }