/* vim: set sw=2: -*- Mode: C; tab-width: 2; indent-tabs-mode: t; c-basic-offset: 2; c-indent-level: 2 -*- */

/* GStreamer MythTV Plug-in

 * Copyright (C) <2006> Rosfran Borges <rosfran.borges@indt.org.br>

 *

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

 * modify it under the terms of the GNU Library 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 library 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

 * Library General Public License for more 

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include "gstmythtvsrc.h"

#include "gmyth_file_transfer.h"

#include "gmyth_livetv.h"

#include <gmyth/gmyth_socket.h>

#include <gmyth/gmyth_tvchain.h>

#include <string.h>

#include <unistd.h>

GST_DEBUG_CATEGORY_STATIC (mythtvsrc_debug);

#define GST_CAT_DEFAULT mythtvsrc_debug

#define GST_GMYTHTV_ID_NUM		1

#define GMYTHTV_VERSION_DEFAULT		30

#define GMYTHTV_TRANSFER_MAX_WAITS	100

#define GMYTHTV_TRANSFER_MAX_BUFFER	1024*1024

//( 32*1024  )

/* 4*1024 ??? */

#define MAX_READ_SIZE              	12*1024

//( 32*1024 )

#define ENABLE_TIMING_POSITION		0

/* stablish a maximum iteration value to the IS_RECORDING message */

static guint wait_to_transfer = 0;

static const GstElementDetails gst_mythtv_src_details =

GST_ELEMENT_DETAILS ( "MythTV client source",

    "Source/Network",

    "Control and receive data as a client over the network via raw socket connections using the MythTV protocol",

    "Rosfran Borges <rosfran.borges@indt.org.br>" );

static GstStaticPadTemplate srctemplate = GST_STATIC_PAD_TEMPLATE ("src",

    GST_PAD_SRC,

    GST_PAD_ALWAYS,

    GST_STATIC_CAPS ("video/x-nuv") );

    //GST_STATIC_CAPS_ANY);

static GThread *update_size_task = NULL;

static GStaticMutex update_size_mutex = G_STATIC_MUTEX_INIT;

enum

{

  PROP_0,

  PROP_LOCATION,

  PROP_URI,

#ifndef GST_DISABLE_GST_DEBUG

  PROP_GMYTHTV_DBG,

#endif

  PROP_GMYTHTV_VERSION,

  PROP_GMYTHTV_LIVE,

  PROP_GMYTHTV_LIVEID,

  PROP_GMYTHTV_LIVE_CHAINID

};

static void gst_mythtv_src_finalize (GObject * gobject);

static GstFlowReturn gst_mythtv_src_create (GstBaseSrc * psrc, guint64 offset, 

	guint size, GstBuffer ** outbuf);

//static GstFlowReturn gst_mythtv_src_create (GstPushSrc * psrc, GstBuffer ** outbuf);

static gboolean gst_mythtv_src_start (GstBaseSrc * bsrc);

static gboolean gst_mythtv_src_stop (GstBaseSrc * bsrc);

static gboolean gst_mythtv_src_get_size (GstBaseSrc * bsrc, guint64 * size);

static gboolean gst_mythtv_src_is_seekable( GstBaseSrc *push_src );

//static gboolean gst_mythtv_new_segment ( GstBaseSrc * psrc );

static void gst_mythtv_src_set_property (GObject * object, guint prop_id,

    const GValue * value, GParamSpec * pspec);

static void gst_mythtv_src_get_property (GObject * object, guint prop_id,

    GValue * value, GParamSpec * pspec);

//static GstFlowReturn gst_mythtv_src_chain (GstPad * pad, GstBuffer * buf);

static void gst_mythtv_src_uri_handler_init (gpointer g_iface, gpointer iface_data);

static gboolean gst_mythtv_src_handle_event (GstPad * pad, GstEvent * event);

//static gboolean gst_mythtv_src_query ( GstPad * pad, GstQuery * query );

static gint do_read_request_response (GstMythtvSrc *src, guint64 offset, 

		guint size, GstBuffer **outbuf);

//static gboolean gst_mythtv_src_sink_activate_pull (GstPad * srcpad, gboolean active);

  static void

_urihandler_init (GType type)

