/*
 * @author Hallyson Melo <hallyson.melo@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
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#include <string.h>
#include <glib.h>
#include <glib/gprintf.h>
#include <glib/gstdio.h>
#include <math.h>

#include <libgnomevfs/gnome-vfs-module.h>
#include <libgnomevfs/gnome-vfs-utils.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_uri.h>
#include <gmyth/gmyth_recorder.h>
#include <gmyth/gmyth_backendinfo.h>
#include <gmyth/gmyth_util.h>
#include <gmyth/gmyth_remote_util.h>
#include <gmyth/gmyth_tvchain.h>
#include <gmyth/gmyth_programinfo.h>

#define GST_MYTHTV_ID_NUM               1
#define MYTHTV_VERSION_DEFAULT          30
#define MYTHTV_TRANSFER_MAX_WAITS       100

/*
 * internal GnomeVFS plug-in buffer size ( 120 Kbytes ) 
 */
#define MYTHTV_BUFFER_SIZE                          80*1024
/*
 * internally sized GnomeVFS plug-in buffer ( 4 Kbytes ) 
 */
#define MYTHTV_MAX_VFS_BUFFER_SIZE                  4096
/*
 * maximum number of bytes to be requested to the MythTV backend ( 64
 * Kbytes ) 
 */
#define MYTHTV_MAX_REQUEST_SIZE                     64*1024

typedef struct {
    GMythFile      *file;
    GMythLiveTV    *livetv;
    GMythBackendInfo *backend_info;
    GMythURI       *gmyth_uri;
    GMythRecorder  *live_recorder;
    gboolean        started;
    gint64          offset;

    gboolean        is_livetv;  /* it is, or not a Live TV content
                                 * transfer */
    gboolean        is_local_file;  /* tell if the file is local to the
                                     * current content transfer */

    gchar          *channel_name;

    gint            mythtv_version;
    gboolean        configured;
} MythtvHandle;


static GnomeVFSResult do_read(GnomeVFSMethod * method,
                              GnomeVFSMethodHandle * method_handle,
                              gpointer buffer,
                              GnomeVFSFileSize num_bytes,
                              GnomeVFSFileSize * bytes_read,
                              GnomeVFSContext * context);

static GnomeVFSResult myth_connection_start(MythtvHandle * method_handle);
static void     myth_destroy_handle(MythtvHandle * method_handle);
static GnomeVFSResult myth_handle_new(GnomeVFSURI * uri,
                                      MythtvHandle ** method_handle);
static GnomeVFSResult myth_get_file_info(MythtvHandle * myth_handle,
                                         GnomeVFSURI * uri,
                                         GnomeVFSFileInfo * info);

static          GnomeVFSResult
myth_handle_new(GnomeVFSURI * uri, MythtvHandle ** method_handle)
{
    gchar          *tmp_str1;
    gchar          *tmp_str2;

    _GNOME_VFS_METHOD_PARAM_CHECK(*method_handle == NULL);

    if (gnome_vfs_uri_get_host_name(uri) == NULL) {
        return GNOME_VFS_ERROR_INVALID_HOST_NAME;
    }

    *method_handle = g_new0(MythtvHandle, 1);
    (*method_handle)->mythtv_version = MYTHTV_VERSION_DEFAULT;

    (*method_handle)->is_livetv = FALSE;
    (*method_handle)->is_local_file = FALSE;

    tmp_str1 = gnome_vfs_uri_to_string(uri, GNOME_VFS_URI_HIDE_NONE);
    tmp_str2 = gnome_vfs_unescape_string(tmp_str1, "");

    gchar          *tmp_str3 = strstr(tmp_str2, ".nuv.avi");
    if (tmp_str3 != NULL) {
        tmp_str3[4] = '\0';
    }

    (*method_handle)->backend_info =
        gmyth_backend_info_new_with_uri(tmp_str2);
    (*method_handle)->gmyth_uri = gmyth_uri_new_with_value(tmp_str2);
    g_free(tmp_str1);
    g_free(tmp_str2);

    return GNOME_VFS_OK;
}

