/* $Id: window-copy.c,v 1.3 2007-11-22 19:26:20 nicm Exp $ */

/*
 * Copyright (c) 2007 Nicholas Marriott <nicm@users.sourceforge.net>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
 * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
 * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>

#include <string.h>

#include "tmux.h"

void	window_copy_init(struct window *);
void	window_copy_resize(struct window *, u_int, u_int);
void	window_copy_draw(struct window *, struct buffer *, u_int, u_int);
void	window_copy_key(struct window *, int);

void	window_copy_draw_position(struct window *, struct screen_draw_ctx *);

u_int	window_copy_line_length(struct window *);
void	window_copy_move_cursor(struct window *);
void	window_copy_cursor_left(struct window *);
void	window_copy_cursor_right(struct window *);
void	window_copy_cursor_up(struct window *);
void	window_copy_cursor_down(struct window *);
void	window_copy_draw_lines(struct window *, u_int, u_int);
void	window_copy_scroll_left(struct window *, u_int);
void	window_copy_scroll_right(struct window *, u_int);
void	window_copy_scroll_up(struct window *, u_int);
void	window_copy_scroll_down(struct window *, u_int);

const struct window_mode window_copy_mode = {
	window_copy_init,
	window_copy_resize,
	window_copy_draw,
	window_copy_key
};

struct window_copy_mode_data {
	u_int	ox;
	u_int	oy;
	u_int	cx;
	u_int	cy;
	u_int	size;

	struct screen_draw_sel sel;
};

void
window_copy_init(struct window *w)
{
	struct window_copy_mode_data	*data;

	w->modedata = data = xmalloc(sizeof *data);
	data->ox = data->oy = 0;
	data->cx = w->screen.cx;
	data->cy = w->screen.cy;
	data->size = w->screen.hsize;

	memset(&data->sel, 0, sizeof data->sel);

	w->screen.mode |= (MODE_BACKGROUND|MODE_BGCURSOR);
}

void
window_copy_resize(unused struct window *w, unused u_int sx, unused u_int sy)
{
}

void
window_copy_draw_position(struct window *w, struct screen_draw_ctx *ctx)
{
	struct window_copy_mode_data	*data = w->modedata;
	char				*ptr, buf[32];
	size_t	 			 len;

	len = xsnprintf(
	    buf, sizeof buf, "[%u,%u/%u]", data->ox, data->oy, data->size);
	if (len <= screen_size_x(ctx->s))
		ptr = buf;
	else {
		ptr = buf + len - screen_size_x(ctx->s);
		len -= len - screen_size_x(ctx->s);
	}
	
	screen_draw_cells(ctx, 0, 0, screen_size_x(ctx->s) - len);
	
	screen_draw_move(ctx, screen_size_x(ctx->s) - len, 0);
	screen_draw_set_attributes(ctx, 0, status_colour);
	buffer_write(ctx->b, ptr, len);
}

void
window_copy_draw(struct window *w, struct buffer *b, u_int py, u_int ny)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct screen			*s = &w->screen;
	struct screen_draw_ctx		 ctx;

	if (s->hsize != data->size) {
		data->oy += s->hsize - data->size;
		data->size = s->hsize;
	}

	screen_draw_start(&ctx, s, b, data->ox, data->oy);
	memcpy(&ctx.sel, &data->sel, sizeof ctx.sel);
	ctx.sel.ex = data->cx + data->ox;
	ctx.sel.ey = data->size + data->cy - data->oy;
	if (py != 0)
		screen_draw_lines(&ctx, py, ny);
	else if (ny > 1)
		screen_draw_lines(&ctx, py + 1, ny - 1);

	if (py == 0)
		window_copy_draw_position(w, &ctx);

	screen_draw_stop(&ctx);

	input_store_two(b, CODE_CURSORMOVE, data->cy + 1, data->cx  + 1);
}

void
window_copy_key(struct window *w, int key)
{
	struct window_copy_mode_data	*data = w->modedata;
	u_int				 oy, sy;
	
	sy = screen_size_y(&w->screen);
	oy = data->oy;

	switch (key) {
	case 'Q':
	case 'q':
		w->mode = NULL;
		xfree(w->modedata);

		w->screen.mode &= ~MODE_BACKGROUND;

		recalculate_sizes();
		server_redraw_window_all(w);
		return;
	case 'h':
	case KEYC_LEFT:
		window_copy_cursor_left(w);
		return;
	case 'l':
	case KEYC_RIGHT:
		window_copy_cursor_right(w);
 		return;
	case 'k':
	case 'K':
	case KEYC_UP:
		window_copy_cursor_up(w);
		return;
	case 'j':
	case 'J':
	case KEYC_DOWN:
		window_copy_cursor_down(w);
		return;
	case '\025':	/* C-u */
	case KEYC_PPAGE:
		if (data->oy + sy > data->size)
			data->oy = data->size;
		else
			data->oy += sy;
		break;
	case '\006':	/* C-f */
	case KEYC_NPAGE:
		if (data->oy < sy)
			data->oy = 0;
		else
			data->oy -= sy;
		break;
	case '\000':	/* C-space */
		data->sel.flag = !data->sel.flag;
		data->sel.sx = data->cx + data->ox;
		data->sel.sy = data->size + data->cy - data->oy;
		oy = -1;	/* XXX */
		break;
	/* XXX start/end of line, next word, prev word */
	}
	if (data->oy != oy) {
		server_redraw_window_all(w);
		window_copy_move_cursor(w);
	}
}