{

  static const GInterfaceInfo urihandler_info = {

    gst_mythtv_src_uri_handler_init,

    NULL,

    NULL

  };

  g_type_add_interface_static (type, GST_TYPE_URI_HANDLER, &urihandler_info);

  GST_DEBUG_CATEGORY_INIT (mythtvsrc_debug, "mythtvsrc", 0,

      "MythTV src");

}

GST_BOILERPLATE_FULL (GstMythtvSrc, gst_mythtv_src, GstBaseSrc,

    GST_TYPE_BASE_SRC, _urihandler_init)

//GST_BOILERPLATE_FULL (GstMythtvSrc, gst_mythtv_src, GstPushSrc,

//    GST_TYPE_PUSH_SRC, _urihandler_init)

  static void

gst_mythtv_src_base_init (gpointer g_class)

{

  GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);

  gst_element_class_add_pad_template (element_class,

      gst_static_pad_template_get (&srctemplate));

  gst_element_class_set_details (element_class, &gst_mythtv_src_details);

}

static void

gst_mythtv_src_class_init (GstMythtvSrcClass * klass)

{

  GObjectClass *gobject_class;

  //GstPushSrcClass *gstpushsrc_class;

  GstBaseSrcClass *gstbasesrc_class;

  gobject_class = (GObjectClass *) klass;

  gstbasesrc_class = (GstBaseSrcClass *) klass;

  //gstpushsrc_class = (GstPushSrcClass *) klass;

  gobject_class->set_property = gst_mythtv_src_set_property;

  gobject_class->get_property = gst_mythtv_src_get_property;

  gobject_class->finalize = gst_mythtv_src_finalize;

  g_object_class_install_property

    (gobject_class, PROP_LOCATION,

     g_param_spec_string ("location", "Location",

       "The location. In the form:"

       "\n\t\t\tmyth://a.com/file.nuv"

       "\n\t\t\tmyth://a.com:23223/file.nuv "

       "\n\t\t\ta.com/file.nuv - default scheme 'myth'",

       "", G_PARAM_READWRITE));

  g_object_class_install_property

    (gobject_class, PROP_URI,

     g_param_spec_string ("uri", "Uri",

       "The location in form of a URI (deprecated; use location)",

       "", G_PARAM_READWRITE));

  g_object_class_install_property

    (gobject_class, PROP_GMYTHTV_VERSION,

     g_param_spec_int ("mythtv-version", "mythtv-version",

       "Change Myth TV version",

       26, 30, 26, G_PARAM_READWRITE));

  g_object_class_install_property

    (gobject_class, PROP_GMYTHTV_LIVEID,

     g_param_spec_int ("mythtv-live-id", "mythtv-live-id",

       "Change Myth TV version",

       0, 200, GST_GMYTHTV_ID_NUM, G_PARAM_READWRITE));

  g_object_class_install_property

    (gobject_class, PROP_GMYTHTV_LIVE_CHAINID,

     g_param_spec_string ("mythtv-live-chainid", "mythtv-live-chainid",

       "Sets the Myth TV chain ID (from TV Chain)",

       "", G_PARAM_READWRITE));

  g_object_class_install_property

    (gobject_class, PROP_GMYTHTV_LIVE,

     g_param_spec_boolean ("mythtv-live", "mythtv-live",

       "Enable MythTV Live TV content streaming",

       FALSE, G_PARAM_READWRITE));

#ifndef GST_DISABLE_GST_DEBUG

  g_object_class_install_property

    (gobject_class, PROP_GMYTHTV_DBG,

     g_param_spec_boolean ("mythtv-debug", "mythtv-debug",

       "Enable MythTV debug messages",

       FALSE, G_PARAM_READWRITE));

#endif

  gstbasesrc_class->start = gst_mythtv_src_start;

  gstbasesrc_class->stop = gst_mythtv_src_stop;

  gstbasesrc_class->get_size = gst_mythtv_src_get_size;

  gstbasesrc_class->is_seekable = gst_mythtv_src_is_seekable;

  //gstpushsrc_class->newsegment = gst_mythtv_src_new_segment;

  gstbasesrc_class->create = gst_mythtv_src_create;

  GST_DEBUG_CATEGORY_INIT (mythtvsrc_debug, "mythtvsrc", 0,

      "MythTV Client Source");

}

static void

gst_mythtv_src_init (GstMythtvSrc * this, GstMythtvSrcClass * g_class)

