/*
 gui-readline.c : irssi

    Copyright (C) 1999 Timo Sirainen

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/

#include "module.h"
#include "signals.h"
#include "server.h"
#include "misc.h"

#include "completion.h"
#include "command-history.h"
#include "keyboard.h"
#include "translation.h"
#include "windows.h"

#include "screen.h"
#include "gui-entry.h"
#include "gui-windows.h"

#include <signal.h>

typedef void (*ENTRY_REDIRECT_KEY_FUNC) (int key, void *data, SERVER_REC *server, WI_ITEM_REC *item);
typedef void (*ENTRY_REDIRECT_ENTRY_FUNC) (const char *line, void *data, SERVER_REC *server, WI_ITEM_REC *item);

typedef struct {
	SIGNAL_FUNC func;
        int key;
	void *data;
} ENTRY_REDIRECT_REC;

static ENTRY_REDIRECT_REC *redir;

static char *savebuffer;
static int readtag, sigint_count = 0;
static time_t idle_time;

static void handle_key_redirect(int key)
{
	ENTRY_REDIRECT_KEY_FUNC func;
	void *data;

	func = (ENTRY_REDIRECT_KEY_FUNC) redir->func;
	data = redir->data;
	g_free_and_null(redir);

	if (func != NULL)
		func(key, data, active_win->active_server, active_win->active);

	gui_entry_remove_perm_prompt();
	window_update_prompt(active_win);
}

static void handle_entry_redirect(const char *line)
{
	ENTRY_REDIRECT_ENTRY_FUNC func;
	void *data;

	func = (ENTRY_REDIRECT_ENTRY_FUNC) redir->func;
	data = redir->data;
	g_free_and_null(redir);

	if (func != NULL)
		func(line, data, active_win->active_server, active_win->active);

	gui_entry_remove_perm_prompt();
	window_update_prompt(active_win);
}

static void window_prev_page(void)
{
	GUI_WINDOW_REC *gui = WINDOW_GUI(active_win);

	gui_window_scroll(active_win, -(gui->parent->last_line-gui->parent->first_line)/2);
}

static void window_next_page(void)
{
	GUI_WINDOW_REC *gui = WINDOW_GUI(active_win);

	gui_window_scroll(active_win, (gui->parent->last_line-gui->parent->first_line)/2);
}

static const char *get_key_name(int key)
{
	static char str[MAX_INT_STRLEN];

	switch (key) {
	case KEY_HOME:
		return "Home";
	case KEY_END:
		return "End";
	case KEY_PPAGE:
		return "Prior";
	case KEY_NPAGE:
		return "Next";
	case KEY_UP:
		return "Up";
	case KEY_DOWN:
		return "Down";
	case KEY_LEFT:
		return "Left";
	case KEY_RIGHT:
		return "Right";
	default:
		ltoa(str, key);
		return str;
	}
}

void handle_key(int key)
{
	const char *text;
	char *str;
	int c;

	/* Quit if we get 5 CTRL-C's in a row. */
	if (key != 3)
		sigint_count = 0;
	else if (++sigint_count >= 5)
		raise(SIGTERM);

	idle_time = time(NULL);

	if (redir != NULL && redir->key) {
		handle_key_redirect(key);
		return;
	}

	switch (key)
	{
	case 27:
		c = getch();
		if (c == toupper(c) && c != tolower(c))
			str = g_strdup_printf("ALT-SHIFT-%c", c);
		else {
			if (c < 256)
				str = g_strdup_printf("ALT-%c", toupper(c));
			else
				str = g_strdup_printf("ALT-%s", get_key_name(c));
		}
		key_pressed(str, NULL);
		g_free(str);
		break;

	case KEY_HOME:
		/* home */
		gui_entry_set_pos(0);
		break;
	case KEY_END:
		/* end */
		gui_entry_set_pos(strlen(gui_entry_get_text()));
		break;
	case KEY_PPAGE:
		/* page up */
		window_prev_page();
		break;
	case KEY_NPAGE:
		/* page down */
		window_next_page();
		break;

	case KEY_UP:
		/* up */
		text = command_history_prev(active_win, gui_entry_get_text());
		gui_entry_set_text(text);
		break;
	case KEY_DOWN:
		/* down */
		text = command_history_next(active_win, gui_entry_get_text());
		gui_entry_set_text(text);
		break;
	case KEY_RIGHT:
		/* right */
		gui_entry_move_pos(1);
		break;
	case KEY_LEFT:
		/* left */
		gui_entry_move_pos(-1);
		break;

	case 21:
		/* Ctrl-U, clear line */
                g_free_not_null(savebuffer);
		savebuffer = g_strdup(gui_entry_get_text());

		gui_entry_set_text("");
		break;
	case 25:
		/* Ctrl-Y, write last ^U'd line */
		gui_entry_insert_text(savebuffer);
		break;
	case 9:
		key_pressed("Tab", NULL);
		break;

	case 8:
	case 127:
	case KEY_BACKSPACE:
		gui_entry_erase(1);
		break;

	case 23:
		/* C-w - erase word to the left of marker */
		gui_entry_erase_word();
		break;

	case 4:
	case KEY_DC:
		if (gui_entry_get_pos() < strlen(gui_entry_get_text())) {
			gui_entry_move_pos(1);
			gui_entry_erase(1);
		}
		break;

	case 11:
		/* C-K - erase the rest of the line */
		c = gui_entry_get_pos();
		gui_entry_set_pos(strlen(gui_entry_get_text()));
		gui_entry_erase(strlen(gui_entry_get_text()) - c);
		break;

	case 0:
		/* Ctrl-space - ignore */
		break;
	case 1:
		/* C-A, home */
		gui_entry_set_pos(0);
		break;
	case 5:
		/* C-E, end */
		gui_entry_set_pos(strlen(gui_entry_get_text()));
		break;

	case '\n':
	case 13:
		key_pressed("Return", NULL);

		str = gui_entry_get_text();
		if (*str == '\0') break;

		translate_output(str);

		if (redir == NULL)
			signal_emit("send command", 3, str, active_win->active_server, active_win->active);
		else
			handle_entry_redirect(str);

		command_history_add(active_win, gui_entry_get_text(), FALSE);
		gui_entry_set_text("");
		command_history_clear_pos(active_win);
		break;

	default:
		if (key > 0 && key < 32) {
			str = g_strdup_printf("CTRL-%c", key == 31 ? '-' : key+'A'-1);
			key_pressed(str, NULL);
			g_free(str);
			break;
		}

		if (key < 256) {
			char str[2];

			str[0] = toupper(key); str[1] = '\0';
			key_pressed(str, NULL);
			gui_entry_insert_char((char) key);
		}
		break;
	}
}

