#include <string.h>
#include <gtk/gtk.h>
#include <vte/vte.h>

#define PCRE2_CODE_UNIT_WIDTH 0
#include <pcre2.h>

#include "portal.h"

#define REGEX_URL "https?://[^\\s<>]*[^\\s\\])}<>.,:;?!\"']"
#define KEY(v, s) (event->keyval == (v) && modifiers == (GDK_CONTROL_MASK|(s)))
#define KEY_S(v) (event->keyval == (v) && (GDK_SHIFT_MASK|modifiers) == (GDK_SHIFT_MASK|GDK_CONTROL_MASK))

#define PALETTE_SIZE 16

GtkWindow *window;
GtkNotebook *notebook;
VteRegex *url_regex;
GdkRGBA palette[PALETTE_SIZE];
double font_scale = 1;

char *default_cmd[2] = {"/bin/bash", NULL};
char *initial_cmd[4] = {"/bin/bash", NULL, NULL, NULL};
const char *colors[PALETTE_SIZE] = {
	"#000", "#c00", "#591", "#b71", "#16c", "#96a", "#299", "#ccc",
	"#333", "#f33", "#7c0", "#ed0", "#6ad", "#c8b", "#0dd", "#fff",
};

void set_font_scale(double value) {
	int i, n;
	GtkWidget *page;

	font_scale = value;

	n = gtk_notebook_get_n_pages(notebook);
	for (i = 0; i < n; i++) {
		page = gtk_notebook_get_nth_page(notebook, i);
		vte_terminal_set_font_scale(VTE_TERMINAL(page), font_scale);
	}
}

void update_show_tabs() {
	if (gtk_notebook_get_n_pages(notebook) > 1) {
		gtk_notebook_set_show_tabs(notebook, TRUE);
	} else {
		gtk_notebook_set_show_tabs(notebook, FALSE);
	}
}

void on_term_title(VteTerminal *term, gpointer user_data) {
	const char *title;
	GtkWidget *label;

	title = vte_terminal_get_termprop_string_by_id(term, VTE_PROPERTY_ID_XTERM_TITLE, NULL);
	label = gtk_notebook_get_tab_label(notebook, GTK_WIDGET(term));
	gtk_label_set_text(GTK_LABEL(label), title);
}

gboolean on_term_click(VteTerminal *term, GdkEventButton *event, gpointer user_data) {
	char *uri;

	if (event->button == 3) {
		uri = vte_terminal_match_check_event(term, (GdkEvent *)event, NULL);
		if (uri != NULL) {
			open_uri(uri);
			g_free(uri);
			return TRUE;
		}
	}
	return FALSE;
}

void on_term_exit(VteTerminal *term, int status, gpointer user_data) {
	gtk_notebook_remove_page(notebook, gtk_notebook_page_num(notebook, GTK_WIDGET(term)));

	if (gtk_notebook_get_n_pages(notebook) == 0) {
		gtk_window_close(window);
	} else {
		update_show_tabs();
	}
}

VteTerminal *get_current_term(void) {
	int page_num;

	page_num = gtk_notebook_get_current_page(notebook);
	GtkWidget *page = gtk_notebook_get_nth_page(notebook, page_num);
	return VTE_TERMINAL(page);
}

const char* get_cwd(VteTerminal *term) {
	// only works if /etc/profile.d/vte-2.91.sh was sourced in .bashrc
	GUri *uri;

	if (term != NULL) {
		uri = vte_terminal_ref_termprop_uri_by_id(term, VTE_PROPERTY_ID_CURRENT_DIRECTORY_URI);
		if (uri != NULL) {
			return g_filename_from_uri(g_uri_to_string(uri), NULL, NULL);
		}
	}

	return NULL;
}

void setup_terminal(VteTerminal *term, char **cmd) {
	int tag;
	const char *cwd;

	cwd = get_cwd(get_current_term());

	vte_terminal_spawn_async(term, VTE_PTY_DEFAULT, cwd, cmd, NULL, G_SPAWN_DEFAULT, NULL, NULL, NULL, -1, NULL, NULL, NULL);
	vte_terminal_set_cursor_blink_mode(term, VTE_CURSOR_BLINK_OFF);
	tag = vte_terminal_match_add_regex(term, url_regex, 0);
	vte_terminal_match_set_cursor_name(term, tag, "pointer");
	vte_terminal_set_colors(term, &palette[15], NULL, palette, PALETTE_SIZE);
	vte_terminal_set_bold_is_bright(term, TRUE);
	vte_terminal_set_font_scale(term, font_scale);
	vte_terminal_set_enable_bidi(term, FALSE);

	g_signal_connect(term, "window-title-changed", G_CALLBACK(on_term_title), NULL);
	g_signal_connect(term, "button-press-event", G_CALLBACK(on_term_click), NULL);
	g_signal_connect(term, "child-exited", G_CALLBACK(on_term_exit), NULL);
}