{

  this->file_transfer = NULL;

  this->unique_setup = FALSE;

  this->mythtv_version = GMYTHTV_VERSION_DEFAULT;

  this->bytes_read = 0;

  this->content_size = -1;

  this->read_offset = 0;

  this->live_tv = FALSE;

  this->user_agent = g_strdup ("mythtvsrc");

  this->mythtv_caps = NULL;  

  gst_base_src_set_format( GST_BASE_SRC( this ), GST_FORMAT_BYTES );  

  gst_base_src_set_live ( GST_BASE_SRC( this ), TRUE );

  //gst_pad_set_chain_function ( GST_BASE_SRC_PAD(GST_BASE_SRC(this)),

  //    gst_mythtv_src_chain );

  gst_pad_set_event_function ( GST_BASE_SRC_PAD(GST_BASE_SRC(this)),

      gst_mythtv_src_handle_event );

/*

  gst_pad_set_query_function ( GST_BASE_SRC_PAD(GST_BASE_SRC(this)),

      gst_mythtv_src_query );

*/

  //gst_pad_set_activatepull_function( GST_BASE_SRC_PAD(GST_BASE_SRC(this)),

  //    gst_mythtv_src_sink_activate_pull );

}

static void

gst_mythtv_src_finalize (GObject * gobject)

{

  GstMythtvSrc *this = GST_MYTHTV_SRC (gobject);

  g_free (this->user_agent);

  if (this->mythtv_caps) {

    gst_caps_unref (this->mythtv_caps);

    this->mythtv_caps = NULL;

  }

  if (this->file_transfer) {

    g_object_unref (this->file_transfer);

    this->file_transfer = NULL;

  }

  if (this->uri_name) {

    g_free (this->uri_name);

  }

  if (this->user_agent) {

    g_free (this->user_agent);

  }

  G_OBJECT_CLASS (parent_class)->finalize (gobject);

}

static gint

do_read_request_response (GstMythtvSrc * src, guint64 offset, guint size, GstBuffer **outbuf)

{

  gint read = 0;

  guint sizetoread = size; //GST_BUFFER_SIZE (outbuf);

  //GstBuffer *buffer = gst_buffer_new_and_alloc( size ); 

  g_print( "Starting: [%s] Reading %d bytes...\n", __FUNCTION__, sizetoread ); 

  /* Loop sending the Myth File Transfer request:

   * Retry whilst authentication fails and we supply it. */

  gint len = 0;

  GST_OBJECT_LOCK(src);

  while ( sizetoread > 0 ) {

  	len = gmyth_file_transfer_read( src->file_transfer,

			GST_BUFFER_DATA( *outbuf ) + read, sizetoread, TRUE );

    if ( len > 0 ) {

      read += len;      

      sizetoread -= len;

    } else if ( len <= 0 ) {

      if ( src->live_tv == FALSE ) {

		goto done;

      } else if ( len == 0 ) {

      	src->update_prog_chain = TRUE;

      	goto done;

      } else if ( /*src->content_size >= src->read_offset && 

      			abs ( src->content_size - src->read_offset ) <= 1024 ) ||*/

      			( src->content_size <= ( src->read_offset + size + GMYTHTV_TRANSFER_MAX_BUFFER ) ) )

  {

#if ENABLE_TIMING_POSITION == 1

    gint64 size_tmp = 0;

    if (src->live_tv == TRUE) {

get_file_pos:

      size_tmp = gmyth_file_transfer_get_file_position( src->file_transfer );

      if ( size_tmp > ( src->content_size + GMYTHTV_TRANSFER_MAX_BUFFER ) )

	src->content_size = size_tmp;

      else if ( size_tmp > 0 )

	goto get_file_pos;

      g_print( "\t[%s]\tGET_POSITION: file_position = %lld\n",

	  __FUNCTION__, size_tmp );

    }

#else

  	gint64 new_offset = gmyth_file_transfer_get_file_position( src->file_transfer );

	if ( new_offset > 0 ) {

	  if ( src->content_size < new_offset ) {

	    src->content_size = new_offset;

	  }

	} else {

	  src->update_prog_chain = TRUE;

	}

#endif

	goto done;

  }

    }

    if ( read == sizetoread )

      break;

  }

  if ( read > 0 ) {

  	src->read_offset += read;

    src->bytes_read += read;

  g_print( "[%s]\tBYTES READ (actual) = %d, BYTES READ (cumulative) = %llu, "\

      "OFFSET = %llu, CONTENT SIZE = %llu.\n", __FUNCTION__, read, src->bytes_read, 

      src->read_offset, src->content_size );

  //GST_BUFFER_TIMESTAMP( buffer ) = GST_BUFFER_TIMESTAMP ( *outbuf );

  GST_BUFFER_SIZE (*outbuf) = read; //GST_BUFFER_SIZE (buffer) = read;

  GST_BUFFER_OFFSET (*outbuf) = offset; //GST_BUFFER_OFFSET (buffer) = offset;

  GST_BUFFER_OFFSET_END (*outbuf) = offset + read;//GST_BUFFER_OFFSET_END (buffer) = offset + read;

  //memcpy( GST_BUFFER_DATA( *outbuf ), GST_BUFFER_DATA( buffer ), read );

  g_print( "Stopping: [%s]\t\tBUFFER --->SIZE = %d, OFFSET = %llu, "\

      "OFFSET_END = %llu.\n\n", __FUNCTION__, GST_BUFFER_SIZE (*outbuf), 

      GST_BUFFER_OFFSET (*outbuf), GST_BUFFER_OFFSET_END (*outbuf) );

  } else if ( !src->live_tv )

  	goto eos;

  goto done;

eos:

  src->eos = TRUE;

done:

  GST_OBJECT_UNLOCK(src);

  return read;

}

