/**

 * GMyth Library

 *

 * @file gmyth/gmyth_scheduler.c

 * 

 * @brief <p> The scheduler encapsulates all functions for browsing, scheduling

 * and modifying the recorded content.

 *

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

 * @author Alexsandro Jose Virginio dos Santos <alexsandro.santos@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

 */

#include <assert.h>

#include "gmyth_scheduler.h"

#include "gmyth_util.h"

#include "gmyth_query.h"

#include "gmyth_socket.h"

#include "gmyth_debug.h"

static void gmyth_scheduler_class_init  (GMythSchedulerClass *klass);

static void gmyth_scheduler_init        (GMythScheduler *object);

static void gmyth_scheduler_dispose  (GObject *object);

static void gmyth_scheduler_finalize (GObject *object);

static gint get_record_id_from_database (GMythScheduler *scheduler);

static void update_backend  (GMythScheduler *scheduler, gint record_id);

G_DEFINE_TYPE(GMythScheduler, gmyth_scheduler, G_TYPE_OBJECT)

static void

gmyth_scheduler_class_init (GMythSchedulerClass *klass)

{

    GObjectClass *gobject_class;

    gobject_class = (GObjectClass *) klass;

    gobject_class->dispose  = gmyth_scheduler_dispose;

    gobject_class->finalize = gmyth_scheduler_finalize;	

}

static void

gmyth_scheduler_init (GMythScheduler *sched)

{

    sched->recordid =0;

    sched->type = 0;

    sched->search = 0;

    sched->profile =  g_string_new("");

    sched->dupin = 0;

    sched->dupmethod = 0;

    sched->autoexpire = 0;

    sched->autotranscode = 0;

    sched->transcoder = 0;

    sched->autocommflag = 0;

    sched->autouserjob1 = 0;

    sched->autouserjob2 = 0;

    sched->autouserjob3 = 0;

    sched->autouserjob4 = 0;

    sched->startoffset = 0;

    sched->endoffset = 0;

    sched->maxepisodes = 0;

    sched->maxnewest = 0;

    sched->recpriority = 0;

    sched->recgroup = 0;

    sched->playgroup = 0;

    sched->prefinput = 0;

    sched->inactive = 0;

    sched->searchType = g_string_new("");

    sched->searchForWhat = g_string_new("");

    sched->msqlquery = gmyth_query_new ();

}

static void

gmyth_scheduler_dispose  (GObject *object)

{

    GMythScheduler *scheduler = GMYTH_SCHEDULER (object);

    if (scheduler->backend_info) { 

	g_object_unref (scheduler->backend_info);

	scheduler->backend_info = NULL;

    }

    G_OBJECT_CLASS (gmyth_scheduler_parent_class)->dispose (object);

}

static void

gmyth_scheduler_finalize (GObject *object)

{

    g_signal_handlers_destroy (object);

    G_OBJECT_CLASS (gmyth_scheduler_parent_class)->finalize (object);

}

/** Creates a new instance of GMythScheduler.

 * 

 * @return a new instance of GMythScheduler.

 */

GMythScheduler*

gmyth_scheduler_new ()

{

    GMythScheduler *scheduler = 

        GMYTH_SCHEDULER (g_object_new(GMYTH_SCHEDULER_TYPE, NULL));

    return scheduler;

}

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

 * is loaded from the GMythSettings instance.

 * 

 * @param scheduler the GMythScheduler instance to be connected.

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

 */

gboolean

gmyth_scheduler_connect (GMythScheduler *scheduler, GMythBackendInfo *backend_info)

{

    assert(scheduler);

    g_return_val_if_fail (backend_info != NULL, FALSE);

    g_object_ref (backend_info);

    scheduler->backend_info = backend_info;

    if (scheduler->msqlquery == NULL) {

        g_warning ("[%s] GMythScheduler db initializing", __FUNCTION__);

	    scheduler->msqlquery = gmyth_query_new ();		

	}

    if (!gmyth_query_connect(scheduler->msqlquery, scheduler->backend_info)) {

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

        return FALSE;

    }

    return TRUE;

}

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

 * 

 * @param scheduler the GMythScheduler instance to be disconnected

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

 */

gboolean

gmyth_scheduler_disconnect (GMythScheduler *scheduler)

{

    assert(scheduler);

    if (scheduler->msqlquery != NULL) {

        gmyth_query_disconnect (scheduler->msqlquery);

        g_object_unref (scheduler->msqlquery);

    }

    return TRUE;

}

