#include "mov-writer.h"
#include "mov-format.h"
#include "mov-udta.h"
#include "mpeg4-aac.h"
#include "opus-head.h"
#include "mp3-header.h"
#include "flv-proto.h"
#include "flv-reader.h"
#include "flv-parser.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>

extern "C" const struct mov_buffer_t* mov_file_buffer(void);
extern "C" int mov_writer_add_udta(mov_writer_t * mov, const void* data, size_t size);

static uint8_t s_buffer[2 * 1024 * 1024];
static int s_width, s_height;

static int onFLV(void* param, int codec, const void* data, size_t bytes, uint32_t pts, uint32_t dts, int flags)
{
	mov_writer_t* mov = (mov_writer_t*)param;
	static int s_audio_track = -1;
	static int s_video_track = -1;

	switch(codec)
	{
	case FLV_AUDIO_AAC:
	case FLV_AUDIO_OPUS:
		return mov_writer_write(mov, s_audio_track, data, bytes, pts, dts, 1 == flags ? MOV_AV_FLAG_KEYFREAME : 0);

	case FLV_AUDIO_MP3:
		if (-1 == s_audio_track)
		{
			struct mp3_header_t mp3;
			if (0 == mp3_header_load(&mp3, data, bytes))
				return -1;
			s_audio_track = mov_writer_add_audio(mov, MOV_OBJECT_MP3, mp3_get_channel(&mp3), 16, mp3_get_frequency(&mp3), NULL, 0);
		}

		if (-1 == s_audio_track)
			return -1;
		return mov_writer_write(mov, s_audio_track, data, bytes, pts, dts, 1 == flags ? MOV_AV_FLAG_KEYFREAME : 0);
	
	case FLV_AUDIO_G711A:
	case FLV_AUDIO_G711U:
		if (-1 == s_audio_track)
			s_audio_track = mov_writer_add_audio(mov, codec==FLV_AUDIO_G711A?MOV_OBJECT_G711a:MOV_OBJECT_G711u, 1, 16, 8000, NULL, 0);
		if (-1 == s_audio_track)
			return -1;
		return mov_writer_write(mov, s_audio_track, data, bytes, pts, dts, 0);

	case FLV_VIDEO_H264:
	case FLV_VIDEO_H265:
	case FLV_VIDEO_AV1:
		return mov_writer_write(mov, s_video_track, data, bytes, pts, dts, flags);

	case FLV_VIDEO_AVCC:
		if (-1 == s_video_track)
			s_video_track = mov_writer_add_video(mov, MOV_OBJECT_H264, s_width, s_height, data, bytes);
		break;

	case FLV_VIDEO_HVCC:
		if (-1 == s_video_track)
			s_video_track = mov_writer_add_video(mov, MOV_OBJECT_HEVC, s_width, s_height, data, bytes);
		break;

	case FLV_VIDEO_AV1C:
		if (-1 == s_video_track)
			s_video_track = mov_writer_add_video(mov, MOV_OBJECT_AV1, s_width, s_height, data, bytes);
		break;

	case FLV_AUDIO_ASC:
		if (-1 == s_audio_track)
		{
			struct mpeg4_aac_t aac;
			mpeg4_aac_audio_specific_config_load((const uint8_t*)data, bytes, &aac);
			int rate = mpeg4_aac_audio_frequency_to((enum mpeg4_aac_frequency)aac.sampling_frequency_index);
			s_audio_track = mov_writer_add_audio(mov, MOV_OBJECT_AAC, aac.channel_configuration, 16, rate, data, bytes);
		}
		break;
	case FLV_AUDIO_OPUS_HEAD:
		if (-1 == s_audio_track)
		{
			struct opus_head_t opus;
			opus_head_load((const uint8_t*)data, bytes, &opus);
			s_audio_track = mov_writer_add_audio(mov, MOV_OBJECT_OPUS, opus.channels, 16, opus.input_sample_rate, data, bytes);
		}
		break;

	default:
		// nothing to do
		assert(FLV_SCRIPT_METADATA == codec);
	}

	printf("\n");
	return 0;
}

static int mov_writer_add_cover(mov_writer_t* mov, const char* cover)
{
	static uint8_t s_cover_data[2 * 1024 * 1024];
	static uint8_t s_udta[2 * 1024 * 1024];
	FILE* fp = fopen(cover, "rb");
	if (!fp)
		return -1;
	int n = fread(s_cover_data, 1, sizeof(s_cover_data), fp);
	fclose(fp);

	assert(n < sizeof(s_cover_data)); // assert load all
	struct mov_udta_meta_t meta;
	memset(&meta, 0, sizeof(meta));
	meta.cover = s_cover_data;
	meta.cover_size = n;
	n = mov_udta_meta_write(&meta, s_udta, sizeof(s_udta));
	assert(n < sizeof(s_udta)); // check buffer size
	return mov_writer_add_udta(mov, s_udta, n);
}

static int mov_writer_add_chapter(mov_writer_t* mov)
{
	struct mov_udta_chapter_t chapters[] = {
		{0 * 10000000,	"chapter 01"},
		{10 * 10000000, "chapter 02"},
		{20 * 10000000, "chapter 03"},
		{30 * 10000000, "chapter 04"},
		{40 * 10000000, "chapter 05"},
	};
	static uint8_t s_udta[1024];
	int n = mov_udta_chapter_write(chapters, sizeof(chapters) / sizeof(chapters[0]), s_udta, sizeof(s_udta));
	assert(n < sizeof(s_udta)); // check buffer size
	return mov_writer_add_udta(mov, s_udta, n);
}

void mov_writer_test(int w, int h, const char* inflv, const char* outmp4)
{
	int r, type;
	size_t taglen;
	uint32_t timestamp;

	FILE* fp = fopen(outmp4, "wb+");
	void* flv = flv_reader_create(inflv);
	mov_writer_t* mov = mov_writer_create(mov_file_buffer(), fp, MOV_FLAG_FASTSTART);

	s_width = w;
	s_height = h;
	while (1 == flv_reader_read(flv, &type, &timestamp, &taglen, s_buffer, sizeof(s_buffer)))
	{
		r = flv_parser_tag(type, s_buffer, taglen, timestamp, onFLV, mov);
		assert(r >= 0);
	}

	//mov_writer_add_cover(mov, "cover.jpg");
	//mov_writer_add_chapter(mov);

	mov_writer_destroy(mov);
	flv_reader_destroy(flv);
	fclose(fp);
}