static void
myth_destroy_handle(MythtvHandle * method_handle)
{
    // TODO: abort if in tranfer state

    if (method_handle->backend_info != NULL) {
        g_object_unref(method_handle->backend_info);
        method_handle->backend_info = NULL;
    }

    if (method_handle->channel_name != NULL) {
        g_free(method_handle->channel_name);
        method_handle->channel_name = NULL;
    }

    if (method_handle->livetv != NULL) {
        g_object_unref(method_handle->livetv);
        method_handle->livetv = NULL;
    }

    if (method_handle->file != NULL) {
        g_object_unref(method_handle->file);
        method_handle->file = NULL;
    }

    if (method_handle->gmyth_uri != NULL) {
        g_object_unref(method_handle->gmyth_uri);
        method_handle->gmyth_uri = NULL;
    }

    g_free(method_handle);
}

static          GnomeVFSResult
myth_get_file_info(MythtvHandle * myth_handle,
                   GnomeVFSURI * uri, GnomeVFSFileInfo * info)
{
    GMythURI       *gmyth_uri;
    GMythBackendInfo *backend_info;
    gboolean        is_livetv;
    gboolean        is_local;

    _GNOME_VFS_METHOD_PARAM_CHECK(info != NULL);

    g_debug("%s - %d", __FUNCTION__, __LINE__);

    if (myth_handle == NULL) {
        gchar          *tmp_str1;
        gchar          *tmp_str2;

        tmp_str1 = gnome_vfs_uri_to_string(uri, GNOME_VFS_URI_HIDE_NONE);
        tmp_str2 = gnome_vfs_unescape_string(tmp_str1, "");

        backend_info = gmyth_backend_info_new_with_uri(tmp_str2);
        gmyth_uri = gmyth_uri_new_with_value(tmp_str2);

        g_free(tmp_str1);
        g_free(tmp_str2);
    } else {
        backend_info = g_object_ref(myth_handle->backend_info);
        gmyth_uri = g_object_ref(myth_handle->gmyth_uri);
    }

    info->valid_fields = 0;
    info->valid_fields = GNOME_VFS_FILE_INFO_FIELDS_TYPE |
        GNOME_VFS_FILE_INFO_FIELDS_MIME_TYPE |
        GNOME_VFS_FILE_INFO_FIELDS_PERMISSIONS;

    info->type = GNOME_VFS_FILE_TYPE_REGULAR;

    /*
     * fixme: get from file extension? 
     */
    info->mime_type = g_strdup("video/x-nuv");
    info->permissions = GNOME_VFS_PERM_USER_READ |
        GNOME_VFS_PERM_OTHER_READ | GNOME_VFS_PERM_GROUP_READ;

    info->name = g_strdup(gmyth_uri_get_path(gmyth_uri));

    /*
     * file size for remote files 
     */
    is_livetv = gmyth_uri_is_livetv(gmyth_uri);

    if (is_livetv == FALSE) {
        GMythFile      *file = NULL;
        gboolean        ret = FALSE;

        /*
         * Verifies if the file exists 
         */
        if (!gmyth_util_file_exists(backend_info,
                                    gmyth_uri_get_path(gmyth_uri))) {
            g_object_unref(file);
            g_object_unref(backend_info);
            g_debug("NOT FOUND %s/%d", __FUNCTION__, __LINE__);
            return GNOME_VFS_ERROR_NOT_FOUND;
        }

        is_local = gmyth_uri_is_local_file(gmyth_uri);
        if (is_local == TRUE) {
            file = GMYTH_FILE(gmyth_file_local_new(backend_info));
            ret = gmyth_file_local_open(GMYTH_FILE_LOCAL(file));
        } else {
            file = GMYTH_FILE(gmyth_file_transfer_new(backend_info));
            ret = gmyth_file_transfer_open(GMYTH_FILE_TRANSFER(file),
                                           gmyth_uri_get_path(gmyth_uri));
        }

        if (!ret) {
            g_object_unref(file);
            g_object_unref(backend_info);
            g_debug("NOT FOUND %s/%d", __FUNCTION__, __LINE__);
            return GNOME_VFS_ERROR_NOT_FOUND;
        }

        info->size = gmyth_file_get_filesize(file);
        info->valid_fields |= GNOME_VFS_FILE_INFO_FIELDS_SIZE;
        g_object_unref(file);
    }

    g_object_unref(backend_info);
    g_object_unref(gmyth_uri);

    return GNOME_VFS_OK;
}