/** Retrieves from the backend Mysql database the list of recording schedules.

 * 

 * @param scheduler The GMythScheduler instance.

 * @param schedule_list the GList pointer to be filled with the loaded list of ScheduleInfo items.

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

 */

gint

gmyth_scheduler_get_schedule_list ( GMythScheduler *scheduler, GList **schedule_list)

{

    ScheduleInfo *schedule;

    MYSQL_RES *msql_res;

    GString *query_str = g_string_new ("");

    GString *date_time = g_string_new ("");

    assert(scheduler);

    g_string_printf (query_str, 

    	"SELECT recordid,programid,chanid,starttime,startdate,"

    	"endtime,enddate,title,subtitle,description,category FROM record;");

    if (scheduler->msqlquery == NULL) {

	g_warning ("[%s] Scheduler db connection not initialized", __FUNCTION__);

	return -1;

    }

    msql_res = gmyth_query_process_statement (scheduler->msqlquery, query_str->str);

    if (msql_res == NULL) {

        g_warning ("DB retrieval of schedule list failed");

        return -1;

    } else {

        MYSQL_ROW row;

        *schedule_list = NULL;

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

        	schedule = g_new0(ScheduleInfo, 1);

        	schedule->record_id  = g_ascii_strtoull (row[0], NULL, 10);

        	schedule->program_id = g_ascii_strtoull (row[1], NULL, 10);

            schedule->channel_id = g_ascii_strtoull (row[2], NULL, 10);

            /* generate a time_t from a time and a date db field */

            g_string_printf (date_time, "%s %s", row[4], row[3]);

            schedule->start_time = gmyth_util_string_to_time (date_time);

            /* generate a time_t from a time and a date db field */

            g_string_printf (date_time, "%s %s", row[6], row[5]);

            schedule->end_time = gmyth_util_string_to_time (date_time);

            schedule->title = g_string_new (row[7]);

            schedule->subtitle = g_string_new (row[8]);

            schedule->description = g_string_new (row[9]);

            schedule->category = g_string_new (row[10]);

           (*schedule_list) = g_list_append (*(schedule_list), schedule);

    	}

    }

    mysql_free_result (msql_res);

    g_string_free(query_str, TRUE);

    g_string_free(date_time, TRUE);

    return (*schedule_list == NULL) ? 0 : g_list_length (*schedule_list);

}

/** Retrieves from the backend Mysql database the list of recorded programs.

 * 

 * @param scheduler The GMythScheduler instance.

 * @param recorded_list the GList pointer to be filled with the loaded RecordInfo items.

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

 */

gint

gmyth_scheduler_get_recorded_list (GMythScheduler *scheduler, GList **recorded_list)

{

    RecordedInfo *record;

    MYSQL_RES *msql_res;

    GString *query_str = g_string_new ("");

    GString *date_time = g_string_new ("");

    assert(scheduler);

    g_string_printf (query_str, 

    	"SELECT recordid,programid,chanid,starttime,progstart,"

    	"endtime,progend,title,subtitle,description,category,filesize,basename FROM recorded WHERE recgroup != 'LiveTV'");

	if (scheduler->msqlquery == NULL) {

		g_warning ("[%s] Scheduler db connection not initialized", __FUNCTION__);

		return -1;

	}

    msql_res = gmyth_query_process_statement (scheduler->msqlquery, query_str->str);

    if (msql_res == NULL) {

        g_warning ("DB retrieval of recording list failed");

        return -1;

	} else {

        MYSQL_ROW row;

        *recorded_list = NULL;

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

        	record = g_new0(RecordedInfo, 1);

            record->record_id  = (guint) g_ascii_strtoull (row[0], NULL, 10);

            record->program_id = (guint) g_ascii_strtoull (row[1], NULL, 10);

            record->channel_id = (guint) g_ascii_strtoull (row[2], NULL, 10);

            /* the db field time already contains the date. therefore

             * we are not using the date field */

            /* generate a time_t from a time and a date db field */

            /* g_string_printf (date_time, "%s %s", row[4], row[3]); */

            g_string_printf (date_time, "%s", row[3]); 

            record->start_time = gmyth_util_string_to_time (date_time);

            /* the db field time already contains the date. therefore

             * we are not using the date field */

            /* generate a time_t from a time and a date db field */

            /* g_string_printf (date_time, "%s %s", row[6], row[5]); */

            g_string_printf (date_time, "%s", row[5]); 

            record->end_time = gmyth_util_string_to_time (date_time);

            record->title = g_string_new (row[7]);

            record->subtitle = g_string_new (row[8]);

            record->description = g_string_new (row[9]);

            record->category = g_string_new (row[10]);

            record->filesize = g_ascii_strtoull (row[11], NULL, 10);

            record->basename = g_string_new (row[12]);

 	    *recorded_list = g_list_append (*recorded_list, record);

    	}

    }

    mysql_free_result (msql_res);

    g_string_free(query_str, TRUE);

    g_string_free(date_time, TRUE);

    return (*recorded_list == NULL) ? 0 : g_list_length (*recorded_list);

}

