#include<gtk/gtk.h>
#include<glib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <string.h>
#include <stdio.h>
#include <stdlib.h>

#include "mmyth_ui.h"
#include "mmyth_recordui.h"

/* GMyth library includes */
#include "gmyth_scheduler.h"
#include "gmyth_util.h"

enum {
	START_DATE_COLUMN = 0,
	TITLE_COLUMN,
	CHAN_ID_COLUMN,
	END_TIME_COLUMN,
	RECORD_ID_COLUMN,
	BASENAME_COLUMN,
	N_COLUMNS
};

gboolean
mmyth_recordui_reload_all (MMythRecordUI *recordui)
{
	gboolean res = FALSE;

	res = mmyth_recordui_reload_schedule (recordui);
		
	res = res & mmyth_recordui_reload_record (recordui);
	
	
	if (!res)
		g_warning ("[%s] Error while reloading schedule and recording content", __FUNCTION__);
		
	return res;
}

gboolean
mmyth_recordui_reload_schedule (MMythRecordUI *recordui)
{
	gint new_row = 0;
	ScheduleInfo *schedule_info;
	GList *schedule_list;
	GtkTreeIter iter;
    GString *start_date_time = NULL;
    GString *end_date_time = NULL;
    GString *str_aux = g_string_new("");
	gint res;

	gtk_tree_store_clear(recordui->sch_tree_store);	

	res = gmyth_scheduler_get_schedule_list(recordui->scheduler, &(schedule_list));
	if (res < 0) {
		g_warning ("[%s] Retrieved NULL list of scheduled data from database", 
                   __FUNCTION__);
		return FALSE;
	}

	for ( ; schedule_list; schedule_list = schedule_list->next) {
 	  	schedule_info = (ScheduleInfo*) schedule_list->data;

  	  	gtk_tree_store_insert(recordui->sch_tree_store, &iter, NULL, new_row++);
        
        start_date_time = gmyth_util_time_to_string(schedule_info->start_time);
        end_date_time = gmyth_util_time_to_string(schedule_info->end_time);
        
        g_string_printf(str_aux, "%d", schedule_info->channel_id);
        
	  	gtk_tree_store_set(recordui->sch_tree_store, &iter,
	  		START_DATE_COLUMN, start_date_time->str, 
	  		TITLE_COLUMN, schedule_info->title->str,
	  		CHAN_ID_COLUMN, str_aux->str,
	  		END_TIME_COLUMN, end_date_time->str, //It doesn't appear
	  		RECORD_ID_COLUMN, schedule_info->record_id, 
			-1); //the last line is a hidden item to be used in searching tasks
 	}
  	
  	g_debug ("[%s] %d lines added to schedule list UI", __FUNCTION__, new_row);
  	
    /* free allocated memory */
    if(!start_date_time)
        g_string_free(start_date_time, FALSE);
    if(!end_date_time)
        g_string_free(end_date_time, FALSE);
    g_string_free(str_aux,  FALSE);
    
  	return TRUE;
}

gboolean
mmyth_recordui_reload_record (MMythRecordUI *recordui)
{
	gint new_row = 0;
	RecordedInfo *recorded_info;
	GList *record_list = NULL;
	GtkTreeIter iter;
    GString *start_date_time = NULL;
    GString *end_date_time = NULL;
    GString *str_aux = g_string_new("");
	gint res;
	
	gtk_tree_store_clear(recordui->rec_tree_store);		

	res = gmyth_scheduler_get_recorded_list(recordui->scheduler, &record_list);
	if (res < 0) {
		g_warning ("[%s] Retrieved NULL list of recorded data from database", __FUNCTION__);
		return FALSE;
	}
	
	for (; record_list; record_list = record_list->next) {
 	  	recorded_info = (RecordedInfo*) record_list->data;

  	  	gtk_tree_store_insert(recordui->rec_tree_store, &iter, NULL, new_row++);

        start_date_time = gmyth_util_time_to_string(recorded_info->start_time);
        end_date_time = gmyth_util_time_to_string(recorded_info->end_time);

        g_string_printf(str_aux, "%d", recorded_info->channel_id);

	  	gtk_tree_store_set(recordui->rec_tree_store, &iter,
	  		START_DATE_COLUMN, start_date_time->str, 
	  		TITLE_COLUMN, recorded_info->title->str,
	  		CHAN_ID_COLUMN, str_aux->str,
	  		END_TIME_COLUMN, end_date_time->str, //It doesn't appear
	  		RECORD_ID_COLUMN, recorded_info->record_id,
			BASENAME_COLUMN, recorded_info->basename->str, -1); 
            //the last line is a hidden item to be used in searching tasks	  		
  	}
  	
  	g_debug ("[%s] %d lines added to record list UI", __FUNCTION__, new_row);
  	  	
  	return TRUE;
}


