/* Clip handling functions.

   Copyright (C) 1993-1998 Sebastiano Vigna 
   Copyright (C) 1999-2022 Todd M. Lewis and Sebastiano Vigna

   This file is part of ne, the nice editor.

   This library 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 3 of the License, or (at your
   option) any later version.

   This library 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, see <http://www.gnu.org/licenses/>.  */


#include "ne.h"
#include "support.h"


/* A clip is a numbered node in the global clip list. The contents of the clip
   are handled through the stream functions contained in streams.c.

   At creation time, a clip is marked with an encoding. Clips, of course, may
   be pasted only in buffers with a compatible encoding. 

   Note that pasting a clip in an ASCII buffer may change its encoding.
*/


/* Allocates a clip descriptor. */

clip_desc *alloc_clip_desc(int n, int64_t size) {
	assert(n >= 0);
	assert(size >= 0);

	clip_desc * const cd = calloc(1, sizeof(clip_desc));
	if (cd) {
		cd->n = n;
		if (cd->cs = alloc_char_stream(size)) return cd;
		free(cd);
	}
	return NULL;
}



/* Reallocates a clip descriptor of the given size. If cd is NULL, this is
   equivalent to calling alloc_clip_desc. */

clip_desc *realloc_clip_desc(clip_desc *cd, int n, int64_t size) {
	assert(n >= 0);
	assert(size >= 0);

	if (!cd) return alloc_clip_desc(n, size);

	assert_clip_desc(cd);

	if (cd->n != n) return NULL;

	char_stream * const cs = realloc_char_stream(cd->cs, size);
	if (cs) {
		cd->cs = cs;
		return cd;
	}

	return NULL;
}


/* Frees a clip descriptor. */

void free_clip_desc(clip_desc *cd) {

	if (!cd) return;

	assert_clip_desc(cd);

	free_char_stream(cd->cs);
	free(cd);
}


/* Scans the global clip list, searching for a specific numbered clip. Returns
   NULL on failure. */

clip_desc *get_nth_clip(int n) {
	for(clip_desc *cd = (clip_desc *)clips.head; cd->cd_node.next; cd = (clip_desc *)cd->cd_node.next) {
		assert_clip_desc(cd);
		if (cd->n == n) return cd;
	}

	return NULL;
}



/* Copies the characters between the cursor and the block marker of the given
   buffer to the nth clip. If the cut flag is true, the characters are also
   removed from the text. The code scans the text two times: the first time in
   order to determine the exact length of the text, the second time in order to
   actually copy it. */

