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