/* 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 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 details.

 *

 * You should have received a copy of the GNU Library General Public

 * License along with this library; if not, write to the

 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,

 * Boston, MA 02111-1307, USA.

 */

/**

 * SECTION:element-mythtvsrc

 *

 * <refsect2>

 * <para>

 * MythTVSrc allows to access a remote MythTV backend streaming Video/Audio server,

 * and to render audio and video content through a TCP/IP connection to a specific

 * port on this server, and based on a known MythTV protocol that is based on 

 * some message passing, such as REQUEST_BLOCK on a specified number of bytes, to get

 * some chunk of remote file data.

 * You should pass the information aboute the remote MythTV backend server 

 * through the <link linkend="GstMythTVSrc--location">location</link> property.

 * </para>

 * <title>Examples</title>

 * <para>

 * If you want to get the LiveTV content (set channel, TV tuner, RemoteEncoder, 

 * Recorder),

 * put the following URI:

 * 

 * <programlisting> 

 *  myth://xxx.xxx.xxx.xxx:6543/livetv?channel=BBC

 * </programlisting>

 *

 * This URI will say to the gmyth library to configure the Recorder instance (used to

 * change the channel, start the TV multimedia content transmition, etc.), using

 * the IP address (xxx.xxx.xxx.xxx) and port number (6543) of the MythTV backend 

 * server, and setting the channel name to "BBC". 

 * 

 * To get a already recorded the MythTV NUV file, put the following URI:

 * 

 * <programlisting>

 *  myth://xxx.xxx.xxx.xxx:6543/filename.nuv

 * </programlisting>

 *

 * This URI will say to the gmyth library to configure the Recorder instance (used to

 * change the channel, start the TV multimedia content transmition, etc.), using

 * the IP address (xxx.xxx.xxx.xxx) and port number (6543) of the MythTV backend 

 * server, and setting the channel name to "BBC".

 * 

 * Another possible way to use the LiveTV content, and just in the case you want to 

 * use the mysql database, put the location URI in the following format:

 * 

 * <programlisting> 

 *  myth://mythtv:mythtv@xxx.xxx.xxx.xxx:6543/?mythconverg&channel=9

 * </programlisting>

 * 

 * Where the first field is the protocol (myth), the second and third are user 

 * name (mythtv) and password (mythtv), then backend host name and port number, 

 * and the last field is the database name (mythconverg).

 * </para>

 * </refsect2>

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include "gstmythtvsrc.h"

#include <gmyth/gmyth_file.h>

#include <gmyth/gmyth_file_transfer.h>

#include <gmyth/gmyth_file_local.h>

#include <gmyth/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 GST_GMYTHTV_CHANNEL_DEFAULT_NUM     (-1)

#define GMYTHTV_VERSION_DEFAULT			        30

#define GMYTHTV_TRANSFER_MAX_WAITS          100

#define GMYTHTV_TRANSFER_MAX_RESENDS        2

#define GMYTHTV_TRANSFER_MAX_BUFFER         (128*1024)

#define MAX_READ_SIZE                       (4*1024)

#define GST_FLOW_ERROR_NO_DATA              (-101)

#define REQUEST_MAX_SIZE                    (64*1024)

#define INTERNAL_BUFFER_SIZE                (90*1024)

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"));

enum

{

  PROP_0,

  PROP_LOCATION,

#ifndef GST_DISABLE_GST_DEBUG

  PROP_GMYTHTV_DBG,

#endif

  PROP_GMYTHTV_VERSION,

  PROP_GMYTHTV_LIVE,

  PROP_GMYTHTV_LIVEID,

  PROP_GMYTHTV_LIVE_CHAINID,

  PROP_GMYTHTV_ENABLE_TIMING_POSITION,

  PROP_GMYTHTV_CHANNEL_NUM,

  PROP_GMYTHTV_MAX_TRY

};

static void gst_mythtv_src_clear (GstMythtvSrc *mythtv_src);

static void gst_mythtv_src_finalize (GObject * gobject);

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_src_do_seek (GstBaseSrc * base,

    GstSegment * segment);

static GstStateChangeReturn

gst_mythtv_src_change_state (GstElement * element, GstStateChange transition);

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 void gst_mythtv_src_uri_handler_init (gpointer g_iface,

    gpointer iface_data);

static gboolean gst_mythtv_src_handle_query (GstPad * pad, GstQuery * query);

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

static GMythFileReadResult do_read_request_response (GstMythtvSrc * src, guint size,

    GByteArray * data_ptr);

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, 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);

  element_class->change_state = gst_mythtv_src_change_state;

}

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_GMYTHTV_VERSION,

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

          "Change MythTV 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 MythTV 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 MythTV 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));

  g_object_class_install_property

      (gobject_class, PROP_GMYTHTV_ENABLE_TIMING_POSITION,

      g_param_spec_boolean ("mythtv-enable-timing-position",

          "mythtv-enable-timing-position",

          "Enable MythTV Live TV content size continuous updating", FALSE,

          G_PARAM_READWRITE));

  g_object_class_install_property

      (gobject_class, PROP_GMYTHTV_CHANNEL_NUM,

      g_param_spec_string ("mythtv-channel", "mythtv-channel",

          "Change MythTV channel number",

          "", G_PARAM_READWRITE));

  g_object_class_install_property

      (gobject_class, PROP_GMYTHTV_MAX_TRY,

      g_param_spec_int ("max-try", "max-try",

          "Set the max try for get MythTV free recorder",

          0, G_MAXINT, 10,  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;

  gstbasesrc_class->do_seek = gst_mythtv_src_do_seek;

  gstpushsrc_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 = NULL;

  this->unique_setup = FALSE;

  this->mythtv_version = GMYTHTV_VERSION_DEFAULT;

  this->state = GST_MYTHTV_SRC_FILE_TRANSFER;

  this->bytes_read = 0;

  this->prev_content_size = 0;

  this->content_size = 0;

  this->read_offset = 0;

  this->content_size_last = 0;

  this->live_tv = FALSE;

  this->enable_timing_position = FALSE;

  this->update_prog_chain = FALSE;

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

  this->update_prog_chain = FALSE;

  this->channel_name = NULL;

  this->eos = FALSE;

  this->bytes_queue = NULL;

  this->wait_to_transfer = 0;

  gst_base_src_set_format (GST_BASE_SRC (this), GST_FORMAT_BYTES);

  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_handle_query);

}

static void

gst_mythtv_src_clear (GstMythtvSrc *mythtv_src)

{

  mythtv_src->unique_setup = FALSE;

  if (mythtv_src->spawn_livetv) {

    g_object_unref (mythtv_src->spawn_livetv);

    mythtv_src->spawn_livetv = NULL;

  }

  if (mythtv_src->file) {

    g_object_unref (mythtv_src->file);

    mythtv_src->file = NULL;

  }

  if (mythtv_src->backend_info) {

    g_object_unref (mythtv_src->backend_info);

    mythtv_src->backend_info = NULL;

  }

  if (mythtv_src->bytes_queue) {

    g_byte_array_free (mythtv_src->bytes_queue, TRUE);

    mythtv_src->bytes_queue = NULL;

  }

}

static void

gst_mythtv_src_finalize (GObject * gobject)

{

  GstMythtvSrc *this = GST_MYTHTV_SRC (gobject);

  gst_mythtv_src_clear (this);

  if (this->uri_name) {

    g_free (this->uri_name);

    this->uri_name = NULL;

  }

  if (this->user_agent) {

    g_free (this->user_agent);

    this->user_agent = NULL;

  }

  G_OBJECT_CLASS (parent_class)->finalize (gobject);

}

static GMythFileReadResult

do_read_request_response (GstMythtvSrc * src, guint size, GByteArray *data_ptr)

{

  gint read = 0;

  guint sizetoread = size;

  gint max_iters = GMYTHTV_TRANSFER_MAX_RESENDS;

  GMythFileReadResult result;

  GST_LOG_OBJECT (src, "Starting: Reading %d bytes...", sizetoread);

  /* Loop sending the Myth File Transfer request:

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

  while (sizetoread == size && --max_iters > 0) {

    /* if ( gmyth_backend_info_is_local_file(src->backend_info) ) */

    if ( IS_GMYTH_FILE_LOCAL(src->file) )

      result = gmyth_file_local_read ( GMYTH_FILE_LOCAL(src->file),

          data_ptr, sizetoread, src->live_tv);      

    else if ( IS_GMYTH_FILE_TRANSFER(src->file) )

      result = gmyth_file_transfer_read ( GMYTH_FILE_TRANSFER(src->file),

          data_ptr, sizetoread, src->live_tv );

    if (data_ptr->len > 0) {

      read += data_ptr->len;

      sizetoread -= data_ptr->len;

    } else if (data_ptr->len <= 0) {

      if (src->live_tv == FALSE) {

        result = GMYTH_FILE_READ_EOF;

        goto eos;

      } else {

        if (result == GMYTH_FILE_READ_ERROR) {  /* -314 */

          GST_INFO_OBJECT (src, "[LiveTV] FileTransfer READ_ERROR!");

          goto done;

        } else if (result == GMYTH_FILE_READ_NEXT_PROG_CHAIN) {      /* -315 */

          GST_INFO_OBJECT (src,

              "[LiveTV] FileTransfer - Go to athe next program chain!");

	  src->update_prog_chain = TRUE;

          continue;

        }

        goto done;

      }

    } /* else if (data_ptr->len == 0)

      goto done;*/

    if (read == sizetoread)

      goto done;

  }

  if ((read < 0 && !src->live_tv) || max_iters == 0) {

    result = GMYTH_FILE_READ_EOF;

    goto eos;

  }

  goto done;