static GstFlowReturn

gst_mythtv_src_create ( GstBaseSrc * psrc, guint64 offset, guint size, GstBuffer **outbuf)

{

  GstMythtvSrc *src;

  GstFlowReturn ret = GST_FLOW_OK;

  gint read = -1;

  src = GST_MYTHTV_SRC (psrc);

  /* The caller should know the number of bytes and not read beyond EOS. */

  if (G_UNLIKELY (src->eos))

    goto eos;

  if ( G_UNLIKELY (src->update_prog_chain) )

    goto change_progchain;

  //GST_OBJECT_LOCK(src);

  if (G_UNLIKELY (src->read_offset != offset)) {

    gint64 new_offset = gmyth_file_transfer_seek(src->file_transfer, offset, SEEK_SET);

    g_print( "[%s] SRC Offset = %lld, NEW actual backend SEEK Offset = %lld.\n",

    	__FUNCTION__, src->read_offset, new_offset );

    if (G_UNLIKELY (new_offset < 0 ) )//|| new_offset != src->read_offset)) {

    {

      //GST_OBJECT_UNLOCK(src);

      goto change_progchain;

    }

    src->read_offset = offset;

  }

  //GST_OBJECT_UNLOCK(src);

  /* Create the buffer. */

  ret = gst_pad_alloc_buffer (GST_BASE_SRC_PAD (GST_BASE_SRC (psrc)),

      src->read_offset, size,

      //src->icy_caps ? src->icy_caps :

      GST_PAD_CAPS (GST_BASE_SRC_PAD (GST_BASE_SRC (psrc))), outbuf);

  if (G_UNLIKELY (ret != GST_FLOW_OK))

    goto done;

  //g_static_mutex_lock( &update_size_mutex );

  read = do_read_request_response ( src, src->read_offset, size, outbuf );

  //g_static_mutex_unlock( &update_size_mutex );  

  if (G_UNLIKELY (src->update_prog_chain) )

    goto change_progchain;

  if (G_UNLIKELY (read < 0) || *outbuf == NULL) {

  	//if ( src->live_tv )

    //	goto done;

    //else

    goto read_error;

  }

/* 

  if (G_UNLIKELY(src->eos))

    goto eos;

  else

    goto done;

*/

done:

 {

    const gchar *reason = gst_flow_get_name (ret);

    GST_DEBUG_OBJECT (src, "DONE task, reason %s", reason);

  	return ret;

 }

eos:

  {

    const gchar *reason = gst_flow_get_name (ret);

    GST_DEBUG_OBJECT (src, "pausing task, reason %s", reason);

    return GST_FLOW_UNEXPECTED;

  }

  /* ERRORS */

read_error:

  {

    GST_ELEMENT_ERROR (src, RESOURCE, READ,

	(NULL), ("Could not read any bytes (%i, %s)", read,

	  src->uri_name));

    return GST_FLOW_ERROR;

  }

change_progchain:

  {

    GST_ELEMENT_ERROR (src, RESOURCE, READ,

	(NULL), ("Seek failed, go to the next program info... (%i, %s)", read,

	  src->uri_name));

	// go to the next program chain

	src->unique_setup = FALSE;

	gst_mythtv_src_start( psrc );

    return GST_FLOW_OK;

  }

}

