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