Adding small fix by Alex Puterbaugh. 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 <signal.h>
6 #include <X11/X.h>
7 #include <X11/Xatom.h>
8 #include <gtk/gtk.h>
9 #include <gdk/gdkx.h>
10 #include <gdk/gdk.h>
11 #include <gdk/gdkkeysyms.h>
12 #include <string.h>
13 #include <sys/types.h>
14 #include <sys/wait.h>
15 #include <unistd.h>
16 #include <stdlib.h>
17 #include <stdio.h>
18 #include <webkit/webkit.h>
19 #include <glib/gstdio.h>
20 #include <JavaScriptCore/JavaScript.h>
21
22 #define LENGTH(x)               (sizeof x / sizeof x[0])
23 #define CLEANMASK(mask)         (mask & ~(GDK_MOD2_MASK))
24
25 typedef union Arg Arg;
26 union Arg {
27         gboolean b;
28         gint i;
29         const void *v;
30 };
31
32 typedef struct Client {
33         GtkWidget *win, *scroll, *vbox, *indicator;
34         GtkWidget **items;
35         WebKitWebView *view;
36         WebKitDownload *download;
37         char *title, *linkhover;
38         const char *uri, *needle;
39         gint progress;
40         struct Client *next;
41         gboolean zoomed;
42 } Client;
43
44 typedef struct {
45         char *label;
46         void (*func)(Client *c, const Arg *arg);
47         const Arg arg;
48 } Item;
49
50 typedef struct {
51         guint mod;
52         guint keyval;
53         void (*func)(Client *c, const Arg *arg);
54         const Arg arg;
55 } Key;
56
57 static Display *dpy;
58 static Atom uriprop, findprop;
59 static Client *clients = NULL;
60 static GdkNativeWindow embed = 0;
61 static gboolean showxid = FALSE;
62 static int ignorexprop = 0;
63 static char winid[64];
64 static char *progname;
65 static gboolean loadimage = 1, plugin = 1, script = 1;
66
67 static char *buildpath(const char *path);
68 static void cleanup(void);
69 static void clipboard(Client *c, const Arg *arg);
70 static void context(WebKitWebView *v, GtkMenu *m, Client *c);
71 static char *copystr(char **str, const char *src);
72 static WebKitWebView *createwindow(WebKitWebView *v, WebKitWebFrame *f, Client *c);
73 static gboolean decidedownload(WebKitWebView *v, WebKitWebFrame *f, WebKitNetworkRequest *r, gchar *m,  WebKitWebPolicyDecision *p, Client *c);
74 static gboolean decidewindow(WebKitWebView *v, WebKitWebFrame *f, WebKitNetworkRequest *r, WebKitWebNavigationAction *n, WebKitWebPolicyDecision *p, Client *c);
75 static void destroyclient(Client *c);
76 static void destroywin(GtkWidget* w, Client *c);
77 static void die(char *str);
78 static void download(Client *c, const Arg *arg);
79 static void drawindicator(Client *c);
80 static gboolean exposeindicator(GtkWidget *w, GdkEventExpose *e, Client *c);
81 static void find(Client *c, const Arg *arg);
82 static const char *getatom(Client *c, Atom a);
83 static const char *getcookies(SoupURI *uri);
84 static char *geturi(Client *c);
85 void gotheaders(SoupMessage *msg, gpointer user_data);
86 static gboolean initdownload(WebKitWebView *v, WebKitDownload *o, Client *c);
87 static void itemclick(GtkMenuItem *mi, Client *c);
88 static gboolean keypress(GtkWidget *w, GdkEventKey *ev, Client *c);
89 static void linkhover(WebKitWebView *v, const char* t, const char* l, Client *c);
90 static void loadstatuschange(WebKitWebView *view, GParamSpec *pspec, Client *c);
91 static void loaduri(Client *c, const Arg *arg);
92 static void navigate(Client *c, const Arg *arg);
93 static Client *newclient(void);
94 static void newwindow(Client *c, const Arg *arg);
95 static void newrequest(SoupSession *s, SoupMessage *msg, gpointer v);
96 static void pasteuri(GtkClipboard *clipboard, const char *text, gpointer d);
97 static void print(Client *c, const Arg *arg);
98 static GdkFilterReturn processx(GdkXEvent *xevent, GdkEvent *event, gpointer d);
99 static void progresschange(WebKitWebView *view, GParamSpec *pspec, Client *c);
100 static void reload(Client *c, const Arg *arg);
101 static void resize(GtkWidget *w, GtkAllocation *a, Client *c);
102 static void scroll(Client *c, const Arg *arg);
103 static void setatom(Client *c, Atom a, const char *v);
104 static void setcookie(SoupCookie *c);
105 static void setup(void);
106 static void sigchld(int unused);
107 static void source(Client *c, const Arg *arg);
108 static void spawn(Client *c, const Arg *arg);
109 static void stop(Client *c, const Arg *arg);
110 static void titlechange(WebKitWebView *v, WebKitWebFrame* frame, const char* title, Client *c);
111 static void update(Client *c);
112 static void updatedownload(WebKitDownload *o, GParamSpec *pspec, Client *c);
113 static void updatewinid(Client *c);
114 static void usage(void);
115 static void windowobjectcleared(GtkWidget *w, WebKitWebFrame *frame, JSContextRef js, JSObjectRef win, Client *c);
116 static void zoom(Client *c, const Arg *arg);
117
118 /* configuration, allows nested code to access above variables */
119 #include "config.h"
120
121 char *
122 buildpath(const char *path) {
123         char *apath, *p;
124         FILE *f;
125
126         /* creating directory */
127         if(path[0] == '/')
128                 apath = g_strdup(path);
129         else
130                 apath = g_strconcat(g_get_home_dir(), "/", path, NULL);
131         if((p = strrchr(apath, '/'))) {
132                 *p = '\0';
133                 g_mkdir_with_parents(apath, 0755);
134                 *p = '/';
135         }
136         /* creating file (gives error when apath ends with "/") */
137         if((f = g_fopen(apath, "a")))
138                 fclose(f);
139         return apath;
140 }
141
142 void
143 cleanup(void) {
144         while(clients)
145                 destroyclient(clients);
146         g_free(cookiefile);
147         g_free(dldir);
148         g_free(scriptfile);
149         g_free(stylefile);
150 }
151
152 void
153 clipboard(Client *c, const Arg *arg) {
154         gboolean paste = *(gboolean *)arg;
155
156         if(paste)
157                 gtk_clipboard_request_text(gtk_clipboard_get(GDK_SELECTION_PRIMARY), pasteuri, c);
158         else
159                 gtk_clipboard_set_text(gtk_clipboard_get(GDK_SELECTION_PRIMARY), c->linkhover ? c->linkhover : geturi(c), -1);
160 }
161
162 void
163 context(WebKitWebView *v, GtkMenu *m, Client *c) {
164         int i;
165         GtkContainer *parent;
166
167         gtk_widget_hide_all(GTK_WIDGET(m));
168         gtk_widget_show(GTK_WIDGET(m));
169         for(i = 0; i < LENGTH(items); i++) {
170                 parent = GTK_CONTAINER(gtk_widget_get_parent(c->items[i]));
171                 if(parent)
172                         gtk_container_remove(parent, c->items[i]);
173                 gtk_menu_shell_append(GTK_MENU_SHELL(m), c->items[i]);
174                 gtk_widget_show(c->items[i]);
175         }
176 }
177
178 char *
179 copystr(char **str, const char *src) {
180         char *tmp;
181         tmp = g_strdup(src);
182
183         if(str && *str) {
184                 g_free(*str);
185                 *str = tmp;
186         }
187         return tmp;
188 }
189
190 WebKitWebView *
191 createwindow(WebKitWebView  *v, WebKitWebFrame *f, Client *c) {
192         Client *n = newclient();
193         return n->view;
194 }
195
196 gboolean
197 decidedownload(WebKitWebView *v, WebKitWebFrame *f, WebKitNetworkRequest *r, gchar *m,  WebKitWebPolicyDecision *p, Client *c) {
198         if(!webkit_web_view_can_show_mime_type(v, m)) {
199                 webkit_web_policy_decision_download(p);
200                 return TRUE;
201         }
202         return FALSE;
203 }
204
205 gboolean
206 decidewindow(WebKitWebView *view, WebKitWebFrame *f, WebKitNetworkRequest *r, WebKitWebNavigationAction *n, WebKitWebPolicyDecision *p, Client *c) {
207         Arg arg;
208
209         if(webkit_web_navigation_action_get_reason(n) == WEBKIT_WEB_NAVIGATION_REASON_LINK_CLICKED) {
210                 webkit_web_policy_decision_ignore(p);
211                 arg.v = (void *)webkit_network_request_get_uri(r);
212                 newwindow(NULL, &arg);
213                 return TRUE;
214         }
215         return FALSE;
216 }
217
218 void
219 destroyclient(Client *c) {
220         int i;
221         Client *p;
222
223         gtk_widget_destroy(c->indicator);
224         gtk_widget_destroy(GTK_WIDGET(c->view));
225         gtk_widget_destroy(c->scroll);
226         gtk_widget_destroy(c->vbox);
227         gtk_widget_destroy(c->win);
228         for(i = 0; i < LENGTH(items); i++)
229                 gtk_widget_destroy(c->items[i]);
230         free(c->items);
231
232         for(p = clients; p && p->next != c; p = p->next);
233         if(p)
234                 p->next = c->next;
235         else
236                 clients = c->next;
237         free(c);
238         if(clients == NULL)
239                 gtk_main_quit();
240 }
241
242 void
243 destroywin(GtkWidget* w, Client *c) {
244         destroyclient(c);
245 }
246
247 void
248 die(char *str) {
249         fputs(str, stderr);
250         exit(EXIT_FAILURE);
251 }
252
253 void
254 download(Client *c, const Arg *arg) {
255         char *uri;
256         WebKitNetworkRequest *r;
257         WebKitDownload *dl;
258
259         if(arg->v)
260                 uri = (char *)arg->v;
261         else
262                 uri = c->linkhover ? c->linkhover : geturi(c);
263         r = webkit_network_request_new(uri);
264         dl = webkit_download_new(r);
265         initdownload(c->view, dl, c);
266         webkit_download_start(c->download);
267 }
268
269 void
270 drawindicator(Client *c) {
271         gint width;
272         const char *uri;
273         GtkWidget *w;
274         GdkGC *gc;
275         GdkColor fg;
276
277         uri = getatom(c, uriprop);
278         w = c->indicator;
279         width = c->progress * w->allocation.width / 100;
280         gc = gdk_gc_new(w->window);
281         gdk_color_parse(strstr(uri, "https://") == uri ?
282                         progress_trust : progress, &fg);
283         gdk_gc_set_rgb_fg_color(gc, &fg);
284         gdk_draw_rectangle(w->window,
285                         w->style->bg_gc[GTK_WIDGET_STATE(w)],
286                         TRUE, 0, 0, w->allocation.width, w->allocation.height);
287         gdk_draw_rectangle(w->window, gc, TRUE, 0, 0, width,
288                         w->allocation.height);
289         g_object_unref(gc);
290 }
291
292 gboolean
293 exposeindicator(GtkWidget *w, GdkEventExpose *e, Client *c) {
294         drawindicator(c);
295         return TRUE;
296 }
297
298 void
299 find(Client *c, const Arg *arg) {
300         const char *s;
301
302         s = getatom(c, findprop);
303         gboolean forward = *(gboolean *)arg;
304         webkit_web_view_search_text(c->view, s, FALSE, forward, TRUE);
305 }
306
307 const char *
308 getcookies(SoupURI *uri) {
309         const char *c;
310         SoupCookieJar *j = soup_cookie_jar_text_new(cookiefile, TRUE);
311         c = soup_cookie_jar_get_cookies(j, uri, TRUE);
312         g_object_unref(j);
313         return c;
314 }
315
316 const char *
317 getatom(Client *c, Atom a) {
318         static char buf[BUFSIZ];
319         Atom adummy;
320         int idummy;
321         unsigned long ldummy;
322         unsigned char *p = NULL;
323
324         XGetWindowProperty(dpy, GDK_WINDOW_XID(GTK_WIDGET(c->win)->window),
325                         a, 0L, BUFSIZ, False, XA_STRING,
326                         &adummy, &idummy, &ldummy, &ldummy, &p);
327         if(p)
328                 strncpy(buf, (char *)p, LENGTH(buf)-1);
329         else
330                 buf[0] = '\0';
331         XFree(p);
332         return buf;
333 }
334
335 char *
336 geturi(Client *c) {
337         char *uri;
338
339         if(!(uri = (char *)webkit_web_view_get_uri(c->view)))
340                 uri = "about:blank";
341         return uri;
342 }
343
344 void
345 gotheaders(SoupMessage *msg, gpointer v) {
346         SoupURI *uri;
347         GSList *l, *p;
348
349         uri = soup_message_get_uri(msg);
350         for(p = l = soup_cookies_from_response(msg); p;
351                 p = g_slist_next(p))  {
352                 setcookie((SoupCookie *)p->data);
353         }
354         soup_cookies_free(l);
355 }
356
357 gboolean
358 initdownload(WebKitWebView *view, WebKitDownload *o, Client *c) {
359         const char *filename;
360         char *uri;
361         WebKitWebBackForwardList *h;
362         WebKitWebHistoryItem *i;
363
364         c->download = o;
365         filename = webkit_download_get_suggested_filename(o);
366         if(!strcmp("", filename))
367                 filename = "index.html";
368         uri = g_strconcat("file://", dldir, "/", filename, NULL);
369         webkit_download_set_destination_uri(c->download, uri);
370         c->progress = 0;
371         h = webkit_web_view_get_back_forward_list(c->view);
372         i = webkit_web_history_item_new_with_data(webkit_download_get_uri(c->download), filename);
373         webkit_web_back_forward_list_add_item(h, i);
374         g_signal_connect(c->download, "notify::progress", G_CALLBACK(updatedownload), c);
375         g_signal_connect(c->download, "notify::status", G_CALLBACK(updatedownload), c);
376         
377         c->title = copystr(&c->title, filename);
378         update(c);
379         return TRUE;
380 }
381
382 void
383 itemclick(GtkMenuItem *mi, Client *c) {
384         int i;
385         const char *label;
386
387         label = gtk_menu_item_get_label(mi);
388         for(i = 0; i < LENGTH(items); i++)
389                 if(!strcmp(items[i].label, label))
390                         items[i].func(c, &(items[i].arg));
391 }
392
393 gboolean
394 keypress(GtkWidget* w, GdkEventKey *ev, Client *c) {
395         guint i;
396         gboolean processed = FALSE;
397
398         updatewinid(c);
399         for(i = 0; i < LENGTH(keys); i++) {
400                 if(gdk_keyval_to_lower(ev->keyval) == keys[i].keyval
401                                 && CLEANMASK(ev->state) == keys[i].mod
402                                 && keys[i].func) {
403                         keys[i].func(c, &(keys[i].arg));
404                         processed = TRUE;
405                 }
406         }
407         return processed;
408 }
409
410 void
411 linkhover(WebKitWebView *v, const char* t, const char* l, Client *c) {
412         if(l)
413                 c->linkhover = copystr(&c->linkhover, l);
414         else if(c->linkhover) {
415                 free(c->linkhover);
416                 c->linkhover = NULL;
417         }
418         update(c);
419 }
420
421 void
422 loadstatuschange(WebKitWebView *view, GParamSpec *pspec, Client *c) {
423         switch(webkit_web_view_get_load_status (c->view)) {
424         case WEBKIT_LOAD_COMMITTED:
425         if(c->download)
426                 stop(c, NULL);
427         setatom(c, uriprop, geturi(c));
428                 break;
429         case WEBKIT_LOAD_FINISHED:
430                 c->progress = 0;
431                 update(c);
432                 break;
433         default:
434                 break;
435         }
436 }
437
438 void
439 loaduri(Client *c, const Arg *arg) {
440         char *u;
441         const char *uri = (char *)arg->v;
442         Arg a = { .b = FALSE };
443
444         if(strcmp(uri, "") == 0)
445                 return;
446         u = g_strrstr(uri, "://") ? g_strdup(uri)
447                 : g_strdup_printf("http://%s", uri);
448         /* prevents endless loop */
449         if(c->uri && strcmp(u, c->uri) == 0) {
450                 reload(c, &a);
451         }
452         else {
453                 webkit_web_view_load_uri(c->view, u);
454                 c->progress = 0;
455                 c->title = copystr(&c->title, u);
456                 g_free(u);
457                 update(c);
458         }
459 }
460
461 void
462 navigate(Client *c, const Arg *arg) {
463         int steps = *(int *)arg;
464         webkit_web_view_go_back_or_forward(c->view, steps);
465 }
466
467 Client *
468 newclient(void) {
469         int i;
470         Client *c;
471         WebKitWebSettings *settings;
472         GdkGeometry hints = { 1, 1 };
473         char *uri, *ua;
474
475         if(!(c = calloc(1, sizeof(Client))))
476                 die("Cannot malloc!\n");
477         /* Window */
478         if(embed) {
479                 c->win = gtk_plug_new(embed);
480         }
481         else {
482                 c->win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
483                 /* TA:  20091214:  Despite what the GNOME docs say, the ICCCM
484                  * is always correct, so we should still call this function.
485                  * But when doing so, we *must* differentiate between a
486                  * WM_CLASS and a resource on the window.  By convention, the
487                  * window class (WM_CLASS) is capped, while the resource is in
488                  * lowercase.   Both these values come as a pair.
489                  */
490                 gtk_window_set_wmclass(GTK_WINDOW(c->win), "surf", "surf");
491
492                 /* TA:  20091214:  And set the role here as well -- so that
493                  * sessions can pick this up.
494                  */
495                 gtk_window_set_role(GTK_WINDOW(c->win), "Surf");
496         }
497         gtk_window_set_default_size(GTK_WINDOW(c->win), 800, 600);
498         g_signal_connect(G_OBJECT(c->win), "destroy", G_CALLBACK(destroywin), c);
499         g_signal_connect(G_OBJECT(c->win), "key-press-event", G_CALLBACK(keypress), c);
500         g_signal_connect(G_OBJECT(c->win), "size-allocate", G_CALLBACK(resize), c);
501
502         if(!(c->items = calloc(1, sizeof(GtkWidget *) * LENGTH(items))))
503                 die("Cannot malloc!\n");
504
505         /* contextmenu */
506         for(i = 0; i < LENGTH(items); i++) {
507                 c->items[i] = gtk_menu_item_new_with_label(items[i].label);
508                 g_signal_connect(G_OBJECT(c->items[i]), "activate",
509                                 G_CALLBACK(itemclick), c);
510         }
511
512         /* VBox */
513         c->vbox = gtk_vbox_new(FALSE, 0);
514
515         /* Scrolled Window */
516         c->scroll = gtk_scrolled_window_new(NULL, NULL);
517         gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(c->scroll),
518                         GTK_POLICY_NEVER, GTK_POLICY_NEVER);
519
520         /* Webview */
521         c->view = WEBKIT_WEB_VIEW(webkit_web_view_new());
522         g_signal_connect(G_OBJECT(c->view), "title-changed", G_CALLBACK(titlechange), c);
523         g_signal_connect(G_OBJECT(c->view), "hovering-over-link", G_CALLBACK(linkhover), c);
524         g_signal_connect(G_OBJECT(c->view), "create-web-view", G_CALLBACK(createwindow), c);
525         g_signal_connect(G_OBJECT(c->view), "new-window-policy-decision-requested", G_CALLBACK(decidewindow), c);
526         g_signal_connect(G_OBJECT(c->view), "mime-type-policy-decision-requested", G_CALLBACK(decidedownload), c);
527         g_signal_connect(G_OBJECT(c->view), "download-requested", G_CALLBACK(initdownload), c);
528         g_signal_connect(G_OBJECT(c->view), "window-object-cleared", G_CALLBACK(windowobjectcleared), c);
529         g_signal_connect(G_OBJECT(c->view), "populate-popup", G_CALLBACK(context), c);
530         g_signal_connect(G_OBJECT(c->view), "notify::load-status", G_CALLBACK(loadstatuschange), c);
531         g_signal_connect(G_OBJECT(c->view), "notify::progress", G_CALLBACK(progresschange), c);
532
533         /* Indicator */
534         c->indicator = gtk_drawing_area_new();
535         gtk_widget_set_size_request(c->indicator, 0, 2);
536         g_signal_connect (G_OBJECT (c->indicator), "expose_event",
537                         G_CALLBACK (exposeindicator), c);
538
539         /* Arranging */
540         gtk_container_add(GTK_CONTAINER(c->scroll), GTK_WIDGET(c->view));
541         gtk_container_add(GTK_CONTAINER(c->win), c->vbox);
542         gtk_container_add(GTK_CONTAINER(c->vbox), c->scroll);
543         gtk_container_add(GTK_CONTAINER(c->vbox), c->indicator);
544
545         /* Setup */
546         gtk_box_set_child_packing(GTK_BOX(c->vbox), c->indicator, FALSE, FALSE, 0, GTK_PACK_START);
547         gtk_box_set_child_packing(GTK_BOX(c->vbox), c->scroll, TRUE, TRUE, 0, GTK_PACK_START);
548         gtk_widget_grab_focus(GTK_WIDGET(c->view));
549         gtk_widget_show(c->vbox);
550         gtk_widget_show(c->indicator);
551         gtk_widget_show(c->scroll);
552         gtk_widget_show(GTK_WIDGET(c->view));
553         gtk_widget_show(c->win);
554         gtk_window_set_geometry_hints(GTK_WINDOW(c->win), NULL, &hints, GDK_HINT_MIN_SIZE);
555         gdk_window_set_events(GTK_WIDGET(c->win)->window, GDK_ALL_EVENTS_MASK);
556         gdk_window_add_filter(GTK_WIDGET(c->win)->window, processx, c);
557         webkit_web_view_set_full_content_zoom(c->view, TRUE);
558         settings = webkit_web_view_get_settings(c->view);
559         if(!(ua = getenv("SURF_USERAGENT")))
560                 ua = useragent;
561         g_object_set(G_OBJECT(settings), "user-agent", ua, NULL);
562         uri = g_strconcat("file://", stylefile, NULL);
563         g_object_set(G_OBJECT(settings), "user-stylesheet-uri", uri, NULL);
564         g_object_set(G_OBJECT(settings), "auto-load-images", loadimage, NULL);
565         g_object_set(G_OBJECT(settings), "enable-plugins", plugin, NULL);
566         g_object_set(G_OBJECT(settings), "enable-scripts", script, NULL);
567         g_free(uri);
568         setatom(c, findprop, "");
569         setatom(c, uriprop, "");
570
571         c->download = NULL;
572         c->title = NULL;
573         c->next = clients;
574         clients = c;
575         if(showxid) {
576                 gdk_display_sync(gtk_widget_get_display(c->win));
577                 printf("%u\n", (guint)GDK_WINDOW_XID(GTK_WIDGET(c->win)->window));
578                 fflush(NULL);
579         }
580         return c;
581 }
582
583 void
584 newrequest(SoupSession *s, SoupMessage *msg, gpointer v) {
585         SoupMessageHeaders *h = msg->request_headers;
586         SoupURI *uri;
587         const char *c;
588
589         soup_message_headers_remove(h, "Cookie");
590         uri = soup_message_get_uri(msg);
591         if((c = getcookies(uri)))
592                 soup_message_headers_append(h, "Cookie", c);
593         g_signal_connect_after(G_OBJECT(msg), "got-headers", G_CALLBACK(gotheaders), NULL);
594 }
595
596 void
597 newwindow(Client *c, const Arg *arg) {
598         guint i = 0;
599         const char *cmd[10], *uri;
600         const Arg a = { .v = (void *)cmd };
601         char tmp[64];
602
603         cmd[i++] = progname;
604         if(embed) {
605                 cmd[i++] = "-e";
606                 snprintf(tmp, LENGTH(tmp), "%u\n", (int)embed);
607                 cmd[i++] = tmp;
608         }
609         if(!script)
610                 cmd[i++] = "-s";
611         if(!plugin)
612                 cmd[i++] = "-p";
613         if(!loadimage)
614                 cmd[i++] = "-i";
615         if(showxid)
616                 cmd[i++] = "-x";
617         cmd[i++] = "--";
618         uri = arg->v ? (char *)arg->v : c->linkhover;
619         if(uri)
620                 cmd[i++] = uri;
621         cmd[i++] = NULL;
622         spawn(NULL, &a);
623 }
624
625 void
626 pasteuri(GtkClipboard *clipboard, const char *text, gpointer d) {
627         Arg arg = {.v = text };
628         if(text != NULL)
629                 loaduri((Client *) d, &arg);
630 }
631
632 void
633 print(Client *c, const Arg *arg) {
634         webkit_web_frame_print(webkit_web_view_get_main_frame(c->view));
635 }
636
637 GdkFilterReturn
638 processx(GdkXEvent *e, GdkEvent *event, gpointer d) {
639         Client *c = (Client *)d;
640         XPropertyEvent *ev;
641         Arg arg;
642
643         if(((XEvent *)e)->type == PropertyNotify) {
644                 ev = &((XEvent *)e)->xproperty;
645                 if(ignorexprop)
646                         ignorexprop--;
647                 else if(ev->state == PropertyNewValue) {
648                         if(ev->atom == uriprop) {
649                                 arg.v = getatom(c, uriprop);
650                                 loaduri(c, &arg);
651                         }
652                         else if(ev->atom == findprop) {
653                                 arg.b = TRUE;
654                                 find(c, &arg);
655                         }
656                         return GDK_FILTER_REMOVE;
657                 }
658         }
659         return GDK_FILTER_CONTINUE;
660 }
661
662 void
663 progresschange(WebKitWebView *view, GParamSpec *pspec, Client *c) {
664         c->progress = webkit_web_view_get_progress(c->view) * 100;
665         update(c);
666 }
667
668 void
669 reload(Client *c, const Arg *arg) {
670         gboolean nocache = *(gboolean *)arg;
671         if(nocache)
672                  webkit_web_view_reload_bypass_cache(c->view);
673         else
674                  webkit_web_view_reload(c->view);
675 }
676
677 void
678 resize(GtkWidget *w, GtkAllocation *a, Client *c) {
679         double zoom;
680
681         if(c->zoomed)
682                 return;
683         zoom = webkit_web_view_get_zoom_level(c->view);
684         if(a->width * a->height < 300 * 400 && zoom != 0.2)
685                 webkit_web_view_set_zoom_level(c->view, 0.2);
686         else if(zoom != 1.0)
687                 webkit_web_view_set_zoom_level(c->view, 1.0);
688 }
689
690 void
691 scroll(Client *c, const Arg *arg) {
692         gdouble v;
693         GtkAdjustment *a;
694
695         a = gtk_scrolled_window_get_vadjustment(GTK_SCROLLED_WINDOW(c->scroll));
696         v = gtk_adjustment_get_value(a);
697         v += gtk_adjustment_get_step_increment(a) * arg->i;
698         v = MAX(v, 0.0);
699         v = MIN(v, gtk_adjustment_get_upper(a) - gtk_adjustment_get_page_size(a));
700         gtk_adjustment_set_value(a, v);
701 }
702
703 void
704 setcookie(SoupCookie *c) {
705         SoupDate *e;
706         SoupCookieJar *j = soup_cookie_jar_text_new(cookiefile, FALSE);
707         c = soup_cookie_copy(c);
708         if(c->expires == NULL) {
709                 e = soup_date_new_from_time_t(time(NULL) + sessiontime);
710                 soup_cookie_set_expires(c, e);
711         }
712         soup_cookie_jar_add_cookie(j, c);
713         g_object_unref(j);
714 }
715
716 void
717 setatom(Client *c, Atom a, const char *v) {
718         XSync(dpy, False);
719         ignorexprop++;
720         XChangeProperty(dpy, GDK_WINDOW_XID(GTK_WIDGET(c->win)->window), a,
721                         XA_STRING, 8, PropModeReplace, (unsigned char *)v,
722                         strlen(v) + 1);
723 }
724
725 void
726 setup(void) {
727         char *proxy;
728         char *new_proxy;
729         SoupURI *puri;
730         SoupSession *s;
731
732         /* clean up any zombies immediately */
733         sigchld(0);
734         gtk_init(NULL, NULL);
735         if (!g_thread_supported())
736                 g_thread_init(NULL);
737
738         dpy = GDK_DISPLAY();
739         s = webkit_get_default_session();
740         uriprop = XInternAtom(dpy, "_SURF_URI", False);
741         findprop = XInternAtom(dpy, "_SURF_FIND", False);
742
743         /* dirs and files */
744         cookiefile = buildpath(cookiefile);
745         dldir = buildpath(dldir);
746         scriptfile = buildpath(scriptfile);
747         stylefile = buildpath(stylefile);
748
749         /* request handler */
750         s = webkit_get_default_session();
751         soup_session_remove_feature_by_type(s, soup_cookie_get_type());
752         soup_session_remove_feature_by_type(s, soup_cookie_jar_get_type());
753         g_signal_connect_after(G_OBJECT(s), "request-started", G_CALLBACK(newrequest), NULL);
754
755         /* proxy */
756         if((proxy = getenv("http_proxy")) && strcmp(proxy, "")) {
757                 new_proxy = g_strrstr(proxy, "http://") ? g_strdup(proxy) :
758                         g_strdup_printf("http://%s", proxy);
759
760                 puri = soup_uri_new(new_proxy);
761                 g_object_set(G_OBJECT(s), "proxy-uri", puri, NULL);
762                 soup_uri_free(puri);
763                 g_free(new_proxy);
764         }
765 }
766
767 void
768 sigchld(int unused) {
769         if(signal(SIGCHLD, sigchld) == SIG_ERR)
770                 die("Can't install SIGCHLD handler");
771         while(0 < waitpid(-1, NULL, WNOHANG));
772 }
773
774 void
775 source(Client *c, const Arg *arg) {
776         Arg a = { .b = FALSE };
777         gboolean s;
778
779         s = webkit_web_view_get_view_source_mode(c->view);
780         webkit_web_view_set_view_source_mode(c->view, !s);
781         reload(c, &a);
782 }
783
784 void
785 spawn(Client *c, const Arg *arg) {
786         if(fork() == 0) {
787                 if(dpy)
788                         close(ConnectionNumber(dpy));
789                 setsid();
790                 execvp(((char **)arg->v)[0], (char **)arg->v);
791                 fprintf(stderr, "surf: execvp %s", ((char **)arg->v)[0]);
792                 perror(" failed");
793                 exit(0);
794         }
795 }
796
797 void
798 stop(Client *c, const Arg *arg) {
799         if(c->download)
800                 webkit_download_cancel(c->download);
801         else
802                 webkit_web_view_stop_loading(c->view);
803         c->download = NULL;
804 }
805
806 void
807 titlechange(WebKitWebView *v, WebKitWebFrame *f, const char *t, Client *c) {
808         c->title = copystr(&c->title, t);
809         update(c);
810 }
811
812 void
813 update(Client *c) {
814         char *t;
815
816         if(c->progress != 100)
817                 t = g_strdup_printf("[%i%%] %s", c->progress, c->title);
818         else if(c->linkhover)
819                 t = g_strdup(c->linkhover);
820         else
821                 t = g_strdup(c->title);
822         drawindicator(c);
823         gtk_window_set_title(GTK_WINDOW(c->win), t);
824         g_free(t);
825 }
826
827 void
828 updatedownload(WebKitDownload *o, GParamSpec *pspec, Client *c) {
829         WebKitDownloadStatus status;
830
831         status = webkit_download_get_status(c->download);
832         if(status == WEBKIT_DOWNLOAD_STATUS_STARTED || status == WEBKIT_DOWNLOAD_STATUS_CREATED) {
833                 c->progress = (gint)(webkit_download_get_progress(c->download)*100);
834         }
835         else
836                 stop(c, NULL);
837         update(c);
838 }
839
840 void
841 updatewinid(Client *c) {
842         snprintf(winid, LENGTH(winid), "%u",
843                         (int)GDK_WINDOW_XID(GTK_WIDGET(c->win)->window));
844 }
845
846 void
847 usage(void) {
848         fputs("surf - simple browser\n", stderr);
849         die("usage: surf [-e xid] [-i] [-p] [-s] [-v] [-x] [uri]\n");
850 }
851
852 void
853 windowobjectcleared(GtkWidget *w, WebKitWebFrame *frame, JSContextRef js, JSObjectRef win, Client *c) {
854         JSStringRef jsscript;
855         char *script;
856         JSValueRef exception = NULL;
857         GError *error;
858         
859         if(g_file_get_contents(scriptfile, &script, NULL, &error)) {
860                 jsscript = JSStringCreateWithUTF8CString(script);
861                 JSEvaluateScript(js, jsscript, JSContextGetGlobalObject(js), NULL, 0, &exception);
862         }
863 }
864
865 void
866 zoom(Client *c, const Arg *arg) {
867         c->zoomed = TRUE;
868         if(arg->i < 0)          /* zoom out */
869                 webkit_web_view_zoom_out(c->view);
870         else if(arg->i > 0)     /* zoom in */
871                 webkit_web_view_zoom_in(c->view);
872         else {                  /* reset */
873                 c->zoomed = FALSE;
874                 webkit_web_view_set_zoom_level(c->view, 1.0);
875         }
876 }
877
878 int
879 main(int argc, char *argv[]) {
880         int i;
881         Arg arg;
882
883         progname = argv[0];
884         /* command line args */
885         for(i = 1, arg.v = NULL; i < argc && argv[i][0] == '-' &&
886                         argv[i][1] != '\0' && argv[i][2] == '\0'; i++) {
887                 if(!strcmp(argv[i], "--")) {
888                         i++;
889                         break;
890                 }
891                 switch(argv[i][1]) {
892                 case 'e':
893                         if(++i < argc)
894                                 embed = atoi(argv[i]);
895                         else
896                                 usage();
897                         break;
898                 case 'i':
899                         loadimage = 0;
900                         break;
901                 case 'p':
902                         plugin = 0;
903                         break;
904                 case 's':
905                         script = 0;
906                         break;
907                 case 'x':
908                         showxid = TRUE;
909                         break;
910                 case 'v':
911                         die("surf-"VERSION", © 2009 surf engineers, see LICENSE for details\n");
912                 default:
913                         usage();
914                 }
915         }
916         if(i < argc)
917                 arg.v = argv[i];
918         setup();
919         newclient();
920         if(arg.v)
921                 loaduri(clients, &arg);
922         gtk_main();
923         cleanup();
924         return EXIT_SUCCESS;
925 }