Applied print patch from pancake
[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                 case GDK_P:
201                         if((ev->state & GDK_SHIFT_MASK))
202                                 webkit_web_frame_print(webkit_web_view_get_main_frame(c->view));
203                         else
204                                 gtk_clipboard_request_text(gtk_clipboard_get(GDK_SELECTION_PRIMARY), pasteurl, c);
205                         return TRUE;
206
207                 case GDK_y:
208                         gtk_clipboard_set_text(gtk_clipboard_get(GDK_SELECTION_PRIMARY), webkit_web_view_get_uri(c->view), -1);
209                         return TRUE;
210                 case GDK_r:
211                 case GDK_R:
212                         if((ev->state & GDK_SHIFT_MASK))
213                                  webkit_web_view_reload_bypass_cache(c->view);
214                         else
215                                  webkit_web_view_reload(c->view);
216                         return TRUE;
217                 case GDK_b:
218                         return TRUE;
219                 case GDK_g:
220                         showurl(c);
221                         return TRUE;
222                 case GDK_slash:
223                         showsearch(c);
224                         return TRUE;
225                 case GDK_plus:
226                 case GDK_equal:
227                         webkit_web_view_zoom_in(c->view);
228                         return TRUE;
229                 case GDK_minus:
230                         webkit_web_view_zoom_out(c->view);
231                         return TRUE;
232                 case GDK_0:
233                         webkit_web_view_set_zoom_level(c->view, 1.0);
234                         return TRUE;
235                 case GDK_n:
236                 case GDK_N:
237                         webkit_web_view_search_text(c->view,
238                                         gtk_entry_get_text(GTK_ENTRY(c->searchbar)),
239                                         FALSE,
240                                         !(ev->state & GDK_SHIFT_MASK),
241                                         TRUE);
242                         return TRUE;
243                 case GDK_Left:
244                         webkit_web_view_go_back(c->view);
245                         return TRUE;
246                 case GDK_Right:
247                         webkit_web_view_go_forward(c->view);
248                         return TRUE;
249                 }
250         }
251         else {
252                 switch(ev->keyval) {
253                 case GDK_Escape:
254                         stop(c);
255                         return TRUE;
256                 }
257         }
258         return FALSE;
259 }
260
261 void
262 linkhover(WebKitWebView* page, const gchar* t, const gchar* l, Client *c) {
263         if(l)
264                 gtk_window_set_title(GTK_WINDOW(c->win), l);
265         else
266                 updatetitle(c, NULL);
267 }
268
269 void
270 loadcommit(WebKitWebView *view, WebKitWebFrame *f, Client *c) {
271         gchar *uri;
272
273         ignore_once = TRUE;
274         uri = geturi(c);
275         XChangeProperty(dpy, GDK_WINDOW_XID(GTK_WIDGET(c->win)->window), urlprop,
276                         XA_STRING, 8, PropModeReplace, (unsigned char *)uri,
277                         strlen(uri) + 1);
278 }
279
280 void
281 loadstart(WebKitWebView *view, WebKitWebFrame *f, Client *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
425         if(((XEvent *)e)->type == PropertyNotify) {
426                 ev = &((XEvent *)e)->xproperty;
427                 if(ev->atom == urlprop && ev->state == PropertyNewValue) {
428                         if(ignore_once)
429                                ignore_once = FALSE;
430                         else {
431                                 XGetWindowProperty(dpy, ev->window, urlprop, 0L, BUFSIZ, False, XA_STRING,
432                                         &adummy, &idummy, &ldummy, &ldummy, &buf);
433                                 loaduri(c, (gchar *)buf);
434                                 XFree(buf);
435                         }
436                         return GDK_FILTER_REMOVE;
437                 }
438         }
439         return GDK_FILTER_CONTINUE;
440 }
441
442 void
443 progresschange(WebKitWebView* view, gint p, Client *c) {
444         c->progress = p;
445         updatetitle(c, NULL);
446 }
447
448 void setup(void) {
449         dpy = GDK_DISPLAY();
450         urlprop = XInternAtom(dpy, "_SURF_URL", False);
451 }
452
453 void
454 showsearch(Client *c) {
455         hideurl(c);
456         gtk_widget_show(c->searchbar);
457         gtk_widget_grab_focus(c->searchbar);
458 }
459
460 void
461 showurl(Client *c) {
462         gchar *uri;
463
464         hidesearch(c);
465         uri = geturi(c);
466         gtk_entry_set_text(GTK_ENTRY(c->urlbar), uri);
467         gtk_widget_show(c->urlbar);
468         gtk_widget_grab_focus(c->urlbar);
469 }
470
471 void
472 stop(Client *c) {
473         if(c->download)
474                 webkit_download_cancel(c->download);
475         else
476                 webkit_web_view_stop_loading(c->view);
477         c->download = NULL;
478 }
479
480 void
481 titlechange(WebKitWebView *v, WebKitWebFrame *f, const gchar *t, Client *c) {
482         updatetitle(c, t);
483 }
484
485 void
486 usage() {
487         fputs("surf - simple browser\n", stderr);
488         die("usage: surf [-e] [-x] [-u uri] [-f file]\n");
489 }
490
491 void
492 updatetitle(Client *c, const char *title) {
493         gchar *t;
494
495         if(title) {
496                 if(c->title)
497                         g_free(c->title);
498                 c->title = g_strdup(title);
499         }
500         if(c->progress == 100)
501                 t = g_strdup(c->title);
502         else
503                 t = g_strdup_printf("%s [%i%%]", c->title, c->progress);
504         gtk_window_set_title(GTK_WINDOW(c->win), t);
505         g_free(t);
506
507 }
508
509 int main(int argc, char *argv[]) {
510         SoupSession *s;
511         Client *c;
512         int o;
513         const gchar *home, *filename;
514
515         gtk_init(NULL, NULL);
516         if (!g_thread_supported())
517                 g_thread_init(NULL);
518         setup();
519         while((o = getopt(argc, argv, "vhxeu:f:")) != -1)
520                 switch(o) {
521                 case 'x':
522                         showxid = TRUE;
523                         break;
524                 case 'e':
525                         showxid = TRUE;
526                         embed = TRUE;
527                         break;
528                 case 'u':
529                         c = newclient();
530                         loaduri(c, optarg);
531                         break;
532                 case 'f':
533                         c = newclient();
534                         loadfile(c, optarg);
535                         break;
536                 case 'v':
537                         die("surf-"VERSION", © 2009 surf engineers, see LICENSE for details\n");
538                         break;
539                 default:
540                         usage();
541                 }
542         if(optind != argc)
543                 usage();
544         if(!clients)
545                 newclient();
546
547         /* make dirs */
548         home = g_get_home_dir();
549         filename = g_build_filename(home, ".surf", NULL);
550         g_mkdir_with_parents(filename, 0711);
551         filename = g_build_filename(home, ".surf", "dl", NULL);
552         g_mkdir_with_parents(filename, 0755);
553
554         /* cookie persistance */
555         s = webkit_get_default_session();
556         filename = g_build_filename(home, ".surf", "cookies", NULL);
557         cookiejar = soup_cookie_jar_text_new(filename, FALSE);
558         soup_session_add_feature(s, SOUP_SESSION_FEATURE(cookiejar));
559
560         gtk_main();
561         cleanup();
562         return EXIT_SUCCESS;
563 }