melunko@38: /*
melunko@38:  * @author Hallyson Melo <hallyson.melo@indt.org.br>
melunko@38:  *
melunko@38:  * This program is free software; you can redistribute it and/or modify
melunko@38:  * it under the terms of the GNU Lesser General Public License as published by
melunko@38:  * the Free Software Foundation; either version 2 of the License, or
melunko@38:  * (at your option) any later version.
melunko@38:  *
melunko@38:  * This program is distributed in the hope that it will be useful,
melunko@38:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
melunko@38:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
melunko@38:  * GNU General Public License for more details.
melunko@38:  *
melunko@38:  * You should have received a copy of the GNU Lesser General Public License
melunko@38:  * along with this program; if not, write to the Free Software
melunko@38:  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
melunko@38:  */
melunko@38: 
melunko@38: #ifdef HAVE_CONFIG_H
melunko@38: #include <config.h>
melunko@38: #endif
melunko@38: 
melunko@38: #include <string.h>
melunko@38: #include <glib.h>
melunko@38: #include <math.h>
melunko@38: 
melunko@38: #include <libgnomevfs/gnome-vfs-module.h>
melunko@38: #include <libgnomevfs/gnome-vfs-utils.h>
melunko@38: 
rosfran@277: #include <gmyth/gmyth_file_transfer.h>
rosfran@277: #include <gmyth/gmyth_livetv.h>
rosfran@277: #include <gmyth/gmyth_uri.h>
rosfran@277: #include <gmyth/gmyth_recorder.h>
rosfran@277: #include <gmyth/gmyth_backendinfo.h>
rosfran@277: #include <gmyth/gmyth_util.h>
melunko@38: 
melunko@38: #define GST_MYTHTV_ID_NUM               1
melunko@38: #define MYTHTV_VERSION_DEFAULT          30
rosfran@277: #define MYTHTV_TRANSFER_MAX_WAITS       100
melunko@38: 
melunko@111: #define MYTHTV_BUFFER_SIZE		1024*64
melunko@111: 
melunko@38: static GnomeVFSResult do_read (GnomeVFSMethod *method,
melunko@38:                                GnomeVFSMethodHandle *method_handle,
melunko@38:                                gpointer buffer,
melunko@38:                                GnomeVFSFileSize num_bytes,
melunko@38:                                GnomeVFSFileSize *bytes_read,
melunko@38:                                GnomeVFSContext *context);
melunko@38: 
melunko@38: typedef struct {
melunko@48:     GMythFileTransfer *file_transfer;
rosfran@277:     GMythLiveTV 			*livetv;
rosfran@277:     gint 							channel_num;
melunko@38:     
melunko@38:     gint mythtv_version;
melunko@38:     guint64 content_size;
melunko@38:     guint64 bytes_read;
melunko@111: 
rosfran@116:     GByteArray *buffer;
melunko@111:     gsize buffer_remain;
melunko@38: } MythtvHandle;
melunko@38: 
melunko@38: 
melunko@38: 
melunko@38: static GnomeVFSResult
melunko@38: do_open (GnomeVFSMethod *method,
melunko@38:          GnomeVFSMethodHandle **method_handle,
melunko@38:          GnomeVFSURI *uri,
melunko@38:          GnomeVFSOpenMode mode,
melunko@38:          GnomeVFSContext *context)
melunko@38: {
melunko@111:     MythtvHandle *myth_handle;
melunko@160:     GMythBackendInfo *backend_info;
rosfran@277:     GMythURI *gmyth_uri = NULL;
rosfran@277:     gboolean ret = TRUE;
rosfran@277:     gboolean is_livetv = FALSE;
rosfran@277:     gint wait_to_transfer = 0;
melunko@38: 
melunko@38:     _GNOME_VFS_METHOD_PARAM_CHECK (method_handle != NULL);
melunko@38:     _GNOME_VFS_METHOD_PARAM_CHECK (uri != NULL);
melunko@38: 
melunko@111:     myth_handle = g_new0 (MythtvHandle, 1);
melunko@111: 
melunko@38:     if (mode & GNOME_VFS_OPEN_WRITE) {
melunko@38:         return GNOME_VFS_ERROR_NOT_PERMITTED;
melunko@38:     }
melunko@38: 
renatofilho@149:     if (gnome_vfs_uri_get_host_name (uri) == NULL) {
renatofilho@149:     	return GNOME_VFS_ERROR_INVALID_HOST_NAME;
melunko@38:     }
melunko@38: 
melunko@38:     /* Initialize mythtv handler*/
melunko@38:     myth_handle->file_transfer = NULL;
rosfran@277:     myth_handle->livetv = NULL;
melunko@38:     myth_handle->mythtv_version = MYTHTV_VERSION_DEFAULT;
melunko@38:     myth_handle->bytes_read = 0;
melunko@38:     myth_handle->content_size = -1;
melunko@38: 
rosfran@277:     /* Creates and fills out the backend info structure */    
rosfran@277: 	  backend_info = gmyth_backend_info_new_with_uri ( gnome_vfs_uri_to_string( uri, GNOME_VFS_URI_HIDE_NONE ) );
rosfran@277: 	  
rosfran@277: 	  /* creates an instance of  */  
rosfran@277: 	  gmyth_uri = gmyth_uri_new_with_value( gnome_vfs_uri_to_string( uri, GNOME_VFS_URI_HIDE_NONE ) );
rosfran@277: 	  
rosfran@277: 	  /* Connect to the backend */	  
rosfran@277: 	  if ( gmyth_uri != NULL && ( is_livetv = gmyth_uri_is_livetv( gmyth_uri ) ) == TRUE ) {
rosfran@277: 	    myth_handle->livetv = gmyth_livetv_new ();
rosfran@180: 	    
rosfran@277: 	    myth_handle->channel_num = gmyth_uri_get_channel_num( gmyth_uri );
rosfran@277: 	
rosfran@277: 	    if ( myth_handle->channel_num != -1 ) {
rosfran@277: 	      if (gmyth_livetv_channel_setup (myth_handle->livetv, myth_handle->channel_num,
rosfran@277: 	              backend_info) == FALSE) {
rosfran@277: 	        g_object_unref( gmyth_uri );
rosfran@277: 	        ret = FALSE;
rosfran@277: 	      }
rosfran@277: 	    } else {
rosfran@277: 	      if ( gmyth_livetv_setup (myth_handle->livetv, backend_info) == FALSE ) {
rosfran@277: 	      	g_object_unref( gmyth_uri );
rosfran@277: 	        ret = FALSE;
rosfran@277: 	      }
rosfran@277: 	    }
rosfran@277: 	
rosfran@277: 	    myth_handle->file_transfer = gmyth_livetv_create_file_transfer (myth_handle->livetv);
rosfran@277: 	
rosfran@277: 	    if (NULL == myth_handle->file_transfer) {
rosfran@277: 	      ret = FALSE;
rosfran@277: 	    }
rosfran@277: 	    
rosfran@277: 	    if ( gmyth_uri != NULL )
rosfran@277: 	    	g_object_unref( gmyth_uri );
rosfran@277: 	    	
rosfran@277: 	  } else {
rosfran@277: 	
rosfran@277: 	    myth_handle->file_transfer = gmyth_file_transfer_new (backend_info);
rosfran@277: 	    
rosfran@277: 	    /* Verifies if the file exists */
rosfran@277: 	    if (!gmyth_util_file_exists (backend_info, gnome_vfs_uri_get_path (uri))) {
rosfran@277: 	        g_object_unref (backend_info);
rosfran@277: 					return GNOME_VFS_ERROR_NOT_FOUND;
rosfran@277: 	    }
rosfran@277: 	    
rosfran@277: 	    /* sets the Playback monitor connection */
rosfran@277: 	    ret = gmyth_file_transfer_open ( myth_handle->file_transfer, gnome_vfs_uri_get_path (uri) );
rosfran@277: 		
rosfran@277: 	  } /* if - LiveTV or not? */
rosfran@277: 	  
rosfran@277:     if (ret == FALSE) {
rosfran@277: 	    g_warning ("MythTV FileTransfer open error\n");
rosfran@277: 	    return GNOME_VFS_ERROR_NOT_OPEN;
rosfran@277: 	  } 
rosfran@277: 	
rosfran@277: 	  if ( is_livetv == TRUE && ret == TRUE ) {
rosfran@277: 	    /* loop finished, set the max tries variable to zero again... */
rosfran@277: 	    wait_to_transfer = 0;
rosfran@277: 	
rosfran@277: 	    while ( wait_to_transfer++ < MYTHTV_TRANSFER_MAX_WAITS &&
rosfran@277: 	        (gmyth_livetv_is_recording (myth_handle->livetv) == FALSE))
rosfran@277: 	      g_usleep (500);
rosfran@277: 	
rosfran@277: 	    /* IS_RECORDING again, just like the MythTV backend does... */
rosfran@277: 	    gmyth_livetv_is_recording (myth_handle->livetv);
rosfran@277: 	
rosfran@277: 	    sleep (4);                  /* FIXME: this is evil (tpm) */
rosfran@277: 	  }
melunko@251: 
renatofilho@188:     g_object_unref (backend_info);
melunko@38: 
melunko@111:     g_return_val_if_fail (myth_handle->file_transfer != NULL, GNOME_VFS_ERROR_NOT_OPEN);
melunko@38: 
melunko@38:     myth_handle->content_size = myth_handle->file_transfer->filesize;
melunko@111:     
rosfran@116:     myth_handle->buffer = g_byte_array_sized_new (MYTHTV_BUFFER_SIZE);
melunko@111:     myth_handle->buffer_remain = 0;
melunko@38: 
melunko@38:     *method_handle = (GnomeVFSMethodHandle *) myth_handle;
melunko@38: 
melunko@38:     return GNOME_VFS_OK;
melunko@38: }
melunko@38: 
melunko@38: static GnomeVFSResult
melunko@38: do_read (GnomeVFSMethod *method,
melunko@38:          GnomeVFSMethodHandle *method_handle,
melunko@38:          gpointer buffer,
melunko@38:          GnomeVFSFileSize num_bytes,
melunko@38:          GnomeVFSFileSize *bytes_read,
melunko@38:          GnomeVFSContext *context)
melunko@38: {
melunko@38:     MythtvHandle *myth_handle = (MythtvHandle *) method_handle;
melunko@38:     GnomeVFSFileSize bytes_to_read;
melunko@38: 
melunko@38:     *bytes_read = 0;
melunko@38: 
melunko@38:     if (myth_handle->bytes_read >= myth_handle->content_size)
melunko@38:         return GNOME_VFS_ERROR_EOF;
melunko@38: 
melunko@38:     // fixme: change this to min math function
melunko@38:     if (num_bytes > myth_handle->content_size - myth_handle->bytes_read)
melunko@38: 	    bytes_to_read = myth_handle->content_size - myth_handle->bytes_read;
melunko@38:     else
melunko@38: 	    bytes_to_read = num_bytes;
melunko@38: 
melunko@38:     /* Loop sending the Myth File Transfer request:
melunko@38:     * Retry whilst authentication fails and we supply it. */
rosfran@116:   
renatofilho@149:     if ( bytes_to_read > myth_handle->buffer_remain ) {
renatofilho@149:         GByteArray *tmp_buffer = g_byte_array_new();
melunko@38: 
renatofilho@149:         gint len = gmyth_file_transfer_read (myth_handle->file_transfer,
renatofilho@149:               tmp_buffer, MYTHTV_BUFFER_SIZE - myth_handle->buffer_remain, TRUE);
melunko@38: 
melunko@251: 	if (len < 0) {
renatofilho@149:             g_byte_array_free (tmp_buffer, TRUE);
renatofilho@164:             g_warning ("Fail to read bytes");
melunko@251: 	    return GNOME_VFS_ERROR_IO;
melunko@251:         } else if (len == 0) {
melunko@251: 	    g_byte_array_free (tmp_buffer, TRUE);
melunko@251: 	    g_warning ("End of file probably achieved");
melunko@251: 	    return GNOME_VFS_ERROR_EOF;
melunko@251: 	}
melunko@111: 
renatofilho@149:         myth_handle->buffer = g_byte_array_append (myth_handle->buffer,
renatofilho@149:     		tmp_buffer->data, len);
rosfran@116: 
rosfran@116: 		myth_handle->buffer_remain += len;
rosfran@116: 
renatofilho@169:         g_byte_array_free (tmp_buffer, TRUE);
rosfran@116:     	tmp_buffer = NULL;    	
rosfran@116:     }
melunko@38:     
melunko@111:     bytes_to_read = (bytes_to_read > myth_handle->buffer_remain) ? myth_handle->buffer_remain : bytes_to_read;
renatofilho@164:     /* gets the first buffer_size bytes from the byte array buffer variable */ 
rosfran@116: 
renatofilho@149:     g_memmove (buffer, myth_handle->buffer->data, bytes_to_read);
rosfran@116: 
melunko@111:     myth_handle->bytes_read += bytes_to_read;
rosfran@116:     myth_handle->buffer_remain -= bytes_to_read;    
rosfran@116: 
rosfran@116:   	/* flushs the newly buffer got from byte array */
renatofilho@149:   	myth_handle->buffer = g_byte_array_remove_range (myth_handle->buffer, 0, bytes_to_read);
melunko@111:     *bytes_read = bytes_to_read;
melunko@38:   
melunko@38:     return GNOME_VFS_OK;
melunko@38: }
melunko@38: 
melunko@38: static GnomeVFSResult
melunko@38: do_close (GnomeVFSMethod *method,
melunko@38:           GnomeVFSMethodHandle *method_handle,
melunko@38:           GnomeVFSContext *context)
melunko@38: {
melunko@111: 
melunko@38:     MythtvHandle *myth_handle = (MythtvHandle *) method_handle;
melunko@38: 
melunko@111:     if (myth_handle->file_transfer) {
renatofilho@164: 	    gmyth_file_transfer_close (myth_handle->file_transfer);
melunko@38:         g_object_unref (myth_handle->file_transfer);
renatofilho@164:     	myth_handle->file_transfer = NULL;
melunko@111:     }
rosfran@127:     
rosfran@277:     if (myth_handle->livetv) {
rosfran@277:       g_object_unref (myth_handle->livetv);
rosfran@277:     	myth_handle->livetv = NULL;
rosfran@277:     }
rosfran@277:     
rosfran@127:     if (myth_handle->buffer) {
renatofilho@164: 		g_byte_array_free (myth_handle->buffer, TRUE);
renatofilho@164: 		myth_handle->buffer = NULL;
rosfran@127:     }
melunko@38: 
melunko@38:     g_free (myth_handle);
melunko@38: 
melunko@38:     return GNOME_VFS_OK;
melunko@38: }
melunko@38: 
melunko@38: static GnomeVFSResult
melunko@38: do_get_file_info (GnomeVFSMethod *method,
melunko@38:                   GnomeVFSURI *uri,
melunko@38:                   GnomeVFSFileInfo *file_info,
melunko@38:                   GnomeVFSFileInfoOptions options,
melunko@38:                   GnomeVFSContext *context)
melunko@38: {
renatofilho@188:     GMythFileTransfer *file_transfer = NULL;
renatofilho@188:     GMythBackendInfo *backend_info = NULL;
renatofilho@188: 
renatofilho@188:     file_info->name = g_strdup (gnome_vfs_uri_get_path (uri));
melunko@38:     file_info->valid_fields = file_info->valid_fields
melunko@38:         | GNOME_VFS_FILE_INFO_FIELDS_TYPE
melunko@38:         | GNOME_VFS_FILE_INFO_FIELDS_MIME_TYPE
melunko@38:         | GNOME_VFS_FILE_INFO_FIELDS_PERMISSIONS;
melunko@38:     file_info->type = GNOME_VFS_FILE_TYPE_REGULAR;
melunko@111:     // fixme: get from file extension?
melunko@111:     file_info->mime_type = g_strdup ("video/x-nuv");
melunko@38:     file_info->permissions =
melunko@38:         GNOME_VFS_PERM_USER_READ |
melunko@38:         GNOME_VFS_PERM_OTHER_READ |
melunko@38:         GNOME_VFS_PERM_GROUP_READ;
melunko@38: 
renatofilho@188:     backend_info = gmyth_backend_info_new_full (gnome_vfs_uri_get_host_name (uri),
renatofilho@188:             gnome_vfs_uri_get_user_name (uri),
renatofilho@188:             gnome_vfs_uri_get_password (uri),
renatofilho@188:             NULL,
renatofilho@188:             gnome_vfs_uri_get_host_port (uri));
renatofilho@188: 
renatofilho@188:     file_transfer = gmyth_file_transfer_new (backend_info);
renatofilho@188:     if (gmyth_file_transfer_open (file_transfer, gnome_vfs_uri_get_path (uri)) == TRUE) {
renatofilho@188:         file_info->size = gmyth_file_transfer_get_filesize (file_transfer);
renatofilho@188:         file_info->block_count = GNOME_VFS_FILE_INFO_FIELDS_BLOCK_COUNT;
renatofilho@188:         file_info->io_block_size = GNOME_VFS_FILE_INFO_FIELDS_IO_BLOCK_SIZE;
renatofilho@188:     }
renatofilho@188: 
renatofilho@188:     g_object_unref (file_transfer);
renatofilho@188:     g_object_unref (backend_info);
melunko@38:     return GNOME_VFS_OK;
melunko@38: }
melunko@38: 
melunko@38: static gboolean
melunko@38: do_is_local (GnomeVFSMethod *method,
melunko@38:              const GnomeVFSURI *uri)
melunko@38: {
melunko@38: 	return FALSE;
melunko@38: }
melunko@38: 
melunko@38: static GnomeVFSMethod method = {
melunko@38: 	sizeof (GnomeVFSMethod),
melunko@38: 	do_open,
melunko@38: 	NULL,
melunko@38: 	do_close,
melunko@38: 	do_read,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	do_get_file_info,
melunko@38: 	NULL,
melunko@38: 	do_is_local,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: 	NULL,
melunko@38: };
melunko@38: 
melunko@38: 
melunko@38: GnomeVFSMethod *
melunko@38: vfs_module_init (const char *method_name, const char *args)
melunko@38: {
melunko@38: 	return &method;
melunko@38: }
melunko@38: 
melunko@38: void
melunko@38: vfs_module_shutdown (GnomeVFSMethod *method)
melunko@38: {
melunko@38: }