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