void
window_copy_move_cursor(struct window *w)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct client			*c;
	u_int		 		 i;
	struct hdr			 hdr;
	size_t				 size;

	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
		c = ARRAY_ITEM(&clients, i);
		if (c == NULL || c->session == NULL)
			continue;
		if (!session_has(c->session, w))
			continue;

		buffer_ensure(c->out, sizeof hdr);
		buffer_add(c->out, sizeof hdr);
		size = BUFFER_USED(c->out);

		input_store_two(
		    c->out, CODE_CURSORMOVE, data->cy + 1, data->cx + 1);

		size = BUFFER_USED(c->out) - size;
		hdr.type = MSG_DATA;
		hdr.size = size;
		memcpy(BUFFER_IN(c->out) - size - sizeof hdr, &hdr, sizeof hdr);
	}
}

void
window_copy_cursor_left(struct window *w)
{
	struct window_copy_mode_data	*data = w->modedata;

	if (data->ox + data->cx == 0)
		return;

	if (data->cx == 0)
		window_copy_scroll_right(w, 1); 
	else {
		data->cx--;
		if (data->sel.flag)
			window_copy_draw_lines(w, data->cy, 1);
	}
	window_copy_move_cursor(w);
}

void
window_copy_cursor_right(struct window *w)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct screen			*s = &w->screen;

	if (data->ox + data->cx == SHRT_MAX)
		return;
	
	if (data->cx == screen_last_x(s))
		window_copy_scroll_left(w, 1);
	else {
		data->cx++;
		if (data->sel.flag)
			window_copy_draw_lines(w, data->cy, 1);
	}
	window_copy_move_cursor(w);
}

void
window_copy_cursor_up(struct window *w)
{
	struct window_copy_mode_data	*data = w->modedata;

	if (data->cy == 0 && data->oy == data->size)
		return;

	if (data->cy == 0)
		window_copy_scroll_down(w, 1);
	else {
		data->cy--;
		if (data->sel.flag)	
			window_copy_draw_lines(w, data->cy, 2);
	}
	window_copy_move_cursor(w);
}

void
window_copy_cursor_down(struct window *w)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct screen			*s = &w->screen;

	if (data->cy == screen_last_y(s) && data->oy == 0)
		return;

	if (data->cy == screen_last_y(s))
		window_copy_scroll_up(w, 1);
	else {
		data->cy++;
		if (data->sel.flag)	
			window_copy_draw_lines(w, data->cy - 1, 2);
	}
	window_copy_move_cursor(w);
}

void
window_copy_draw_lines(struct window *w, u_int py, u_int ny)
{
	struct client		*c;
	struct buffer		*b;
	u_int		 	 i;
	struct hdr		 hdr;
	size_t			 size;

	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
		c = ARRAY_ITEM(&clients, i);
		if (c == NULL || c->session == NULL)
			continue;
		if (!session_has(c->session, w))
			continue;
		b = c->out;

		buffer_ensure(b, sizeof hdr);
		buffer_add(b, sizeof hdr);
		size = BUFFER_USED(b);
		
		window_copy_draw(w, b, py, ny);

		size = BUFFER_USED(b) - size;
		hdr.type = MSG_DATA;
		hdr.size = size;
		memcpy(BUFFER_IN(b) - size - sizeof hdr, &hdr, sizeof hdr);
	}	
}

void
window_copy_scroll_left(struct window *w, u_int nx)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct screen			*s = &w->screen;
	struct screen_draw_ctx		 ctx;
	struct client			*c;
	struct buffer			*b;
	u_int		 		 i, j;
	struct hdr			 hdr;
	size_t				 size;

	if (data->ox >= SHRT_MAX - nx)
		nx = SHRT_MAX - data->ox;
	if (nx == 0)
		return;
	data->ox += nx;

	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
		c = ARRAY_ITEM(&clients, i);
		if (c == NULL || c->session == NULL)
			continue;
		if (!session_has(c->session, w))
			continue;
		b = c->out;

		buffer_ensure(b, sizeof hdr);
		buffer_add(b, sizeof hdr);
		size = BUFFER_USED(b);
		
		screen_draw_start(&ctx, s, b, data->ox, data->oy);
		memcpy(&ctx.sel, &data->sel, sizeof ctx.sel);
		ctx.sel.ex = data->cx + data->ox;
		ctx.sel.ey = data->size + data->cy - data->oy;
		for (j = 1; j < screen_size_y(s); j++) {
			screen_draw_move(&ctx, 0, j);
			input_store_one(b, CODE_DELETECHARACTER, nx);
		}
		for (j = 0; j < nx; j++)
			screen_draw_column(&ctx, screen_last_x(s) - j);
		window_copy_draw_position(w, &ctx);
		screen_draw_stop(&ctx);
		
		size = BUFFER_USED(b) - size;
		hdr.type = MSG_DATA;
		hdr.size = size;
		memcpy(BUFFER_IN(b) - size - sizeof hdr, &hdr, sizeof hdr);
	}	
}