int copy_to_clip(buffer *b, int n, bool cut) {
	if (!b->marking) return MARK_BLOCK_FIRST;

	if (b->block_start_line >= b->num_lines) return MARK_OUT_OF_BUFFER;

	/* If the mark and the cursor are on the same line and on the same position
	(or both beyond the line length), we can't copy anything. */

	line_desc *ld = b->cur_line_desc;
	clip_desc *cd = get_nth_clip(n);
	int64_t y = b->cur_line;

	if (y == b->block_start_line &&
		(b->cur_pos == b->block_start_pos ||
		 b->cur_pos >= ld->line_len && b->block_start_pos >= ld->line_len)) {

		clip_desc * const new_cd = realloc_clip_desc(cd, n, 0);
		if (!new_cd) return OUT_OF_MEMORY;
		if (!cd) add_head(&clips, &new_cd->cd_node);
		return OK;
	}


	/* We have two different loops for direct or inverse copying. Making this
		conditional code would be cumbersome, awkward, and definitely inefficient. */

	bool chaining;
	char *p = NULL;
	if (y > b->block_start_line || y == b->block_start_line && b->cur_pos > b->block_start_pos) {
		/* mark before/above cursor */
		int64_t clip_len = 0;
		chaining = false;
		for(int pass = 0; pass < 2; pass++) {

			ld = b->cur_line_desc;

			for(int64_t i = y; i >= b->block_start_line; i--) {
				int64_t start_pos = 0;

				if (i == b->block_start_line) {
					if (!pass && cut && ld->line_len < b->block_start_pos) {
						if (!chaining) {
							chaining = true;
							start_undo_chain(b);
						}
						const int64_t bsp = b->block_start_pos; /* because the mark will move when we insert_spaces()! */
						insert_spaces(b, ld, i, ld->line_len, b->block_start_pos - ld->line_len);
						b->block_start_pos = bsp;
					}
					start_pos = min(ld->line_len, b->block_start_pos);
				}

				const int64_t end_pos = i == y ? min(ld->line_len, b->cur_pos) : ld->line_len;
				const int64_t len = end_pos - start_pos;

				if (pass) {
					assert(!(len != 0 && ld->line == NULL));

					if (i != y) *--p = 0;
					p -= len;
					if (ld->line) memcpy(p, ld->line + start_pos, len);
				}
				else clip_len += len + (i != y);

				ld = (line_desc *)ld->ld_node.prev;
			}

			if (pass) {
				cd->cs->len = clip_len;
				set_stream_encoding(cd->cs, b->encoding);
				assert_clip_desc(cd);

				if (cut) {
					goto_line_pos(b, b->block_start_line, b->block_start_pos);
					delete_stream(b, b->cur_line_desc, b->cur_line, b->cur_pos, clip_len);
					update_syntax_states_delay(b, b->cur_line_desc, NULL);
				}

				if (chaining) end_undo_chain(b);
				return OK;
			}

			clip_desc * const new_cd = realloc_clip_desc(cd, n, clip_len);
			if (!new_cd) {
				if (chaining) end_undo_chain(b);
				return OUT_OF_MEMORY;
			}
			if (!cd) add_head(&clips, &new_cd->cd_node);
			cd = new_cd;
			p = cd->cs->stream + clip_len;
		}
	}
	else {
		/* mark after cursor */
		int64_t clip_len = 0;
		chaining = false;
		for(int pass = 0; pass < 2; pass++) {

			ld = b->cur_line_desc;

			for(int64_t i = y; i <= b->block_start_line; i++) {

				int64_t start_pos = 0;

				if (i == y) {
					if (!pass && cut && b->cur_pos > ld->line_len) {
						if (!chaining) {
							chaining = true;
							start_undo_chain(b);
						}
						insert_spaces(b, ld, i, ld->line_len, b->cur_pos - ld->line_len);
					}
					start_pos = b->cur_pos > ld->line_len ? ld->line_len : b->cur_pos;
				}

				const int64_t end_pos = i == b->block_start_line ? min(b->block_start_pos, ld->line_len) : ld->line_len;
				const int64_t len = end_pos - start_pos;

				if (pass) {
					assert(!(len != 0 && ld->line == NULL));

					if (ld->line) memcpy(p, ld->line + start_pos, len);
					p += len;
					if (i != b->block_start_line) *(p++) = 0;
				}
				else clip_len += len + (i != y);

				ld = (line_desc *)ld->ld_node.next;
			}

			if (pass) {
				cd->cs->len = clip_len;
				set_stream_encoding(cd->cs, b->encoding);
				assert_clip_desc(cd);
				if (cut) delete_stream(b, b->cur_line_desc, b->cur_line, b->cur_pos, clip_len);
				update_syntax_states_delay(b, b->cur_line_desc, NULL);
				if (chaining) end_undo_chain(b);
				return OK;
			}

			clip_desc * const new_cd = realloc_clip_desc(cd, n, clip_len);
			if (!new_cd) {
				if (chaining) end_undo_chain(b);
				return OUT_OF_MEMORY;
			}
			if (!cd) add_head(&clips, &new_cd->cd_node);
			cd = new_cd;
			p = cd->cs->stream;
		}
	}
	if (chaining) end_undo_chain(b);
	return OK;
}



/* Simply erases a block, without putting it in a clip. Calls
   update_syntax_states_delay() if update is true. */

