main.cpp

#include "utils/headers/key_func.h"

llist text;
iter it;
vector<cut_s> cut;
vector<bool> overflows;
WINDOW *header_win, *ln_win, *text_win;
wchar_t s[4];
char s2[4], *filename;
cchar_t mark;
uint ry, rx;
uint y, x, maxy, maxx, flag;
uint ofy;

int main(int argc, char *argv[])
{
	if (argc > 1 && (strcmp(argv[1], "-h") == 0 ||
	strcmp(argv[1], "--help") == 0)) {
		puts(name);
		puts("A simple, compact and fast text editor.\n"
		"Source code: https://github.com/ikozyris/kri\n"
		"Wiki: https://github.com/ikozyris/kri/wiki\n"
		"License: GNU GPL v3+\n");
		puts("Usage:\n"
		"	--help, -h	Show this help\n"
		"<file>			Open file and allow edits\n"
		"-			Ask for file name on save\n\n"
		"Keybindings:\n"
		"Save:			Ctrl-S\n"
		"Exit:			Ctrl-X\n"
		"Go to start of line:	Ctrl-A\n"
		"Go to end of line:	Ctrl-E\n"
		"Built-in terminal:	Alt-C\n"
		"Previous/Next word	Shift + Left/Right arrow\n"
		"Show debbuging info:	Alt-I (also command stats in built-in terminal)\n\n"
		"Built-in terminal commands:\n"
		"scroll		Scroll to line\n"
		"suspend	Suspend and wait for signal\n"
		"find		Find string, takes 2 parameters, can be combined\n"
		"		  find <str> \\n \\n -> highlight on all lines\n"
		"		  find <str> \\n c 0-10 -> count on lines [0,10]\n"
		"replace 	Replace string, 2 parameters from, to (default 0, max)\n"
		"		  replace \\n <str1> \\n <str2> -> replace all str1 with str2\n"
		"		  replace 0-20 \\n <str1> \\n <str2> \\n -> in range [0,20]\n"
		"help		List commands");
		return 0;
	}
	text.head = create_chunk();
	text.tail = create_chunk();
	text.tail->num_lines = pow2(30); // large enough to avoid out of bounds iteration
	text.nodes = 1;
{
	chunk *new_chunk_tmp = create_chunk();
	connect(text.head, new_chunk_tmp); // insert without updating size
	connect(new_chunk_tmp, text.tail);
	point2begin(&it);
}

	if (argc > 1) {
		filename = argv[1];
		FILE *in = fopen(argv[1], "r");
#ifdef HIGHLIGHT
		eligible = detect_lang(argv[1]); // syntax highlighting
#endif
		if (!in) {
			print2header("New file", 1);
			goto init;
		}
		read_file2(in);
		fclose(in);
	} else {
init:
		insert_c(*it.orig, 0);
		append_len(it.parent(), 1);
	}

	init_curses();
	getmaxyx(stdscr, maxy, maxx);

	// initialize windows
	init_header();
	init_lines();
	init_text();
	setcchar(&mark, L">", A_STANDOUT, COLOR_BLACK, nullptr);

	getmaxyx(text_win, maxy, maxx);
	wnoutrefresh(ln_win);
	wnoutrefresh(header_win);
	overflows.resize(maxy);

	print_text(0);
//loop:
	wmove(text_win, 0, 0);

	while (1) {
		getyx(text_win, y, x);
		ry = y + ofy;
		// if out of bounds: move (to avoid bugs)
		if (x > min(it.len() - 1 - ofx, maxx))
			wmove(text_win, y, x = min(it.len() - ofx - 1, maxx));
		rx = x + ofx;
		mv_curs(*it.orig, rx + it.offset);

#ifndef RELEASE
		stats();
#endif
#ifdef DEBUG
		print_text(y);
		wmove(text_win, y, x);
#endif
	//goto stop;
		wget_wch(text_win, (wint_t*)s);
		switch (s[0]) {
		case DOWN:
			if (ry >= text.lines) // do not scroll indefinetly
				break;
			if (!cut.empty()) { // revert cut
				mvprint_line(y, 0, &it, 0, 0);
				highlight(y, &it);
			}
			cut.clear();
			ofx = 0; // invalidated
			if (y == maxy - 1)
				scrolldown();
			else {
				iterate_fw(&it, 1);
				ofx = calc_offset_dis(x, 0, &it);
				if (flag < maxx)
					wmove(text_win, y + 1, flag);
				else { // tab cut
					y++;
					x = flag;
					ofx += prevdchar();
				}
			}
			break;

		case UP:
			if (ofy == 0 && y == 0)
				break;
			if (!cut.empty()) { // revert cut
				mvprint_line(y, 0, &it, 0, 0);
				highlight(y, &it);
			}
			if (y == 0 && ofy != 0)
				scrollup();
			else {
				iterate_bw(&it, 1);
				ofx = calc_offset_dis(x, 0, &it);
				if (flag < maxx)
					wmove(text_win, y - 1, flag);
				else { // tab cut
					y--;
					x = flag;
					ofx += prevdchar();
				}
			}
			cut.clear();
			break;

		case LEFT:
			left();
			break;

		case KEY_SLEFT:
			prnxt_word(left);
			break;

		case RIGHT:
			right();
			break;

		case KEY_SRIGHT:
			prnxt_word(right);
			break;

		case BACKSPACE:
			if (x > 0) {
				erase_ch(&it);
				if (it.orig->buffer()[it.orig->gps] == '\t') { // deleted a tab
					ofx += prevdchar();
					x = getcurx(text_win);
					mvprint_line(y, 0, &it, 0, 0);
					wclrtoeol(text_win);
				} else {
					mvwdelch(text_win, y, --x);
					if (overflows[y])
						print_new_mark();
				}
				clear_attrs;
				highlight(y, &it);
				wmove(text_win, y, x);
			} else if (!cut.empty()) { // delete x_-1 on cut line
				erase_ch(&it);
				left();
			} else if (y != 0) { // x = 0 && cut.empty(); merge lines
				iter b = it;
				iterate_bw(&it, 1);
				uint tmp = it.len();
				merge_lines(&text, &it, &b);
				--text.lines;
				print_text(--y);
				mvr_scurs(tmp);
			}
			break;

		case DELETE:
			if (it.orig->buffer()[it.orig->gpe + 1] == '\n') { // similar to backspace
				iter b = it;
				iterate_fw(&b, 1);
				mv_curs(*it.orig, rx + it.offset + 1);
				merge_lines(&text, &it, &b);
				--text.lines;
				mvprint_line(y, 0, &it, cut.empty() ? 0 : cut.back().byte, 0);
				highlight(y, &it);
				print_text(y + 1);
				wmove(text_win, y, x);
			} else if (rx + 1 < it.len()) {
				// or mblen(it->buffer + it->gpe + 1, 3);
				uint len = it.orig->buffer()[it.orig->gpe + 1] < 0 ? 2 : 1;
				if (it.parent()->len)
					it.parent()->len[it.relative_pos] -= len;
				it.orig->gpe += len; // delete forward

				if (it.orig->buffer()[it.orig->gps] == '\t') {
					wclrtoeol(text_win);
					mvprint_line(y, x, &it, rx, 0);
				} else {
					wdelch(text_win);
					clear_attrs;
					if (overflows[y])
						print_new_mark();
				}
				highlight(y, &it);
				wmove(text_win, y, x);
			}
			break;

		/*case DELLINE: // TODO: fix this
			if (text.lines > 0 && text.nodes > 2) {
				rm_mline(it.parent(), it.relative_pos, &it);
				text.lines--;
				print_text(y);
				wmove(text_win, y, 0);
			} else { // clear line buffer
				it.orig->gps = 0;
				it.orig->gpe = it.orig->cpt() - 1;
				it.parent()->len[0] = 1;
				it.parent()->num_lines = 1;
				clearline;
			}
			break;*/

		case ENTER:
			enter();
			break;

		case HOME:
			sol();
			break;

		case END:
			eol();
			break;

		case SAVE:
			save();
			s2[0] = 0; // no new char has been inserted since last save
			argc = 3;
			break;

		case 27: { // ALT or ESC
			wtimeout(text_win, 1000);
			int ch = wgetch(text_win);
			wtimeout(text_win, -1);
			if (ch == INFO)
				stats();
			else if (ch == CMD)
				command();
			wmove(text_win, y, x);
			break;
		}

		case REFRESH:
			reset_view();
			break;

		case KEY_RESIZE:
			endwin();
			refresh();
			getmaxyx(stdscr, maxy, maxx);
			delwin(text_win);
			init_text();
			getmaxyx(text_win, maxy, maxx);
			reset_view();
			break;

		case EXIT:
			// has char been inserted, new file
			if (s2[0] != 0 || argc < 2) {
				char *in = input_header("Exit and Save changes? (y/n/c) ");
				flag = in[0]; // tmp var to free branchlessly | TODO: getch()
				free(in);
				if (flag == 'y')
					save();
				else if (flag != 'n') {
					reset_header();
					wmove(text_win, y, x);
					break;
				}
			}
			goto stop;

		case KEY_TAB:
			insert_c(*it.orig, '\t');
			if (it.parent()->len)
				it.parent()->len[it.relative_pos] += 1;
			mvprint_line(y, x, &it, rx, 0);
			highlight(y, &it);
			ofx -= 7 - x % 8;
			wmove(text_win, y, x + 8 - x % 8);
			break;

		default:
			if (s[0] > 0 && s[0] < 32) // not a character
				break;
			if (x == maxx - 1) { // cut line
				cut.push_back({maxx - 1, (uint)ofx});
				clearline;
				ofx += maxx - 1;
				print_line(&it, ofx, 0, y);
				wmove(text_win, y, x = 0);
				rx = ofx;
			} if (it.orig->buffer()[it.orig->gpe + 1] == '\t') { // next character is a tab
				waddnwstr(text_win, s, 1);
				if (x % 8 >= 7) // filled the empty tab space; reprint tab
					winsch(text_win, '\t');
			} else {
				wins_nwstr(text_win, s, 1);
				clear_attrs;
				highlight(y, &it);
				wmove(text_win, y, x + 1);
			}
			uint len = wcstombs(s2, s, 4);
			insert_s(*it.orig, s2, len);
			if (len > 1)
				ofx += len - 1; // UTF-8 character
			if (it.parent()->len) {
				it.parent()->len[it.relative_pos] += len;
				if (it.orig->len() + 1 >= MAX_CHUNK_SIZE) {
					split_mline(&text, it.parent());
					if (it.relative_pos == it.parent()->num_lines) // last line of chunk
						point2chunk(&it, it.parent()->next);
				}
			}
			break;
		}
	}
stop:
	free(lnbuf);
	delwin(text_win);
	delwin(ln_win);
	delwin(header_win);
	endwin();
	return 0;
}