1 /*
2  * Copyright (c) 2021 Omar Polo <op@omarpolo.com>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <curses.h>
18 #include <limits.h>
19 #include <stdlib.h>
20 #include <string.h>
21 #include <unistd.h>
22 
23 #include "defaults.h"
24 #include "telescope.h"
25 #include "ui.h"
26 
27 char *new_tab_url = NULL;
28 int fill_column = 80;
29 int olivetti_mode = 1;
30 int enable_colors = 1;
31 int hide_pre_context = 0;
32 int hide_pre_blocks = 0;
33 int hide_pre_closing_line = 0;
34 int dont_wrap_pre = 0;
35 int emojify_link = 1;
36 
37 struct lineprefix line_prefixes[] = {
38 	[LINE_TEXT] =		{ "",		"" },
39 	[LINE_LINK] =		{ "=> ",	"   " },
40 	[LINE_TITLE_1] =	{ "# ",		"  " },
41 	[LINE_TITLE_2] =	{ "## ",	"   " },
42 	[LINE_TITLE_3] =	{ "### ",	"    " },
43 	[LINE_ITEM] =		{ "* ",		"  " },
44 	[LINE_QUOTE] =		{ "> ",		"  " },
45 	[LINE_PRE_START] =	{ "```",	"   " },
46 	[LINE_PRE_CONTENT] =	{ "",		"" },
47 	[LINE_PRE_END] =	{ "```",	"```" },
48 
49 	[LINE_COMPL] =		{"", ""},
50 	[LINE_COMPL_CURRENT] =	{"", ""},
51 };
52 
53 struct line_face line_faces[] = {
54 	[LINE_TEXT] =		{
55 		.prfx_pair = PT_PRFX,
56 		.pair = PT,
57 		.trail_pair = PT_TRAIL,
58 	},
59 	[LINE_LINK] =		{
60 		.prfx_pair = PL_PRFX,
61 		.pair = PL,
62 		.trail_pair = PL_TRAIL,
63 	},
64 	[LINE_TITLE_1] =	{
65 		.prfx_pair = PT1_PRFX,
66 		.pair = PT1,
67 		.trail_pair = PT1_TRAIL,
68 		.attr = A_BOLD,
69 	},
70 	[LINE_TITLE_2] =	{
71 		.prfx_pair = PT2_PRFX,
72 		.pair = PT2,
73 		.trail_pair = PT2_TRAIL,
74 		.attr = A_BOLD,
75 	},
76 	[LINE_TITLE_3] =	{
77 		.prfx_pair = PT3_PRFX,
78 		.pair = PT3,
79 		.trail_pair = PT3_TRAIL,
80 		.attr = A_BOLD,
81 	},
82 	[LINE_ITEM] =		{
83 		.prfx_pair = PI_PRFX,
84 		.pair = PI,
85 		.trail_pair = PI_TRAIL,
86 	},
87 	[LINE_QUOTE] =		{
88 		.prfx_pair = PQ_PRFX,
89 		.pair = PQ,
90 		.trail_pair = PQ_TRAIL,
91 		.attr = A_DIM,
92 	},
93 	[LINE_PRE_START] =	{
94 		.prfx_pair = PPSTART_PRFX,
95 		.pair = PPSTART,
96 		.trail_pair = PPSTART_TRAIL,
97 	},
98 	[LINE_PRE_CONTENT] =	{
99 		.prfx_pair = PP_PRFX,
100 		.pair = PP,
101 		.trail_pair = PP_TRAIL,
102 	},
103 	[LINE_PRE_END] =	{
104 		.prfx_pair = PPEND_PRFX,
105 		.pair = PPEND,
106 		.trail_pair = PPEND_TRAIL,
107 	},
108 
109 	/* minibuffer */
110 	[LINE_COMPL] = {
111 		.prfx_pair = PCOMPL_PRFX,
112 		.pair = PCOMPL,
113 		.trail_pair = PCOMPL_TRAIL,
114 	},
115 	[LINE_COMPL_CURRENT] = {
116 		.prfx_pair = PCOMPL_CURR_PRFX,
117 		.pair = PCOMPL_CURR,
118 		.trail_pair = PCOMPL_CURR_TRAIL,
119 		.attr = A_REVERSE,
120 		.trail_attr = A_REVERSE,
121 	},
122 };
123 
124 struct tab_face tab_face = {
125 	.bg_attr = A_REVERSE, .bg_bg = -1, .bg_fg = -1,
126 	.t_attr  = A_REVERSE, .t_bg  = -1, .t_fg  = -1,
127 	.c_attr  = A_NORMAL,  .c_bg  = -1, .c_fg  = -1,
128 
129 	/*
130 	 * set these so that even when enable-color=0 the bar has some
131 	 * sane defaults.
132 	 */
133 	.background =	A_REVERSE,
134 	.tab =		A_REVERSE,
135 	.current =	A_NORMAL,
136 };
137 
138 struct body_face body_face = {
139 	.lbg = -1, .lfg = -1,
140 	.bg  = -1, .fg  = -1,
141 	.rbg = -1, .rfg = -1,
142 };
143 
144 struct modeline_face modeline_face = {
145 	.bg  = -1,
146 	.fg  = -1,
147 	.attr = A_REVERSE,
148 };
149 
150 struct minibuffer_face minibuffer_face = {
151 	.bg = -1,
152 	.fg = -1,
153 	.attr = A_NORMAL,
154 };
155 
156 struct mapping {
157 	const char	*label;
158 	int		 linetype;
159 } mappings[] = {
160 	{"text",	LINE_TEXT},
161 	{"link",	LINE_LINK},
162 	{"title1",	LINE_TITLE_1},
163 	{"title2",	LINE_TITLE_2},
164 	{"title3",	LINE_TITLE_3},
165 	{"item",	LINE_ITEM},
166 	{"quote",	LINE_QUOTE},
167 	{"pre.start",	LINE_PRE_START},
168 	{"pre",		LINE_PRE_CONTENT},
169 	{"pre.end",	LINE_PRE_END},
170 
171 	/* minibuffer */
172 	{"compl",	LINE_COMPL},
173 	{"compl.current", LINE_COMPL_CURRENT}
174 };
175 
176 static struct mapping *
177 mapping_by_name(const char *name)
178 {
179 	size_t i;
180 
181 	for (i = 0; i < sizeof(mappings)/sizeof(mappings[0]); ++i) {
182 		if (!strcmp(name, mappings[i].label))
183 			return &mappings[i];
184 	}
185 
186 	return NULL;
187 }
188 
189 
190 static inline void
191 global_set_key(const char *key, void (*fn)(struct buffer*))
192 {
193 	if (!kmap_define_key(&global_map, key, fn))
194 		_exit(1);
195 }
196 
197 static inline void
198 minibuffer_set_key(const char *key, void (*fn)(struct buffer*))
199 {
200 	if (!kmap_define_key(&minibuffer_map, key, fn))
201 		_exit(1);
202 }
203 
204 static void
205 load_default_keys(void)
206 {
207 	/* === global map === */
208 
209 	/* emacs */
210 	global_set_key("C-p",		cmd_previous_line);
211 	global_set_key("C-n",		cmd_next_line);
212 	global_set_key("C-f",		cmd_forward_char);
213 	global_set_key("C-b",		cmd_backward_char);
214 	global_set_key("M-{",		cmd_backward_paragraph);
215 	global_set_key("M-}",		cmd_forward_paragraph);
216 	global_set_key("C-a",		cmd_move_beginning_of_line);
217 	global_set_key("C-e",		cmd_move_end_of_line);
218 
219 	global_set_key("M-v",		cmd_scroll_up);
220 	global_set_key("C-v",		cmd_scroll_down);
221 	global_set_key("M-space",	cmd_scroll_up);
222 	global_set_key("space",		cmd_scroll_down);
223 
224 	global_set_key("M-<",		cmd_beginning_of_buffer);
225 	global_set_key("M->",		cmd_end_of_buffer);
226 
227 	global_set_key("C-x C-c",	cmd_kill_telescope);
228 
229 	global_set_key("C-g",		cmd_clear_minibuf);
230 
231 	global_set_key("M-x",		cmd_execute_extended_command);
232 
233 	global_set_key("C-c {",		cmd_dec_fill_column);
234 	global_set_key("C-c }",		cmd_inc_fill_column);
235 
236 	global_set_key("C-c p",		cmd_previous_heading);
237 	global_set_key("C-c n",		cmd_next_heading);
238 
239 	global_set_key(">",		cmd_load_url);
240 	global_set_key("<",		cmd_load_current_url);
241 	global_set_key("C-x C-f",	cmd_load_url);
242 	global_set_key("C-x M-f",	cmd_load_current_url);
243 
244 	global_set_key("C-x t 0",	cmd_tab_close);
245 	global_set_key("C-x t 1",	cmd_tab_close_other);
246 	global_set_key("C-x t 2",	cmd_tab_new);
247 	global_set_key("C-x t o",	cmd_tab_next);
248 	global_set_key("C-x t O",	cmd_tab_previous);
249 	global_set_key("C-x t m",	cmd_tab_move);
250 	global_set_key("C-x t M",	cmd_tab_move_to);
251 
252 	global_set_key("B",		cmd_previous_page);
253 	global_set_key("C-M-b",		cmd_previous_page);
254 	global_set_key("F",		cmd_next_page);
255 	global_set_key("C-M-f",		cmd_next_page);
256 
257 	global_set_key("<f7> a",	cmd_bookmark_page);
258 	global_set_key("<f7> <f7>",	cmd_list_bookmarks);
259 
260 	global_set_key("C-z",		cmd_suspend_telescope);
261 
262 	/* vi/vi-like */
263 	global_set_key("k",		cmd_previous_line);
264 	global_set_key("j",		cmd_next_line);
265 	global_set_key("l",		cmd_forward_char);
266 	global_set_key("h",		cmd_backward_char);
267 	global_set_key("{",		cmd_backward_paragraph);
268 	global_set_key("}",		cmd_forward_paragraph);
269 	global_set_key("^",		cmd_move_beginning_of_line);
270 	global_set_key("$",		cmd_move_end_of_line);
271 
272 	global_set_key("K",		cmd_scroll_line_up);
273 	global_set_key("J",		cmd_scroll_line_down);
274 
275 	global_set_key("g g",		cmd_beginning_of_buffer);
276 	global_set_key("G",		cmd_end_of_buffer);
277 
278 	global_set_key("g D",		cmd_tab_close);
279 	global_set_key("g N",		cmd_tab_new);
280 	global_set_key("g t",		cmd_tab_next);
281 	global_set_key("g T",		cmd_tab_previous);
282 	global_set_key("g M-t",		cmd_tab_move);
283 	global_set_key("g M-T",		cmd_tab_move_to);
284 
285 	global_set_key("H",		cmd_previous_page);
286 	global_set_key("L",		cmd_next_page);
287 
288 	/* tmp */
289 	global_set_key("q",		cmd_kill_telescope);
290 
291 	global_set_key("esc",		cmd_clear_minibuf);
292 
293 	global_set_key(":",		cmd_execute_extended_command);
294 
295 	/* cua */
296 	global_set_key("<up>",		cmd_previous_line);
297 	global_set_key("<down>",	cmd_next_line);
298 	global_set_key("<right>",	cmd_forward_char);
299 	global_set_key("<left>",	cmd_backward_char);
300 	global_set_key("<home>",	cmd_move_beginning_of_line);
301 	global_set_key("<end>",		cmd_move_end_of_line);
302 	global_set_key("<prior>",	cmd_scroll_up);
303 	global_set_key("<next>",	cmd_scroll_down);
304 
305 	global_set_key("C-w",		cmd_tab_close);
306 	global_set_key("C-t",		cmd_tab_new);
307 	global_set_key("M-<prior>",	cmd_tab_previous);
308 	global_set_key("M-<next>",	cmd_tab_next);
309 
310 	global_set_key("M-<left>",	cmd_previous_page);
311 	global_set_key("M-<right>",	cmd_next_page);
312 
313 	global_set_key("<f5>",		cmd_reload_page);
314 	global_set_key("r",		cmd_reload_page);
315 
316 	/* "ncurses standard" */
317 	global_set_key("C-l",		cmd_redraw);
318 
319 	/* global */
320 	global_set_key("<f1>",		cmd_toggle_help);
321 	global_set_key("C-m",		cmd_push_button);
322 	global_set_key("M-enter",	cmd_push_button_new_tab);
323 	global_set_key("M-tab",		cmd_previous_button);
324 	global_set_key("backtab",	cmd_previous_button);
325 	global_set_key("tab",		cmd_next_button);
326 	global_set_key("M-t",		cmd_tab_select);
327 	global_set_key("[",		cmd_tab_previous);
328 	global_set_key("]",		cmd_tab_next);
329 	global_set_key("M-l",		cmd_link_select);
330 	global_set_key("M-/",		cmd_swiper);
331 	global_set_key("t",		cmd_toc);
332 
333 	/* === minibuffer map === */
334 	minibuffer_set_key("ret",		cmd_mini_complete_and_exit);
335 	minibuffer_set_key("C-g",		cmd_mini_abort);
336 	minibuffer_set_key("esc",		cmd_mini_abort);
337 	minibuffer_set_key("C-d",		cmd_mini_delete_char);
338 	minibuffer_set_key("del",		cmd_mini_delete_backward_char);
339 	minibuffer_set_key("backspace",		cmd_mini_delete_backward_char);
340 	minibuffer_set_key("C-h",		cmd_mini_delete_backward_char);
341 
342 	minibuffer_set_key("C-b",		cmd_backward_char);
343 	minibuffer_set_key("C-f",		cmd_forward_char);
344 	minibuffer_set_key("<left>",		cmd_backward_char);
345 	minibuffer_set_key("<right>",		cmd_forward_char);
346 	minibuffer_set_key("C-e",		cmd_move_end_of_line);
347 	minibuffer_set_key("C-a",		cmd_move_beginning_of_line);
348 	minibuffer_set_key("<end>",		cmd_move_end_of_line);
349 	minibuffer_set_key("<home>",		cmd_move_beginning_of_line);
350 	minibuffer_set_key("C-k",		cmd_mini_kill_line);
351 
352 	minibuffer_set_key("M-p",		cmd_mini_previous_history_element);
353 	minibuffer_set_key("M-n",		cmd_mini_next_history_element);
354 
355 	minibuffer_set_key("C-p",		cmd_previous_completion);
356 	minibuffer_set_key("C-n",		cmd_next_completion);
357 	minibuffer_set_key("<up>",		cmd_previous_completion);
358 	minibuffer_set_key("<down>",		cmd_next_completion);
359 
360 	minibuffer_set_key("M-<",		cmd_mini_goto_beginning);
361 	minibuffer_set_key("M->",		cmd_mini_goto_end);
362 
363 	minibuffer_set_key("tab",		cmd_insert_current_candidate);
364 }
365 
366 void
367 config_init(void)
368 {
369 	struct line_face *f;
370 	size_t i, len;
371 
372 	len = sizeof(line_faces)/sizeof(line_faces[0]);
373 	for (i = 0; i < len; ++i) {
374 		f = &line_faces[i];
375 
376 		f->prfx_bg = f->bg = f->trail_bg = -1;
377 		f->prfx_fg = f->fg = f->trail_fg = -1;
378 	}
379 
380 	line_faces[LINE_LINK].fg = COLOR_BLUE;
381 
382 	load_default_keys();
383 }
384 
385 int
386 config_setprfx(const char *name, const char *prfx, const char *cont)
387 {
388 	struct lineprefix *p;
389 	struct mapping *m;
390 
391 	if (!has_prefix(name, "line."))
392 		return 0;
393 	name += 5;
394 
395 	if ((m = mapping_by_name(name)) == NULL)
396 		return 0;
397 
398 	p = &line_prefixes[m->linetype];
399 	p->prfx1 = prfx;
400 	p->prfx2 = cont;
401 
402 	return 1;
403 }
404 
405 int
406 config_setvari(const char *var, int val)
407 {
408 	if (!strcmp(var, "fill-column")) {
409 		if ((fill_column = val) <= 0)
410 			fill_column = INT_MAX;
411 	} else if (!strcmp(var, "olivetti-mode")) {
412 		olivetti_mode = !!val;
413 	} else if (!strcmp(var, "enable-colors")) {
414 		enable_colors = !!val;
415 	} else if (!strcmp(var, "hide-pre-context")) {
416 		hide_pre_context = !!val;
417 	} else if (!strcmp(var, "hide-pre-blocks")) {
418 		hide_pre_blocks = !!val;
419 	} else if (!strcmp(var, "hide-pre-closing-line")) {
420 		hide_pre_closing_line = !!val;
421 	} else if (!strcmp(var, "dont-wrap-pre")) {
422 		dont_wrap_pre = !!val;
423 	} else if (!strcmp(var, "emojify-link")) {
424 		emojify_link = !!val;
425 	} else {
426 		return 0;
427 	}
428 
429 	return 1;
430 }
431 
432 int
433 config_setvars(const char *var, char *val)
434 {
435 	if (!strcmp(var, "new-tab-url")) {
436 		if (new_tab_url != NULL)
437 			free(new_tab_url);
438 		new_tab_url = val;
439 	} else
440 		return 0;
441 	return 1;
442 }
443 
444 int
445 config_setcolor(int bg, const char *name, int prfx, int line, int trail)
446 {
447         struct mapping *m;
448 	struct line_face *f;
449 
450 	if (!strcmp(name, "tabline")) {
451 		if (bg)
452 			tab_face.bg_bg = prfx;
453 		else
454 			tab_face.bg_fg = prfx;
455 	} else if (has_prefix(name, "tabline.")) {
456 		name += 8;
457 
458 		if (!strcmp(name, "tab")) {
459 			if (bg)
460 				tab_face.t_bg = prfx;
461 			else
462 				tab_face.t_fg = prfx;
463 		} else if (!strcmp(name, "current")) {
464 			if (bg)
465 				tab_face.c_bg = prfx;
466 			else
467 				tab_face.c_fg = prfx;
468 		} else
469 			return 0;
470 	} else if (has_prefix(name, "line.")) {
471 		name += 5;
472 
473 		if ((m = mapping_by_name(name)) == NULL)
474 			return 0;
475 
476 		f = &line_faces[m->linetype];
477 
478 		if (bg) {
479 			f->prfx_bg = prfx;
480 			f->bg = line;
481 			f->trail_bg = trail;
482 		} else {
483 			f->prfx_fg = prfx;
484 			f->fg = line;
485 			f->trail_fg = trail;
486 		}
487 	} else if (!strcmp(name, "line")) {
488 		if (bg) {
489 			body_face.lbg = prfx;
490 			body_face.bg  = line;
491 			body_face.rbg = trail;
492 		} else {
493 			body_face.lfg = prfx;
494 			body_face.fg = line;
495 			body_face.rfg = trail;
496 		}
497 	} else if (!strcmp(name, "minibuffer")) {
498 		if (bg)
499 			minibuffer_face.bg = prfx;
500 		else
501 			minibuffer_face.fg = prfx;
502 	} else if (!strcmp(name, "modeline")) {
503 		if (bg)
504 			modeline_face.bg = prfx;
505 		else
506 			modeline_face.fg = prfx;
507 	} else {
508 		return 0;
509 	}
510 
511 	return 1;
512 }
513 
514 int
515 config_setattr(const char *name, int prfx, int line, int trail)
516 {
517 	struct mapping *m;
518 	struct line_face *f;
519 
520 	if (!strcmp(name, "tabline")) {
521 		tab_face.bg_attr = prfx;
522 	} else if (has_prefix(name, "tabline.")) {
523 		name += 8;
524 
525 		if (!strcmp(name, "tab"))
526 			tab_face.t_attr = prfx;
527 		else if (!strcmp(name, "current"))
528 			tab_face.c_attr = prfx;
529 		else
530 			return 0;
531 	} else if (has_prefix(name, "line.")) {
532 		name += 5;
533 
534 		if ((m = mapping_by_name(name)) == NULL)
535 			return 0;
536 
537 		f = &line_faces[m->linetype];
538 
539 		f->prfx_attr = prfx;
540 		f->attr = line;
541 		f->trail_attr = trail;
542 	} else if (!strcmp(name, "minibuffer")) {
543 		minibuffer_face.attr = prfx;
544 	} else if (!strcmp(name, "modeline")) {
545 		modeline_face.attr = prfx;
546 	} else {
547 		return 0;
548 	}
549 
550 	return 1;
551 }
552 
553 static inline void
554 tl_init_pair(int colors, int pair, int f, int b)
555 {
556 	if (f >= colors || !enable_colors)
557 		f = -1;
558 	if (b >= colors || !enable_colors)
559 		b = -1;
560 	init_pair(pair, f, b);
561 }
562 
563 void
564 config_apply_style(void)
565 {
566 	size_t i, colors, len;
567 	struct line_face *f;
568 
569 	colors = COLORS;
570 
571 	len = sizeof(line_faces)/sizeof(line_faces[0]);
572 	for (i = 0; i < len; ++i) {
573 		f = &line_faces[i];
574 
575 		tl_init_pair(colors, f->prfx_pair, f->prfx_fg, f->prfx_bg);
576 		f->prefix = COLOR_PAIR(f->prfx_pair) | f->prfx_attr;
577 
578 		tl_init_pair(colors, f->pair, f->fg, f->bg);
579 		f->text = COLOR_PAIR(f->pair) | f->attr;
580 
581 		tl_init_pair(colors, f->trail_pair, f->trail_fg, f->trail_bg);
582 		f->trail = COLOR_PAIR(f->trail_pair) | f->trail_attr;
583 	}
584 
585 	/* tab line */
586 	tl_init_pair(colors, PTL_BG, tab_face.bg_fg, tab_face.bg_bg);
587 	tab_face.background = COLOR_PAIR(PTL_BG) | tab_face.bg_attr;
588 
589 	tl_init_pair(colors, PTL_TAB, tab_face.t_fg, tab_face.t_bg);
590 	tab_face.tab = COLOR_PAIR(PTL_TAB) | tab_face.t_attr;
591 
592 	tl_init_pair(colors, PTL_CURR, tab_face.c_fg, tab_face.c_bg);
593 	tab_face.current = COLOR_PAIR(PTL_CURR) | tab_face.c_attr;
594 
595 	/* body */
596 	tl_init_pair(colors, PBODY, body_face.fg, body_face.bg);
597 	body_face.body = COLOR_PAIR(PBODY);
598 
599 	tl_init_pair(colors, PBLEFT, body_face.lfg, body_face.lbg);
600 	body_face.left = COLOR_PAIR(PBLEFT);
601 
602 	tl_init_pair(colors, PBRIGHT, body_face.rfg, body_face.rbg);
603 	body_face.right = COLOR_PAIR(PBRIGHT);
604 
605 	/* modeline */
606 	tl_init_pair(colors, PMODELINE, modeline_face.fg, modeline_face.bg);
607 	modeline_face.background = COLOR_PAIR(PMODELINE) | modeline_face.attr;
608 
609 	/* minibuffer */
610 	tl_init_pair(colors, PMINIBUF, minibuffer_face.fg, minibuffer_face.bg);
611 	minibuffer_face.background = COLOR_PAIR(PMINIBUF) | minibuffer_face.attr;
612 
613 }