int erase_block(buffer *b, const bool update) {
	if (!b->marking) return MARK_BLOCK_FIRST;
	if (b->block_start_line >= b->num_lines) return MARK_OUT_OF_BUFFER;

	int64_t y = b->cur_line, erase_len = 0;
	line_desc *ld = b->cur_line_desc;

	if (y == b->block_start_line &&
		(b->cur_pos == b->block_start_pos ||
		 b->cur_pos >= ld->line_len && b->block_start_pos >= ld->line_len))
		return OK;

	bool chaining = false;
	if (y > b->block_start_line || y == b->block_start_line && b->cur_pos > b->block_start_pos) {
		for(int64_t i = y; i >= b->block_start_line; i--) {
			int64_t start_pos = 0;
			if (i == b->block_start_line) {
				if (ld->line_len < b->block_start_pos) {
					if (!chaining) {
						chaining = true;
						start_undo_chain(b);
					}
					const int64_t bsp = b->block_start_pos; /* because the mark will move when we insert_spaces()! */
					insert_spaces(b, ld, i, ld->line_len, b->block_start_pos - ld->line_len);
					b->block_start_pos = bsp;
				}
				start_pos = min(ld->line_len, b->block_start_pos);
			}
			const int64_t end_pos = i == y ? min(ld->line_len, b->cur_pos) : ld->line_len;
			const int64_t len = end_pos - start_pos;

			erase_len += len + 1;
			ld = (line_desc *)ld->ld_node.prev;
		}
		goto_line_pos(b, b->block_start_line, b->block_start_pos);
	}
	else {
		for(int64_t i = y; i <= b->block_start_line; i++) {
			int64_t start_pos = 0;
			if (i == y) {
				if (b->cur_pos > ld->line_len) {
					if (!chaining) {
						chaining = true;
						start_undo_chain(b);
					}
					insert_spaces(b, ld, i, ld->line_len, b->cur_pos - ld->line_len);
				}
				start_pos = b->cur_pos > ld->line_len ? ld->line_len : b->cur_pos;
			}
			const int64_t end_pos = i == b->block_start_line ? min(b->block_start_pos, ld->line_len) : ld->line_len;
			const int64_t len = end_pos - start_pos;

			erase_len += len + 1;
			ld = (line_desc *)ld->ld_node.next;
		}
	}

	delete_stream(b, b->cur_line_desc, b->cur_line, b->cur_pos, erase_len - 1);
	if (chaining) end_undo_chain(b);
	if (update) update_syntax_states_delay(b, b->cur_line_desc, NULL);
	return OK;
}




/* Pastes a clip into a buffer. Since clips are streams, the operation is
   definitely straightforward. */

int paste_to_buffer(buffer *b, int n) {

	clip_desc * const cd = get_nth_clip(n);
	if (!cd) return CLIP_DOESNT_EXIST;

	if (!cd->cs->len) return OK;
	if (cd->cs->encoding == ENC_ASCII || b->encoding == ENC_ASCII || cd->cs->encoding == b->encoding) {
		line_desc * const ld = b->cur_line_desc, * const end_ld = (line_desc *)b->cur_line_desc->ld_node.next;
		if (b->encoding == ENC_ASCII) b->encoding = cd->cs->encoding;

		b->bookmark[PASTE_START_BOOKMARK].pos = b->cur_pos;
		b->bookmark[PASTE_START_BOOKMARK].line = b->cur_line;
		b->bookmark[PASTE_START_BOOKMARK].cur_y = b->cur_y;
		b->bookmark_mask |= (1 << PASTE_START_BOOKMARK);

		b->bookmark[PASTE_END_BOOKMARK].pos = next_pos(b->cur_line_desc->line, b->cur_pos, b->encoding);
		b->bookmark[PASTE_END_BOOKMARK].line = b->cur_line;
		b->bookmark_mask |= (1 << PASTE_END_BOOKMARK);

		start_undo_chain(b);
		if (b->cur_pos > ld->line_len) 
			insert_spaces(b, ld, b->cur_line, ld->line_len, b->win_x + b->cur_x - calc_width(ld, ld->line_len, b->opt.tab_size, b->encoding));

		insert_stream(b, ld, b->cur_line, b->cur_pos, cd->cs->stream, cd->cs->len);
		end_undo_chain(b);
		b->bookmark[PASTE_END_BOOKMARK].pos = prev_pos(nth_line_desc(b, b->bookmark[PASTE_END_BOOKMARK].line)->line, b->bookmark[PASTE_END_BOOKMARK].pos, b->encoding);
		b->bookmark[PASTE_END_BOOKMARK].cur_y = min(ne_lines - 2, b->bookmark[PASTE_START_BOOKMARK].cur_y + b->bookmark[PASTE_END_BOOKMARK].line - b->bookmark[PASTE_START_BOOKMARK].line);

		assert(ld == b->cur_line_desc);
		update_syntax_states_delay(b, ld, end_ld);
		return OK;
	}

	return INCOMPATIBLE_CLIP_ENCODING;
}



