#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <string.h>

#include <gst/gst.h>
#include <glib.h>

#include "gmencoder.h"

#define FILE_OUT  1

static GMainLoop *mainloop = NULL;
/*
 * Options 
 */
static gchar   *input_file = NULL;
static gchar   *video_encode = NULL;
static gchar   *video_opts = NULL;
static gdouble  video_fps = 0.0;
static gint     video_rate = 0;
static gint     video_width = 0;
static gint     video_height = 0;
static gchar   *audio_encode = NULL;
static gchar   *audio_opts = NULL;
static double   audio_rate = 0.0;
static gchar   *mux_name = NULL;
static gchar   *output_uri = NULL;
static gboolean chunked = FALSE;



static          gboolean
_quit(gpointer data)
{
    g_object_unref(data);
    g_main_loop_quit(mainloop);
    return FALSE;
}

static void
_mencoder_eos_cb(GMencoder * mencoder, gpointer data)
{
    g_print("PROGRESS: 100\n");
    g_idle_add(_quit, mencoder);
}


static void
_mencoder_error_cb(GMencoder * mencoder, const gchar * msg, gpointer data)
{
    g_print("Error: %s\n", msg);
    g_idle_add(_quit, mencoder);
}

static          gboolean
_io_channel_cb(GIOChannel * ch, GIOCondition condition, gpointer data)
{
    GString        *cmd = g_string_new("");
    g_io_channel_read_line_string(ch, cmd, NULL, NULL);

    if (strcmp(cmd->str, "PLAY\n") == 0) {
        g_mencoder_play_stream(G_MENCODER(data));
    } else if (strcmp(cmd->str, "PAUSE\n") == 0) {
        g_mencoder_pause_stream(G_MENCODER(data));
    } else if (strcmp(cmd->str, "STOP\n") == 0) {
        g_mencoder_close_stream(G_MENCODER(data));
    } else if (strcmp(cmd->str, "QUIT\n") == 0) {
        g_mencoder_close_stream(G_MENCODER(data));
        g_main_loop_quit(mainloop);
    }
    g_string_free(cmd, TRUE);
    return TRUE;
}

int
main(int argc, char **argv)
{
    GMencoder      *coder = NULL;
    GIOChannel     *ch;
    gchar         **vopts;
    gchar         **aopts;
    gchar         **files;
    gint            i;
    gboolean      ret;

    GOptionContext *context;
    static const GOptionEntry options[] = {
        {"input-files", 'i', 0, G_OPTION_ARG_STRING, &input_file,
         "Input File", NULL},

        {"video-encode", 0, 0, G_OPTION_ARG_STRING, &video_encode,
         "GstElementName for used to video encode", NULL},

        {"video-opts", 0, 0, G_OPTION_ARG_STRING, &video_opts,
         "Properties to set on video element", NULL},

        {"video-fps", 0, 0, G_OPTION_ARG_DOUBLE, &video_fps,
         "Video FPS", NULL},

        {"video-rate", 0, 0, G_OPTION_ARG_INT, &video_rate,
         "Video rate", NULL},

        {"video-width", 0, 0, G_OPTION_ARG_INT, &video_width,
         "Video width", NULL},

        {"video-height", 0, 0, G_OPTION_ARG_INT, &video_height,
         "Video height", NULL},

        {"audio-encode", 0, 0, G_OPTION_ARG_STRING, &audio_encode,
         "GstElementName for use to audio encode", NULL},

        {"audio-opts", 0, 0, G_OPTION_ARG_STRING, &audio_opts,
         "Properties to set on audio element", NULL},

        {"audio-rate", 0, 0, G_OPTION_ARG_INT, &audio_rate,
         "Audio rate", NULL},

        {"mux-element", 0, 0, G_OPTION_ARG_STRING, &mux_name,
         "GstElementName for use to mux file", NULL},

        {"output-uri", 'o', 0, G_OPTION_ARG_STRING, &output_uri,
         "Uri to output", NULL},

	{"chunked", 'c', 0, G_OPTION_ARG_NONE, &chunked, 
         "Send package chunked", NULL},

        {NULL}
    };

    g_type_init();
    g_thread_init(NULL);
    mainloop = g_main_loop_new(NULL, FALSE);

    g_set_prgname("gmemcoder");
    context = g_option_context_new(NULL);
    g_option_context_set_help_enabled(context, TRUE);
    g_option_context_add_main_entries(context, options, NULL);
    g_option_context_add_group(context, gst_init_get_option_group());
    g_option_context_parse(context, &argc, &argv, NULL);

    gst_init(&argc, &argv);

    if (output_uri == NULL) {
        g_print("You need to specify output-uri.\nTry --help "
                "for more information.\n");
        return 1;
    }

    if (input_file == NULL) {
        g_print("You need to specify input file\nTry --help "
                "for more information.\n");
    }

    coder = g_mencoder_new();
    ch = g_io_channel_unix_new(0);

    if (audio_opts != NULL)
        aopts = g_strsplit(audio_opts, ",", 0);
    else
        aopts = NULL;

    if (video_opts != NULL)
        vopts = g_strsplit(video_opts, ",", 0);
    else
        vopts = NULL;

    
    ret = g_mencoder_setup_stream(coder, chunked, mux_name,
                                  video_encode, vopts, video_fps,
                                  video_rate, video_width, video_height,
                                  audio_encode, aopts, audio_rate, output_uri);

    if (ret == TRUE) {
        files = g_strsplit(input_file, ",", 0);
        for (i = 0; i < g_strv_length(files); i++) {
            if (!g_mencoder_append_uri(coder, files[i])) {
                g_debug("Invalid uri: %s", files[i]);
            }
        }
        g_strfreev(files);
    }

    g_strfreev(aopts);
    g_strfreev(vopts);

    if (ret == TRUE) {
        g_io_add_watch(ch, G_IO_IN, _io_channel_cb, coder);
        g_signal_connect(G_OBJECT(coder),
                         "eos", G_CALLBACK(_mencoder_eos_cb), mainloop);

        g_signal_connect(G_OBJECT(coder),
                         "error", G_CALLBACK(_mencoder_error_cb), mainloop);

        g_mencoder_play_stream(coder);
        g_main_loop_run(mainloop);
    }

    g_object_unref(coder);
    return 0;
}