void

update_size_func( void *mythtv_data ) 

{

  GstMythtvSrc *src;

  g_return_if_fail( mythtv_data != NULL );

  src = GST_MYTHTV_SRC ( mythtv_data );

  g_static_mutex_lock( &update_size_mutex );

  if ( src->do_start ) {

#if ENABLE_TIMING_POSITION == 1

    gint64 size_tmp = 0;

    if (src->live_tv == TRUE) {

get_file_pos:

      g_usleep( 50 );

      size_tmp = gmyth_file_transfer_get_file_position( src->file_transfer );

      if ( size_tmp > ( src->content_size + GMYTHTV_TRANSFER_MAX_BUFFER ) )

	src->content_size = size_tmp;

      else if ( size_tmp > 0 )

	goto get_file_pos;

      g_print( "\t[%s]\tGET_POSITION: file_position = %lld\n",

	  __FUNCTION__, size_tmp );

    }

#endif

  }

  g_static_mutex_unlock( &update_size_mutex );

}

guint64

gst_mythtv_src_get_position ( GstMythtvSrc* src ) 

{

  if ( src->do_start ) {

#if ENABLE_TIMING_POSITION == 1

    gint64 size_tmp = 0;

    if (src->live_tv == TRUE) {

get_file_pos:

      g_usleep( 50 );

      size_tmp = gmyth_file_transfer_get_file_position( src->file_transfer );

      if ( size_tmp > ( src->content_size + GMYTHTV_TRANSFER_MAX_BUFFER ) )

		src->content_size = size_tmp;

      else if ( size_tmp > 0 )

	goto get_file_pos;

      g_print( "\t[%s]\tGET_POSITION: file_position = %lld\n",

	  __FUNCTION__, size_tmp );

    }

#endif

  }

  return src->content_size;	

}

/* create a socket for connecting to remote server */

static gboolean

gst_mythtv_src_start ( GstBaseSrc * bsrc )

{

  GstMythtvSrc *src = GST_MYTHTV_SRC (bsrc);

  GString *chain_id_local = NULL;

  gboolean ret = TRUE;

  if (src->unique_setup == FALSE) {

    src->unique_setup = TRUE;

  } else {

    goto done;

  }

  GST_OBJECT_LOCK(src);

  if ( src->live_tv ) {

    src->spawn_livetv = gmyth_livetv_new( );

    if ( gmyth_livetv_setup( src->spawn_livetv ) == FALSE ) {

      ret = FALSE;

      goto init_failed;

    }

    /* set up the uri variable */

    src->uri_name = g_strdup( src->spawn_livetv->proginfo->pathname->str );

    chain_id_local = gmyth_tvchain_get_id( src->spawn_livetv->tvchain );

    if ( chain_id_local != NULL ) {

      src->live_chain_id = g_strdup( chain_id_local->str );

      g_print( "\t[%s]\tLocal chain ID = %s.\n", __FUNCTION__, src->live_chain_id );

    }

    src->live_tv_id = src->spawn_livetv->remote_encoder->recorder_num;

    g_print ( "[%s] LiveTV id = %d, URI path = %s.\n", __FUNCTION__, src->live_tv_id, src->uri_name );

  }

  src->file_transfer = gmyth_file_transfer_new( src->live_tv_id, 

      g_string_new( src->uri_name ), -1, src->mythtv_version );

  if ( src->file_transfer == NULL ) {

    GST_OBJECT_UNLOCK(src);

    goto init_failed;

  }

  /* sets the Playback monitor connection */

  ret = gmyth_file_transfer_playback_setup( &(src->file_transfer), src->live_tv );

  if ( src->live_tv == TRUE && ret == TRUE ) {

    /* loop finished, set the max tries variable to zero again... */

    wait_to_transfer = 0;

    while ( wait_to_transfer++ < GMYTHTV_TRANSFER_MAX_WAITS && ( gmyth_file_transfer_is_recording( src->file_transfer ) == FALSE 

	  /*|| ( gmyth_file_transfer_get_file_position( src->file_transfer ) < ( src->content_size + 327680 ) )*/ ) )

      g_usleep( 100 );

  }

  /* sets the FileTransfer instance connection (video/audio download) */

  ret = gmyth_file_transfer_setup( &(src->file_transfer), src->live_tv );

  if ( ret == FALSE ) {

    GST_OBJECT_UNLOCK(src);

#ifndef GST_DISABLE_GST_DEBUG  

    if ( src->mythtv_msgs_dbg )

      g_printerr( "MythTV FileTransfer request failed when setting up socket connection!\n" );  	  

#endif

    goto begin_req_failed;

  }

  src->content_size = src->file_transfer->filesize;

  GST_OBJECT_UNLOCK(src);

  if ( src->live_tv ) {

    //GError* error;

    //update_size_task = g_thread_create( (GThreadFunc)update_size_func, src, FALSE, &error );

    g_print( "[%s] Update Size task = %s\n", __FUNCTION__, update_size_task != NULL ?  "OK !" : "ERROR!!!" );

  }

  src->do_start = TRUE;  

done:

  return TRUE;

  /* ERRORS */

init_failed:

  {

    if (src->spawn_livetv != NULL )

      g_object_unref( src->spawn_livetv );

    GST_ELEMENT_ERROR (src, LIBRARY, INIT,

	(NULL), ("Could not initialize MythTV library (%i, %s)", ret, src->uri_name));

    return FALSE;

  }

begin_req_failed:

  {

    GST_ELEMENT_ERROR (src, LIBRARY, INIT,

	(NULL), ("Could not begin request sent to MythTV server (%i, %s)", ret, src->uri_name));

    return FALSE;

  }

}