/* Works like copy_to_clip(), but the region to copy is the rectangle defined
   by the cursor and the marker. Same comments apply. Note that in case of a
   cut we use start_undo_chain() in order to make the various deletions a
   single undo operation. */

int copy_vert_to_clip(buffer *b, int n, bool cut) {
	if (!b->marking) return MARK_BLOCK_FIRST;
	if (b->block_start_line >= b->num_lines) return MARK_OUT_OF_BUFFER;

	int64_t y = b->cur_line;
	line_desc *ld = b->cur_line_desc;
	clip_desc *cd = get_nth_clip(n);

	if (b->cur_pos == b->block_start_pos ||
		 y == b->block_start_line && b->cur_pos >= ld->line_len && 
		 b->block_start_pos >= ld->line_len) {

		clip_desc * const new_cd = realloc_clip_desc(cd, n, 0);
		if (!new_cd) return OUT_OF_MEMORY;
		set_stream_encoding(new_cd->cs, ENC_ASCII);
		if (!cd) add_head(&clips, &new_cd->cd_node);
		return OK;
	}

	int64_t start_x = calc_width(nth_line_desc(b, b->block_start_line), b->block_start_pos, b->opt.tab_size, b->encoding);
	int64_t end_x = b->win_x + b->cur_x;

	if (end_x < start_x) {
		const uint64_t t = start_x;
		start_x = end_x;
		end_x = t;
	}

	const int64_t cur_pos = b->cur_pos;
	if (cut) {
		b->cur_pos = -1;
		start_undo_chain(b);
	}

	char *p = NULL;
	if (y > b->block_start_line) {
		int64_t clip_len = 0;
		for(int pass = 0; pass < 2; pass++) {

			ld = b->cur_line_desc;

			for(int64_t i = y; i >= b->block_start_line; i--) {

				const int64_t start_pos = calc_pos(ld, start_x, b->opt.tab_size, b->encoding);
				const int64_t len = calc_pos(ld, end_x, b->opt.tab_size, b->encoding) - start_pos;

				if (pass) {
					*--p = 0;
					p -= len;
					if (len) memcpy(p, ld->line + start_pos, len);
					if (cut) delete_stream(b, ld, i, start_pos, len);
				}

				else clip_len += len + 1;
				ld = (line_desc *)ld->ld_node.prev;
			}

			if (pass) {
				cd->cs->len = clip_len;
				set_stream_encoding(cd->cs, b->encoding);
				assert_clip_desc(cd);
				if (cut) {
					update_syntax_states_delay(b, (line_desc *)ld->ld_node.next, b->cur_line_desc);
					goto_line_pos(b, min(b->block_start_line, b->cur_line), min(b->block_start_pos, cur_pos));
					end_undo_chain(b);
				}
				return OK;
			}

			clip_desc * const new_cd = realloc_clip_desc(cd, n, clip_len);
			if (!new_cd) return OUT_OF_MEMORY;
			if (!cd) add_head(&clips, &new_cd->cd_node);
			cd = new_cd;
			p = cd->cs->stream + clip_len;
		}
	}
	else {
		int64_t clip_len = 0;
		for(int pass = 0; pass < 2; pass++) {

			ld = b->cur_line_desc;

			for(int64_t i = y; i <= b->block_start_line; i++) {

				const int64_t start_pos = calc_pos(ld, start_x, b->opt.tab_size, b->encoding);
				const int64_t len = calc_pos(ld, end_x, b->opt.tab_size, b->encoding) - start_pos;

				if (pass) {
					if (len) memcpy(p, ld->line + start_pos, len);
					p += len;
					*(p++) = 0;
					if (cut) delete_stream(b, ld, i, start_pos, len);
				}

				else clip_len += len + 1;
				ld = (line_desc *)ld->ld_node.next;
			}

			if (pass) {
				cd->cs->len = clip_len;
				set_stream_encoding(cd->cs, b->encoding);
				assert_clip_desc(cd);
				if (cut) {
					update_syntax_states_delay(b, b->cur_line_desc, (line_desc *)ld->ld_node.prev);
					goto_line_pos(b, min(b->block_start_line, b->cur_line), min(b->block_start_pos, cur_pos));
					end_undo_chain(b);
				}
				return OK;
			}

			clip_desc * const new_cd = realloc_clip_desc(cd, n, clip_len);
			if (!new_cd) return OUT_OF_MEMORY;
			if (!cd) add_head(&clips, &new_cd->cd_node);
			cd = new_cd;
			p = cd->cs->stream;
		}
	}

	if (cut) end_undo_chain(b);
	return OK;
}