MMythRecordUI*
mmyth_recordui_new(void)
{
	MMythRecordUI *recordui = g_new0 (MMythRecordUI, 1);
	
	recordui->scrolled_window = gtk_scrolled_window_new (NULL, NULL);
	gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (recordui->scrolled_window),
			GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);

	recordui->viewport = gtk_viewport_new (NULL, NULL);
  	gtk_container_add (GTK_CONTAINER (recordui->scrolled_window), recordui->viewport);

  	recordui->notebook = gtk_notebook_new ();
  	gtk_container_set_border_width (GTK_CONTAINER (recordui->notebook), 1);
  	gtk_notebook_set_scrollable (GTK_NOTEBOOK (recordui->notebook), TRUE);
  	gtk_notebook_popup_enable (GTK_NOTEBOOK (recordui->notebook));
  	gtk_container_add (GTK_CONTAINER (recordui->viewport), recordui->notebook);
  	gtk_notebook_popup_disable(GTK_NOTEBOOK (recordui->notebook));

	/* Schedule tab */
  	recordui->sch_scrolled_window = gtk_scrolled_window_new (NULL, NULL);
  	gtk_container_add (GTK_CONTAINER (recordui->notebook), recordui->sch_scrolled_window);
  	gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (recordui->sch_scrolled_window), 
                                    GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
  	gtk_scrolled_window_set_shadow_type (GTK_SCROLLED_WINDOW (recordui->sch_scrolled_window), 
                                         GTK_SHADOW_IN);

    /* The basename column in the sched_tree_store is not being used*/
  	recordui->sch_tree_store = 
        gtk_tree_store_new (N_COLUMNS, G_TYPE_STRING, G_TYPE_STRING,
  		                    G_TYPE_STRING, G_TYPE_STRING, G_TYPE_INT, G_TYPE_STRING );
    
  	recordui->sch_treeview = 
        gtk_tree_view_new_with_model(GTK_TREE_MODEL(recordui->sch_tree_store));
  	gtk_container_add (GTK_CONTAINER (recordui->sch_scrolled_window), 
                       recordui->sch_treeview);
  	recordui->sch_renderer = gtk_cell_renderer_text_new();
  	//g_object_set(G_OBJECT(renderer1), "foreground", "green", "background", "black", NULL);
  	recordui->sch_column1 = 
        gtk_tree_view_column_new_with_attributes("Start time", recordui->sch_renderer, 
                                                 "text", START_DATE_COLUMN, NULL);
  	gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->sch_treeview), 
                                recordui->sch_column1); 
  	recordui->sch_column2 = 
        gtk_tree_view_column_new_with_attributes("Title", recordui->sch_renderer, 
                                                 "text", TITLE_COLUMN, NULL);
  	gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->sch_treeview), 
                                recordui->sch_column2);
  	recordui->sch_column3 = 
        gtk_tree_view_column_new_with_attributes("Channel", recordui->sch_renderer, 
                                                 "text", CHAN_ID_COLUMN, NULL);
  	gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->sch_treeview), 
                                recordui->sch_column3);
  	gtk_tree_view_column_set_resizable(recordui->sch_column1, TRUE);
  	gtk_tree_view_column_set_resizable(recordui->sch_column2, TRUE);
  	gtk_tree_view_column_set_resizable(recordui->sch_column3, TRUE);
  	gtk_tree_view_column_set_reorderable(recordui->sch_column1, TRUE);
  	gtk_tree_view_column_set_reorderable(recordui->sch_column2, TRUE);
  	gtk_tree_view_column_set_reorderable(recordui->sch_column3, TRUE);  	  	  	  	
