#include <SDL2/SDL.h>
#include <SDL2/SDL_audio.h>
#include <SDL2/SDL_image.h>
#include <SDL2/SDL_ttf.h>
#include <stdio.h>

#include "log.h"

const int SCREEN_WIDTH = 640;
const int SCREEN_HEIGHT = 480;

const int MAX_RECORDING_DEVICES = 10;
const int MAX_RECORDING_SECONDS = 1;
const int RECORDING_BUFFER_SECONDS = MAX_RECORDING_SECONDS + 1;
/*
 * Application
 */
int init();
void destroy();
int select_device(SDL_KeyboardEvent* key_event, int max_index);

typedef enum {
	SELECTING_DEVICE,
	RECORDING,
	RECORDED,
	PLAYBACK,
} application_state;

typedef struct {
	application_state application_state;
	int device_index;
	int recording_device_id;
	int playback_device_id;
} state;

void application_state_to_string(application_state state, char* string);
/*
 * Visual
 */
SDL_Window* window = NULL;
SDL_Renderer* renderer = NULL;
TTF_Font* global_font = NULL;

typedef struct {
	SDL_Texture* texture;
	int width;
	int height;
} text_texture;

int text_texture_init(text_texture* text_texture);
int text_texture_load(text_texture* text_texture, char* string);
int text_texture_render(text_texture* text_texture, int x, int y);
int text_texture_free(text_texture* text_texture);
/*
 * Audio
 */
int num_devices = 0;

SDL_AudioSpec recording_spec;
SDL_AudioSpec playback_spec;
SDL_AudioSpec received_recording_spec;
SDL_AudioSpec received_playback_spec;
Uint8* recording_buffer = NULL;
unsigned int recording_buffer_size = 0;
unsigned int recording_buffer_position = 0;
unsigned int recording_buffer_position_max = 0;

SDL_AudioDeviceID audio_recording_init(int index, void* userdata);
SDL_AudioDeviceID audio_playback_init(void* userdata);
void audio_buffer_destroy();
void audio_recording_callback( void* userdata, Uint8* stream, int len );
void audio_playback_callback( void* userdata, Uint8* stream, int len );

void handle_input(SDL_Event* event, state* state);
void handle_task(state* state);
int main(int argc, char* argv[])
{
	if (!init()) return 1;

	num_devices = SDL_GetNumAudioDevices(1);
	printf("Num audio devices: %i\n", num_devices);
	char char_buffer[64];
	char default_fmt[] = "Device selected: %i";

	state state = {
		SELECTING_DEVICE,
		-1,
		0,
		0,
	};

	text_texture display_text_texture;
	text_texture_init(&display_text_texture);
	sprintf(char_buffer, default_fmt, state.device_index);
	text_texture_load(&display_text_texture, char_buffer);

	text_texture device_textures[num_devices];
	for (int i = 0; i < num_devices; i++) {
		text_texture_init(&device_textures[i]);
		sprintf(char_buffer, "%i: %s", i, SDL_GetAudioDeviceName(i, 1));
		text_texture_load(&device_textures[i], char_buffer);
	}

	text_texture state_text_texture;
	text_texture_init(&state_text_texture);
	application_state_to_string(state.application_state, char_buffer);
	text_texture_load(&state_text_texture, char_buffer);

	SDL_Event event;
	int running = 1;
	while (running) {
		while (SDL_PollEvent(&event)) {
			SDL_Keycode keysym = event.key.keysym.sym;
			switch (event.type) {

			case SDL_QUIT:
				running = 0;
				break;

			case SDL_KEYDOWN:
				handle_input(&event, &state);
				break;

			}
		}

		handle_task(&state);

		sprintf(char_buffer, default_fmt, state.device_index);
		text_texture_load(&display_text_texture, char_buffer);
		application_state_to_string(state.application_state, char_buffer);
		text_texture_load(&state_text_texture, char_buffer);

		// Clear the screen
		SDL_SetRenderDrawColor(renderer, 0, 0, 0, 255);
		SDL_RenderClear(renderer);

		// Render
		text_texture_render(&display_text_texture, 0, 0);

		if (state.application_state == SELECTING_DEVICE) {
			int text_y_offset_step = display_text_texture.height * 2;
			int text_y_offset = text_y_offset_step;
			for (int i = 0; i < num_devices; i++) {
				text_texture_render(&device_textures[i], 0, text_y_offset);
				text_y_offset += text_y_offset_step;
			}
		}

		text_texture_render(&state_text_texture, 0, SCREEN_HEIGHT - state_text_texture.height);

		// Update the screen
		SDL_RenderPresent(renderer);
	}

	text_texture_free(&display_text_texture);
	for (int i = 0; i < num_devices; i++) {
		text_texture_free(&device_textures[i]);
	}
	destroy();

	return 0;
}

