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