eos:

  src->eos = TRUE;

done:

  return result;

}

static GstFlowReturn

gst_mythtv_src_create (GstPushSrc * psrc, GstBuffer ** outbuf)

{

  GstMythtvSrc *src;

  GstFlowReturn ret = GST_FLOW_OK;

  guint buffer_size_inter = 0;

  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_DEBUG_OBJECT (src, "offset = %" G_GUINT64_FORMAT ", size = %d...",

      src->read_offset, MAX_READ_SIZE);

  GST_DEBUG_OBJECT (src, "Create: buffer_remain: %d, buffer_size = %d.",

      (gint) src->buffer_remain, src->bytes_queue->len);

program_chain_changed:

  /* just get from the byte array, no network effort... */

  if ((src->buffer_remain = src->bytes_queue->len) < MAX_READ_SIZE) {

  	GByteArray *buffer;

    GMythFileReadResult result = GMYTH_FILE_READ_OK;

    buffer = NULL;

    buffer_size_inter = (INTERNAL_BUFFER_SIZE - src->buffer_remain);

    if (buffer_size_inter > REQUEST_MAX_SIZE)

      buffer_size_inter = REQUEST_MAX_SIZE;

    buffer = g_byte_array_new ();

    result = do_read_request_response (src, buffer_size_inter, buffer);

    /* got the next program info? */

    if (G_UNLIKELY (src->update_prog_chain)) {

      if (result == GMYTH_FILE_READ_NEXT_PROG_CHAIN) {

         if  (buffer != NULL) {

           g_byte_array_free (buffer, TRUE);

           buffer = NULL;

         }

         goto program_chain_changed; 

      } else if (result == GMYTH_FILE_READ_OK) {

        /* remove wasteful, NUV file header data */

        buffer = g_byte_array_remove_range( buffer, 0, 512 );

	/* TODO: need to send a new segment event to NUVDemux? */

	gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC (psrc)),

			gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_BYTES, 0, -1, 0));

        /* goto change_progchain; */

      }

    }

    if (G_UNLIKELY (buffer->len < 0)) {

    	if (buffer != NULL) {

	      g_byte_array_free (buffer, TRUE);

	      buffer = NULL;

	    }

      if (src->live_tv || ( result == GMYTH_FILE_READ_NEXT_PROG_CHAIN ))

        goto change_progchain;

      else

        goto read_error;

    } else if (G_UNLIKELY (read == 0)) {

      if (buffer != NULL) {

	  g_byte_array_free (buffer, TRUE);

	  buffer = NULL;

      }

      if (!src->live_tv)

        goto done;

      else

        goto program_chain_changed;

    }

    src->bytes_queue =

        g_byte_array_append (src->bytes_queue, buffer->data, buffer->len);

    if (buffer->len > buffer_size_inter)

      GST_WARNING_OBJECT (src,

          "INCREASED buffer size! Backend sent more than we ask him... (%d)",

          abs (buffer->len - buffer_size_inter));

    src->buffer_remain += buffer->len;

    if (buffer != NULL) {

      g_byte_array_free (buffer, TRUE);

      buffer = NULL;

    }

	/*

    GST_DEBUG_OBJECT (src,

        "BYTES READ (actual) = %d, BYTES READ (cumulative) = %llu, "

        "OFFSET = %llu, CONTENT SIZE = %llu.", read,

        src->bytes_read, src->read_offset, src->content_size);

	*/		

  }

  guint buffer_size =

      (src->buffer_remain < MAX_READ_SIZE) ? src->buffer_remain : MAX_READ_SIZE;

  *outbuf = gst_buffer_new ();

  /*

  GST_DEBUG_OBJECT (src, "read from network? %s!, buffer_remain = %d",

      (buffer_size_inter ==

          0) ? "NO, got from buffer" : "YES, go see the backend's log file",

      src->buffer_remain);

	  */

  GST_BUFFER_SIZE (*outbuf) = buffer_size;

  GST_BUFFER_MALLOCDATA (*outbuf) = g_malloc0 (GST_BUFFER_SIZE (*outbuf));

  GST_BUFFER_DATA (*outbuf) = GST_BUFFER_MALLOCDATA (*outbuf);

  g_memmove (GST_BUFFER_DATA ((*outbuf)), src->bytes_queue->data,

      GST_BUFFER_SIZE (*outbuf));

  GST_BUFFER_OFFSET (*outbuf) = src->read_offset;

  GST_BUFFER_OFFSET_END (*outbuf) =

      src->read_offset + GST_BUFFER_SIZE (*outbuf);

  src->buffer_remain -= GST_BUFFER_SIZE (*outbuf);

  src->read_offset += GST_BUFFER_SIZE (*outbuf);

  src->bytes_read += GST_BUFFER_SIZE (*outbuf);

  //GST_DEBUG_OBJECT (src, "Buffer output with size: %d",

  //    GST_BUFFER_SIZE (*outbuf));

  /* flushs the newly buffer got from byte array */

  src->bytes_queue =

      g_byte_array_remove_range (src->bytes_queue, 0, buffer_size);

  if ( G_UNLIKELY (src->eos) || ( !src->live_tv

          && ( src->bytes_read >= src->content_size ) ) )

    goto eos;