/* Simply erases a vertical block, without putting it in a clip. Calls
   update_syntax_states_delay() if update is true. */

int erase_vert_block(buffer *b, const bool update) {

	if (!b->marking) return MARK_BLOCK_FIRST;
	if (b->block_start_line >= b->num_lines) return MARK_OUT_OF_BUFFER;

	int64_t y = b->cur_line;
	line_desc *ld = b->cur_line_desc;

	if (b->cur_pos == b->block_start_pos ||
		y == b->block_start_line && b->cur_pos >= ld->line_len && 
		b->block_start_pos >= ld->line_len) 
		return OK;

	const int64_t cur_pos = b->cur_pos;
	b->cur_pos = -1;

	int64_t start_x = calc_width(nth_line_desc(b, b->block_start_line), b->block_start_pos, b->opt.tab_size, b->encoding);
	int64_t end_x = b->win_x + b->cur_x;

	if (end_x < start_x) {
		const uint64_t t = start_x;
		start_x = end_x;
		end_x = t;
	}

	start_undo_chain(b);

	if (y > b->block_start_line) {
		for(int64_t i = y; i >= b->block_start_line; i--) {
			const int64_t start_pos = calc_pos(ld, start_x, b->opt.tab_size, b->encoding);
			const int64_t len = calc_pos(ld, end_x, b->opt.tab_size, b->encoding) - start_pos;
			delete_stream(b, ld, i, start_pos, len);
			ld = (line_desc *)ld->ld_node.prev;
		}
		if (update) update_syntax_states_delay(b, (line_desc *)ld->ld_node.next, b->cur_line_desc);
	}
	else {
		for(int64_t i = y; i <= b->block_start_line; i++) {
			const int64_t start_pos = calc_pos(ld, start_x, b->opt.tab_size, b->encoding);
			const int64_t len = calc_pos(ld, end_x, b->opt.tab_size, b->encoding)-start_pos;
			delete_stream(b, ld, i, start_pos, len);
			ld = (line_desc *)ld->ld_node.next;
		}
		if (update) update_syntax_states_delay(b, b->cur_line_desc, (line_desc *)ld->ld_node.prev);
	}

	end_undo_chain(b);

	goto_line_pos(b, min(b->block_start_line, b->cur_line), min(b->block_start_pos, cur_pos));

	return OK;
}



/* Performs a vertical paste. It has to be done via an insert_stream() for each
   string of the clip. Again, the undo linking feature makes all these
   operations a single undo step. */

