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 "compat.h"
18 
19 #include <stdio.h>
20 #include <stdlib.h>
21 #include <string.h>
22 
23 #include "minibuffer.h"
24 #include "ui.h"
25 #include "utf8.h"
26 
27 static void		*minibuffer_metadata(void);
28 static void		 minibuffer_hist_save_entry(void);
29 static void		 yornp_self_insert(void);
30 static void		 yornp_abort(void);
31 static void		 read_self_insert(void);
32 static void		 read_abort(void);
33 static void		 read_select(void);
34 static void		 handle_clear_echoarea(int, short, void *);
35 
36 static struct event	clechoev;
37 static struct timeval	clechoev_timer = { 5, 0 };
38 
39 static void (*yornp_cb)(int, struct tab *);
40 static struct tab *yornp_data;
41 
42 static void (*read_cb)(const char*, struct tab *);
43 static struct tab *read_data;
44 
45 struct histhead eecmd_history,
46 	ir_history,
47 	lu_history,
48 	read_history;
49 
50 struct ministate ministate;
51 
52 struct buffer minibufferwin;
53 
54 int in_minibuffer;
55 
56 /*
57  * Recompute the visible completions.  If add is 1, don't consider the
58  * ones already hidden.
59  */
60 void
61 recompute_completions(int add)
62 {
63 	struct line	*l;
64 	struct vline	*vl;
65 	struct buffer	*b;
66 
67 	if (in_minibuffer != MB_COMPREAD)
68 		return;
69 
70 	b = &ministate.compl.buffer;
71 	TAILQ_FOREACH(l, &b->page.head, lines) {
72 		l->type = LINE_COMPL;
73 		if (add && l->flags & L_HIDDEN)
74 			continue;
75 		if (strcasestr(l->line, ministate.buf) != NULL ||
76 		    (l->alt != NULL && strcasestr(l->alt, ministate.buf) != NULL)) {
77 			if (l->flags & L_HIDDEN)
78 				b->line_max++;
79 			l->flags &= ~L_HIDDEN;
80 		} else {
81 			if (!(l->flags & L_HIDDEN))
82 				b->line_max--;
83 			l->flags |= L_HIDDEN;
84 		}
85 	}
86 
87 	if (b->current_line == NULL)
88 		b->current_line = TAILQ_FIRST(&b->head);
89 	b->current_line = adjust_line(b->current_line, b);
90 	vl = b->current_line;
91 	if (vl != NULL)
92 		vl->parent->type = LINE_COMPL_CURRENT;
93 }
94 
95 static void *
96 minibuffer_metadata(void)
97 {
98 	struct vline	*vl;
99 
100 	vl = ministate.compl.buffer.current_line;
101 
102 	if (vl == NULL || vl->parent->flags & L_HIDDEN)
103 		return NULL;
104 
105 	return vl->parent->data;
106 }
107 
108 static void
109 minibuffer_hist_save_entry(void)
110 {
111 	struct hist	*hist;
112 
113 	if (ministate.history == NULL)
114 		return;
115 
116 	if ((hist = calloc(1, sizeof(*hist))) == NULL)
117 		abort();
118 
119 	strlcpy(hist->h, ministate.buf, sizeof(hist->h));
120 
121 	if (TAILQ_EMPTY(&ministate.history->head))
122 		TAILQ_INSERT_HEAD(&ministate.history->head, hist, entries);
123 	else
124 		TAILQ_INSERT_TAIL(&ministate.history->head, hist, entries);
125 	ministate.history->len++;
126 }
127 
128 /*
129  * taint the minibuffer cache: if we're currently showing a history
130  * element, copy that to the current buf and reset the "history
131  * navigation" thing.
132  */
133 void
134 minibuffer_taint_hist(void)
135 {
136 	if (ministate.hist_cur == NULL)
137 		return;
138 
139 	strlcpy(ministate.buf, ministate.hist_cur->h, sizeof(ministate.buf));
140 	ministate.hist_cur = NULL;
141 	ministate.buffer.current_line->line = ministate.buf;
142 }
143 
144 void
145 minibuffer_self_insert(void)
146 {
147 	char	*c, tmp[5] = {0};
148 	size_t	len;
149 
150 	minibuffer_taint_hist();
151 
152 	if (thiskey.cp == 0)
153 		return;
154 
155 	len = utf8_encode(thiskey.cp, tmp);
156 	c = utf8_nth(ministate.buffer.current_line->line, ministate.buffer.cpoff);
157 	if (c + len > ministate.buf + sizeof(ministate.buf) - 1)
158 		return;
159 
160 	memmove(c + len, c, strlen(c)+1);
161 	memcpy(c, tmp, len);
162 	ministate.buffer.cpoff++;
163 
164 	recompute_completions(1);
165 }
166 
167 void
168 sensible_self_insert(void)
169 {
170 	if (thiskey.meta ||
171 	    (unicode_isspace(thiskey.key) && thiskey.key != ' ')) {
172 		global_key_unbound();
173 		return;
174 	}
175 
176 	minibuffer_self_insert();
177 }
178 
179 void
180 eecmd_self_insert(void)
181 {
182 	if (thiskey.meta || unicode_isspace(thiskey.cp) ||
183 	    !unicode_isgraph(thiskey.cp)) {
184 		global_key_unbound();
185 		return;
186 	}
187 
188 	minibuffer_self_insert();
189 }
190 
191 void
192 eecmd_select(void)
193 {
194 	struct cmd *cmd;
195 
196 	for (cmd = cmds; cmd->cmd != NULL; ++cmd) {
197 		if (!strcmp(cmd->cmd, ministate.buf)) {
198 			exit_minibuffer();
199 			minibuffer_hist_save_entry();
200 			cmd->fn(current_buffer());
201 			return;
202 		}
203 	}
204 
205 	message("No match");
206 }
207 
208 void
209 ir_self_insert(void)
210 {
211 	minibuffer_self_insert();
212 }
213 
214 void
215 ir_select(void)
216 {
217 	char		 buf[1025] = {0};
218 	struct phos_uri	 uri;
219 	struct tab	*tab = current_tab;
220 
221 	exit_minibuffer();
222 	minibuffer_hist_save_entry();
223 
224 	/* a bit ugly but... */
225 	memcpy(&uri, &tab->uri, sizeof(tab->uri));
226 	phos_uri_set_query(&uri, ministate.buf);
227 	phos_serialize_uri(&uri, buf, sizeof(buf));
228 	load_url_in_tab(tab, buf, NULL);
229 }
230 
231 void
232 lu_select(void)
233 {
234 	exit_minibuffer();
235 	minibuffer_hist_save_entry();
236 	load_url_in_tab(current_tab, ministate.buf, NULL);
237 }
238 
239 void
240 bp_select(void)
241 {
242 	exit_minibuffer();
243 	if (*ministate.buf != '\0')
244 		add_to_bookmarks(ministate.buf);
245 	else
246 		message("Abort.");
247 }
248 
249 void
250 ts_select(void)
251 {
252 	struct tab	*tab;
253 
254 	if ((tab = minibuffer_metadata()) == NULL) {
255 		message("No tab selected");
256 		return;
257 	}
258 
259 	exit_minibuffer();
260 	switch_to_tab(tab);
261 }
262 
263 void
264 ls_select(void)
265 {
266 	struct line	*l;
267 
268 	if ((l = minibuffer_metadata()) == NULL) {
269 		message("No link selected");
270 		return;
271 	}
272 
273 	exit_minibuffer();
274 	load_url_in_tab(current_tab, l->alt, NULL);
275 }
276 
277 static inline void
278 jump_to_line(struct line *l)
279 {
280 	struct vline	*vl;
281 	struct buffer	*buffer;
282 
283 	buffer = current_buffer();
284 
285 	TAILQ_FOREACH(vl, &buffer->head, vlines) {
286 		if (vl->parent == l)
287 			break;
288 	}
289 
290 	if (vl == NULL)
291 		message("Ops, %s error!  Please report to %s",
292 		    __func__, PACKAGE_BUGREPORT);
293 	else {
294 		buffer->top_line = vl;
295 		buffer->current_line = vl;
296 	}
297 }
298 
299 void
300 swiper_select(void)
301 {
302 	struct line	*l;
303 
304 	if ((l = minibuffer_metadata()) == NULL) {
305 		message("No line selected");
306 		return;
307 	}
308 
309 	exit_minibuffer();
310 	jump_to_line(l);
311 }
312 
313 void
314 toc_select(void)
315 {
316 	struct line	*l;
317 
318 	if ((l = minibuffer_metadata()) == NULL) {
319 		message("No line selected");
320 		return;
321 	}
322 
323 	exit_minibuffer();
324 	jump_to_line(l);
325 }
326 
327 static void
328 yornp_self_insert(void)
329 {
330 	if (thiskey.key != 'y' && thiskey.key != 'n') {
331 		message("Please answer y or n");
332 		return;
333 	}
334 
335 	exit_minibuffer();
336 	yornp_cb(thiskey.key == 'y', yornp_data);
337 }
338 
339 static void
340 yornp_abort(void)
341 {
342 	exit_minibuffer();
343 	yornp_cb(0, yornp_data);
344 }
345 
346 static void
347 read_self_insert(void)
348 {
349 	if (thiskey.meta || !unicode_isgraph(thiskey.cp)) {
350 		global_key_unbound();
351 		return;
352 	}
353 
354 	minibuffer_self_insert();
355 }
356 
357 static void
358 read_abort(void)
359 {
360 	exit_minibuffer();
361 	read_cb(NULL, read_data);
362 }
363 
364 static void
365 read_select(void)
366 {
367         exit_minibuffer();
368 	minibuffer_hist_save_entry();
369 	read_cb(ministate.buf, read_data);
370 }
371 
372 /*
373  * TODO: we should collect this asynchronously...
374  */
375 static inline void
376 populate_compl_buffer(complfn *fn, void *data)
377 {
378 	const char	*s, *descr;
379 	struct line	*l;
380 	struct buffer	*b;
381 	struct parser	*p;
382 	void		*linedata;
383 
384 	b = &ministate.compl.buffer;
385 	p = &b->page;
386 
387 	linedata = NULL;
388 	descr = NULL;
389 	while ((s = fn(&data, &linedata, &descr)) != NULL) {
390 		if ((l = calloc(1, sizeof(*l))) == NULL)
391 			abort();
392 
393 		l->type = LINE_COMPL;
394 		l->data = linedata;
395 		l->alt = (char*)descr;
396 		if ((l->line = strdup(s)) == NULL)
397 			abort();
398 
399 		if (TAILQ_EMPTY(&p->head))
400 			TAILQ_INSERT_HEAD(&p->head, l, lines);
401 		else
402 			TAILQ_INSERT_TAIL(&p->head, l, lines);
403 
404 		linedata = NULL;
405 		descr = NULL;
406 	}
407 
408 	if ((l = TAILQ_FIRST(&p->head)) != NULL)
409 		l->type = LINE_COMPL_CURRENT;
410 }
411 
412 void
413 enter_minibuffer(void (*self_insert_fn)(void), void (*donefn)(void),
414     void (*abortfn)(void), struct histhead *hist,
415     complfn *complfn, void *compldata)
416 {
417 	in_minibuffer = complfn == NULL ? MB_READ : MB_COMPREAD;
418 	if (in_minibuffer == MB_COMPREAD) {
419 		ui_schedule_redraw();
420 
421 		ministate.compl.fn = complfn;
422 		ministate.compl.data = compldata;
423 		populate_compl_buffer(complfn, compldata);
424 	}
425 
426 	base_map = &minibuffer_map;
427 	current_map = &minibuffer_map;
428 
429 	base_map->unhandled_input = self_insert_fn;
430 
431 	ministate.donefn = donefn;
432 	ministate.abortfn = abortfn;
433 	memset(ministate.buf, 0, sizeof(ministate.buf));
434 	ministate.buffer.current_line = &ministate.vline;
435 	ministate.buffer.current_line->line = ministate.buf;
436 	ministate.buffer.cpoff = 0;
437 	strlcpy(ministate.buf, "", sizeof(ministate.prompt));
438 
439 	ministate.history = hist;
440 	ministate.hist_cur = NULL;
441 	ministate.hist_off = 0;
442 }
443 
444 void
445 exit_minibuffer(void)
446 {
447 	if (in_minibuffer == MB_COMPREAD) {
448 		erase_buffer(&ministate.compl.buffer);
449 		ui_schedule_redraw();
450 	}
451 
452 	in_minibuffer = 0;
453 	base_map = &global_map;
454 	current_map = &global_map;
455 }
456 
457 void
458 yornp(const char *prompt, void (*fn)(int, struct tab*),
459     struct tab *data)
460 {
461 	size_t len;
462 
463 	if (in_minibuffer) {
464 		fn(0, data);
465 		return;
466 	}
467 
468 	yornp_cb = fn;
469 	yornp_data = data;
470 	enter_minibuffer(yornp_self_insert, yornp_self_insert,
471 	    yornp_abort, NULL, NULL, NULL);
472 
473 	len = sizeof(ministate.prompt);
474 	strlcpy(ministate.prompt, prompt, len);
475 	strlcat(ministate.prompt, " (y or n) ", len);
476 }
477 
478 void
479 minibuffer_read(const char *prompt, void (*fn)(const char *, struct tab *),
480     struct tab *data)
481 {
482 	size_t len;
483 
484 	if (in_minibuffer)
485 		return;
486 
487 	read_cb = fn;
488 	read_data = data;
489 	enter_minibuffer(read_self_insert, read_select, read_abort,
490 	    &read_history, NULL, NULL);
491 
492 	len = sizeof(ministate.prompt);
493 	strlcpy(ministate.prompt, prompt, len);
494 	strlcat(ministate.prompt, ": ", len);
495 }
496 
497 static void
498 handle_clear_echoarea(int fd, short ev, void *d)
499 {
500 	free(ministate.curmesg);
501 	ministate.curmesg = NULL;
502 
503 	ui_after_message_hook();
504 }
505 
506 void
507 vmessage(const char *fmt, va_list ap)
508 {
509 	if (evtimer_pending(&clechoev, NULL))
510 		evtimer_del(&clechoev);
511 
512 	free(ministate.curmesg);
513 	ministate.curmesg = NULL;
514 
515 	if (fmt != NULL) {
516 		evtimer_set(&clechoev, handle_clear_echoarea, NULL);
517 		evtimer_add(&clechoev, &clechoev_timer);
518 
519 		/* TODO: what to do if the allocation fails here? */
520 		if (vasprintf(&ministate.curmesg, fmt, ap) == -1)
521 			ministate.curmesg = NULL;
522 	}
523 
524 	ui_after_message_hook();
525 }
526 
527 void
528 message(const char *fmt, ...)
529 {
530 	va_list ap;
531 
532 	va_start(ap, fmt);
533 	vmessage(fmt, ap);
534 	va_end(ap);
535 }
536 
537 void
538 minibuffer_init(void)
539 {
540 	TAILQ_INIT(&eecmd_history.head);
541 	TAILQ_INIT(&ir_history.head);
542 	TAILQ_INIT(&lu_history.head);
543 
544 	ministate.line.type = LINE_TEXT;
545 	ministate.vline.parent = &ministate.line;
546 	ministate.buffer.current_line = &ministate.vline;
547 
548 	evtimer_set(&clechoev, handle_clear_echoarea, NULL);
549 }