done:

  {

    const gchar *reason = gst_flow_get_name (ret);

    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));

/* TODO: need to send a new segment event to NUVDemux? */

    gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC (psrc)),

        gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0, -1, 0));

    goto program_chain_changed;

  }

}

gint64

gst_mythtv_src_get_position (GstMythtvSrc * src)

{

  gint64 size_tmp = 0;

  guint max_tries = 2;

  if (src->live_tv == TRUE && (abs (src->content_size - src->bytes_read) <

          GMYTHTV_TRANSFER_MAX_BUFFER)) {

  get_file_pos:

    g_usleep (10);

    size_tmp = gmyth_recorder_get_file_position (src->spawn_livetv->recorder);

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

      src->content_size = size_tmp;

    else if (size_tmp > 0 && --max_tries > 0)

      goto get_file_pos;

    GST_LOG_OBJECT (src, "GET_POSITION: file_position = %lld", size_tmp);

    /* sets the last content size amount before it can be updated */

    src->prev_content_size = src->content_size;

  }

  return src->content_size;

}

static gboolean

gst_mythtv_src_do_seek (GstBaseSrc * base, GstSegment * segment)

{

  GstMythtvSrc *src = GST_MYTHTV_SRC (base);

  gint64 new_offset = -1;

  gint64 actual_seek = segment->start;

  gboolean ret = TRUE;

  GST_LOG_OBJECT (src, "seek, segment: %" GST_SEGMENT_FORMAT, segment);

  if (segment->format == GST_FORMAT_TIME) {

    goto done;

  }

  GST_LOG_OBJECT (src,

      "Trying to seek at the value (actual_seek = %lld, read_offset = %lld)",

      actual_seek, src->read_offset);

  /* verify if it needs to seek */

  if (src->read_offset != actual_seek) {

    /* if ( gmyth_backend_info_is_local_file(src->backend_info) ) */

    if ( IS_GMYTH_FILE_LOCAL(src->file) )

      new_offset =

          gmyth_file_local_seek ( GMYTH_FILE_LOCAL(src->file), segment->start, G_SEEK_SET);

    else if ( IS_GMYTH_FILE_TRANSFER(src->file) )

      new_offset =

          gmyth_file_transfer_seek ( GMYTH_FILE_TRANSFER(src->file), segment->start, SEEK_SET);

    GST_LOG_OBJECT (src,

        "Segment offset start = %lld, SRC Offset = %lld, NEW actual backend SEEK Offset = %lld.",

        segment->start, src->read_offset, new_offset);

    if (G_UNLIKELY (new_offset < 0)) {

      ret = FALSE;

      if (!src->live_tv)

        goto eos;

    }

    src->read_offset = new_offset;

    if (ret == FALSE) {

      GST_INFO_OBJECT (src, "Failed to set the SEEK on segment!");

    }

  }

done:

  return ret;

eos:

  {

    GST_DEBUG_OBJECT (src, "EOS found on seeking!!!");

    return FALSE;

  }

}