void handle_input(SDL_Event* event, state* state)
{
	SDL_Keycode keysym = event->key.keysym.sym;
	switch (state->application_state) {

	case SELECTING_DEVICE:
		switch (keysym) {
		case SDLK_y:
		case SDLK_RETURN:
			SDL_CloseAudioDevice(state->recording_device_id);
			state->recording_device_id = audio_recording_init(state->device_index, NULL);
			if (state->device_index == -1 || !state->recording_device_id) break;
			recording_buffer_position = 0;
			SDL_PauseAudioDevice(state->recording_device_id, 0);
			/*SDL_UnlockAudioDevice(state->recording_device_id);*/
			state->application_state = RECORDING;
			break;
		case SDLK_ESCAPE:
			state->device_index = -1;
			SDL_CloseAudioDevice(state->recording_device_id);
			break;
		}
		int device_index_new = select_device(&event->key, num_devices);
		if (device_index_new == -1) break;
		if (state->device_index != -1 && device_index_new == -1) break;
		state->device_index = device_index_new;
		break;

	case RECORDING:
		switch (keysym) {
		case SDLK_y:
			SDL_PauseAudioDevice(state->recording_device_id, 0);
			break;
		case SDLK_q:
		case SDLK_ESCAPE:
			SDL_PauseAudioDevice(state->recording_device_id, 1);
			audio_buffer_destroy();
			state->application_state = SELECTING_DEVICE;
			break;
		}
		break;

	case RECORDED:
		switch (keysym) {
		case SDLK_y:
		case SDLK_RETURN:
			SDL_CloseAudioDevice(state->playback_device_id);
			state->playback_device_id = audio_playback_init(NULL);
			recording_buffer_position = 0;
			SDL_PauseAudioDevice(state->playback_device_id, 0);
			/*SDL_UnlockAudioDevice(state->playback_device_id);*/
			state->application_state = PLAYBACK;
			break;
		case SDLK_q:
		case SDLK_ESCAPE:
			audio_buffer_destroy();
			state->application_state = SELECTING_DEVICE;
			break;
		}
		break;

	case PLAYBACK:
		switch (keysym) {
		case SDLK_q:
		case SDLK_ESCAPE:
			state->application_state = RECORDED;
			break;
		}
		break;

	}
}

void handle_task(state* state)
{
	switch (state->application_state) {

	case SELECTING_DEVICE:
		break;

	case RECORDING:
		SDL_LockAudioDevice(state->recording_device_id);
		/*printf("buf pos: %u\n", recording_buffer_position);*/
		if (recording_buffer_position >= recording_buffer_position_max) {
			SDL_UnlockAudioDevice(state->recording_device_id);
			SDL_PauseAudioDevice(state->recording_device_id, 1);
			SDL_CloseAudioDevice(state->recording_device_id);
			state->application_state = RECORDED;
			break;
		}
		SDL_UnlockAudioDevice(state->recording_device_id);
		break;

	case RECORDED:
		break;

	case PLAYBACK:
		SDL_LockAudioDevice(state->playback_device_id);
		if (recording_buffer_position >= recording_buffer_position_max) {
			SDL_UnlockAudioDevice(state->playback_device_id);
			SDL_PauseAudioDevice(state->playback_device_id, 1);
			SDL_CloseAudioDevice(state->playback_device_id);
			state->application_state = RECORDED;
			break;
		}
		SDL_UnlockAudioDevice(state->playback_device_id);
		break;

	}
}

