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