static          GnomeVFSResult
myth_connection_start(MythtvHandle * method_handle)
{
    GnomeVFSResult  result = GNOME_VFS_OK;

    _GNOME_VFS_METHOD_PARAM_CHECK(method_handle != NULL);
    _GNOME_VFS_METHOD_PARAM_CHECK(method_handle->backend_info != NULL);

    /*
     * Connect to the backend 
     */
    if ((method_handle->is_livetv =
         gmyth_uri_is_livetv(method_handle->gmyth_uri)) == TRUE) {
        method_handle->livetv =
            gmyth_livetv_new(method_handle->backend_info);
        method_handle->channel_name =
            gmyth_uri_get_channel_name(method_handle->gmyth_uri);

        if (method_handle->channel_name != NULL) {
            if (gmyth_livetv_channel_name_setup(method_handle->livetv,
                                                method_handle->
                                                channel_name) == FALSE) {
                result = GNOME_VFS_ERROR_INVALID_URI;
                goto error;
            }
        } else if (gmyth_livetv_setup(method_handle->livetv) == FALSE) {
            result = GNOME_VFS_ERROR_INVALID_URI;
            goto error;
        }


        method_handle->file =
            GMYTH_FILE(gmyth_livetv_create_file_transfer
                       (method_handle->livetv));

        if (method_handle->file == NULL) {
            result = GNOME_VFS_ERROR_INVALID_URI;
            g_debug("MythTV FileTransfer is NULL!\n");
            goto error;
        }

        if (!gmyth_file_transfer_open
            (GMYTH_FILE_TRANSFER(method_handle->file),
             method_handle->livetv->uri !=
             NULL ? gmyth_uri_get_path(method_handle->livetv->
                                       uri) : method_handle->livetv->
             proginfo->pathname->str)) {

            g_debug("Couldn't open MythTV FileTransfer is NULL!\n");
            result = GNOME_VFS_ERROR_NOT_OPEN;
            goto error;
        }
    } else {
        gboolean        ret = TRUE;
        /*
         * Verifies if the file exists 
         */
        if (!gmyth_util_file_exists(method_handle->backend_info,
                                    gmyth_uri_get_path(method_handle->
                                                       gmyth_uri))) {

            g_debug("NOT FOUND %s/%d", __FUNCTION__, __LINE__);
            goto error;
        }

        if ((method_handle->is_local_file =
             gmyth_uri_is_local_file(method_handle->gmyth_uri)) == TRUE) {
            method_handle->file =
                GMYTH_FILE(gmyth_file_local_new
                           (method_handle->backend_info));
            ret =
                gmyth_file_local_open(GMYTH_FILE_LOCAL
                                      (method_handle->file));
        } else {
            method_handle->file =
                GMYTH_FILE(gmyth_file_transfer_new
                           (method_handle->backend_info));
            ret =
                gmyth_file_transfer_open(GMYTH_FILE_TRANSFER
                                         (method_handle->file),
                                         gmyth_uri_get_path(method_handle->
                                                            gmyth_uri));
        }

        /*
         * sets the Playback monitor connection 
         */
        if (!ret) {

            g_debug("NOT FOUND %s/%d", __FUNCTION__, __LINE__);
            result = GNOME_VFS_ERROR_NOT_FOUND;
            goto error;
        }
    }                           /* if - LiveTV or not? */

    method_handle->configured = TRUE;

    if (method_handle->file == NULL) {
        result = GNOME_VFS_ERROR_NOT_OPEN;
    }

  error:

    return result;
}

static          GnomeVFSResult
do_open(GnomeVFSMethod * method,
        GnomeVFSMethodHandle ** method_handle,
        GnomeVFSURI * uri, GnomeVFSOpenMode mode,
        GnomeVFSContext * context)
{
    MythtvHandle   *myth_handle = NULL;
    GnomeVFSResult  result = GNOME_VFS_OK;

    _GNOME_VFS_METHOD_PARAM_CHECK(method_handle != NULL);
    _GNOME_VFS_METHOD_PARAM_CHECK(uri != NULL);

    if (mode & GNOME_VFS_OPEN_WRITE) {
        return GNOME_VFS_ERROR_INVALID_OPEN_MODE;
    }

    result = myth_handle_new(uri, &myth_handle);
    if (result != GNOME_VFS_OK)
        return result;

    result = myth_connection_start(myth_handle);
    if (result != GNOME_VFS_OK) {
        myth_destroy_handle(myth_handle);
        myth_handle = NULL;
        return result;
    }

    *method_handle = (GnomeVFSMethodHandle *) myth_handle;

    return result;
}