/* 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;  

  GMythURI *gmyth_uri = NULL;

  gboolean ret = TRUE;

  GstMessage *msg;

  if (src->unique_setup == FALSE) {

    src->unique_setup = TRUE;

  } else {

    goto done;

  }

  gmyth_uri = gmyth_uri_new_with_value (src->uri_name);

  src->backend_info = gmyth_backend_info_new_with_uri (src->uri_name);

  src->live_tv = gmyth_uri_is_livetv( gmyth_uri );

  /* testing UPnP... */

  /* gmyth_backend_info_set_hostname( src->backend_info, NULL ); */

  if ( src->live_tv ) {

    src->spawn_livetv = gmyth_livetv_new (src->backend_info);

    gchar* ch = gmyth_uri_get_channel_name( gmyth_uri );

    if ( ch != NULL )

    	src->channel_name = ch;

    if (src->channel_name != NULL) {

	  gboolean result;

	  result = gmyth_livetv_channel_name_setup (src->spawn_livetv, src->channel_name);

	  if (result == FALSE) {

	     GST_INFO_OBJECT (src, "LiveTV setup felt down on error");

	   	 ret = FALSE;

	  	 goto init_failed;

	  }

    } else {

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

        GST_INFO_OBJECT (src, "LiveTV setup felt down on error");

        ret = FALSE;

        goto init_failed;

      }

    }

    /* testing change channel... */

    /* gmyth_recorder_change_channel( src->spawn_livetv->recorder, CHANNEL_DIRECTION_UP ); */

    src->file = GMYTH_FILE( gmyth_livetv_create_file_transfer (src->spawn_livetv) );

    if (NULL == src->file) {

      GST_INFO_OBJECT (src, "[LiveTV] FileTransfer equals to NULL");

      ret = FALSE;

      goto init_failed;

    }

    /* Check if the file is local to this specific client renderer */

    if ( gmyth_uri_is_local_file(gmyth_uri) )

      ret = gmyth_file_local_open( GMYTH_FILE_LOCAL(src->file) );

    else

      ret = gmyth_file_transfer_open( GMYTH_FILE_TRANSFER(src->file), src->spawn_livetv->uri != NULL ? 

                gmyth_uri_get_path(src->spawn_livetv->uri) : 

                src->spawn_livetv->proginfo->pathname->str );

    /* sets the mythtvsrc "location" property */

    g_object_set (src, "location", gmyth_file_get_uri (src->file), NULL);

		if ( !ret )

		{

			GST_INFO_OBJECT (src, "Error: couldn't open the FileTransfer from LiveTV source!" );

			g_object_unref( src->file );

			src->file = NULL;

			goto init_failed;

		}

  } else {

    /* Check if the file is local to this specific client renderer, and tries to open

     * a local connection

     */

    if ( gmyth_uri_is_local_file(gmyth_uri) )

    {

      src->file = GMYTH_FILE(gmyth_file_local_new(src->backend_info));

      ret = gmyth_file_local_open ( GMYTH_FILE_LOCAL( src->file ) );

    } else {

      src->file = GMYTH_FILE(gmyth_file_transfer_new(src->backend_info));

      ret = gmyth_file_transfer_open ( GMYTH_FILE_TRANSFER(src->file), src->uri_name );

    }

  } /* if (else) - recorded FileTransfer */

  if (NULL == src->file) {

    GST_INFO_OBJECT (src, "FileTransfer is NULL");

    goto init_failed;

  }

  /*GST_INFO_OBJECT( src, "uri = %s", src->spawn_livetv->file); */

  if (ret == FALSE) {

#ifndef GST_DISABLE_GST_DEBUG

    if (src->mythtv_msgs_dbg)

      GST_INFO_OBJECT (src,

          "MythTV FileTransfer request failed when setting up socket connection!");

#endif

    goto begin_req_failed;

  }

  GST_INFO_OBJECT (src,

      "MythTV FileTransfer filesize = %lld, content_size = %lld!",

      gmyth_file_get_filesize( src->file ), src->content_size);

  src->content_size = gmyth_file_get_filesize (src->file);

  msg = gst_message_new_duration (GST_OBJECT (src), GST_FORMAT_BYTES, src->content_size);

  gst_element_post_message (GST_ELEMENT (src), msg);

  src->do_start = FALSE;

  /* this is used for the buffer cache */

  src->bytes_queue = g_byte_array_sized_new (INTERNAL_BUFFER_SIZE);

  src->buffer_remain = 0;

  gst_pad_push_event (GST_BASE_SRC_PAD (GST_BASE_SRC (src)),

      gst_event_new_new_segment (TRUE, 1.0, GST_FORMAT_TIME, 0,

          src->content_size, 0));