int init()
{
	if (SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO) < 0) {
		printf("SDL could not initialize! SDL Error: %s\n", SDL_GetError());
		return 0;
	}

	if (TTF_Init() == -1) {
		printf("TTF could not initialize! TTF Error: %s\n", TTF_GetError());
		return 0;
	}

	window = SDL_CreateWindow("SDL record", SDL_WINDOWPOS_UNDEFINED,
	                          SDL_WINDOWPOS_UNDEFINED, SCREEN_WIDTH,
	                          SCREEN_HEIGHT, SDL_WINDOW_SHOWN);

	if (!window) {
		printf("Window could not be created! SDL Error: %s\n", SDL_GetError());
		return 0;
	}

	renderer = SDL_CreateRenderer(window, -1, SDL_RENDERER_ACCELERATED);
	if (!renderer) {
		printf("Renderer could not be created! SDL_Error: %s\n", SDL_GetError());
		return 0;
	}

	global_font = TTF_OpenFont("FiraCode-Regular.ttf", 24);
	if (!global_font) {
		printf("Failed to load font! TTF_Error: %s\n", TTF_GetError());
		return 0;
	}

	return 1;
}

void destroy()
{
	free(recording_buffer);

	TTF_CloseFont(global_font);
	SDL_DestroyRenderer(renderer);
	SDL_DestroyWindow(window);
	TTF_Quit();
	SDL_Quit();
}

int select_device(SDL_KeyboardEvent* key_event, int max_index)
{
	SDL_Keycode keycode = key_event->keysym.sym;
	SDL_Keycode mod = key_event->keysym.mod;

	int keycode_is_num = (keycode >= SDLK_0 && keycode <= SDLK_9);
	int no_mod = (mod == 0);
	if (!keycode_is_num || !no_mod) {
		printf("Not number: %c <%u>\n", keycode, keycode);
		return -1;
	} else {
		printf("Is number: %c <%u>\n", keycode, keycode);
	}

	int index = keycode - SDLK_0;
	if (index >= max_index) return -1;

	return index;
}

void application_state_to_string(application_state state, char* string)
{
	switch (state) {
	case SELECTING_DEVICE:
		sprintf(string, "Selecting Device [0-9] -> [y/RET], ");
		break;
	case RECORDING:
		sprintf(string, "Recording");
		break;
	case RECORDED:
		sprintf(string, "Recorded");
		break;
	case PLAYBACK:
		sprintf(string, "Playing");
		break;
	}
}

int text_texture_init(text_texture* text_texture)
{
	text_texture->texture = NULL;
	text_texture->width = 0;
	text_texture->height = 0;
	return 1;
}

int text_texture_load(text_texture* text_texture, char* string)
{
	if (text_texture->texture) {
		SDL_DestroyTexture(text_texture->texture);
		text_texture->texture = NULL;
	}

	SDL_Color text_color = { 255, 255, 255 };
	SDL_Surface* text_surface = TTF_RenderText_Solid(global_font, string, text_color);
	if (!text_surface) {
		printf("Unable to render text surface! TTF_Error: %s\n", TTF_GetError());
		return 0;
	}

	SDL_Texture* texture = SDL_CreateTextureFromSurface(renderer, text_surface);
	if (!texture) {
		printf("Unable to create texture from rendered text! SDL_Error: %s\n", SDL_GetError());
		return 0;
	}

	text_texture->texture = SDL_CreateTextureFromSurface(renderer, text_surface);
	text_texture->width = text_surface->w;
	text_texture->height = text_surface->h;

	SDL_FreeSurface(text_surface);
	return 1;
}