void add_tab(char **cmd) {
	GtkWidget *page, *label;
	int page_num;

	label = gtk_label_new("");
	gtk_widget_set_hexpand(label, TRUE);
	gtk_label_set_ellipsize(GTK_LABEL(label), PANGO_ELLIPSIZE_END);

	page = vte_terminal_new();
	page_num = gtk_notebook_get_current_page(notebook) + 1;
	page_num = gtk_notebook_insert_page(notebook, page, label, page_num);
	gtk_notebook_set_tab_reorderable(notebook, page, TRUE);

	update_show_tabs();
	gtk_widget_show(page);

	// needs to execute after gtk_widget_show() (for proper dimensions)
	// and before gtk_notebook_get_current_page() (so we can access the
	// previous term to get cwd)
	setup_terminal(VTE_TERMINAL(page), cmd);

	gtk_notebook_set_current_page(notebook, page_num);
	gtk_widget_grab_focus(page);
}

void move_tab(int offset) {
	int k = gtk_notebook_get_current_page(notebook);
	GtkWidget *widget = gtk_notebook_get_nth_page(notebook, k);
	if (k + offset >= 0) {
		gtk_notebook_reorder_child(notebook, widget, k + offset);
	}
}

gboolean on_key(GtkWidget *widget, GdkEventKey *event, gpointer user_data) {
	VteTerminal *term;
	GdkModifierType modifiers;

	modifiers = event->state & gtk_accelerator_get_default_mod_mask();

	if (!(modifiers & GDK_CONTROL_MASK)) {
		return FALSE;
	} else if (KEY(GDK_KEY_T, GDK_SHIFT_MASK)) {
		add_tab(default_cmd);
	} else if (KEY(GDK_KEY_Page_Up, 0)) {
		gtk_notebook_prev_page(notebook);
	} else if (KEY(GDK_KEY_Page_Down, 0)) {
		gtk_notebook_next_page(notebook);
	} else if (KEY(GDK_KEY_Page_Up, GDK_SHIFT_MASK)) {
		move_tab(-1);
	} else if (KEY(GDK_KEY_Page_Down, GDK_SHIFT_MASK)) {
		move_tab(1);
	} else if (KEY(GDK_KEY_C, GDK_SHIFT_MASK)) {
		term = get_current_term();
		vte_terminal_copy_clipboard_format(term, VTE_FORMAT_TEXT);
	} else if (KEY(GDK_KEY_V, GDK_SHIFT_MASK)) {
		term = get_current_term();
		vte_terminal_paste_clipboard(term);
	} else if (KEY_S(GDK_KEY_plus)) {
		set_font_scale(font_scale * 1.2);
	} else if (KEY_S(GDK_KEY_minus)) {
		set_font_scale(font_scale / 1.2);
	} else if (KEY_S(GDK_KEY_0)) {
		set_font_scale(1);
	} else {
		return FALSE;
	}
	return TRUE;
}

int main(int argc, char **argv) {
	int i;
	char command[128] = "";
	GError *err = NULL;
	GtkWidget *widget;

	if (argc > 1) {
		if (strcmp(argv[1], "-e") != 0) {
			fprintf(stderr, "Usage: xiterm [-e CMD]\n");
			exit(EXIT_FAILURE);
		}
		for (i = 2; i < argc; i++) {
			strncat(command, argv[i], 128 - 1 - strlen(command));
			if (i + 1 < argc) {
				strncat(command, " ", 128 - 1 - strlen(command));
			}
		}
		initial_cmd[0] = "/bin/sh";
		initial_cmd[1] = "-c";
		initial_cmd[2] = command;
	}

	url_regex = vte_regex_new_for_match(REGEX_URL, -1, PCRE2_MULTILINE, &err);
	g_assert(err == NULL);

	for (i = 0; i < PALETTE_SIZE; i++) {
		gdk_rgba_parse(palette + i, colors[i]);
	}

	gtk_init(&argc, &argv);
	widget = gtk_window_new(GTK_WINDOW_TOPLEVEL);
	window = GTK_WINDOW(widget);
	gtk_window_set_default_icon_name("utilities-terminal");
	gtk_window_set_default_size(window, 620, 340);
	gtk_window_set_title(window, "XiTerm");
	g_signal_connect(GTK_WIDGET(window), "key-press-event", G_CALLBACK(on_key), NULL);
	g_signal_connect(GTK_WIDGET(window), "destroy", G_CALLBACK(gtk_main_quit), NULL);

	portal_setup();

	widget = gtk_notebook_new();
	gtk_container_add(GTK_CONTAINER(window), widget);
	notebook = GTK_NOTEBOOK(widget);

	gtk_notebook_set_show_border(notebook, FALSE);
	gtk_widget_show_all(GTK_WIDGET(window));

	add_tab(initial_cmd);
	gtk_main();
	vte_regex_unref(url_regex);
	portal_finalize();

	return 0;
}