/** Requests the Mysql database in the backend to add a new schedule.

 * 

 * @param scheduler the GMythScheduler instance.

 * @param schedule_info the ScheduleInfo with recording schedule information

 * to be added. record_id = -1 to add a new schedule, otherwise this

 * function will update the schedule in the db

 * @return gboolean returns FALSE if some error occurs, TRUE otherwise

 */

gboolean

gmyth_scheduler_add_schedule (GMythScheduler *scheduler,

                        	  ScheduleInfo *schedule_info)

{

    struct tm start_tm;

    struct tm end_tm;

    MYSQL_RES *msql_res;

    GString *query_str = g_string_new ("");

    assert(scheduler);

    if (scheduler->msqlquery == NULL) {

	g_warning ("[%s] Scheduler db connection not initialized", __FUNCTION__);

	return 0;

    }

	/* manipulating time */

    if(localtime_r(&schedule_info->start_time, &start_tm) == NULL) {

        g_warning ("localtime_r error in libgmyth scheduler!\n");

        return FALSE;                 

    }

    if(localtime_r(&schedule_info->end_time, &end_tm) == NULL) {

        g_warning ("localtime_r error in libgmyth scheduler!\n");

        return FALSE;

    }	

    //TODO: verify if this funtion realy does what it should do!

    g_string_printf (query_str, "REPLACE INTO record "

	    "(recordid, type, chanid, starttime, "

	    "startdate, endtime, enddate, title,"

	    "profile, recpriority, maxnewest, inactive, "

	    "maxepisodes, autoexpire, startoffset, endoffset, "

	    "recgroup, dupmethod, dupin, station, "

	    "autocommflag, findday, findtime, findid, "

	    "search, autotranscode, transcoder, tsdefault, "

	    "autouserjob1, autouserjob2, autouserjob3, autouserjob4) "

	    " values ( %d, 1, %d, \"%02d:%02d:00\","	//recordid, type, chanid, starttime

	    " \"%d-%02d-%02d\", \"%02d:%02d:00\", \"%04d-%02d-%02d\", \"%s\","

        //startdate, endtime, enddate, title

	    "DEFAULT, 0, 0, 0, "	//profile, recpriority, maxnewest, inactive

	    "0, 1, 0, 0, "			//maxepisodes, autoexpire, startoffset, endoffset

	    "DEFAULT, 6, 15, %d, " 	//recgroup, dupmethod, dupin, station

	    "1, %d, \"%02d:%02d:00\", %d, "	//autocommflag, findday, findtime, findid

	    "5, 0, 29, 1, "			//search, autotranscode, transcoder, tsdefault

	    "0, 0, 0, 0 );",		//autouserjob1, autouserjob2, autouserjob3, autouserjob4

	    schedule_info->record_id, schedule_info->channel_id,

        start_tm.tm_hour, start_tm.tm_min,

	    start_tm.tm_year+1900, start_tm.tm_mon+1, 

        start_tm.tm_mday,

        end_tm.tm_hour, end_tm.tm_min,

        end_tm.tm_year+1900, end_tm.tm_mon+1, 

        end_tm.tm_mday, schedule_info->title->str, //title

    	schedule_info->channel_id,//station

    	start_tm.tm_wday+1, //findday

   		start_tm.tm_hour, start_tm.tm_min, //findtime

   		(gint)(schedule_info->start_time/60/60/24 + 719528)//findid

    );

    msql_res = gmyth_query_process_statement (scheduler->msqlquery, query_str->str);

    /* FIXME: currently no way to detect db error in UPDATE, REPLACES!

    if (msql_res == NULL) {

        g_warning ("DB retrieval of recording list failed");

        return -1;

	}*/

    /* TODO: verify record_id = -1 semantics */

    if (schedule_info->record_id <= 0)

	    schedule_info->record_id = get_record_id_from_database(scheduler);

    /* Notify the backend of changes */

    update_backend(scheduler, schedule_info->record_id);

    /* free allocated memory */

    mysql_free_result (msql_res);

    g_string_free(query_str, TRUE);

    return 1;

}

/** Requests the Mysql database in the backend to remove an existing schedule.

 * 

 * @param scheduler the GMythScheduler instance.

 * @param record_id The schedule's record id to be removed

 * @return gboolean TRUE if success, FALSE if error

 */