//  recordui->sch_column4 = 
//      gtk_tree_view_column_new_with_attributes("", recordui->sch_renderer, "text", END_TIME_COLUMN, NULL);
//  gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->sch_treeview), 
//                                            recordui->sch_column4);

  	recordui->sch_label = gtk_label_new (("Schedule"));
  	gtk_notebook_set_tab_label (GTK_NOTEBOOK (recordui->notebook), 
                                gtk_notebook_get_nth_page (
                                               GTK_NOTEBOOK (recordui->notebook), 0), 
                                recordui->sch_label);

  // Record items  tab
  //   g_object_set(G_OBJECT(renderer2), "foreground", "blue", NULL);
  	recordui->rec_scrolled_window = gtk_scrolled_window_new (NULL, NULL);
  	gtk_container_add (GTK_CONTAINER (recordui->notebook), 
                       recordui->rec_scrolled_window);
  	gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (recordui->rec_scrolled_window), 
                                    GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
  	gtk_scrolled_window_set_shadow_type (GTK_SCROLLED_WINDOW (recordui->rec_scrolled_window), 
                                         GTK_SHADOW_IN);

  	recordui->rec_tree_store = 
        gtk_tree_store_new (N_COLUMNS, G_TYPE_STRING, G_TYPE_STRING,
                	        G_TYPE_STRING, G_TYPE_STRING, G_TYPE_INT, G_TYPE_STRING);
  	recordui->rec_treeview = 
        gtk_tree_view_new_with_model(GTK_TREE_MODEL(recordui->rec_tree_store));
  	gtk_container_add (GTK_CONTAINER (recordui->rec_scrolled_window), 
                       recordui->rec_treeview);
  	recordui->rec_renderer = gtk_cell_renderer_text_new();
  	//g_object_set(G_OBJECT(renderer1), "foreground", "green", "background", "black", NULL);
 
  	recordui->rec_column1 = 
        gtk_tree_view_column_new_with_attributes("Start time", recordui->rec_renderer, 
                                                 "text", START_DATE_COLUMN, NULL);
  	gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->rec_treeview), 
                                recordui->rec_column1);
  	recordui->rec_column2 = 
        gtk_tree_view_column_new_with_attributes("Title", recordui->rec_renderer, 
                                                 "text", TITLE_COLUMN, NULL);
  	gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->rec_treeview), 
                                recordui->rec_column2);
  	recordui->rec_column3 = 
        gtk_tree_view_column_new_with_attributes("Channel", recordui->rec_renderer, 
                                                 "text", CHAN_ID_COLUMN, NULL);
  	gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->rec_treeview), 
                                recordui->rec_column3);
  	gtk_tree_view_column_set_resizable(recordui->rec_column1, TRUE);
  	gtk_tree_view_column_set_resizable(recordui->rec_column2, TRUE);
  	gtk_tree_view_column_set_resizable(recordui->rec_column3, TRUE);
  	gtk_tree_view_column_set_reorderable(recordui->rec_column1, TRUE);
  	gtk_tree_view_column_set_reorderable(recordui->rec_column2, TRUE);
  	gtk_tree_view_column_set_reorderable(recordui->rec_column3, TRUE);
//  recordui->rec_column4 = gtk_tree_view_column_new_with_attributes("", recordui->rec_renderer, "text", END_TIME_COLUMN, NULL);
//  gtk_tree_view_append_column(GTK_TREE_VIEW(recordui->rec_treeview), recordui->rec_column4);

  	recordui->rec_label = gtk_label_new (("Recorded"));
  	gtk_notebook_set_tab_label (GTK_NOTEBOOK (recordui->notebook), 
                                gtk_notebook_get_nth_page (
                                      GTK_NOTEBOOK (recordui->notebook), 1), 
                                recordui->rec_label);
  
  	// Gets the mmyth scheduler manager
  	recordui->scheduler = gmyth_scheduler_new ();
    
    /* init connection to the backend */
    gmyth_scheduler_connect (recordui->scheduler);
    
	return recordui;
}

void
mmyth_recordui_free (MMythRecordUI *recordui)
{
	// FIXME: Release memory here!
    /* close connection to the backend */
    gmyth_scheduler_disconnect (recordui->scheduler);
}

void
mmyth_recordui_delete_selected (GtkButton *button, MMythRecordUI *recordui)
{
	GtkTreeSelection *selection;
	GtkTreeModel *list_store;
	GtkTreeIter iter;
	int index;
	int curr_page = 0;
	
	curr_page = gtk_notebook_get_current_page(GTK_NOTEBOOK(recordui->notebook));

	if ( curr_page == 0) {
		selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(recordui->sch_treeview));
		if (selection != NULL) {
			gtk_tree_selection_get_selected(selection, &list_store, &iter);
			gtk_tree_model_get(list_store, &iter, RECORD_ID_COLUMN, &index, -1);
			gmyth_scheduler_delete_schedule(recordui->scheduler, index);
			mmyth_recordui_reload_schedule (recordui);
			return;
		}
			
	} else if (curr_page == 1) {
		selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(recordui->rec_treeview));
		if (selection != NULL) {
			gtk_tree_selection_get_selected(selection, &list_store, &iter);
			gtk_tree_model_get(list_store, &iter, RECORD_ID_COLUMN, &index, -1);
			gmyth_scheduler_delete_recorded(recordui->scheduler, index);
			mmyth_recordui_reload_record (recordui);
			return;
		}
	}
	
	g_warning ("[%s] None element was removed from the list", __FUNCTION__);
}

/* FIXME: change this function name, it is returning the 
 * basename_column that represents the nuv filename of 
 * the recorded content */
gchar*
mmyth_recordui_get_selected_recorded (MMythRecordUI *recordui)
{
	GtkTreeSelection *selection = NULL;
	GtkTreeModel *list_store = NULL;
	GtkTreeIter iter;
	gchar *path = NULL;

	/* returning nuv filename, basename_column */
	selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(recordui->rec_treeview));
	if (gtk_tree_selection_get_selected (selection, &list_store, &iter)) {
		gtk_tree_model_get(list_store, &iter, BASENAME_COLUMN, &path, -1);
	}

	// FIXME: MOVE THIS TO OTHER PLACE
	return path;
}