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