gboolean

gmyth_scheduler_delete_schedule (GMythScheduler *scheduler, gint record_id)

{

    MYSQL_RES *msql_res;

    GString *query_str = g_string_new ("");

    assert(scheduler);

    if (scheduler->msqlquery == NULL) {

	g_warning ("[%s] Scheduler db connection not initialized", __FUNCTION__);

	return FALSE;

    }

    //========================================

    g_string_printf (query_str, 

    	"DELETE FROM record WHERE recordid=%d", record_id);

    msql_res = gmyth_query_process_statement (scheduler->msqlquery, query_str->str);

    if (msql_res == NULL) {

    	g_warning ("[%s] Error while trying to delete a schedule in the database", __FUNCTION__);

    	return FALSE;

    }

    update_backend(scheduler, record_id);// Notify the backend of the changes

    mysql_free_result (msql_res);

    g_string_free(query_str, TRUE);

    return TRUE;

}

/** Requests the Mysql database in the backend to remove an existing recorded item.

 * 

 * @param scheduler the GMythScheduler instance.

 * @param record_id The recorded item id to be removed

 * @return gboolean TRUE if success, FALSE if error

 */

gboolean

gmyth_scheduler_delete_recorded (GMythScheduler *scheduler, gint record_id)

{

    MYSQL_RES *msql_res;

    GString *query_str = g_string_new ("");

    assert(scheduler);

	if (scheduler->msqlquery == NULL) {

		g_warning ("[%s] Scheduler db connection not initialized", __FUNCTION__);

		return FALSE;

	}

    //========================================

    g_string_printf (query_str, 

    	"DELETE FROM recorded WHERE recordid=%d", record_id);

    msql_res = gmyth_query_process_statement (scheduler->msqlquery, query_str->str);

    update_backend(scheduler, record_id);// Notify the backend of the changes

    mysql_free_result (msql_res);

    g_string_free(query_str, TRUE);

    return TRUE;

}

/** Retrieves an existing recorded item information from database. The information

 * is used to fill the returned GMythProgramInfo.

 * 

 * @param scheduler The GMythScheduler instance.

 * @param channel The channel associated to the record

 * @param starttime The record start time

 * @return A GMythProgramInfo struct with the requested record item

 * information, or NULL if error.

 */

GMythProgramInfo*

gmyth_scheduler_get_recorded (GMythScheduler *scheduler, 

                              GString *channel, time_t starttime)