int paste_vert_to_buffer(buffer *b, int n) {

	line_desc * ld = b->cur_line_desc;
	clip_desc * const cd = get_nth_clip(n);
	if (!cd) return CLIP_DOESNT_EXIST;
	if (!cd->cs->len) return OK;

	if (cd->cs->encoding != ENC_ASCII && b->encoding != ENC_ASCII && cd->cs->encoding != b->encoding) return INCOMPATIBLE_CLIP_ENCODING;
	if (b->encoding == ENC_ASCII) b->encoding = cd->cs->encoding;

	char * const stream = cd->cs->stream;
	char *p = stream;
	const int64_t stream_len = cd->cs->len;
	const uint64_t x = b->cur_x + b->win_x;
	int64_t line = b->cur_line;

	start_undo_chain(b);

	b->bookmark[PASTE_START_BOOKMARK].pos = b->cur_pos;
	b->bookmark[PASTE_START_BOOKMARK].line = b->cur_line;
	b->bookmark[PASTE_START_BOOKMARK].cur_y = b->cur_y;
	b->bookmark_mask |= (1 << PASTE_START_BOOKMARK);

	while(p - stream < stream_len) {
		if (!ld->ld_node.next) {
			insert_one_line(b, (line_desc *)ld->ld_node.prev, line - 1, ((line_desc *)ld->ld_node.prev)->line_len);
			ld = (line_desc *)ld->ld_node.prev;
		}

		const int64_t len = strnlen_ne(p, stream_len - (p - stream));
		if (len) {
			uint64_t pos, n;
			for(n = pos = 0; pos < ld->line_len && n < x; pos = next_pos(ld->line, pos, b->encoding)) {
				if (ld->line[pos] == '\t') n += b->opt.tab_size - n % b->opt.tab_size;
				else n += get_char_width(&ld->line[pos], b->encoding);
			}

			b->bookmark[PASTE_END_BOOKMARK].line = line;
			b->bookmark_mask |= (1 << PASTE_END_BOOKMARK);

			if (pos == ld->line_len && n < x) {
				/* We miss x - n characters after the end of the line. */
				insert_spaces(b, ld, line, ld->line_len, x - n);
				b->bookmark[PASTE_END_BOOKMARK].pos = next_pos(ld->line, ld->line_len, b->encoding);
				insert_stream(b, ld, line, ld->line_len, p, len);
			}
			else {
				b->bookmark[PASTE_END_BOOKMARK].pos = next_pos(ld->line, pos, b->encoding);
				insert_stream(b, ld, line, pos, p, len);
			}
		}

		p += len + 1;
		ld = (line_desc *)ld->ld_node.next;
		line++;
	}
	b->bookmark[PASTE_END_BOOKMARK].pos = prev_pos(((line_desc *)ld->ld_node.prev)->line, b->bookmark[PASTE_END_BOOKMARK].pos, b->encoding);
	b->bookmark[PASTE_END_BOOKMARK].cur_y = min(ne_lines - 2, b->bookmark[PASTE_START_BOOKMARK].cur_y + b->bookmark[PASTE_END_BOOKMARK].line - b->bookmark[PASTE_START_BOOKMARK].line);
	
	end_undo_chain(b);
	update_syntax_states_delay(b, b->cur_line_desc, ld);
	return OK;
}


/* Loads a clip. It is just a load_stream, plus an insertion in the clip
   list. If preserve_cr is true, CRs are preserved. */

int load_clip(int n, const char *name, const bool preserve_cr, const bool binary) {
	clip_desc *cd = get_nth_clip(n);
	if (!cd) {
		if (!(cd = alloc_clip_desc(n, 0))) return OUT_OF_MEMORY;
		add_head(&clips, &cd->cd_node);
	}

	const int error = load_stream(cd->cs, name, preserve_cr, binary) ? OK : CANT_OPEN_FILE;

	if (error == OK) set_stream_encoding(cd->cs, ENC_ASCII);

	return error;
}


/* Saves a clip to a file. If CRLF is true, the clip is saved with CR/LF pairs
   as line terminators. */

int save_clip(int n, const char *name, const bool CRLF, const bool binary) {
	clip_desc * const cd = get_nth_clip(n);
	if (!cd) return CLIP_DOESNT_EXIST;
	return save_stream(cd->cs, name, CRLF, binary);
}
