#include <vector>
#include <string>
#include <obs-module.h>
#include <util/config-file.h>
#include <util/platform.h>
#include <obs-avc.h>

#ifndef RTSP_HELPER_H
#define RTSP_HELPER_H

#define CONFIG_SECTIION "RstpOutput"
#define HOTKEY_CONFIG_SECTIION "Hotkeys"

static bool make_config_dir()
{
	auto path = obs_module_config_path("");
	auto ret = os_mkdirs(path);
	bfree(path);
	return ret == MKDIR_SUCCESS || ret == MKDIR_EXISTS;
}

static obs_data_t *rtsp_output_read_data()
{
	obs_data_t *data;
	auto path = obs_module_config_path("rtsp_output.json");
	data = obs_data_create_from_json_file_safe(path, "bak");
	bfree(path);
	return data;
}

static bool rtsp_output_save_data(obs_data_t *data)
{
	if (!make_config_dir())
		return false;
	auto path = obs_module_config_path("rtsp_output.json");
	auto ret = obs_data_save_json_safe(data, path, "tmp", "bak");
	bfree(path);
	return ret;
}

static config_t *rtsp_properties_open_config()
{
	if (!make_config_dir())
		return nullptr;
	auto path = obs_module_config_path("config.ini");
	config_t *config;
	auto ret = config_open(&config, path, CONFIG_OPEN_ALWAYS);
	bfree(path);
	config_set_default_bool(config, CONFIG_SECTIION, "AutoStart", false);
	config_set_default_bool(config, CONFIG_SECTIION, "AudioTrack1", true);
	config_set_default_bool(config, CONFIG_SECTIION, "AudioTrack2", false);
	config_set_default_bool(config, CONFIG_SECTIION, "AudioTrack3", false);
	config_set_default_bool(config, CONFIG_SECTIION, "AudioTrack4", false);
	config_set_default_bool(config, CONFIG_SECTIION, "AudioTrack5", false);
	config_set_default_bool(config, CONFIG_SECTIION, "AudioTrack6", false);
	return config;
}

static void rtsp_output_avc_get_sps_pps(const uint8_t *data, size_t size,
					const uint8_t **sps, size_t *sps_size,
					const uint8_t **pps, size_t *pps_size)
{
	const uint8_t *nal_start, *nal_end;
	const uint8_t *end = data + size;
	int type;

	nal_start = obs_avc_find_startcode(data, end);
	while (true) {
		while (nal_start < end && !*(nal_start++))
			;

		if (nal_start == end)
			break;

		nal_end = obs_avc_find_startcode(nal_start, end);

		type = nal_start[0] & 0x1F;
		if (type == OBS_NAL_SPS) {
			*sps = nal_start;
			*sps_size = nal_end - nal_start;
		} else if (type == OBS_NAL_PPS) {
			*pps = nal_start;
			*pps_size = nal_end - nal_start;
		}

		nal_start = nal_end;
	}
}

static std::string string_format(std::string format, ...)
{
	va_list argp;
	va_start(argp, format);
	auto size = (size_t)vsnprintf(nullptr, 0, format.c_str(), argp) + 1;
	va_end(argp);
	auto buf = std::vector<char>(size);
	va_start(argp, format);
	vsnprintf(buf.data(), size, format.c_str(), argp);
	va_end(argp);
	return std::string(buf.data(), buf.data() + size - 1);
}

static std::string rtsp_properties_get_data_volume_display(uint64_t total_bytes) {
	const uint64_t kb = 1024;
	const uint64_t mb = kb * 1024;
	const uint64_t gb = mb * 1024;
	const uint64_t tb = gb * 1024;
	if (total_bytes == 0)
		return "0.0 MB";
	if (total_bytes < kb) {
		return string_format("%lu bytes", total_bytes);
	}
	if (total_bytes < mb) {
		return string_format("%.1f KB", double(total_bytes) / kb);
	}
	if (total_bytes < gb) {
		return string_format("%.1f MB", double(total_bytes) / mb);
	}
	if (total_bytes < tb) {
		return string_format("%.1f GB", double(total_bytes) / gb);
	}
	return string_format("%.1f TB", double(total_bytes) / tb);
}

#endif // RTSP_HELPER_H