cookie persistance
[surf.git] / surf.c
1 /* See LICENSE file for copyright and license details.
2  *
3  * To understand surf, start reading main().
4  */
5 #include <X11/X.h>
6 #include <X11/Xatom.h>
7 #include <gtk/gtk.h>
8 #include <gdk/gdkx.h>
9 #include <gdk/gdk.h>
10 #include <gdk/gdkkeysyms.h>
11 #include <string.h>
12 #include <unistd.h>
13 #include <getopt.h>
14 #include <stdlib.h>
15 #include <stdio.h>
16 #include <webkit/webkit.h>
17
18 #define LENGTH(x) (sizeof x / sizeof x[0])
19
20 Display *dpy;
21 Atom urlprop;
22 typedef struct Client {
23         GtkWidget *win, *scroll, *vbox, *urlbar, *urllist, *searchbar;
24         WebKitWebView *view;
25         gchar *title;
26         gint progress;
27         struct Client *next;
28 } Client;
29 SoupCookieJar *cookiejar;
30 Client *clients = NULL;
31 gboolean embed = FALSE;
32 gboolean showxid = FALSE;
33 gboolean ignore_once = FALSE;
34 extern char *optarg;
35 extern int optind;
36
37 static void cleanup(void);
38 static gboolean decidewindow(WebKitWebView *view, WebKitWebFrame *f,
39                 WebKitNetworkRequest *r, WebKitWebNavigationAction *n,
40                 WebKitWebPolicyDecision *p, gpointer d);
41 static void destroyclient(Client *c);
42 static void destroywin(GtkWidget* w, gpointer d);
43 static void die(char *str);
44 static gboolean download(WebKitWebView *view, WebKitDownload *o, gpointer d);
45 static gchar *geturi(Client *c);
46 static void hidesearch(Client *c);
47 static void hideurl(Client *c);
48 static gboolean keypress(GtkWidget* w, GdkEventKey *ev, gpointer d);
49 static void linkhover(WebKitWebView* page, const gchar* t, const gchar* l, gpointer d);
50 static void loadcommit(WebKitWebView *view, WebKitWebFrame *f, gpointer d);
51 static void loadfile(const Client *c, const gchar *f);
52 static void loaduri(const Client *c, const gchar *uri);
53 static Client *newclient();
54 static WebKitWebView *newwindow(WebKitWebView  *v, WebKitWebFrame *f, gpointer d);
55 static void progresschange(WebKitWebView *view, gint p, gpointer d);
56 static GdkFilterReturn processx(GdkXEvent *xevent, GdkEvent *event, gpointer data);
57 static void setup(void);
58 static void showsearch(Client *c);
59 static void showurl(Client *c);
60 static void titlechange(WebKitWebView* view, WebKitWebFrame* frame, const gchar* title, gpointer d);
61 static void updatetitle(Client *c);
62
63 void
64 cleanup(void) {
65         while(clients)
66                 destroyclient(clients);
67 }
68
69 gboolean
70 decidewindow(WebKitWebView *view, WebKitWebFrame *f,
71                 WebKitNetworkRequest *r, WebKitWebNavigationAction *n,
72                 WebKitWebPolicyDecision *p, gpointer d) {
73         /* TODO */
74         return TRUE;
75 }
76
77 void
78 destroyclient(Client *c) {
79         Client *p;
80
81         gtk_widget_destroy(GTK_WIDGET(webkit_web_view_new()));
82         gtk_widget_destroy(c->scroll);
83         gtk_widget_destroy(c->urlbar);
84         gtk_widget_destroy(c->urllist);
85         gtk_widget_destroy(c->searchbar);
86         gtk_widget_destroy(c->vbox);
87         gtk_widget_destroy(c->win);
88         for(p = clients; p && p->next != c; p = p->next);
89         if(p)
90                 p->next = c->next;
91         else
92                 clients = c->next;
93         free(c);
94         if(clients == NULL)
95                 gtk_main_quit();
96 }
97
98 void
99 destroywin(GtkWidget* w, gpointer d) {
100         Client *c = (Client *)d;
101
102         destroyclient(c);
103 }
104
105 void die(char *str) {
106         fputs(str, stderr);
107         exit(EXIT_FAILURE);
108 }
109
110 gboolean
111 download(WebKitWebView *view, WebKitDownload *o, gpointer d) {
112         /* TODO */
113         const gchar *home;
114         gchar *uri, *filename;
115
116         home = g_get_home_dir();
117         filename = g_build_filename(home, "Desktop",
118                         webkit_download_get_suggested_filename(o), NULL);
119         uri = g_strconcat("file://", filename, NULL);
120         webkit_download_set_destination_uri(o, uri);
121         g_free(filename);
122         g_free(uri);
123         webkit_download_start(o);
124         return TRUE;
125 }
126
127 gchar *
128 geturi(Client *c) {
129         gchar *uri;
130
131         if(!(uri = (gchar *)webkit_web_view_get_uri(c->view)))
132                 uri = g_strdup("about:blank");
133         return uri;
134 }
135
136 void
137 hidesearch(Client *c) {
138         gtk_widget_hide(c->searchbar);
139         gtk_widget_grab_focus(GTK_WIDGET(c->view));
140 }
141
142 void
143 hideurl(Client *c) {
144         gtk_widget_hide(c->urlbar);
145         gtk_widget_grab_focus(GTK_WIDGET(c->view));
146 }
147
148 gboolean
149 keypress(GtkWidget* w, GdkEventKey *ev, gpointer d) {
150         Client *c = (Client *)d;
151
152         if(ev->type != GDK_KEY_PRESS)
153                 return FALSE;
154         if(GTK_WIDGET_HAS_FOCUS(c->searchbar)) {
155                 switch(ev->keyval) {
156                 case GDK_Escape:
157                         hidesearch(c);
158                         return TRUE;
159                 case GDK_Return:
160                         webkit_web_view_search_text(c->view,
161                                         gtk_entry_get_text(GTK_ENTRY(c->searchbar)),
162                                         FALSE,
163                                         !(ev->state & GDK_SHIFT_MASK),
164                                         TRUE);
165                         return TRUE;
166                 case GDK_Left:
167                 case GDK_Right:
168                 case GDK_r:
169                         return FALSE;
170                 }
171         }
172         else if(GTK_WIDGET_HAS_FOCUS(c->urlbar)) {
173                 switch(ev->keyval) {
174                 case GDK_Escape:
175                         hideurl(c);
176                         return TRUE;
177                 case GDK_Return:
178                         loaduri(c, gtk_entry_get_text(GTK_ENTRY(c->urlbar)));
179                         hideurl(c);
180                         return TRUE;
181                 case GDK_Left:
182                 case GDK_Right:
183                 case GDK_r:
184                         return FALSE;
185                 }
186         }
187         else if(ev->state == GDK_CONTROL_MASK || ev->state == (GDK_CONTROL_MASK | GDK_SHIFT_MASK)) {
188                 switch(ev->keyval) {
189                 case GDK_r:
190                 case GDK_R:
191                         if((ev->state & GDK_SHIFT_MASK))
192                                  webkit_web_view_reload_bypass_cache(c->view);
193                         else
194                                  webkit_web_view_reload(c->view);
195                         return TRUE;
196                 case GDK_b:
197                         return TRUE;
198                 case GDK_g:
199                         showurl(c);
200                         return TRUE;
201                 case GDK_slash:
202                         showsearch(c);
203                         return TRUE;
204                 case GDK_Left:
205                         webkit_web_view_go_back(c->view);
206                         return TRUE;
207                 case GDK_Right:
208                         webkit_web_view_go_forward(c->view);
209                         return TRUE;
210                 }
211         }
212         else {
213                 switch(ev->keyval) {
214                 case GDK_k:
215                         webkit_web_view_move_cursor(c->view, GTK_MOVEMENT_DISPLAY_LINES, -1);
216                         return TRUE;
217                 case GDK_j:
218                         webkit_web_view_move_cursor(c->view, GTK_MOVEMENT_DISPLAY_LINES, 1);
219                         return TRUE;
220                 }
221         }
222         return FALSE;
223 }
224
225 void
226 linkhover(WebKitWebView* page, const gchar* t, const gchar* l, gpointer d) {
227         Client *c = (Client *)d;
228
229         if(l)
230                 gtk_window_set_title(GTK_WINDOW(c->win), l);
231         else
232                 updatetitle(c);
233 }
234
235 void
236 loadcommit(WebKitWebView *view, WebKitWebFrame *f, gpointer d) {
237         Client *c = (Client *)d;
238         gchar *uri;
239
240         uri = geturi(c);
241         ignore_once = TRUE;
242         XChangeProperty(dpy, GDK_WINDOW_XID(GTK_WIDGET(c->win)->window), urlprop,
243                         XA_STRING, 8, PropModeReplace, (unsigned char *)uri,
244                         strlen(uri) + 1);
245 }
246
247 void
248 loadfile(const Client *c, const gchar *f) {
249         GIOChannel *chan = NULL;
250         GError *e = NULL;
251         GString *code = g_string_new("");
252         GString *uri = g_string_new(f);
253         gchar *line;
254
255         if(strcmp(f, "-") == 0) {
256                 chan = g_io_channel_unix_new(STDIN_FILENO);
257                 if (chan) {
258                         while(g_io_channel_read_line(chan, &line, NULL, NULL, &e) == G_IO_STATUS_NORMAL) {
259                                 g_string_append(code, line);
260                                 g_free(line);
261                         }
262                         webkit_web_view_load_html_string(c->view, code->str, NULL);
263                         g_io_channel_shutdown(chan, FALSE, NULL);
264                 }
265         }
266         else {
267                 g_string_prepend(uri, "file://");
268                 loaduri(c, uri->str);
269         }
270
271 }
272
273 void
274 loaduri(const Client *c, const gchar *uri) {
275         GString* u = g_string_new(uri);
276         if(g_strrstr(u->str, ":") == NULL)
277                 g_string_prepend(u, "http://");
278         webkit_web_view_load_uri(c->view, u->str);
279         g_string_free(u, TRUE);
280 }
281
282 Client *
283 newclient(void) {
284         Client *c;
285         if(!(c = calloc(1, sizeof(Client))))
286                 die("Cannot malloc!\n");
287         /* Window */
288         if(embed) {
289                 c->win = gtk_plug_new(0);
290         }
291         else {
292                 c->win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
293                 gtk_window_set_wmclass(GTK_WINDOW(c->win), "surf", "surf");
294         }
295         gtk_window_set_default_size(GTK_WINDOW(c->win), 800, 600);
296         g_signal_connect(G_OBJECT(c->win), "destroy", G_CALLBACK(destroywin), c);
297         g_signal_connect(G_OBJECT(c->win), "key-press-event", G_CALLBACK(keypress), c);
298
299         /* VBox */
300         c->vbox = gtk_vbox_new(FALSE, 0);
301
302         /* scrolled window */
303         c->scroll = gtk_scrolled_window_new(NULL, NULL);
304         gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(c->scroll),
305                         GTK_POLICY_NEVER, GTK_POLICY_NEVER);
306
307         /* webview */
308         c->view = WEBKIT_WEB_VIEW(webkit_web_view_new());
309         g_signal_connect(G_OBJECT(c->view), "title-changed", G_CALLBACK(titlechange), c);
310         g_signal_connect(G_OBJECT(c->view), "load-progress-changed", G_CALLBACK(progresschange), c);
311         g_signal_connect(G_OBJECT(c->view), "load-committed", G_CALLBACK(loadcommit), c);
312         g_signal_connect(G_OBJECT(c->view), "hovering-over-link", G_CALLBACK(linkhover), c);
313         g_signal_connect(G_OBJECT(c->view), "create-web-view", G_CALLBACK(newwindow), c);
314         g_signal_connect(G_OBJECT(c->view), "new-window-policy-decision-requested", G_CALLBACK(decidewindow), c);
315         g_signal_connect(G_OBJECT(c->view), "download-requested", G_CALLBACK(download), c);
316
317         /* urlbar */
318         c->urlbar = gtk_entry_new();
319         gtk_entry_set_has_frame(GTK_ENTRY(c->urlbar), FALSE);
320
321         /* searchbar */
322         c->searchbar = gtk_entry_new();
323         gtk_entry_set_has_frame(GTK_ENTRY(c->searchbar), FALSE);
324
325         /* Arranging */
326         gtk_container_add(GTK_CONTAINER(c->scroll), GTK_WIDGET(c->view));
327         gtk_container_add(GTK_CONTAINER(c->win), c->vbox);
328         gtk_container_add(GTK_CONTAINER(c->vbox), c->scroll);
329         gtk_container_add(GTK_CONTAINER(c->vbox), c->searchbar);
330         gtk_container_add(GTK_CONTAINER(c->vbox), c->urlbar);
331
332         /* Setup */
333         gtk_box_set_child_packing(GTK_BOX(c->vbox), c->urlbar, FALSE, FALSE, 0, GTK_PACK_START);
334         gtk_box_set_child_packing(GTK_BOX(c->vbox), c->searchbar, FALSE, FALSE, 0, GTK_PACK_START);
335         gtk_box_set_child_packing(GTK_BOX(c->vbox), c->scroll, TRUE, TRUE, 0, GTK_PACK_START);
336         gtk_widget_grab_focus(GTK_WIDGET(c->view));
337         gtk_widget_hide_all(c->searchbar);
338         gtk_widget_hide_all(c->urlbar);
339         gtk_widget_show(c->vbox);
340         gtk_widget_show(c->scroll);
341         gtk_widget_show(GTK_WIDGET(c->view));
342         gtk_widget_show(c->win);
343         gdk_window_set_events(GTK_WIDGET(c->win)->window, GDK_ALL_EVENTS_MASK);
344         gdk_window_add_filter(GTK_WIDGET(c->win)->window, processx, c);
345         c->next = clients;
346         clients = c;
347         if(showxid)
348                 printf("%u\n", (unsigned int)GDK_WINDOW_XID(GTK_WIDGET(c->win)->window));
349         return c;
350 }
351
352 WebKitWebView *
353 newwindow(WebKitWebView  *v, WebKitWebFrame *f, gpointer d) {
354         /* TODO */
355         Client *c = newclient();
356         return c->view;
357 }
358
359 void
360 progresschange(WebKitWebView* view, gint p, gpointer d) {
361         Client *c = (Client *)d;
362
363         c->progress = p;
364         updatetitle(c);
365 }
366
367 GdkFilterReturn
368 processx(GdkXEvent *e, GdkEvent *event, gpointer d) {
369         XPropertyEvent *ev;
370         Client *c = (Client *)d;
371         Atom adummy;
372         int idummy;
373         unsigned long ldummy;
374         unsigned char *buf = NULL;
375         if(((XEvent *)e)->type == PropertyNotify) {
376                 ev = &((XEvent *)e)->xproperty;
377                 if(ignore_once == FALSE && ev->atom == urlprop && ev->state == PropertyNewValue) {
378                         XGetWindowProperty(dpy, ev->window, urlprop, 0L, BUFSIZ, False, XA_STRING,
379                                 &adummy, &idummy, &ldummy, &ldummy, &buf);
380                         loaduri(c, (gchar *)buf);
381                         XFree(buf);
382                         return GDK_FILTER_REMOVE;
383                 }
384         }
385         return GDK_FILTER_CONTINUE;
386 }
387
388 void setup(void) {
389         dpy = GDK_DISPLAY();
390         urlprop = XInternAtom(dpy, "_SURF_URL", False);
391 }
392
393 void
394 showsearch(Client *c) {
395         hideurl(c);
396         gtk_widget_show(c->searchbar);
397         gtk_widget_grab_focus(c->searchbar);
398 }
399
400 void
401 showurl(Client *c) {
402         gchar *uri;
403
404         hidesearch(c);
405         uri = geturi(c);
406         gtk_entry_set_text(GTK_ENTRY(c->urlbar), uri);
407         gtk_widget_show(c->urlbar);
408         gtk_widget_grab_focus(c->urlbar);
409 }
410
411 void
412 titlechange(WebKitWebView *v, WebKitWebFrame *f, const gchar *t, gpointer d) {
413         Client *c = (Client *)d;
414
415         if(c->title)
416                 g_free(c->title);
417         c->title = g_strdup(t);
418         updatetitle(c);
419 }
420
421 void
422 updatetitle(Client *c) {
423         char t[512];
424         if(c->progress == 100)
425                 snprintf(t, LENGTH(t), "%s", c->title);
426         else
427                 snprintf(t, LENGTH(t), "%s [%i%%]", c->title, c->progress);
428         gtk_window_set_title(GTK_WINDOW(c->win), t);
429 }
430
431 int main(int argc, char *argv[]) {
432         gchar *uri = NULL, *file = NULL;
433         SoupSession *s;
434         Client *c;
435         int o;
436         const gchar *home, *filename;
437
438         gtk_init(NULL, NULL);
439         if (!g_thread_supported())
440                 g_thread_init(NULL);
441         setup();
442         while((o = getopt(argc, argv, "vhxeu:f:")) != -1)
443                 switch(o) {
444                 case 'x':
445                         showxid = TRUE;
446                         break;
447                 case 'e':
448                         showxid = TRUE;
449                         embed = TRUE;
450                         break;
451                 case 'u':
452                         if(!(uri = optarg))
453                                 goto argerr;
454                         c = newclient();
455                         loaduri(c, uri);
456                         updatetitle(c);
457                         break;
458                 case 'f':
459                         if(!(file = optarg))
460                                 goto argerr;
461                         c = newclient();
462                         loadfile(c, file);
463                         updatetitle(c);
464                         break;
465                 case 'v':
466                         die("surf-"VERSION", © 2009 surf engineers, see LICENSE for details\n");
467                         break;
468                 argerr:
469                 default:
470                         puts("surf - simple browser");
471                         die("usage: surf [-e] [-x] [-u uri] [-f file]\n");
472                         return EXIT_FAILURE;
473                 }
474         if(optind != argc)
475                 goto argerr;
476         if(!clients)
477                 newclient();
478
479         /* cookie persistance */
480         s = webkit_get_default_session();
481         home = g_get_home_dir();
482         filename = g_build_filename(home, ".surf-cookies", NULL);
483         cookiejar = soup_cookie_jar_text_new(filename, FALSE);
484         soup_session_add_feature(s, SOUP_SESSION_FEATURE(cookiejar));
485
486         gtk_main();
487         cleanup();
488         return EXIT_SUCCESS;
489 }