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