void
window_copy_scroll_right(struct window *w, u_int nx)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct screen			*s = &w->screen;
	struct screen_draw_ctx		 ctx;
	struct client			*c;
	struct buffer			*b;
	u_int		 		 i, j;
	struct hdr			 hdr;
	size_t				 size;

	if (data->ox < nx)
		nx = data->ox;
	if (nx == 0)
		return;
	data->ox -= nx;

	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
		c = ARRAY_ITEM(&clients, i);
		if (c == NULL || c->session == NULL)
			continue;
		if (!session_has(c->session, w))
			continue;
		b = c->out;

		buffer_ensure(b, sizeof hdr);
		buffer_add(b, sizeof hdr);
		size = BUFFER_USED(b);
		
		screen_draw_start(&ctx, s, b, data->ox, data->oy);
		memcpy(&ctx.sel, &data->sel, sizeof ctx.sel);
		ctx.sel.ex = data->cx + data->ox;
		ctx.sel.ey = data->size + data->cy - data->oy;
		for (j = 1; j < screen_size_y(s); j++) {
			screen_draw_move(&ctx, 0, j);
			input_store_one(b, CODE_INSERTCHARACTER, nx);
		}
		for (j = 0; j < nx; j++)
			screen_draw_column(&ctx, j);
		window_copy_draw_position(w, &ctx);
		screen_draw_stop(&ctx);
		
		size = BUFFER_USED(b) - size;
		hdr.type = MSG_DATA;
		hdr.size = size;
		memcpy(BUFFER_IN(b) - size - sizeof hdr, &hdr, sizeof hdr);
	}	
}

void
window_copy_scroll_up(struct window *w, u_int ny)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct screen			*s = &w->screen;
	struct screen_draw_ctx		 ctx;
	struct client			*c;
	u_int		 		 i;
	struct hdr			 hdr;
	size_t				 size;

	if (data->oy < ny)
		ny = data->oy;
	if (ny == 0)
		return;
	data->oy -= ny;

	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
		c = ARRAY_ITEM(&clients, i);
		if (c == NULL || c->session == NULL)
			continue;
		if (!session_has(c->session, w))
			continue;

		buffer_ensure(c->out, sizeof hdr);
		buffer_add(c->out, sizeof hdr);
		size = BUFFER_USED(c->out);
		
		screen_draw_start(&ctx, s, c->out, data->ox, data->oy);
		memcpy(&ctx.sel, &data->sel, sizeof ctx.sel);
		ctx.sel.ex = data->cx + data->ox;
		ctx.sel.ey = data->size + data->cy - data->oy;
		screen_draw_move(&ctx, 0, 0);
		input_store_one(c->out, CODE_DELETELINE, ny);
		for (i = 0; i < ny; i++)
			screen_draw_line(&ctx, screen_last_y(s) - i);
		if (data->sel.flag)
			screen_draw_line(&ctx, screen_last_y(s) - ny);
		window_copy_draw_position(w, &ctx);
		screen_draw_stop(&ctx);
		
		size = BUFFER_USED(c->out) - size;
		hdr.type = MSG_DATA;
		hdr.size = size;
		memcpy(BUFFER_IN(c->out) - size - sizeof hdr, &hdr, sizeof hdr);
	}	
}

void
window_copy_scroll_down(struct window *w, u_int ny)
{
	struct window_copy_mode_data	*data = w->modedata;
	struct screen			*s = &w->screen;
	struct screen_draw_ctx		 ctx;
	struct client			*c;
	u_int		 		 i;
	struct hdr			 hdr;
	size_t				 size;

	if (data->oy >= data->size - ny)
		ny = data->size - data->oy;
	if (ny == 0)
		return;
	data->oy += ny;

	for (i = 0; i < ARRAY_LENGTH(&clients); i++) {
		c = ARRAY_ITEM(&clients, i);
		if (c == NULL || c->session == NULL)
			continue;
		if (!session_has(c->session, w))
			continue;

		buffer_ensure(c->out, sizeof hdr);
		buffer_add(c->out, sizeof hdr);
		size = BUFFER_USED(c->out);

		screen_draw_start(&ctx, s, c->out, data->ox, data->oy);
		memcpy(&ctx.sel, &data->sel, sizeof ctx.sel);
		ctx.sel.ex = data->cx + data->ox;
		ctx.sel.ey = data->size + data->cy - data->oy;
		screen_draw_move(&ctx, 0, 0);
		input_store_one(c->out, CODE_INSERTLINE, ny);
		for (i = 1; i < ny + 1; i++)
			screen_draw_line(&ctx, i);
		window_copy_draw_position(w, &ctx);
		screen_draw_stop(&ctx);

		size = BUFFER_USED(c->out) - size;
		hdr.type = MSG_DATA;
		hdr.size = size;
		memcpy(BUFFER_IN(c->out) - size - sizeof hdr, &hdr, sizeof hdr);
	}	
}