#if 0

/* handles queries for location in the stream in the requested format */

static gboolean

gst_mythtv_src_query ( GstPad * pad, GstQuery * query )

{

  gboolean res = TRUE;

  GstMythtvSrc *mythtv;

  guint64 size = 0;

  mythtv = GST_GMYTHTV_SRC( GST_PAD_PARENT (pad) );

  size = gst_mythtv_src_get_position (mythtv);

  switch (GST_QUERY_TYPE (query)) {

    case GST_QUERY_POSITION:

      {

	//GstFormat format;

	gint64 cur = 0;

	/* save requested format */

	gst_query_parse_position (query, NULL, &cur);

	/* query peer for current position in time */

	g_print( "[%s] Actual size is %s than current size from sink. [ %lld, %lld ]\n", __FUNCTION__, 

	    ( size > cur ) ? "greater" : "lower", size, cur );

	gst_query_set_position (query, GST_FORMAT_BYTES, size);

	if ( size < cur )

	  goto error;

	break;

      }

     #if 0

    case GST_QUERY_DURATION:

      {

	//GstFormat format;

	gint64 cur = 0;

	/* save requested format */

	gst_query_parse_position (query, NULL, &cur);

	/* query peer for current position in time */

	g_print( "[%s] Actual size is %s than current size from sink. [ %lld, %lld ]\n", __FUNCTION__, 

	    ( size * GST_SECOND > cur * GST_SECOND ) ? "greater" : "lower", size * GST_SECOND, 

	    cur * GST_SECOND );

	gst_query_set_position (query, GST_FORMAT_TIME, size * GST_SECOND );

	if ( size * GST_SECOND < cur * GST_SECOND )

	  goto error;

	break;

      }

	#endif

    default:

      res = FALSE;

      break;

  }

  return res;

error:

  return FALSE;

}

#endif

static gboolean

gst_mythtv_src_get_size (GstBaseSrc * bsrc, guint64 * size)

{

  GstMythtvSrc *src = GST_MYTHTV_SRC (bsrc);

  gboolean ret = TRUE;

  if (src->content_size <= 0) {

    ret= FALSE;

  } else if ( abs ( src->content_size - src->read_offset ) <= GMYTHTV_TRANSFER_MAX_BUFFER ) {

    //g_static_mutex_lock( &update_size_mutex );

    GST_OBJECT_LOCK(src);

  	gint64 new_offset = gmyth_file_transfer_get_file_position( src->file_transfer );

	if ( new_offset > 0 ) {

	  if ( src->content_size < new_offset ) {

	    src->content_size = new_offset;

	  }

	} else {

	  src->update_prog_chain = TRUE;

	  src->content_size = 0;

	}

#if ENABLE_TIMING_POSITION == 1

    gint64 size_tmp = 0;

    if (src->live_tv == TRUE) {

get_file_pos:

      g_usleep( 5 );

      size_tmp = gmyth_file_transfer_get_file_position( src->file_transfer );

      if ( size_tmp > ( src->content_size + GMYTHTV_TRANSFER_MAX_BUFFER ) )

	src->content_size = size_tmp;

      else if ( size_tmp > 0  )

	goto get_file_pos;

      g_print( "\t[%s]\tGET_POSITION: file_position = %lld\n",

	  __FUNCTION__, size_tmp );

    }

#endif

	GST_OBJECT_UNLOCK(src);

    //g_static_mutex_unlock( &update_size_mutex );

  }

  *size = src->content_size;

  g_print( "[%s] Content size = %llu\n", __FUNCTION__, src->content_size );

  return ret;

}