int text_texture_render(text_texture* text_texture, int x, int y)
{
	if (!text_texture) {
		printf("text_texture uninitialized");
		return 0;
	}
	if (!text_texture->texture) {
		printf("text_texture->texture uninitialized");
		return 0;
	}

	int quad_width = text_texture->width;
	int quad_height = text_texture->height;
	SDL_Rect render_quad = { x, y, quad_width, quad_height };
	SDL_RenderCopy(renderer, text_texture->texture, NULL, &render_quad);

	return 1;
}

int text_texture_free(text_texture* text_texture)
{
	if (!text_texture->texture) {
		return 0;
	}
	SDL_DestroyTexture(text_texture->texture);
	return 1;
}

SDL_AudioDeviceID audio_recording_init(int index, void* userdata)
{
	SDL_zero(recording_spec);
	recording_spec.freq = 44100;
	recording_spec.format = AUDIO_F32;
	recording_spec.channels = 2;
	recording_spec.samples = 4096;
	recording_spec.callback = audio_recording_callback;
	recording_spec.userdata = userdata;

	SDL_AudioDeviceID device_id = SDL_OpenAudioDevice(SDL_GetAudioDeviceName(index, 1), 1, &recording_spec, &received_recording_spec, SDL_AUDIO_ALLOW_FORMAT_CHANGE);
	printf("Recording Device: %s\n", SDL_GetAudioDeviceName(index, 1));
	if (!device_id) {
		printf("Failed to open recording device! SDL Error: %s", SDL_GetError());
		return 0;
	}

	unsigned int bytes_per_sample = received_recording_spec.channels * (SDL_AUDIO_BITSIZE(received_recording_spec.format)/8);
	unsigned int bytes_per_second = received_recording_spec.freq * bytes_per_sample;
	recording_buffer_size = RECORDING_BUFFER_SECONDS * bytes_per_second;
	recording_buffer_position_max = MAX_RECORDING_SECONDS * bytes_per_second;

	recording_buffer = malloc(recording_buffer_size);
	memset(recording_buffer, 0, recording_buffer_size);

	return device_id;
}

SDL_AudioDeviceID audio_playback_init(void* userdata)
{
	if (!recording_buffer) {
		printf("Audio buffer not initialized");
		return 0;
	}

	SDL_zero(playback_spec);
	playback_spec.freq = 44100;
	playback_spec.format = AUDIO_F32;
	playback_spec.channels = 2;
	playback_spec.samples = 4096;
	playback_spec.callback = audio_playback_callback;
	playback_spec.callback = userdata;

	SDL_AudioDeviceID device_id = SDL_OpenAudioDevice(NULL, 0, &playback_spec, &received_playback_spec, SDL_AUDIO_ALLOW_FORMAT_CHANGE);
	if (!device_id) {
		printf("Failed to open playback device! SDL Error: %s", SDL_GetError());
		return 0;
	}

	return device_id;
}

int audio_device_stop(int device_id)
{
	SDL_PauseAudioDevice(device_id, 1);
	SDL_CloseAudioDevice(device_id);
	return 1;
}

void audio_buffer_destroy()
{
	free(recording_buffer);
	recording_buffer = NULL;
	recording_buffer_size = 0;
	recording_buffer_position = 0;
}

void audio_recording_callback(void* userdata, Uint8* stream, int len)
{
	memcpy(&recording_buffer[recording_buffer_position], stream, len);
	recording_buffer_position += len;
	char string[64];
	sprintf(string, "Record pos: %u %u", recording_buffer_position, len);
	printf(string, "Record pos: %u %u\n", recording_buffer_position, len);
	log_message(LOG_INFO, string);
}

void audio_playback_callback(void* userdata, Uint8* stream, int len)
{
	memcpy(stream, &recording_buffer[recording_buffer_position], len);
	recording_buffer_position += len;
	char string[64];
	sprintf(string, "Playback pos: %u %u", recording_buffer_position, len);
	printf(string, "Playback pos: %u %u", recording_buffer_position, len);
	log_message(LOG_INFO, string);
}