done:

	if (gmyth_uri != NULL)

	{

  	    g_object_unref (gmyth_uri);

      	gmyth_uri = NULL;

	}

    if (chain_id_local != NULL) {

        g_string_free (chain_id_local, TRUE);

        chain_id_local = NULL;

    }

    return TRUE;

    /* ERRORS */

init_failed:

	if (gmyth_uri != NULL)

	{

  	    g_object_unref (gmyth_uri);

      	gmyth_uri = NULL;

	}

    if (src->spawn_livetv != NULL) {

        g_object_unref (src->spawn_livetv);

        src->spawn_livetv = NULL;

    }

    GST_ELEMENT_ERROR (src, LIBRARY, INIT,

            (NULL), ("Could not initialize MythTV library (%i, %s)", ret,

            src->uri_name));

    gst_mythtv_src_clear (src);

    return FALSE;

begin_req_failed:

	if (gmyth_uri != NULL)

	{

  	    g_object_unref (gmyth_uri);

      	gmyth_uri = NULL;

	}

    GST_ELEMENT_ERROR (src, LIBRARY, INIT,

        (NULL), ("Could not begin request sent to MythTV server (%i, %s)", ret,

            src->uri_name));

    return FALSE;

}

static gboolean

gst_mythtv_src_get_size (GstBaseSrc * bsrc, guint64 * size)

