op public repos

Blob

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