void readline(void)
{
	int key;

	for (;;) {
		key = getch();
		if (key == ERR) break;

		handle_key(key);
	}
}

time_t get_idle_time(void)
{
	return idle_time;
}

static void sig_prev_page(void)
{
	window_prev_page();
}

static void sig_next_page(void)
{
	window_next_page();
}

static void sig_change_window(const char *data)
{
	signal_emit("command window goto", 3, data, active_win->active_server, active_win->active);
}

static void sig_completion(void)
{
	char *line;
	int pos;

	pos = gui_entry_get_pos();

	line = word_complete(active_win, gui_entry_get_text(), &pos);
	if (line != NULL) {
		gui_entry_set_text(line);
		gui_entry_set_pos(pos);
		g_free(line);
	}
}

static void sig_replace(void)
{
	char *line;
	int pos;

	pos = gui_entry_get_pos();

	line = auto_word_complete(gui_entry_get_text(), &pos);
	if (line != NULL) {
		gui_entry_set_text(line);
		gui_entry_set_pos(pos);
		g_free(line);
	}
}

static void sig_prev_window(void)
{
	signal_emit("command window prev", 3, "", active_win->active_server, active_win->active);
}

static void sig_next_window(void)
{
	signal_emit("command window next", 3, "", active_win->active_server, active_win->active);
}

static void sig_up_window(void)
{
	signal_emit("command window up", 3, "", active_win->active_server, active_win->active);
}

static void sig_down_window(void)
{
	signal_emit("command window down", 3, "", active_win->active_server, active_win->active);
}

static void sig_window_goto_active(void)
{
	signal_emit("command window goto", 3, "active", active_win->active_server, active_win->active);
}

static void sig_prev_window_item(void)
{
	SERVER_REC *server;
	GSList *pos;

	if (active_win->items != NULL)
		signal_emit("command window item prev", 3, "", active_win->active_server, active_win->active);
	else if (active_win->active_server != NULL) {
		/* change server */
		pos = g_slist_find(servers, active_win->active_server);
		server = pos->next != NULL ? pos->next->data : servers->data;
		signal_emit("command window server", 3, server->tag, active_win->active_server, active_win->active);
	}
}

static void sig_next_window_item(void)
{
	SERVER_REC *server;
	int index;

	if (active_win->items != NULL)
		signal_emit("command window item next", 3, "", active_win->active_server, active_win->active);
	else if (active_win->active_server != NULL) {
		/* change server */
		index = g_slist_index(servers, active_win->active_server);
		server = index > 0 ? g_slist_nth(servers, index-1)->data :
			g_slist_last(servers)->data;
		signal_emit("command window server", 3, server->tag, active_win->active_server, active_win->active);
	}
}

static void sig_addchar(const char *data)
{
	gui_entry_insert_char(*data);
}

static void sig_window_auto_changed(WINDOW_REC *window)
{
	command_history_next(active_win, gui_entry_get_text());
	gui_entry_set_text("");
}

