// deals with all of ted's menus ("open" menu, "save as" menu, etc.)

#include "ted.h"

static void menu_close_with_next(Ted *ted, Menu next) {
	ted_switch_to_buffer(ted, ted->prev_active_buffer);
	TextBuffer *buffer = ted->active_buffer;
	ted->prev_active_buffer = NULL;
	if (buffer) {
		buffer->scroll_x = ted->prev_active_buffer_scroll.x;
		buffer->scroll_y = ted->prev_active_buffer_scroll.y;
	}
	switch (ted->menu) {
	case MENU_NONE: assert(0); break;
	case MENU_OPEN:
	case MENU_SAVE_AS:
		file_selector_free(&ted->file_selector);
		buffer_clear(&ted->line_buffer);
		break;
	case MENU_WARN_UNSAVED:
		if (next != MENU_WARN_UNSAVED) {
			ted->warn_unsaved = 0;
			*ted->warn_unsaved_names = 0;
		}
		break;
	case MENU_ASK_RELOAD:
		*ted->ask_reload = 0;
		break;
	case MENU_GOTO_DEFINITION:
		definitions_selector_close(ted);
		break;
	case MENU_GOTO_LINE:
		buffer_clear(&ted->line_buffer);
		break;
	case MENU_COMMAND_SELECTOR: {
		Selector *selector = &ted->command_selector;
		buffer_clear(&ted->line_buffer);
		buffer_clear(&ted->argument_buffer);
		free(selector->entries); selector->entries = NULL; selector->n_entries = 0;
	} break;
	case MENU_SHELL:
		buffer_clear(&ted->line_buffer);
		break;
	}
	ted->menu = MENU_NONE;
	ted->selector_open = NULL;
}

void menu_close(Ted *ted) {
	menu_close_with_next(ted, 0);
}

void menu_open(Ted *ted, Menu menu) {
	if (ted->menu)
		menu_close_with_next(ted, menu);
	if (ted->find) find_close(ted);
	autocomplete_close(ted);
	ted->menu = menu;
	TextBuffer *prev_buf = ted->prev_active_buffer = ted->active_buffer;
	if (prev_buf)
		ted->prev_active_buffer_scroll = Vec2d(prev_buf->scroll_x, prev_buf->scroll_y);
	
	ted_switch_to_buffer(ted, NULL);
	*ted->warn_overwrite = 0; // clear warn_overwrite
	buffer_clear(&ted->line_buffer);
	switch (menu) {
	case MENU_NONE: assert(0); break;
	case MENU_OPEN:
		ted_switch_to_buffer(ted, &ted->line_buffer);
		ted->file_selector.create_menu = false;
		break;
	case MENU_SAVE_AS:
		ted_switch_to_buffer(ted, &ted->line_buffer);
		ted->file_selector.create_menu = true;
		break;
	case MENU_WARN_UNSAVED:
		assert(ted->warn_unsaved);
		assert(*ted->warn_unsaved_names);
		break;
	case MENU_ASK_RELOAD:
		assert(*ted->ask_reload);
		break;
	case MENU_GOTO_DEFINITION:
		definitions_selector_open(ted);
		break;
	case MENU_GOTO_LINE:
		ted_switch_to_buffer(ted, &ted->line_buffer);
		break;
	case MENU_COMMAND_SELECTOR: {
		ted_switch_to_buffer(ted, &ted->line_buffer);
		buffer_insert_char_at_cursor(&ted->argument_buffer, '1');
		Selector *selector = &ted->command_selector;
		selector->enable_cursor = true;
		selector->cursor = 0;
	} break;
	case MENU_SHELL:
		ted_switch_to_buffer(ted, &ted->line_buffer);
		ted->shell_history_pos = arr_len(ted->shell_history);
		break;
	}
}

void menu_escape(Ted *ted) {
	if (*ted->warn_overwrite) {
		// just close "are you sure you want to overwrite?"
		*ted->warn_overwrite = 0;
		ted_switch_to_buffer(ted, &ted->line_buffer);
	} else {
		menu_close(ted);
	}
}

float menu_get_width(Ted *ted) {
	const Settings *settings = ted_active_settings(ted);
	return minf(settings->max_menu_width, ted->window_width - 2.0f * settings->padding);
}

Rect menu_rect(Ted *ted) {
	Settings *settings = ted_active_settings(ted);
	float window_width = ted->window_width, window_height = ted->window_height;
	float padding = settings->padding;
	float menu_width = menu_get_width(ted);
	return rect(
		Vec2(window_width * 0.5f - 0.5f * menu_width, padding),
		Vec2(menu_width, window_height - 2 * padding)
	);
}