{

  GstMythtvSrc *src = GST_MYTHTV_SRC (bsrc);

  gboolean ret = TRUE;

  GST_LOG_OBJECT (src, "Differs from previous content size: %d (max.: %d)",

      abs (src->content_size - src->prev_content_size),

      GMYTHTV_TRANSFER_MAX_BUFFER);

  if (src->live_tv) {

    ret = FALSE;

  } else if (src->live_tv && src->enable_timing_position

      && (abs (src->content_size - src->bytes_read) <

          GMYTHTV_TRANSFER_MAX_BUFFER)) {

    gint64 new_offset =

        gmyth_recorder_get_file_position (src->spawn_livetv->recorder);

    if (new_offset > 0 && new_offset > src->content_size) {

      src->content_size = new_offset;

    } else if (new_offset < src->content_size) {

      src->update_prog_chain = TRUE;

    }

  }

  *size = src->content_size;

  GST_LOG_OBJECT (src, "Content size = %lld", 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 = GST_MYTHTV_SRC (bsrc);

  gst_mythtv_src_clear (src);

  /* 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));

  gint64 cont_size = 0;

  gboolean ret = FALSE;

  switch (GST_EVENT_TYPE (event)) {

    case GST_EVENT_EOS:

      GST_WARNING_OBJECT (src, "Got EOS event");

      if (src->live_tv) {

        cont_size = gst_mythtv_src_get_position (src);

        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);

        }

      }

      break;

    default:

      ret = gst_pad_event_default (pad, event);

  }

  return ret;

}

static gboolean

gst_mythtv_src_is_seekable (GstBaseSrc * push_src)

{

  return TRUE;

}

static gboolean

gst_mythtv_src_handle_query (GstPad * pad, GstQuery * query)

{

  gboolean res = FALSE;

  GstMythtvSrc *myth = GST_MYTHTV_SRC (gst_pad_get_parent (pad));

  GstFormat formt;

  switch (GST_QUERY_TYPE (query)) {

    case GST_QUERY_POSITION:

    {

      gst_query_parse_position (query, &formt, NULL);

      if (formt == GST_FORMAT_BYTES) {

        gst_query_set_position (query, formt, myth->read_offset);

        GST_DEBUG_OBJECT (myth, "POS %" G_GINT64_FORMAT, myth->read_offset);

        res = TRUE;

      } else if (formt == GST_FORMAT_TIME) {

        res = gst_pad_query_default (pad, query);

      }

      break;

    }

    case GST_QUERY_DURATION:

    {

      gst_query_parse_duration (query, &formt, NULL);

      if (formt == GST_FORMAT_BYTES) {

		gint64 size = myth->content_size;			  

        gst_query_set_duration (query, GST_FORMAT_BYTES, 10);

        GST_DEBUG_OBJECT (myth, "SIZE %" G_GINT64_FORMAT, size);

        res = TRUE;

      } else if (formt == GST_FORMAT_TIME) {

        res = gst_pad_query_default (pad, query);

      }

      break;

    }

    default:

    {

      res = gst_pad_query_default (pad, query);

      break;

    }

  }

  gst_object_unref (myth);

  return res;