{

	MYSQL_RES *msql_res;

	GMythProgramInfo *proginfo = NULL;

	GString *query_str = g_string_new("");

	GString *time_str = gmyth_util_time_to_string (starttime);

    assert(scheduler);

	if (scheduler->msqlquery == NULL) {

		g_warning ("[%s] Scheduler db connection not initialized", __FUNCTION__);

		return NULL;

	}

	g_string_printf (query_str, "SELECT recorded.chanid,starttime,endtime,title, "

                  "subtitle,description,channel.channum, "

                  "channel.callsign,channel.name,channel.commfree, "

                  "channel.outputfilters,seriesid,programid,filesize, "

                  "lastmodified,stars,previouslyshown,originalairdate, "

                  "hostname,recordid,transcoder,playgroup, "

                  "recorded.recpriority,progstart,progend,basename,recgroup "

                  "FROM recorded "

                  "LEFT JOIN channel "

                  "ON recorded.chanid = channel.chanid "

                  "WHERE recorded.chanid = \"%s\" "

                  "AND starttime = \"%s\" ;",

                  channel->str, time_str->str);

    msql_res = gmyth_query_process_statement (scheduler->msqlquery, query_str->str);

    if (msql_res /*&& query.size() > 0*/) {

    	MYSQL_ROW msql_row = mysql_fetch_row (msql_res);

    	if (msql_row) {

    		proginfo = g_new0 (GMythProgramInfo, 1);

	        proginfo->chanid = g_string_new (msql_row[0]);

	        proginfo->startts = gmyth_util_string_to_time (g_string_new (msql_row[23]));

	        proginfo->endts = gmyth_util_string_to_time (g_string_new (msql_row[24]));

	        proginfo->recstartts = gmyth_util_string_to_time (g_string_new (msql_row[1]));

	        proginfo->recendts = gmyth_util_string_to_time (g_string_new (msql_row[2]));

	        proginfo->title = g_string_new (msql_row[3]);

	        proginfo->subtitle = g_string_new (msql_row[4]);

	        proginfo->description = g_string_new (msql_row[5]);

	        proginfo->chanstr = g_string_new (msql_row[6]);

	        proginfo->chansign = g_string_new (msql_row[7]);

	        proginfo->channame = g_string_new (msql_row[0]);

	        proginfo->chancommfree = g_ascii_strtoull (msql_row[9], NULL, 10);

	        proginfo->chanOutputFilters = g_string_new (msql_row[10]);

	        proginfo->seriesid = g_string_new (msql_row[11]);

	        proginfo->programid = g_string_new (msql_row[12]);

	        proginfo->filesize = g_ascii_strtoull (msql_row[13], NULL, 10);

	        proginfo->lastmodified = gmyth_util_string_to_time (g_string_new (msql_row[14]));

	        proginfo->stars = g_ascii_strtod (msql_row[15], NULL);

	        proginfo->repeat = g_ascii_strtoull (msql_row[16], NULL, 10);

	        if (msql_row[17] == NULL) {

	            proginfo->originalAirDate = 0;

	            proginfo->hasAirDate = FALSE;

	        } else {

	            proginfo->originalAirDate = gmyth_util_string_to_time (g_string_new (msql_row[17]));

	            proginfo->hasAirDate = TRUE;

	        }

	        proginfo->hostname = g_string_new (msql_row[18]);

	        proginfo->recordid = g_ascii_strtoull (msql_row[19], NULL, 10);

	        proginfo->transcoder = g_ascii_strtoull (msql_row[20], NULL, 10);

	        //proginfo->spread = -1;

			//proginfo->programflags = proginfo->getProgramFlags();

	        proginfo->recgroup = g_string_new (msql_row[26]);

	        proginfo->playgroup = g_string_new (msql_row[21]);

	        proginfo->recpriority = g_ascii_strtoull (msql_row[22], NULL, 10);

	        proginfo->pathname = g_string_new (msql_row[25]);

	        gmyth_debug ("One program info loaded from mysql database\n");

    	}

    }

	mysql_free_result (msql_res);

    g_string_free(query_str, TRUE);

    g_string_free(time_str, TRUE);

    return proginfo;

}

/** Retrieves the next record id.

 * 

 * @param scheduler The GMythScheduler instance.

 * @return gint record_id if success, -1 otherwise 

 */

static gint

get_record_id_from_database (GMythScheduler *scheduler)

{

    gint record_id;

    assert(scheduler);

	if (scheduler->msqlquery == NULL) {

		g_warning ("[%s] Scheduler db connection not initialized", __FUNCTION__);

		return 0;

	}

    record_id = mysql_insert_id (scheduler->msqlquery->conn);

	return record_id;

}	

/** Notifies the backend of an update in the db.

 * 

 * @param record_id the id of the modified recording.

 */

static void

update_backend(GMythScheduler *scheduler, gint record_id)//fixme: put void and discovery record_id inside

{

    GMythSocket *socket;

    GMythStringList *strlist = gmyth_string_list_new ();

    GString *datastr = g_string_new ("RESCHEDULE_RECORDINGS ");

    g_string_append_printf (datastr, "%d", record_id);

    gmyth_string_list_append_string (strlist, datastr);

    socket = gmyth_socket_new ();

    if (gmyth_socket_connect (socket, scheduler->backend_info->hostname,

			    scheduler->backend_info->port)) {

        gmyth_socket_sendreceive_stringlist (socket, strlist);

    } else {

	g_warning ("[%s] Connection to backend failed!", __FUNCTION__);

    }

    g_string_free(datastr, TRUE);

    g_object_unref(strlist);

}

void

gmyth_scheduler_record_info_get_preview (RecordedInfo *info, GByteArray* data)

{

}

void

gmyth_scheduler_record_info_free (RecordedInfo *info)

{

    if (info->title != NULL)

        g_string_free (info->title, TRUE);

    if (info->subtitle != NULL)

        g_string_free (info->subtitle, TRUE);

    if (info->description != NULL)

        g_string_free (info->description, TRUE);

    if (info->category != NULL)

        g_string_free (info->category, TRUE);

    if (info->basename != NULL)

        g_string_free (info->basename, TRUE);

    g_free (info);

}

void

gmyth_scheduler_schedule_info_free (ScheduleInfo *info)

{

    if (info->title != NULL)

        g_string_free (info->title, TRUE);

    if (info->subtitle != NULL)

        g_string_free (info->subtitle, TRUE);

    if (info->description != NULL)

        g_string_free (info->description, TRUE);

    if (info->category != NULL)

        g_string_free (info->category, TRUE);

    g_free (info);

}