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