/*****************************************************************************
#                                                                            #
#    uStreamer - Lightweight and fast MJPEG-HTTP streamer.                   #
#                                                                            #
#    Copyright (C) 2018-2022  Maxim Devaev <mdevaev@gmail.com>               #
#                                                                            #
#    This program is free software: you can redistribute it and/or modify    #
#    it under the terms of the GNU General Public License as published by    #
#    the Free Software Foundation, either version 3 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 General Public License       #
#    along with this program.  If not, see <https://www.gnu.org/licenses/>.  #
#                                                                            #
*****************************************************************************/


#include "client.h"


static void *_video_thread(void *v_client);
static void *_audio_thread(void *v_client);
static void *_common_thread(void *v_client, bool video);


us_janus_client_s *us_janus_client_init(janus_callbacks *gw, janus_plugin_session *session, bool has_audio) {
	us_janus_client_s *client;
	US_CALLOC(client, 1);
	client->gw = gw;
	client->session = session;
	atomic_init(&client->transmit, true);

	atomic_init(&client->stop, false);

	client->video_queue = us_queue_init(1024);
	US_THREAD_CREATE(client->video_tid, _video_thread, client);

	if (has_audio) {
		client->audio_queue = us_queue_init(64);
		US_THREAD_CREATE(client->audio_tid, _audio_thread, client);
	}
	return client;
}

void us_janus_client_destroy(us_janus_client_s *client) {
	atomic_store(&client->stop, true);
	us_queue_put(client->video_queue, NULL, 0);
	if (client->audio_queue != NULL) {
		us_queue_put(client->audio_queue, NULL, 0);
	}

	US_THREAD_JOIN(client->video_tid);
	US_QUEUE_DELETE_WITH_ITEMS(client->video_queue, us_rtp_destroy);
	if (client->audio_queue != NULL) {
		US_THREAD_JOIN(client->audio_tid);
		US_QUEUE_DELETE_WITH_ITEMS(client->audio_queue, us_rtp_destroy);
	}
	free(client);
}

void us_janus_client_send(us_janus_client_s *client, const us_rtp_s *rtp) {
	if (
		!atomic_load(&client->transmit)
		|| (!rtp->video && client->audio_queue == NULL)
	) {
		return;
	}
	us_rtp_s *const new = us_rtp_dup(rtp);
	if (us_queue_put((new->video ? client->video_queue : client->audio_queue), new, 0) != 0) {
		US_JLOG_ERROR("client", "Session %p %s queue is full",
			client->session, (new->video ? "video" : "audio"));
		us_rtp_destroy(new);
	}
}

static void *_video_thread(void *v_client) {
	return _common_thread(v_client, true);
}

static void *_audio_thread(void *v_client) {
	return _common_thread(v_client, false);
}

static void *_common_thread(void *v_client, bool video) {
	us_janus_client_s *const client = (us_janus_client_s *)v_client;
	us_queue_s *const queue = (video ? client->video_queue : client->audio_queue);
	assert(queue != NULL); // Audio may be NULL

	while (!atomic_load(&client->stop)) {
		us_rtp_s *rtp;
		if (!us_queue_get(queue, (void **)&rtp, 0.1)) {
			if (rtp == NULL) {
				break;
			}
			if (atomic_load(&client->transmit)) {
				janus_plugin_rtp packet = {0};
				packet.video = rtp->video;
				packet.buffer = (char *)rtp->datagram;
				packet.length = rtp->used;
				janus_plugin_rtp_extensions_reset(&packet.extensions);
				// FIXME: Это очень эффективный способ уменьшить задержку, но WebRTC стек в хроме и фоксе
				// слишком корявый, чтобы обработать это, из-за чего на кейфреймах начинаются заикания.
				//   - https://github.com/Glimesh/janus-ftl-plugin/issues/101
				if (rtp->zero_playout_delay) {
					packet.extensions.min_delay = 0;
					packet.extensions.max_delay = 0;
				}
				client->gw->relay_rtp(client->session, &packet);
			}
			us_rtp_destroy(rtp);
		}
	}
	return NULL;
}