/* $Id$
   
   Copyright (C) 2020-2022 Aamot Software
   Author(s): Ole Aamot <ole@gnome.org>
   License: GNU GPL version 3
   Version: 6.2.0 (2022-07-09)
   Website: http://www.garagejam.org/
   
*/

#include <string.h>
#include <gst/gst.h>
#include <signal.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>

// v4l2src ! tee name=t t. ! x264enc ! mp4mux ! filesink location=/home/rish/Desktop/okay.264 t. ! videoconvert ! autovideosink

static GMainLoop *loop;
static GstElement *pipeline, *audio_source, *sink, *src, *tee, *encoder,
    *muxer, *filesink, *videoconvert, *videosink, *queue_record,
    *queue_display;
static GstBus *bus;
static GstPad *teepad;
static gboolean recording = FALSE;
static gint counter = 0;
static char *file_path;

static gboolean
message_cb(GstBus *bus, GstMessage *message, gpointer user_data)
{
	switch (GST_MESSAGE_TYPE(message)) {
	case GST_MESSAGE_ERROR:{
			GError *err = NULL;
			gchar *name, *debug = NULL;

			name = gst_object_get_path_string(message->src);
			gst_message_parse_error(message, &err, &debug);

			g_printerr("ERROR: from element %s: %s\n", name,
				   err->message);
			if (debug != NULL)
				g_printerr("Additional debug info:\n%s\n",
					   debug);

			g_error_free(err);
			g_free(debug);
			g_free(name);

			g_main_loop_quit(loop);
			break;
		}
	case GST_MESSAGE_WARNING:{
			GError *err = NULL;
			gchar *name, *debug = NULL;

			name = gst_object_get_path_string(message->src);
			gst_message_parse_warning(message, &err, &debug);

			g_printerr("ERROR: from element %s: %s\n", name,
				   err->message);
			if (debug != NULL)
				g_printerr("Additional debug info:\n%s\n",
					   debug);

			g_error_free(err);
			g_free(debug);
			g_free(name);
			break;
		}
	case GST_MESSAGE_EOS:{
			g_print("Got EOS\n");
			g_main_loop_quit(loop);
			gst_element_set_state(pipeline, GST_STATE_NULL);
			g_main_loop_unref(loop);
			gst_object_unref(pipeline);
			exit(0);
			break;
		}
	default:
		break;
	}

	return TRUE;
}

static GstPadProbeReturn unlink_cb(GstPad *pad, GstPadProbeInfo *info,
				   gpointer user_data)
{
	g_print("Unlinking...");
	GstPad *sinkpad;
	sinkpad = gst_element_get_static_pad(queue_record, "sink");
	gst_pad_unlink(teepad, sinkpad);
	gst_object_unref(sinkpad);

	gst_element_send_event(encoder, gst_event_new_eos());

	sleep(1);
	gst_bin_remove(GST_BIN(pipeline), queue_record);
	gst_bin_remove(GST_BIN(pipeline), encoder);
	gst_bin_remove(GST_BIN(pipeline), muxer);
	gst_bin_remove(GST_BIN(pipeline), filesink);

	gst_element_set_state(queue_record, GST_STATE_NULL);
	gst_element_set_state(encoder, GST_STATE_NULL);
	gst_element_set_state(muxer, GST_STATE_NULL);
	gst_element_set_state(filesink, GST_STATE_NULL);

	gst_object_unref(queue_record);
	gst_object_unref(encoder);
	gst_object_unref(muxer);
	gst_object_unref(filesink);

	gst_element_release_request_pad(tee, teepad);
	gst_object_unref(teepad);

	g_print("Unlinked\n");

	return GST_PAD_PROBE_REMOVE;
}

void stopRecording()
{
	g_print("stopRecording\n");
	gst_pad_add_probe(teepad, GST_PAD_PROBE_TYPE_IDLE, unlink_cb, NULL,
			  (GDestroyNotify) g_free);
	recording = FALSE;
}

void startRecording()
{
	g_print("startRecording\n");
	GstPad *sinkpad;
	GstPadTemplate *templ;

	templ =
	    gst_element_class_get_pad_template(GST_ELEMENT_GET_CLASS(tee),
					       "src_%u");
	teepad = gst_element_request_pad(tee, templ, NULL, NULL);
	queue_record = gst_element_factory_make("queue", "queue_record");
	encoder = gst_element_factory_make("x264enc", NULL);
	muxer = gst_element_factory_make("mp4mux", NULL);
	filesink = gst_element_factory_make("filesink", NULL);
	char *file_name = (char *) malloc(255 * sizeof(char));
	sprintf(file_name, "%s%d.mp4", file_path, counter++);
	g_print("Recording to file %s", file_name);
	g_object_set(filesink, "location", file_name, NULL);
	g_object_set(encoder, "tune", 4, NULL);
	free(file_name);

	gst_bin_add_many(GST_BIN(pipeline), gst_object_ref(queue_record),
			 gst_object_ref(encoder), gst_object_ref(muxer),
			 gst_object_ref(filesink), NULL);
	gst_element_link_many(queue_record, encoder, muxer, filesink,
			      NULL);

	gst_element_sync_state_with_parent(queue_record);
	gst_element_sync_state_with_parent(encoder);
	gst_element_sync_state_with_parent(muxer);
	gst_element_sync_state_with_parent(filesink);

	sinkpad = gst_element_get_static_pad(queue_record, "sink");
	gst_pad_link(teepad, sinkpad);
	gst_object_unref(sinkpad);

	recording = TRUE;
}

int sigintHandler(int unused)
{
	g_print("You ctrl-c!\n");
	if (recording)
		stopRecording();
	else
		startRecording();
	return 0;
}

int gb_record_cb(char *path, gpointer data)
{
	return 0;
}