/**

  * GMyth Library

  *

  * @file gmyth/gmyth_file_transfer.c

  * 

  * @brief <p> GMythFileTransfer deals with the file streaming media remote/local

  * transfering to the MythTV frontend.

  *

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

  * @author Rosfran Lins Borges <rosfran.borges@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

  *

  * GStreamer MythTV plug-in properties:

  * - location (backend server hostname/URL) [ex.: myth://192.168.1.73:28722/1000_1092091.nuv]

  * - path (qurl - remote file to be opened)

  * - port number *   

  */

 #ifdef HAVE_CONFIG_H

 #include "config.h"

 #endif

 #include "gmyth_file_transfer.h"

 #include "gmyth_livetv.h"

 #include "gmyth_util.h"

 #include "gmyth_socket.h"

 #include "gmyth_stringlist.h"

 #include "gmyth_debug.h"

 #include "gmyth_uri.h"

 #include "gmyth_marshal.h"

 #include <unistd.h>

 #include <glib.h>

 #include <arpa/inet.h>

 #include <sys/types.h>

 #include <sys/socket.h>

 #include <netdb.h>

 #include <errno.h>

 #include <stdlib.h>

 #include <assert.h>

 #define GMYTHTV_QUERY_HEADER		"QUERY_FILETRANSFER "

 /* default values to the file transfer parameters */

 #define GMYTHTV_USER_READ_AHEAD			TRUE

 #define GMYTHTV_RETRIES							-1

 #define GMYTHTV_FILE_SIZE						0

 #define GMYTHTV_BUFFER_SIZE					64*1024

 #define GMYTHTV_VERSION							30

 #define GMYTHTV_TRANSFER_MAX_WAITS	700

 #ifdef GMYTHTV_ENABLE_DEBUG

 #define GMYTHTV_ENABLE_DEBUG				1

 #else

 #undef GMYTHTV_ENABLE_DEBUG

 #endif

 /* this NDEBUG is to maintain compatibility with GMyth library */

 #ifndef NDEBUG

 #define GMYTHTV_ENABLE_DEBUG				1

 #endif

 enum myth_sock_types {

   GMYTH_PLAYBACK_TYPE = 0,

   GMYTH_MONITOR_TYPE,

   GMYTH_FILETRANSFER_TYPE,

   GMYTH_RINGBUFFER_TYPE

 };

 #define GMYTH_FILE_TRANSFER_GET_PRIVATE(obj) (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GMYTH_FILE_TRANSFER_TYPE, GMythFileTransferPrivate))

 struct _GMythFileTransferPrivate {

 	GMythLiveTV *livetv;

 	gboolean do_next_program_chain;

 };

 static void gmyth_file_transfer_class_init          (GMythFileTransferClass *klass);

 static void gmyth_file_transfer_init                (GMythFileTransfer *object);

 static void gmyth_file_transfer_dispose  (GObject *object);

 static void gmyth_file_transfer_finalize (GObject *object);

 static void gmyth_file_transfer_program_info_changed( GMythFileTransfer *transfer, 

 										gint msg_code, gpointer livetv_transfer );

 static gboolean gmyth_connect_to_backend (GMythFileTransfer *transfer);

 void gmyth_file_transfer_close( GMythFileTransfer *transfer );

 static gboolean myth_control_acquire_context( gboolean do_wait );

 static gboolean myth_control_release_context( );

 G_DEFINE_TYPE(GMythFileTransfer, gmyth_file_transfer, G_TYPE_OBJECT)

 static void

 gmyth_file_transfer_class_init (GMythFileTransferClass *klass)

 {

   GObjectClass *gobject_class;

   GMythFileTransferClass *gtransfer_class;

   gobject_class = (GObjectClass *) klass;

   gtransfer_class = (GMythFileTransferClass *) gobject_class;

   gobject_class->dispose  = gmyth_file_transfer_dispose;

   gobject_class->finalize = gmyth_file_transfer_finalize;

   g_type_class_add_private (gobject_class, sizeof (GMythFileTransferPrivate));

 	gtransfer_class->program_info_changed_handler = gmyth_file_transfer_program_info_changed;

 	gtransfer_class->program_info_changed_handler_signal_id = 

 		  g_signal_new ( "program-info-changed",

 		                 G_TYPE_FROM_CLASS (gtransfer_class),

 		                 G_SIGNAL_RUN_LAST | G_SIGNAL_NO_RECURSE | G_SIGNAL_NO_HOOKS,

 		                 0,

 										 NULL,

 										 NULL,

 										 gmyth_marshal_VOID__INT_POINTER,

 										 G_TYPE_NONE,

 										 2,

 										 G_TYPE_INT,

 										 G_TYPE_POINTER );

 }

 static void

 gmyth_file_transfer_init (GMythFileTransfer *transfer)

 { 

   g_return_if_fail( transfer != NULL );

   transfer->readposition = 0;

   transfer->filesize = GMYTHTV_FILE_SIZE;

   transfer->priv = GMYTH_FILE_TRANSFER_GET_PRIVATE(transfer);	

   transfer->priv->do_next_program_chain = FALSE;

   transfer->priv->livetv = NULL;

   transfer->control_sock = NULL;

   transfer->sock = NULL;

   transfer->mutex = g_mutex_new();

 	g_signal_connect ( G_OBJECT (transfer), "program-info-changed",

               (GCallback)(GMYTH_FILE_TRANSFER_GET_CLASS(transfer)->program_info_changed_handler),

               NULL );

 }

 static void

 gmyth_file_transfer_dispose  (GObject *object)

 {

 	GMythFileTransfer *transfer = GMYTH_FILE_TRANSFER (object);

   if ( transfer->mutex != NULL )  

   {

   	g_mutex_free( transfer->mutex );

   	transfer->mutex = NULL;

   }

   if ( transfer->control_sock != NULL )  

   {

   	g_object_unref( transfer->control_sock );

   	transfer->control_sock = NULL;

   }  

   if ( transfer->sock != NULL )  

   {

   	g_object_unref( transfer->sock );

   	transfer->sock = NULL;

   }

   if ( transfer->filename != NULL )  

   {

   	g_free( transfer->filename );

   	transfer->filename = NULL;

   }

   G_OBJECT_CLASS (gmyth_file_transfer_parent_class)->dispose (object);

 }

 static void

 gmyth_file_transfer_finalize (GObject *object)

 {

   g_signal_handlers_destroy (object);

   G_OBJECT_CLASS (gmyth_file_transfer_parent_class)->finalize (object);

 }

 // fixme: do we need the card_id????

 GMythFileTransfer*

 gmyth_file_transfer_new ( const GMythBackendInfo *backend_info)

 {

   GMythFileTransfer *transfer = GMYTH_FILE_TRANSFER (g_object_new (GMYTH_FILE_TRANSFER_TYPE, NULL));

   transfer->backend_info = (GMythBackendInfo *)backend_info;

   g_object_ref (transfer->backend_info);

   return transfer;

 }

 GMythFileTransfer* 

 gmyth_file_transfer_new_with_uri (const gchar* uri_str)

 {

   GMythFileTransfer *transfer = GMYTH_FILE_TRANSFER (g_object_new (GMYTH_FILE_TRANSFER_TYPE, NULL));

   transfer->backend_info = gmyth_backend_info_new_with_uri (uri_str);

   return transfer;

 }

 gboolean

 gmyth_file_transfer_open (GMythFileTransfer *transfer, const gchar* filename)

 {

   gboolean ret = TRUE;

   g_return_val_if_fail (transfer != NULL, FALSE);

   g_return_val_if_fail (filename != NULL, FALSE);

   if (transfer->filename != NULL)

   {

     g_free (transfer->filename);

     transfer->filename = NULL;

   }

   transfer->filename = g_strdup (filename);

   /* configure the control socket */

   if (transfer->control_sock == NULL) { 

     if (!gmyth_connect_to_backend (transfer)) {

       gmyth_debug ("Connection to backend failed (Control Socket).\n");

       ret = FALSE;

     }

   } else {

     g_warning("Remote transfer control socket already created.\n");

   }

   gmyth_debug ("Got file with size = %lld.\n", transfer->filesize);

   return ret;

 }

 static gboolean

 gmyth_connect_to_backend (GMythFileTransfer *transfer)

 {

   GString *base_str = NULL;

   GString *hostname = NULL;

   GMythStringList *strlist = NULL; 

   gboolean ret = TRUE;

   g_return_val_if_fail (transfer != NULL, FALSE );

   base_str = g_string_new ("");

   /* Creates the control socket */

   if (transfer->control_sock != NULL) {

     g_object_unref (transfer->control_sock);

     transfer->control_sock = NULL;

   }

   transfer->control_sock = gmyth_socket_new();

   // Connects the socket, send Mythtv ANN command and verify Mythtv protocol version 

   if (!gmyth_socket_connect_to_backend (transfer->control_sock,

         transfer->backend_info->hostname, transfer->backend_info->port, TRUE)) {

     g_object_unref (transfer->control_sock);

     transfer->control_sock = NULL;

     return FALSE;

   }

   /* Creates the data socket */

   if (transfer->sock != NULL) {

     g_object_unref (transfer->sock);

     transfer->sock = NULL;

   }

   transfer->sock = gmyth_socket_new ();

   gmyth_socket_connect (transfer->sock, transfer->backend_info->hostname, transfer->backend_info->port);

   strlist = gmyth_string_list_new();

   hostname = gmyth_socket_get_local_hostname();

   gmyth_debug( "[%s] MythTV version (from backend) = %d.\n", __FUNCTION__, transfer->control_sock->mythtv_version );

   if ( transfer->control_sock->mythtv_version > 26 )

   	g_string_printf( base_str, "ANN FileTransfer %s 1 -1", hostname->str);

   else

   	g_string_printf( base_str, "ANN FileTransfer %s", hostname->str);

   gmyth_string_list_append_string (strlist, base_str );

   gmyth_string_list_append_char_array (strlist, transfer->filename);

   gmyth_socket_write_stringlist (transfer->sock, strlist );

   gmyth_socket_read_stringlist (transfer->sock, strlist );

   /* file identification used in future file transfer requests to backend */

   transfer->file_id = gmyth_string_list_get_int( strlist, 1 );

   /* Myth URI stream file size - decoded using two 8-bytes sequences (64 bits/long long types) */

   transfer->filesize = gmyth_util_decode_long_long( strlist, 2 );

   gmyth_debug ( "[%s] ***** Received: recordernum = %d, filesize = %" G_GUINT64_FORMAT "\n", __FUNCTION__,

           transfer->file_id, transfer->filesize );

   if (transfer->filesize < 0 ) {

       gmyth_debug ( "[%s] Got filesize equals to %llu is lesser than 0 [invalid stream file]\n", __FUNCTION__, transfer->filesize );

       g_object_unref (transfer->sock);

       transfer->sock = NULL;

       ret = FALSE;

       goto cleanup;

   }

 cleanup:

   if ( strlist != NULL )

     g_object_unref( strlist );

   g_string_free (base_str, TRUE);

   g_string_free (hostname, TRUE);

   return ret;

 }    

 void

 gmyth_file_transfer_emit_program_info_changed_signal ( GMythFileTransfer *transfer, gint msg_code,

 			gpointer live_tv ) 

 {

 				/*

 	g_signal_emit_by_name ( G_OBJECT(transfer),

          "program-info-changed",

          msg_code, live_tv );

          */

   gmyth_debug( "Calling signal handler... [FILE_TRANSFER]" );

   g_signal_emit ( transfer,

          GMYTH_FILE_TRANSFER_GET_CLASS (transfer)->program_info_changed_handler_signal_id,

          0, /* details */

          msg_code, live_tv );

 }

 gboolean

 gmyth_file_transfer_is_open (GMythFileTransfer *transfer)

 {

     GMythStringList *strlist;

     GString *query;

     g_return_val_if_fail (transfer->control_sock != NULL, FALSE);

     g_return_val_if_fail (transfer->sock != NULL, FALSE);

     strlist = gmyth_string_list_new();

     query = g_string_new (GMYTHTV_QUERY_HEADER);

     g_string_append_printf (query, "%d", transfer->file_id );

     gmyth_string_list_append_string (strlist, query );

     gmyth_string_list_append_char_array( strlist, "IS_OPEN" );

     gmyth_socket_write_stringlist( transfer->control_sock, strlist );

     gmyth_socket_read_stringlist( transfer->control_sock, strlist );

     g_string_free (query, TRUE);

     g_object_unref (strlist);

     return ( strlist != NULL && gmyth_string_list_get_int( strlist, 0 ) == 1 );

 }

 void

 gmyth_file_transfer_close( GMythFileTransfer *transfer )

 {

   GMythStringList *strlist;

   GString *query;

   g_return_if_fail (transfer->control_sock != NULL);

   strlist = gmyth_string_list_new( );

   query = g_string_new (GMYTHTV_QUERY_HEADER);

   g_string_append_printf( query, "%d", transfer->file_id);

   gmyth_string_list_append_string( strlist, query );

   gmyth_string_list_append_char_array( strlist, "DONE" );

   if ( gmyth_socket_sendreceive_stringlist(transfer->control_sock, strlist) <= 0 ) {

     // fixme: time out???

     g_printerr( "Remote file timeout.\n" );

   }

   g_string_free (query, TRUE);

   g_object_unref (strlist);

   if (transfer->sock) {

     g_object_unref( transfer->sock );

     transfer->sock = NULL;

   }

   if (transfer->control_sock) {

     g_object_unref( transfer->control_sock );

     transfer->control_sock = NULL;

   } 

 }

 gint64

 gmyth_file_transfer_seek(GMythFileTransfer *transfer, guint64 pos, gint whence)

 {

   GMythStringList *strlist = gmyth_string_list_new();

   GString *query;

   g_return_val_if_fail (transfer->sock != NULL, -1);

   g_return_val_if_fail (transfer->control_sock != NULL, -1);

   strlist = gmyth_string_list_new();

   query = g_string_new (GMYTHTV_QUERY_HEADER);

   g_string_append_printf (query, "%d", transfer->file_id);

   myth_control_acquire_context( TRUE );

   gmyth_string_list_append_string( strlist, query );

   gmyth_string_list_append_char_array( strlist, "SEEK" );

   gmyth_string_list_append_uint64( strlist, pos );

   gmyth_string_list_append_int( strlist, whence );  

   if (pos > 0 )

     gmyth_string_list_append_uint64( strlist, pos );

   else

     gmyth_string_list_append_uint64( strlist, transfer->readposition );

   gmyth_socket_sendreceive_stringlist( transfer->control_sock, strlist );

   gint64 retval = gmyth_string_list_get_int64(strlist, 0);

   transfer->readposition = retval;

   gmyth_debug ( "[%s] got reading position pointer from the streaming = %lld\n", 

       __FUNCTION__, retval );

   myth_control_release_context( );

   return retval;

 }

 static gboolean 

 myth_control_acquire_context( gboolean do_wait )

 {

 	gboolean ret = TRUE;	

 	//guint max_iter = 50;

 	//g_mutex_lock( mutex );

   //while ( !has_io_access ) 

   //	g_cond_wait( io_watcher_cond, mutex );

   //has_io_access = FALSE;

   //myth_control_acquire_context (FALSE);

   /* 

   if ( do_wait ) {

   	while ( --max_iter > 0 && !g_main_context_wait( io_watcher_context, io_watcher_cond, mutex ) )

   		ret = FALSE;

   } else if ( !g_main_context_acquire( io_watcher_context ) )

   	ret = FALSE;

   */

   //g_static_mutex_lock( &st_mutex );

   return ret;

 }

 static gboolean 

 myth_control_release_context( ) 

 {

 	gboolean ret = TRUE;

   //g_static_mutex_unlock( &st_mutex );

 	//g_main_context_release( io_watcher_context );

   //g_main_context_wakeup( io_watcher_context );

   //has_io_access = TRUE;

   //g_cond_broadcast( io_watcher_cond );

   //g_mutex_unlock( mutex );  

   return ret;

 }

 gint 

 gmyth_file_transfer_read(GMythFileTransfer *transfer, GByteArray *data, gint size, gboolean read_unlimited)

 {

   gint bytes_sent = 0;

   gsize bytes_read = 0;

   gsize total_read = 0;

   GError *error = NULL;

   GIOChannel *io_channel;

   GIOChannel *io_channel_control;

   GIOCondition io_cond;

   GIOCondition io_cond_control;

   GIOStatus io_status = G_IO_STATUS_NORMAL, io_status_control = G_IO_STATUS_NORMAL;

   gboolean ret = TRUE;   

   GString *query;

   g_return_val_if_fail ( data != NULL, -2 );

   io_channel = transfer->sock->sd_io_ch;

   io_channel_control = transfer->control_sock->sd_io_ch;

   io_status = g_io_channel_set_encoding( io_channel, NULL, &error );

   if ( io_status == G_IO_STATUS_NORMAL )

     gmyth_debug ( "[%s] Setting encoding to binary data socket).\n", __FUNCTION__ );

   io_cond = g_io_channel_get_buffer_condition( io_channel );  

   io_cond_control = g_io_channel_get_buffer_condition( io_channel );

   if ( transfer->sock == NULL || ( io_status == G_IO_STATUS_ERROR ) ) {

     g_printerr( "gmyth_file_transfer_read(): Called with no raw socket.\n" );

     //exit(0); // fixme remove this

     return -1;

   }

   if ( transfer->control_sock == NULL || ( io_status_control == G_IO_STATUS_ERROR ) ) {

     g_printerr( "gmyth_file_transfer_read(): Called with no control socket.\n" );

     //exit(0); // fixme remove this

     return -1;

   }

   query = g_string_new (GMYTHTV_QUERY_HEADER);

   g_string_append_printf ( query, "%d", transfer->file_id );

   gmyth_debug ("[%s] Transfer_query = %s\n", __FUNCTION__, query->str );

   /* send requests to the maximum number of REQUEST_BLOCK messages */

   guint max_tries = 5;

   myth_control_acquire_context( TRUE );

   while (total_read == 0 && --max_tries > 0) {

     GMythStringList *strlist = gmyth_string_list_new();

     gmyth_string_list_append_char_array( strlist, query->str );

     gmyth_string_list_append_char_array( strlist, "REQUEST_BLOCK" );

     gmyth_string_list_append_int( strlist, size );

     // Request the block to the backend

     gmyth_socket_write_stringlist( transfer->control_sock, strlist );

     // Receives the backand answer    

     gmyth_socket_read_stringlist( transfer->control_sock, strlist );

     if ( strlist != NULL && gmyth_string_list_length( strlist ) > 0 ) 

     { 

 	    bytes_sent = gmyth_string_list_get_int( strlist,  0 ); // -1 on backend error

 	    gmyth_debug ( "[%s] got SENT buffer message = %d\n", __FUNCTION__, bytes_sent );

     	if ( bytes_sent >= 0 ) 

     	{

         gchar *data_buffer = g_new0 ( gchar, bytes_sent );    

    	    while ( 0 != bytes_sent ) 

    	    { 

           io_status = g_io_channel_read_chars( io_channel, data_buffer + total_read, 

 	    				(gsize) bytes_sent,	&bytes_read, &error );

 	        total_read += bytes_read;

           bytes_sent -= bytes_read;

             /* append new data to the increasing byte array */

 	    		data = g_byte_array_append (data, (const guint8*)data_buffer, bytes_read);

 	        gmyth_debug ("Total transfer data read: %d\n", total_read);

         }

         g_free (data_buffer);

       } /* if */

     } else if ( !(transfer->priv != NULL && transfer->priv->livetv != NULL &&

 					transfer->priv->do_next_program_chain) ) {

     	total_read = GMYTHTV_FILE_TRANSFER_READ_ERROR;

     	g_object_unref (strlist);

     	strlist = NULL;

     	break;

     }

     g_object_unref (strlist);

     strlist = NULL;

   }

   if ( bytes_sent == 0 && max_tries == 0 ) 

 	{

 		gmyth_debug( "Trying to move to the next program chain..." );

 		transfer->priv = GMYTH_FILE_TRANSFER_GET_PRIVATE(transfer);

 		if ( transfer->priv != NULL && transfer->priv->livetv != NULL &&

 					transfer->priv->do_next_program_chain )

 		{

 			total_read = GMYTHTV_FILE_TRANSFER_NEXT_PROG_CHAIN;	

 	  		g_mutex_lock( transfer->mutex );

 	  		ret = gmyth_livetv_next_program_chain( transfer->priv->livetv );

 	  		g_mutex_unlock( transfer->mutex );

 			if ( !ret )

 				gmyth_debug( "Cannot change to the next program chain!" );

 			else

 				gmyth_debug( "OK!!! MOVED to the next program chain [%s]!", 

 										(gmyth_tvchain_get_id( transfer->priv->livetv->tvchain ))->str );						

 		}

 	} /* if */

   myth_control_release_context( );

   g_string_free (query, TRUE);

   if ( error != NULL ) {

     gmyth_debug ("Cleaning-up ERROR: %s [msg = %s, code = %d]\n", __FUNCTION__, error->message, 

 	    error->code);

     g_error_free (error);

   }

   if ( total_read > 0 )

   	transfer->readposition += total_read;  	

   return total_read;

 }

 static void 

 gmyth_file_transfer_program_info_changed( GMythFileTransfer *transfer, 

 										gint msg_code, gpointer livetv_transfer )

 {

 	GMythLiveTV *livetv = GMYTH_LIVETV( livetv_transfer );

 	gmyth_debug( "Program info changed! ( file transfer orig. = %p, ptr. = [%s], user data = [%s] )", transfer, 

 		livetv_transfer != NULL ? "[NOT NULL]" : "[NULL]", livetv != NULL ? "[NOT NULL]" : "[NULL]" );

 	if ( livetv != NULL && transfer == livetv->file_transfer )

 	{

 		gmyth_debug( "YES, the requested program info movement on the LiveTV transfer is authentical!" );

 	}

 	transfer->priv = GMYTH_FILE_TRANSFER_GET_PRIVATE(transfer);	

 	transfer->priv->livetv = livetv;	

 	transfer->priv->do_next_program_chain = TRUE;

 	//g_object_unref( transfer );

 }

 gboolean 

 gmyth_file_transfer_settimeout( GMythFileTransfer *transfer, gboolean fast )

 {

   GMythStringList *strlist = NULL;

   g_return_val_if_fail (transfer->sock != NULL, FALSE);

   g_return_val_if_fail (transfer->control_sock != NULL, FALSE);

 //  if ( transfer->timeoutisfast == fast )

 //    return;

   strlist = gmyth_string_list_new(); 

   gmyth_string_list_append_char_array( strlist, GMYTHTV_QUERY_HEADER );

   gmyth_string_list_append_char_array( strlist, "SET_TIMEOUT" );

   gmyth_string_list_append_int( strlist, fast ); 

   gmyth_socket_write_stringlist( transfer->control_sock, strlist );

   gmyth_socket_read_stringlist( transfer->control_sock, strlist );

 //  transfer->timeoutisfast = fast;

   return TRUE;

 }

 guint64

 gmyth_file_transfer_get_filesize (GMythFileTransfer *transfer)

 {

     g_return_val_if_fail (transfer != NULL, 0);

     return transfer->filesize;

 }