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