/* close the socket and associated resources

 * used both to recover from errors and go to NULL state */

  static gboolean

gst_mythtv_src_stop (GstBaseSrc * bsrc)

{

  GstMythtvSrc *src;

  src = GST_MYTHTV_SRC (bsrc);

  if (src->uri_name) {

    g_free (src->uri_name);

    src->uri_name = NULL;

  }

  if (src->mythtv_caps) {

    gst_caps_unref (src->mythtv_caps);

    src->mythtv_caps = NULL;

  }

  src->eos = FALSE;

  return TRUE;

}

static gboolean

gst_mythtv_src_handle_event (GstPad * pad, GstEvent * event)

{

  GstMythtvSrc *src = GST_MYTHTV_SRC (GST_PAD_PARENT (pad));

  switch (GST_EVENT_TYPE (event)) {

#if 0

    case GST_EVENT_FLUSH_START:

      src->eos = FALSE;

      break;

      //return TRUE;

    case GST_EVENT_FLUSH_STOP:

      src->do_start = TRUE;

      src->eos = FALSE;

      gst_element_set_state (GST_ELEMENT(src), GST_STATE_NULL);

      //gst_element_set_locked_state (GST_ELEMENT(src), TRUE);

      break;

#endif

    case GST_EVENT_EOS:

      g_print( "[%s] Got EOS event!!!\n", __FUNCTION__ );

	  guint64 cont_size = gst_mythtv_src_get_position (src);

	  if ( !src->live_tv ) {

		  if ( cont_size > src->content_size ) {

		  	src->content_size = cont_size;

		  	src->eos = FALSE;

	  	  } else {

	      	src->eos = TRUE;

	      	gst_element_set_state ( GST_ELEMENT (src), GST_STATE_NULL );

	      	gst_element_set_locked_state ( GST_ELEMENT (src), FALSE );

	  	  }

	  } else 

	  	src->eos = TRUE;

      break;

#if 0

    case GST_EVENT_NEWSEGMENT:

      g_print( "[%s] Got NEWSEGMENT!!!\n", __FUNCTION__ );

      src->eos = FALSE;

      break;

    case GST_EVENT_SEEK:  	  

      {

	g_print( "[%s] Got EVENT_SEEK!!!\n", __FUNCTION__ );

	gdouble rate;

	//gboolean update = TRUE;

	GstFormat format;

	GstSeekType cur_type, stop_type;

	GstSeekFlags flags;

	gint64 cur = 0, stop = 0;

	gst_event_parse_seek ( event, &rate, &format,

	    &flags, &cur_type, &cur,

	    &stop_type, &stop );

	g_print( "[%s] Got EVENT_SEEK.\n", __FUNCTION__ );

	if ( !( flags & GST_SEEK_FLAG_FLUSH ) ) {

	  g_print( "[%s] Could get the FLAG_FLUSH message.\n", __FUNCTION__ );

	}

	//gboolean ret = gst_event_parse_new_segment ( event,

	//    &update, &rate, &format, &start, &stop,

	//    &position );

	//GstFlowReturn flow_ret = gst_mythtv_src_create (GST_BASE_SRC( GST_PAD_PARENT( psrc ) ), 

	//			cur, stop - cur + 1, GstBuffer)

      }

#endif

    default:

      return gst_pad_event_default (pad, event);

  }

  return gst_pad_event_default (pad, event);

}

static gboolean

gst_mythtv_src_is_seekable( GstBaseSrc *push_src )

{

  return TRUE;

}

static void

gst_mythtv_src_set_property (GObject * object, guint prop_id,

    const GValue * value, GParamSpec * pspec)

