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