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