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 <errno.h>
20 #include <stdio.h>
21 #include <stdlib.h>
22 #include <string.h>
23 #include <unistd.h>
24 
25 #include "defaults.h"
26 #include "minibuffer.h"
27 #include "parser.h"
28 #include "session.h"
29 #include "ui.h"
30 
31 static struct event	 autosaveev;
32 
33 void
34 switch_to_tab(struct tab *tab)
35 {
36 	current_tab = tab;
37 	tab->flags &= ~TAB_URGENT;
38 
39 	if (operating && tab->flags & TAB_LAZY)
40 		load_url_in_tab(tab, tab->hist_cur->h, NULL, 0);
41 }
42 
43 unsigned int
44 tab_new_id(void)
45 {
46 	static uint32_t tab_counter;
47 
48 	return tab_counter++;
49 }
50 
51 struct tab *
52 new_tab(const char *url, const char *base, struct tab *after)
53 {
54 	struct tab	*tab;
55 
56 	ui_schedule_redraw();
57 	autosave_hook();
58 
59 	if ((tab = calloc(1, sizeof(*tab))) == NULL) {
60 		event_loopbreak();
61 		return NULL;
62 	}
63 	tab->fd = -1;
64 
65 	TAILQ_INIT(&tab->hist.head);
66 
67 	TAILQ_INIT(&tab->buffer.head);
68 	TAILQ_INIT(&tab->buffer.page.head);
69 
70 	tab->id = tab_new_id();
71 	if (!operating)
72 		tab->flags |= TAB_LAZY;
73 	switch_to_tab(tab);
74 
75 	if (after != NULL)
76 		TAILQ_INSERT_AFTER(&tabshead, after, tab, tabs);
77 	else
78 		TAILQ_INSERT_TAIL(&tabshead, tab, tabs);
79 
80 	load_url_in_tab(tab, url, base, 0);
81 	return tab;
82 }
83 
84 /*
85  * Free every resource linked to the tab, including the tab itself.
86  * Removes the tab from the tablist, but doesn't update the
87  * current_tab though.
88  */
89 void
90 free_tab(struct tab *tab)
91 {
92 	stop_tab(tab);
93 	ui_schedule_redraw();
94 	autosave_hook();
95 
96 	if (evtimer_pending(&tab->loadingev, NULL))
97 		evtimer_del(&tab->loadingev);
98 
99 	TAILQ_REMOVE(&tabshead, tab, tabs);
100 	free(tab);
101 }
102 
103 void
104 stop_tab(struct tab *tab)
105 {
106 	ui_send_net(IMSG_STOP, tab->id, NULL, 0);
107 
108 	if (tab->fd != -1) {
109 		close(tab->fd);
110 		tab->fd = -1;
111 		free(tab->path);
112 		tab->path = NULL;
113 		load_page_from_str(tab, "Stopped.\n");
114 	}
115 }
116 
117 void
118 save_session(void)
119 {
120 	struct tab	*tab;
121 	char		*t;
122 	int		 flags;
123 
124 	ui_send_fs(IMSG_SESSION_START, 0, NULL, 0);
125 
126 	TAILQ_FOREACH(tab, &tabshead, tabs) {
127 		flags = tab->flags;
128 		if (tab == current_tab)
129 			flags |= TAB_CURRENT;
130 
131 		t = tab->hist_cur->h;
132 		ui_send_fs(IMSG_SESSION_TAB, flags, t, strlen(t)+1);
133 
134 		t = tab->buffer.page.title;
135 		ui_send_fs(IMSG_SESSION_TAB_TITLE, 0, t, strlen(t)+1);
136 	}
137 
138 	ui_send_fs(IMSG_SESSION_END, 0, NULL, 0);
139 }
140 
141 /*
142  * Parse a line of the session file.  The format is:
143  *
144  *	URL [flags,...] [title]\n
145  */
146 static void
147 parse_session_line(char *line, const char **title, uint32_t *flags)
148 {
149 	char *s, *t, *ap;
150 
151 	*title = "";
152 	*flags = 0;
153 	if ((s = strchr(line, ' ')) == NULL)
154 		return;
155 
156 	*s++ = '\0';
157 
158 	if ((t = strchr(s, ' ')) != NULL) {
159 		*t++ = '\0';
160 		*title = t;
161 	}
162 
163 	while ((ap = strsep(&s, ",")) != NULL) {
164 		if (*ap == '\0')
165 			;
166 		else if (!strcmp(ap, "current"))
167 			*flags |= TAB_CURRENT;
168 		else
169 			message("unknown tab flag: %s", ap);
170 	}
171 }
172 
173 void
174 load_last_session(void)
175 {
176 	const char	*title;
177 	char		*nl, *line = NULL;
178 	uint32_t	 flags;
179 	size_t		 linesize = 0;
180 	ssize_t		 linelen;
181 	FILE		*session;
182 	struct tab	*tab, *curr = NULL;
183 
184 	if ((session = fopen(session_file, "r")) == NULL) {
185 		/* first time? */
186 		new_tab("about:new", NULL, NULL);
187 		switch_to_tab(new_tab("about:help", NULL, NULL));
188 		return;
189 	}
190 
191 	while ((linelen = getline(&line, &linesize, session)) != -1) {
192 		if ((nl = strchr(line, '\n')) != NULL)
193 			*nl = '\0';
194 		parse_session_line(line, &title, &flags);
195 		if ((tab = new_tab(line, NULL, NULL)) == NULL)
196 			err(1, "new_tab");
197 		strlcpy(tab->buffer.page.title, title,
198 		    sizeof(tab->buffer.page.title));
199 		if (flags & TAB_CURRENT)
200 			curr = tab;
201 	}
202 
203 	if (ferror(session))
204 		message("error reading %s: %s",
205 		    session_file, strerror(errno));
206 	fclose(session);
207 	free(line);
208 
209 	if (curr != NULL)
210 		switch_to_tab(curr);
211 
212 	if (last_time_crashed())
213 		switch_to_tab(new_tab("about:crash", NULL, NULL));
214 
215 	return;
216 }
217 
218 void
219 autosave_init(void)
220 {
221 	evtimer_set(&autosaveev, autosave_timer, NULL);
222 }
223 
224 void
225 autosave_timer(int fd, short event, void *data)
226 {
227 	save_session();
228 }
229 
230 /*
231  * Function to be called in "interesting" places where we may want to
232  * schedule an autosave (like on new tab or before loading an url.)
233  */
234 void
235 autosave_hook(void)
236 {
237 	struct timeval tv;
238 
239 	if (autosave <= 0)
240 		return;
241 
242 	if (!evtimer_pending(&autosaveev, NULL)) {
243 		tv.tv_sec = autosave;
244 		tv.tv_usec = 0;
245 
246 		evtimer_add(&autosaveev, &tv);
247 	}
248 }