{

  GstMythtvSrc *mythtvsrc = GST_MYTHTV_SRC (object);

  GST_OBJECT_LOCK (mythtvsrc);

  switch (prop_id) {

    case PROP_URI:

    case PROP_LOCATION:

      {

	if (!g_value_get_string (value)) {

	  GST_WARNING ("location property cannot be NULL");

	  goto done;

	}

	if (mythtvsrc->uri_name != NULL) {

	  g_free (mythtvsrc->uri_name);

	  mythtvsrc->uri_name = NULL;

	}

	mythtvsrc->uri_name = g_value_dup_string (value);

	break;

      }

#ifndef GST_DISABLE_GST_DEBUG

    case PROP_GMYTHTV_DBG:

      {

	mythtvsrc->mythtv_msgs_dbg = g_value_get_boolean (value);

	break;

      }

#endif

    case PROP_GMYTHTV_VERSION:

      {

	mythtvsrc->mythtv_version = g_value_get_int (value);

	break;

      }

    case PROP_GMYTHTV_LIVEID:

      {

	mythtvsrc->live_tv_id = g_value_get_int (value);

	break;

      }

    case PROP_GMYTHTV_LIVE:

      {

	mythtvsrc->live_tv = g_value_get_boolean (value);

	break;

      }

    case PROP_GMYTHTV_LIVE_CHAINID:

      {

	if (!g_value_get_string (value)) {

	  GST_WARNING ("MythTV Live chainid property cannot be NULL");

	  goto done;

	}

	if (mythtvsrc->live_chain_id != NULL) {

	  g_free (mythtvsrc->live_chain_id);

	  mythtvsrc->live_chain_id = NULL;

	}

	mythtvsrc->live_chain_id = g_value_dup_string (value);

	break;

      }

    default:

      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);

      break;

  }

  GST_OBJECT_UNLOCK (mythtvsrc);

done:

  return;

}

  static void

gst_mythtv_src_get_property (GObject * object, guint prop_id,

    GValue * value, GParamSpec * pspec)

{

  GstMythtvSrc *mythtvsrc = GST_MYTHTV_SRC (object);

  GST_OBJECT_LOCK (mythtvsrc);

  switch (prop_id) {

    case PROP_URI:

    case PROP_LOCATION:

      {

	gchar *str = g_strdup( "" );

	if ( mythtvsrc->uri_name == NULL ) {

	  g_free (mythtvsrc->uri_name);

	  mythtvsrc->uri_name = NULL;

	} else {

	  str = g_strdup( mythtvsrc->uri_name );

	}

	g_value_set_string ( value, str );

	break;

      }

#ifndef GST_DISABLE_GST_DEBUG

    case PROP_GMYTHTV_DBG:

      g_value_set_boolean ( value, mythtvsrc->mythtv_msgs_dbg );

      break;

#endif

    case PROP_GMYTHTV_VERSION:

      {

	g_value_set_int ( value, mythtvsrc->mythtv_version );

	break;

      }

    case PROP_GMYTHTV_LIVEID:

      {

	g_value_set_int ( value, mythtvsrc->live_tv_id );

	break;

      }

    case PROP_GMYTHTV_LIVE:

      g_value_set_boolean ( value, mythtvsrc->live_tv );

      break;

    case PROP_GMYTHTV_LIVE_CHAINID:

      {

	gchar *str = g_strdup( "" );

	if ( mythtvsrc->live_chain_id == NULL ) {

	  g_free (mythtvsrc->live_chain_id);

	  mythtvsrc->live_chain_id = NULL;

	} else {

	  str = g_strdup( mythtvsrc->live_chain_id );

	}

	g_value_set_string ( value, str );

	break;

      }

    default:

      G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);

      break;

  }

  GST_OBJECT_UNLOCK (mythtvsrc);

}

/* entry point to initialize the plug-in

 * initialize the plug-in itself

 * register the element factories and pad templates

 * register the features

 */

static gboolean

plugin_init (GstPlugin * plugin)

{

  return gst_element_register (plugin, "mythtvsrc", GST_RANK_NONE,

      GST_TYPE_MYTHTV_SRC);

}

/* this is the structure that gst-register looks for

 * so keep the name plugin_desc, or you cannot get your plug-in registered */

GST_PLUGIN_DEFINE (GST_VERSION_MAJOR,

    GST_VERSION_MINOR,

    "mythtv",

    "lib MythTV src",

    plugin_init, VERSION, "LGPL", "GStreamer", "http://gstreamer.net/")