static void sig_gui_entry_redirect(SIGNAL_FUNC func, const char *entry, gpointer key, void *data)
{
	redir = g_new0(ENTRY_REDIRECT_REC, 1);
	redir->func = func;
	redir->key = key != NULL;
	redir->data = data;

	gui_entry_set_perm_prompt(entry);
}

void gui_readline_init(void)
{
	static char changekeys[] = "1234567890QWERTYUIO";
	char *key, data[MAX_INT_STRLEN];
	int n;

	savebuffer = NULL;
	redir = NULL;
	idle_time = time(NULL);
	readtag = g_input_add(0, G_INPUT_READ, (GInputFunction) readline, NULL);

	key_bind("completion", "Nick completion", "Tab", NULL, (SIGNAL_FUNC) sig_completion);
	key_bind("check replaces", "Check word replaces", " ", NULL, (SIGNAL_FUNC) sig_replace);
	key_bind("check replaces", NULL, "Return", NULL, (SIGNAL_FUNC) sig_replace);
	key_bind("window prev", "Previous window", "CTRL-P", NULL, (SIGNAL_FUNC) sig_prev_window);
	key_bind("window prev", NULL, "ALT-Left", NULL, (SIGNAL_FUNC) sig_prev_window);
	key_bind("window next", "Next window", "CTRL-N", NULL, (SIGNAL_FUNC) sig_next_window);
	key_bind("window next", NULL, "ALT-Right", NULL, (SIGNAL_FUNC) sig_next_window);
	key_bind("window up", "Upper window", "ALT-Up", NULL, (SIGNAL_FUNC) sig_up_window);
	key_bind("window down", "Lower window", "ALT-Down", NULL, (SIGNAL_FUNC) sig_down_window);
	key_bind("window active", "Go to next window with the highest activity", "ALT-A", NULL, (SIGNAL_FUNC) sig_window_goto_active);
	key_bind("window item next", "Next channel", "CTRL-X", NULL, (SIGNAL_FUNC) sig_next_window_item);
	key_bind("window item prev", "Previous channel", NULL, NULL, (SIGNAL_FUNC) sig_prev_window_item);

	key_bind("redraw", "Redraw window", "CTRL-L", NULL, (SIGNAL_FUNC) irssi_redraw);
	key_bind("prev page", "Previous page", "ALT-P", NULL, (SIGNAL_FUNC) sig_prev_page);
	key_bind("next page", "Next page", "ALT-N", NULL, (SIGNAL_FUNC) sig_next_page);

	key_bind("special char", "Insert special character", "CTRL-B", "\x02", (SIGNAL_FUNC) sig_addchar);
	key_bind("special char", NULL, "CTRL--", "\x1f", (SIGNAL_FUNC) sig_addchar);
	key_bind("special char", NULL, "CTRL-C", "\x03", (SIGNAL_FUNC) sig_addchar);
	key_bind("special char", NULL, "CTRL-V", "\x16", (SIGNAL_FUNC) sig_addchar);
	key_bind("special char", NULL, "CTRL-G", "\x07", (SIGNAL_FUNC) sig_addchar);
	key_bind("special char", NULL, "CTRL-O", "\x0f", (SIGNAL_FUNC) sig_addchar);

	for (n = 0; changekeys[n] != '\0'; n++) {
		key = g_strdup_printf("ALT-%c", changekeys[n]);
		ltoa(data, n+1);
		key_bind("window change", "Change window", key, data, (SIGNAL_FUNC) sig_change_window);
		g_free(key);
	}

	signal_add("window changed automatic", (SIGNAL_FUNC) sig_window_auto_changed);
	signal_add("gui entry redirect", (SIGNAL_FUNC) sig_gui_entry_redirect);
}

void gui_readline_deinit(void)
{
	g_free_not_null(savebuffer);
	g_source_remove(readtag);

	key_unbind("completion", (SIGNAL_FUNC) sig_completion);
	key_unbind("check replaces", (SIGNAL_FUNC) sig_replace);
	key_unbind("window prev", (SIGNAL_FUNC) sig_prev_window);
	key_unbind("window next", (SIGNAL_FUNC) sig_next_window);
	key_unbind("window active", (SIGNAL_FUNC) sig_window_goto_active);
	key_unbind("window item next", (SIGNAL_FUNC) sig_next_window_item);
	key_unbind("window item prev", (SIGNAL_FUNC) sig_prev_window_item);

	key_unbind("redraw", (SIGNAL_FUNC) irssi_redraw);
	key_unbind("prev page", (SIGNAL_FUNC) sig_prev_page);
	key_unbind("next page", (SIGNAL_FUNC) sig_next_page);

	key_unbind("special char", (SIGNAL_FUNC) sig_addchar);
	key_unbind("window change", (SIGNAL_FUNC) sig_change_window);

	signal_remove("window changed automatic", (SIGNAL_FUNC) sig_window_auto_changed);
	signal_remove("gui entry redirect", (SIGNAL_FUNC) sig_gui_entry_redirect);
}