void menu_update(Ted *ted) {
	Menu menu = ted->menu;
	const Settings *settings = ted_active_settings(ted);
	const u32 *colors = settings->colors;
	TextBuffer *line_buffer = &ted->line_buffer;

	assert(menu);
	switch (menu) {
	case MENU_NONE: break;
	case MENU_SAVE_AS: {
		if (*ted->warn_overwrite) {
			switch (popup_update(ted, POPUP_YES_NO_CANCEL)) {
			case POPUP_NONE:
				// no option selected
				break;
			case POPUP_YES: {
				// overwrite it!
				TextBuffer *buffer = ted->prev_active_buffer;
				if (buffer) {
					buffer_save_as(buffer, ted->warn_overwrite);
				}
				menu_close(ted);
			} break;
			case POPUP_NO:
				// back to the file selector
				*ted->warn_overwrite = '\0';
				ted_switch_to_buffer(ted, &ted->line_buffer);
				break;
			case POPUP_CANCEL:
				// close "save as" menu
				menu_close(ted);
				break;
			}
		} else {
			char *selected_file = file_selector_update(ted, &ted->file_selector);
			if (selected_file) {
				TextBuffer *buffer = ted->prev_active_buffer;
				if (buffer) {
					if (fs_path_type(selected_file) != FS_NON_EXISTENT) {
						// file already exists! warn about overwriting it.
						strbuf_cpy(ted->warn_overwrite, selected_file);
						ted_switch_to_buffer(ted, NULL);
					} else {
						// create the new file.
						buffer_save_as(buffer, selected_file);
						menu_close(ted);
					}
				}
				free(selected_file);
			}
		}
	} break;
	case MENU_OPEN: {
		char *selected_file = file_selector_update(ted, &ted->file_selector);
		if (selected_file) {
			// open that file!
			menu_close(ted);
			ted_open_file(ted, selected_file);
			free(selected_file);
		}
	} break;
	case MENU_ASK_RELOAD: {
		TextBuffer *buffer = ted->prev_active_buffer;
		switch (popup_update(ted, POPUP_YES_NO)) {
		case POPUP_NONE: break;
		case POPUP_YES:
			menu_close(ted);
			if (buffer)
				buffer_reload(buffer);
			break;
		case POPUP_NO:
			menu_close(ted);
			if (buffer)
				buffer->last_write_time = timespec_to_seconds(time_last_modified(buffer->path));
			break;
		case POPUP_CANCEL: assert(0); break;
		}
	} break;
	case MENU_WARN_UNSAVED:
		switch (popup_update(ted, POPUP_YES_NO_CANCEL)) {
		case POPUP_NONE: break;
		case POPUP_YES:
			// save changes
			switch (ted->warn_unsaved) {
			case CMD_TAB_CLOSE: {
				menu_close(ted);
				TextBuffer *buffer = ted->active_buffer;
				command_execute(ted, CMD_SAVE, 1); 
				if (!buffer_unsaved_changes(buffer)) {
					command_execute(ted, CMD_TAB_CLOSE, 1);
				}
			} break;
			case CMD_QUIT:
				menu_close(ted);
				if (ted_save_all(ted)) {
					command_execute(ted, CMD_QUIT, 1);
				}
				break;
			default:
				assert(0);
				break;
			}
			break;
		case POPUP_NO: {
			// pass in an argument of 2 to override dialog
			Command cmd = ted->warn_unsaved;
			menu_close(ted);
			command_execute(ted, cmd, 2);
		} break;
		case POPUP_CANCEL:
			menu_close(ted);
			break;
		}
		break;
	case MENU_GOTO_DEFINITION: {
		definitions_selector_update(ted);
	} break;
	case MENU_GOTO_LINE: {
		char *contents = str32_to_utf8_cstr(buffer_get_line(line_buffer, 0));
		char *end;
		long line_number = strtol(contents, &end, 0);
		TextBuffer *buffer = ted->prev_active_buffer;
		if (*contents != '\0' && *end == '\0') {
			if (line_number < 1) line_number = 1;
			if (line_number > (long)buffer->nlines) line_number = (long)buffer->nlines; 
			BufferPos pos = {(u32)line_number - 1, 0};
			
			if (line_buffer->line_buffer_submitted) {
				// let's go there!
				menu_close(ted);
				buffer_cursor_move_to_pos(buffer, pos);
				buffer_center_cursor(buffer);
			} else {
				// scroll to the line
				buffer_scroll_center_pos(buffer, pos);
			}
		}
		line_buffer->line_buffer_submitted = false;
		free(contents);
	} break;
	case MENU_COMMAND_SELECTOR: {
		Selector *selector = &ted->command_selector;
		SelectorEntry *entries = selector->entries = calloc(CMD_COUNT, sizeof *selector->entries);
		char *search_term = str32_to_utf8_cstr(buffer_get_line(line_buffer, 0));
		if (entries) {
			SelectorEntry *entry = entries;
			for (Command c = 0; c < CMD_COUNT; ++c) {
				const char *name = command_to_str(c);
				if (c != CMD_UNKNOWN && *name && strstr_case_insensitive(name, search_term)) {
					entry->name = name;
					entry->color = colors[COLOR_TEXT];
					++entry;
				}
			}
			selector->n_entries = (u32)(entry - entries);
			selector_sort_entries_by_name(selector);
		}

		char *chosen_command = selector_update(ted, &ted->command_selector);
		if (chosen_command) {
			Command c = command_from_str(chosen_command);
			if (c != CMD_UNKNOWN) {
				char *argument = str32_to_utf8_cstr(buffer_get_line(&ted->argument_buffer, 0)), *endp = NULL;
				long long arg = strtoll(argument, &endp, 0);

				if (*endp == '\0') {
					menu_close(ted);
					command_execute(ted, c, arg);
				}

				free(argument);
			}

			free(chosen_command);
		}
		free(search_term);
	} break;
	case MENU_SHELL:
		if (line_buffer->line_buffer_submitted) {
			char *command = str32_to_utf8_cstr(buffer_get_line(line_buffer, 0));
			if (ted->shell_history_pos == arr_len(ted->shell_history) || line_buffer->modified) {
				arr_add(ted->shell_history, command);
			}
			menu_close(ted);
			strbuf_cpy(ted->build_dir, ted->cwd);
			build_start_with_command(ted, command);
		}
		break;
	}
}