static          GnomeVFSResult
do_create(GnomeVFSMethod * method,
          GnomeVFSMethodHandle ** method_handle,
          GnomeVFSURI * uri,
          GnomeVFSOpenMode mode,
          gboolean exclusive, guint perm, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_close(GnomeVFSMethod * method,
         GnomeVFSMethodHandle * method_handle, GnomeVFSContext * context)
{
    MythtvHandle   *myth_handle = (MythtvHandle *) method_handle;

    myth_destroy_handle(myth_handle);

    return GNOME_VFS_OK;
}


static          GnomeVFSResult
do_read(GnomeVFSMethod * method,
        GnomeVFSMethodHandle * method_handle,
        gpointer buffer,
        GnomeVFSFileSize num_bytes,
        GnomeVFSFileSize * bytes_read, GnomeVFSContext * context)
{
    GnomeVFSResult  retval = GNOME_VFS_OK;
    MythtvHandle   *myth_handle;
    GMythFileReadResult result;
    GByteArray     *myth_buffer = g_byte_array_new();

    _GNOME_VFS_METHOD_PARAM_CHECK(method_handle != NULL);

    myth_handle = (MythtvHandle *) method_handle;
    if (myth_handle->is_local_file)
        result = gmyth_file_local_read(GMYTH_FILE_LOCAL(myth_handle->file),
                                       myth_buffer,
                                       num_bytes, myth_handle->is_livetv);
    else
        result =
            gmyth_file_transfer_read(GMYTH_FILE_TRANSFER
                                     (myth_handle->file), myth_buffer,
                                     num_bytes, myth_handle->is_livetv);

    if (result == GMYTH_FILE_READ_ERROR) {
        retval = GNOME_VFS_ERROR_IO;
    }

    if (result == GMYTH_FILE_READ_EOF) {
        retval = GNOME_VFS_ERROR_EOF;
    }

    if (myth_buffer->len > 0) {
        g_memmove(buffer, myth_buffer->data, myth_buffer->len);
        *bytes_read = (GnomeVFSFileSize) myth_buffer->len;
        myth_handle->offset += myth_buffer->len;
        g_byte_array_free(myth_buffer, TRUE);
    }

    return retval;
}

static          GnomeVFSResult
do_write(GnomeVFSMethod * method,
         GnomeVFSMethodHandle * method_handle,
         gconstpointer buffer,
         GnomeVFSFileSize num_bytes,
         GnomeVFSFileSize * bytes_written, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_seek(GnomeVFSMethod * method,
        GnomeVFSMethodHandle * method_handle,
        GnomeVFSSeekPosition whence,
        GnomeVFSFileOffset offset, GnomeVFSContext * context)
{
    MythtvHandle   *myth_handle;
    guint64         whence_p = 0;
    gint64          new_offset = 0;

    _GNOME_VFS_METHOD_PARAM_CHECK(method_handle != NULL);

    myth_handle = (MythtvHandle *) method_handle;

    g_debug("seek offset %" G_GINT64_FORMAT " whence %d", offset, whence);

    if (gmyth_uri_is_livetv(myth_handle->gmyth_uri))
        return GNOME_VFS_ERROR_NOT_SUPPORTED;

    switch (whence) {
    case GNOME_VFS_SEEK_START:
        whence_p = 0;
        break;
    case GNOME_VFS_SEEK_CURRENT:
        whence_p = myth_handle->offset;
        break;
    case GNOME_VFS_SEEK_END:
        return GNOME_VFS_ERROR_NOT_SUPPORTED;
    }

    new_offset =
        gmyth_file_transfer_seek(myth_handle->file, offset, whence_p);
    if (new_offset != 0) {
        myth_handle->offset = new_offset;
        return GNOME_VFS_OK;
    }

    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_tell(GnomeVFSMethod * method,
        GnomeVFSMethodHandle * method_handle,
        GnomeVFSFileSize * offset_return)
{
    MythtvHandle   *myth_handle = NULL;

    _GNOME_VFS_METHOD_PARAM_CHECK(method_handle != NULL);

    myth_handle = (MythtvHandle *) method_handle;
    *offset_return = myth_handle->offset;

    return GNOME_VFS_OK;
}

static          GnomeVFSResult
do_truncate_handle(GnomeVFSMethod * method,
                   GnomeVFSMethodHandle * method_handle,
                   GnomeVFSFileSize where, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_open_directory(GnomeVFSMethod * method,
                  GnomeVFSMethodHandle ** method_handle,
                  GnomeVFSURI * uri,
                  GnomeVFSFileInfoOptions options,
                  GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_close_directory(GnomeVFSMethod * method,
                   GnomeVFSMethodHandle * method_handle,
                   GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_read_directory(GnomeVFSMethod * method,
                  GnomeVFSMethodHandle * method_handle,
                  GnomeVFSFileInfo * file_info, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}


static          GnomeVFSResult
do_get_file_info(GnomeVFSMethod * method,
                 GnomeVFSURI * uri,
                 GnomeVFSFileInfo * file_info,
                 GnomeVFSFileInfoOptions options,
                 GnomeVFSContext * context)
{
    return myth_get_file_info(NULL, uri, file_info);
}

static          GnomeVFSResult
do_get_file_info_from_handle(GnomeVFSMethod * method,
                             GnomeVFSMethodHandle * method_handle,
                             GnomeVFSFileInfo * file_info,
                             GnomeVFSFileInfoOptions options,
                             GnomeVFSContext * context)
{
    MythtvHandle   *myth_handle = (MythtvHandle *) method_handle;

    return myth_get_file_info(myth_handle, NULL, file_info);
}

static          gboolean
do_is_local(GnomeVFSMethod * method, const GnomeVFSURI * uri)
{
    return FALSE;
}

static          GnomeVFSResult
do_make_directory(GnomeVFSMethod * method,
                  GnomeVFSURI * uri, guint perm, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_remove_directory(GnomeVFSMethod * method,
                    GnomeVFSURI * uri, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_move(GnomeVFSMethod * method,
        GnomeVFSURI * old_uri,
        GnomeVFSURI * new_uri,
        gboolean force_replace, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_unlink(GnomeVFSMethod * method,
          GnomeVFSURI * uri, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_check_same_fs(GnomeVFSMethod * method,
                 GnomeVFSURI * a,
                 GnomeVFSURI * b,
                 gboolean * same_fs_return, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_set_file_info(GnomeVFSMethod * method,
                 GnomeVFSURI * uri,
                 const GnomeVFSFileInfo * info,
                 GnomeVFSSetFileInfoMask mask, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_truncate(GnomeVFSMethod * method,
            GnomeVFSURI * uri,
            GnomeVFSFileSize where, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_find_directory(GnomeVFSMethod * method,
                  GnomeVFSURI * near_uri,
                  GnomeVFSFindDirectoryKind kind,
                  GnomeVFSURI ** result_uri,
                  gboolean create_if_needed,
                  gboolean find_if_needed,
                  guint permissions, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_create_symbolic_link(GnomeVFSMethod * method,
                        GnomeVFSURI * uri,
                        const char *target_reference,
                        GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_READ_ONLY;
}

static          GnomeVFSResult
do_monitor_add(GnomeVFSMethod * method,
               GnomeVFSMethodHandle ** method_handle_return,
               GnomeVFSURI * uri, GnomeVFSMonitorType monitor_type)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_monitor_cancel(GnomeVFSMethod * method,
                  GnomeVFSMethodHandle * method_handle)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static          GnomeVFSResult
do_file_control(GnomeVFSMethod * method,
                GnomeVFSMethodHandle * method_handle,
                const char *operation,
                gpointer operation_data, GnomeVFSContext * context)
{
    return GNOME_VFS_ERROR_NOT_SUPPORTED;
}

static GnomeVFSMethod method = {
    sizeof(GnomeVFSMethod),
    do_open,
    do_create,
    do_close,
    do_read,
    do_write,
    do_seek,
    do_tell,
    do_truncate_handle,
    do_open_directory,
    do_close_directory,
    do_read_directory,
    do_get_file_info,
    do_get_file_info_from_handle,
    do_is_local,
    do_make_directory,
    do_remove_directory,
    do_move,
    do_unlink,
    do_check_same_fs,
    do_set_file_info,
    do_truncate,
    do_find_directory,
    do_create_symbolic_link,
    do_monitor_add,
    do_monitor_cancel,
    do_file_control
};


GnomeVFSMethod *
vfs_module_init(const char *method_name, const char *args)
{
    return &method;
}

void
vfs_module_shutdown(GnomeVFSMethod * method)
{
}