void menu_render(Ted *ted) {
	Menu menu = ted->menu;
	assert(menu);
	const Settings *settings = ted_active_settings(ted);
	const u32 *colors = settings->colors;
	const float window_width = ted->window_width, window_height = ted->window_height;
	Font *font_bold = ted->font_bold, *font = ted->font;
	const float char_height = text_font_char_height(font);
	const float char_height_bold = text_font_char_height(font_bold);
	const float line_buffer_height = ted_line_buffer_height(ted);
	
	// render backdrop
	gl_geometry_rect(rect(Vec2(0, 0), Vec2(window_width, window_height)), colors[COLOR_MENU_BACKDROP]);
	gl_geometry_draw();

	if (*ted->warn_overwrite) {
		const char *path = ted->warn_overwrite;
		const char *filename = path_filename(path);
		char title[64] = {0}, body[1024] = {0};
		strbuf_printf(title, "Overwrite %s?", filename);
		strbuf_printf(body, "Are you sure you want to overwrite %s?", path);
		popup_render(ted, POPUP_YES_NO_CANCEL, title, body);
		return;
	}

	
	float padding = settings->padding;
	Rect bounds = menu_rect(ted);
	float x1, y1, x2, y2;
	rect_coords(bounds, &x1, &y1, &x2, &y2);

	if (menu == MENU_OPEN || menu == MENU_SAVE_AS || menu == MENU_GOTO_DEFINITION || menu == MENU_COMMAND_SELECTOR) {
		// menu rectangle & border
		gl_geometry_rect(bounds, colors[COLOR_MENU_BG]);
		gl_geometry_rect_border(bounds, settings->border_thickness, colors[COLOR_BORDER]);
		gl_geometry_draw();
		
		x1 += padding;
		y1 += padding;
		x2 -= padding;
		y2 -= padding;
	}
		

	switch (menu) {
	case MENU_NONE: assert(0); break;
	case MENU_WARN_UNSAVED: {
		char title[64] = {0}, body[1024] = {0};
		strbuf_printf(title, "Save changes?");
		strbuf_printf(body, "Do you want to save your changes to %s?", ted->warn_unsaved_names);
		popup_render(ted, POPUP_YES_NO_CANCEL, title, body);
	} break;
	case MENU_ASK_RELOAD: {
		char title[64] = {0}, body[1024] = {0};
		strbuf_printf(title, "Reload %s?", ted->ask_reload);
		strbuf_printf(body, "%s has been changed by another program. Do you want to reload it?", ted->ask_reload);
		popup_render(ted, POPUP_YES_NO, title, body);
	} break;
	case MENU_OPEN:
	case MENU_SAVE_AS: {

		if (menu == MENU_OPEN) {
			text_utf8(font_bold, "Open...", x1, y1, colors[COLOR_TEXT]);
		} else if (menu == MENU_SAVE_AS) {
			text_utf8(font_bold, "Save as...", x1, y1, colors[COLOR_TEXT]);
		}
		text_render(font_bold);
			
		y1 += char_height_bold * 0.75f + padding;

		FileSelector *fs = &ted->file_selector;
		fs->bounds = rect4(x1, y1, x2, y2);
		file_selector_render(ted, fs);
	} break;
	case MENU_GOTO_DEFINITION: {
		definitions_selector_render(ted, rect4(x1, y1, x2, y2));
	} break;
	case MENU_GOTO_LINE: {
		float menu_height = char_height + 2 * padding;
		Rect r = rect(Vec2(padding, window_height - menu_height - padding), Vec2(window_width - 2 * padding, menu_height));
		gl_geometry_rect(r, colors[COLOR_MENU_BG]);
		gl_geometry_rect_border(r, settings->border_thickness, colors[COLOR_BORDER]);
		const char *text = "Go to line...";
		vec2 text_size = text_get_size_vec2(font_bold, text);
		rect_coords(r, &x1, &y1, &x2, &y2);
		x1 += padding;
		y1 += padding;
		x2 -= padding;
		y2 -= padding;
		// render "Go to line" text
		text_utf8(font_bold, text, x1, 0.5f * (y1 + y2 - text_size.y), colors[COLOR_TEXT]);
		x1 += text_size.x + padding;
		gl_geometry_draw();
		text_render(font_bold);
		
		// line buffer
		buffer_render(&ted->line_buffer, rect4(x1, y1, x2, y2));
	} break;
	case MENU_COMMAND_SELECTOR: {
		// argument field
		const char *text = "Argument";
		text_utf8(font_bold, text, x1, y1, colors[COLOR_TEXT]);
		float x = x1 + text_get_size_vec2(font_bold, text).x + padding;
		buffer_render(&ted->argument_buffer, rect4(x, y1, x2, y1 + line_buffer_height));

		y1 += line_buffer_height + padding;

		Selector *selector = &ted->command_selector;
		selector->bounds = rect4(x1, y1, x2, y2);
		selector_render(ted, selector);

		text_render(font_bold);
	} break;
	case MENU_SHELL: {		
		bounds.size.y = line_buffer_height + 2 * padding;
		rect_coords(bounds, &x1, &y1, &x2, &y2);
		
		gl_geometry_rect(bounds, colors[COLOR_MENU_BG]);
		gl_geometry_rect_border(bounds, settings->border_thickness, colors[COLOR_BORDER]);
		gl_geometry_draw();
		
		x1 += padding;
		y1 += padding;
		x2 -= padding;
		y2 -= padding;
		
		const char *text = "Run";
		text_utf8(font_bold, text, x1, y1, colors[COLOR_TEXT]);
		x1 += text_get_size_vec2(font_bold, text).x + padding;
		text_render(font_bold);
		
		buffer_render(&ted->line_buffer, rect4(x1, y1, x2, y2));
	} break;
	}
}

void menu_shell_move(Ted *ted, int direction) {
	TextBuffer *line_buffer = &ted->line_buffer;
	if (line_buffer->modified) {
		// don't do it if the command has been edited
		return;
	}
	i64 pos = ted->shell_history_pos;
	pos += direction;
	if (pos >= 0 && pos <= arr_len(ted->shell_history)) {
		ted->shell_history_pos = (u32)pos;
		buffer_clear(line_buffer);
		if (pos == arr_len(ted->shell_history)) {
			// bottom of history; just clear line buffer
		} else {
			line_buffer->store_undo_events = false;
			buffer_insert_utf8_at_cursor(line_buffer, ted->shell_history[pos]);
			line_buffer->store_undo_events = true;
			line_buffer->modified = false;
		}
		// line_buffer->x/y1/2 are wrong (all 0), because of buffer_clear
		line_buffer->center_cursor_next_frame = true;
	}
}

void menu_shell_up(Ted *ted) {
	menu_shell_move(ted, -1);
}
void menu_shell_down(Ted *ted) {
	menu_shell_move(ted, +1);
}