browser.c

   1 #include <fcntl.h>
   2 #include <limits.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <sys/stat.h>
   7 #include <sys/types.h>
   8
   9 #include <JavaScriptCore/JavaScript.h>
  10 #include <gdk/gdkkeysyms.h>
  11 #include <gio/gio.h>
  12 #include <gtk/gtk.h>
  13 #include <gtk/gtkx.h>
  14 #include <webkit2/webkit2.h>
  15
  16 void client_destroy(GtkWidget *, gpointer);
  17 WebKitWebView *client_new(const gchar *, WebKitWebView *, gboolean,
  18                                                                                                         gboolean);
  19 WebKitWebView *client_new_request(WebKitWebView *, WebKitNavigationAction *,
  20                                                                                                                                         gpointer);
  21 void cooperation_setup(void);
  22 void changed_download_progress(GObject *, GParamSpec *, gpointer);
  23 void changed_load_progress(GObject *, GParamSpec *, gpointer);
  24 void changed_favicon(GObject *, GParamSpec *, gpointer);
  25 void changed_title(GObject *, GParamSpec *, gpointer);
  26 void changed_uri(GObject *, GParamSpec *, gpointer);
  27 gboolean crashed_web_view(WebKitWebView *, gpointer);
  28 gboolean decide_policy(WebKitWebView *, WebKitPolicyDecision *,
  29                                                                                          WebKitPolicyDecisionType, gpointer);
  30 gboolean download_handle(WebKitDownload *, gchar *, gpointer);
  31 void download_handle_start(WebKitWebView *, WebKitDownload *, gpointer);
  32 void downloadmanager_cancel(GtkToolButton *, gpointer);
  33 gboolean downloadmanager_delete(GtkWidget *, gpointer);
  34 void downloadmanager_setup(void);
  35 gchar *ensure_uri_scheme(const gchar *);
  36 void grab_environment_configuration(void);
  37 void grab_feeds_finished(GObject *, GAsyncResult *, gpointer);
  38 void hover_web_view(WebKitWebView *, WebKitHitTestResult *, guint, gpointer);
  39 void icon_location(GtkEntry *, GtkEntryIconPosition, GdkEvent *, gpointer);
  40 void init_default_web_context(void);
  41 gboolean key_common(GtkWidget *, GdkEvent *, gpointer);
  42 gboolean key_downloadmanager(GtkWidget *, GdkEvent *, gpointer);
  43 gboolean key_location(GtkWidget *, GdkEvent *, gpointer);
  44 gboolean key_tablabel(GtkWidget *, GdkEvent *, gpointer);
  45 gboolean key_web_view(GtkWidget *, GdkEvent *, gpointer);
  46 void mainwindow_setup(void);
  47 void mainwindow_title(gint);
  48 void notebook_switch_page(GtkNotebook *, GtkWidget *, guint, gpointer);
  49 gboolean quit_if_nothing_active(void);
  50 gboolean remote_msg(GIOChannel *, GIOCondition, gpointer);
  51 void run_user_scripts(WebKitWebView *);
  52 void search(gpointer, gint);
  53 void show_web_view(WebKitWebView *, gpointer);
  54 void trust_user_certs(WebKitWebContext *);
  55 GKeyFile *get_ini(void);
  56 GKeyFile *config;
  57
  58 struct Client
  59 {
  60         gchar *external_handler_uri;
  61         gchar *feed_html;
  62         gchar *hover_uri;
  63         GtkWidget *location;
  64         GtkWidget *tabicon;
  65         GtkWidget *tablabel;
  66         GtkWidget *vbox;
  67         GtkWidget *web_view;
  68         gboolean focus_new_tab;
  69 };
  70
  71 struct MainWindow
  72 {
  73         GtkWidget *win;
  74         GtkWidget *notebook;
  75 } mw;
  76
  77 struct DownloadManager
  78 {
  79         GtkWidget *scroll;
  80         GtkWidget *toolbar;
  81         GtkWidget *win;
  82 } dm;
  83
  84 struct Configuration
  85 {
  86         WebKitCookieAcceptPolicy cookie_policy;
  87         const gchar *accepted_language[2];
  88         gboolean cooperative_alone;
  89         gboolean cooperative_instances;
  90         gboolean enable_console_to_stdout;
  91         gboolean javascript_disabled;
  92         gchar *download_dir;
  93         gchar *fifo_suffix;
  94         gchar *history_file;
  95         gchar *home_uri;
  96         gchar *search_engine;
  97         gchar *user_agent;
  98         gdouble global_zoom;
  99         gint tab_width_chars;
 100 } cfg;
 101
 102 gint clients = 0, downloads = 0;
 103 int cooperative_pipe_fp = 0;
 104 gchar *search_text;
 105
 106 void
 107 client_destroy(GtkWidget *widget, gpointer data)
 108 {
 109         struct Client *c = (struct Client *)data;
 110         gint idx;
 111
 112         g_signal_handlers_disconnect_by_func(G_OBJECT(c->web_view),
 113                                                                                                                                                          changed_load_progress, c);
 114
 115         idx = gtk_notebook_page_num(GTK_NOTEBOOK(mw.notebook), c->vbox);
 116         if (idx == -1)
 117                 fprintf(stderr, __NAME__": Tab index was -1, bamboozled\n");
 118         else
 119                 gtk_notebook_remove_page(GTK_NOTEBOOK(mw.notebook), idx);
 120
 121         free(c);
 122         clients--;
 123
 124         quit_if_nothing_active();
 125 }
 126
 127 WebKitWebView *
 128 client_new(const gchar *uri, WebKitWebView *related_wv, gboolean show,
 129                                          gboolean focus_tab)
 130 {
 131         struct Client *c;
 132         gchar *f;
 133         GtkWidget *evbox, *tabbox;
 134
 135         if (uri != NULL && cfg.cooperative_instances && !cfg.cooperative_alone) {
 136                 f = ensure_uri_scheme(uri);
 137                 write(cooperative_pipe_fp, f, strlen(f));
 138                 write(cooperative_pipe_fp, "\n", 1);
 139                 g_free(f);
 140                 return NULL;
 141         }
 142
 143         c = calloc(1, sizeof(struct Client));
 144         if (!c) {
 145                 fprintf(stderr, __NAME__": fatal: calloc failed\n");
 146                 exit(EXIT_FAILURE);
 147         }
 148
 149         c->focus_new_tab = focus_tab;
 150
 151         if (related_wv == NULL)
 152                 c->web_view = webkit_web_view_new();
 153         else
 154                 c->web_view = webkit_web_view_new_with_related_view(related_wv);
 155
 156         webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), cfg.global_zoom);
 157         WebKitSettings *settings = webkit_web_view_get_settings(WEBKIT_WEB_VIEW(c->web_view));
 158         webkit_settings_set_enable_javascript(settings, !cfg.javascript_disabled);
 159         g_signal_connect(G_OBJECT(c->web_view), "notify::favicon",
 160                                                                          G_CALLBACK(changed_favicon), c);
 161         g_signal_connect(G_OBJECT(c->web_view), "notify::title",
 162                                                                          G_CALLBACK(changed_title), c);
 163         g_signal_connect(G_OBJECT(c->web_view), "notify::uri",
 164                                                                          G_CALLBACK(changed_uri), c);
 165         g_signal_connect(G_OBJECT(c->web_view), "notify::estimated-load-progress",
 166                                                                          G_CALLBACK(changed_load_progress), c);
 167         g_signal_connect(G_OBJECT(c->web_view), "create",
 168                                                                          G_CALLBACK(client_new_request), NULL);
 169         g_signal_connect(G_OBJECT(c->web_view), "close",
 170                                                                          G_CALLBACK(client_destroy), c);
 171         g_signal_connect(G_OBJECT(c->web_view), "decide-policy",
 172                                                                          G_CALLBACK(decide_policy), NULL);
 173         g_signal_connect(G_OBJECT(c->web_view), "key-press-event",
 174                                                                          G_CALLBACK(key_web_view), c);
 175         g_signal_connect(G_OBJECT(c->web_view), "button-release-event",
 176                                                                          G_CALLBACK(key_web_view), c);
 177         g_signal_connect(G_OBJECT(c->web_view), "scroll-event",
 178                                                                          G_CALLBACK(key_web_view), c);
 179         g_signal_connect(G_OBJECT(c->web_view), "mouse-target-changed",
 180                                                                          G_CALLBACK(hover_web_view), c);
 181         g_signal_connect(G_OBJECT(c->web_view), "web-process-crashed",
 182                                                                          G_CALLBACK(crashed_web_view), c);
 183
 184         if (cfg.user_agent != NULL)
 185                 g_object_set(G_OBJECT(webkit_web_view_get_settings(WEBKIT_WEB_VIEW(c->web_view))),
 186                                                                  "user-agent", cfg.user_agent, NULL);
 187
 188         if (cfg.enable_console_to_stdout)
 189                 webkit_settings_set_enable_write_console_messages_to_stdout(webkit_web_view_get_settings(WEBKIT_WEB_VIEW(c->web_view)), TRUE);
 190
 191         webkit_settings_set_enable_developer_extras(webkit_web_view_get_settings(WEBKIT_WEB_VIEW(c->web_view)), TRUE);
 192
 193         c->location = gtk_entry_new();
 194         g_signal_connect(G_OBJECT(c->location), "key-press-event",
 195                                                                          G_CALLBACK(key_location), c);
 196         g_signal_connect(G_OBJECT(c->location), "icon-release",
 197                                                                          G_CALLBACK(icon_location), c);
 198         /* XXX This is a workaround. Setting this to NULL (which is done in
 199          * grab_feeds_finished() if no feed has been detected) adds a little
 200          * padding left of the text. Not sure why. The point of this call
 201          * right here is to have that padding right from the start. This
 202          * avoids a graphical artifact. */
 203         gtk_entry_set_icon_from_icon_name(GTK_ENTRY(c->location),
 204                                                                                                                                                 GTK_ENTRY_ICON_PRIMARY,
 205                                                                                                                                                 NULL);
 206
 207         c->vbox = gtk_box_new(GTK_ORIENTATION_VERTICAL, 0);
 208         gtk_box_pack_start(GTK_BOX(c->vbox), c->location, FALSE, FALSE, 0);
 209         gtk_box_pack_start(GTK_BOX(c->vbox), c->web_view, TRUE, TRUE, 0);
 210         gtk_container_set_focus_child(GTK_CONTAINER(c->vbox), c->web_view);
 211
 212         c->tabicon = gtk_image_new_from_icon_name("text-html", GTK_ICON_SIZE_SMALL_TOOLBAR);
 213
 214         c->tablabel = gtk_label_new(__NAME__);
 215         gtk_label_set_ellipsize(GTK_LABEL(c->tablabel), PANGO_ELLIPSIZE_END);
 216         gtk_label_set_width_chars(GTK_LABEL(c->tablabel), cfg.tab_width_chars);
 217         gtk_widget_set_has_tooltip(c->tablabel, TRUE);
 218
 219         /* XXX I don't own a HiDPI screen, so I don't know if scale_factor
 220          * does the right thing. */
 221         tabbox = gtk_box_new(GTK_ORIENTATION_HORIZONTAL,
 222                                                                                          5 * gtk_widget_get_scale_factor(mw.win));
 223         gtk_box_pack_start(GTK_BOX(tabbox), c->tabicon, FALSE, FALSE, 0);
 224         gtk_box_pack_start(GTK_BOX(tabbox), c->tablabel, TRUE, TRUE, 0);
 225
 226         evbox = gtk_event_box_new();
 227         gtk_container_add(GTK_CONTAINER(evbox), tabbox);
 228         g_signal_connect(G_OBJECT(evbox), "button-release-event",
 229                                                                          G_CALLBACK(key_tablabel), c);
 230
 231         gtk_widget_add_events(evbox, GDK_SCROLL_MASK);
 232         g_signal_connect(G_OBJECT(evbox), "scroll-event",
 233                                                                          G_CALLBACK(key_tablabel), c);
 234
 235         /* For easy access, store a reference to our label. */
 236         g_object_set_data(G_OBJECT(evbox), "lariza-tab-label", c->tablabel);
 237
 238         /* This only shows the event box and the label inside, nothing else.
 239          * Needed because the evbox/label is "internal" to the notebook and
 240          * not part of the normal "widget tree" (IIUC). */
 241         gtk_widget_show_all(evbox);
 242
 243         gtk_notebook_insert_page(GTK_NOTEBOOK(mw.notebook), c->vbox, evbox,
 244                                                                                                          gtk_notebook_get_current_page(GTK_NOTEBOOK(mw.notebook)) + 1);
 245         gtk_notebook_set_tab_reorderable(GTK_NOTEBOOK(mw.notebook), c->vbox, TRUE);
 246
 247         if (show)
 248                 show_web_view(NULL, c);
 249         else
 250                 g_signal_connect(G_OBJECT(c->web_view), "ready-to-show",
 251                                                                                  G_CALLBACK(show_web_view), c);
 252
 253         if (uri != NULL) {
 254                 f = ensure_uri_scheme(uri);
 255                 webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
 256                 g_free(f);
 257         }
 258
 259         clients++;
 260
 261         return WEBKIT_WEB_VIEW(c->web_view);
 262 }
 263
 264 WebKitWebView *
 265 client_new_request(WebKitWebView *web_view,
 266                                                                          WebKitNavigationAction *navigation_action, gpointer data)
 267 {
 268         return client_new(NULL, web_view, FALSE, FALSE);
 269 }
 270
 271 void
 272 cooperation_setup(void)
 273 {
 274         GIOChannel *towatch;
 275         gchar *fifofilename, *fifopath;
 276
 277         fifofilename = g_strdup_printf("%s-%s", __NAME__".fifo", cfg.fifo_suffix);
 278         fifopath = g_build_filename(g_get_user_runtime_dir(), fifofilename, NULL);
 279         g_free(fifofilename);
 280
 281         if (!g_file_test(fifopath, G_FILE_TEST_EXISTS))
 282                 mkfifo(fifopath, 0600);
 283
 284         cooperative_pipe_fp = open(fifopath, O_WRONLY | O_NONBLOCK);
 285         if (!cooperative_pipe_fp) {
 286                 fprintf(stderr, __NAME__": Can't open FIFO at all.\n");
 287         } else {
 288                 if (write(cooperative_pipe_fp, "", 0) == -1) {
 289                         /* Could not do an empty write to the FIFO which means there's
 290                          * no one listening. */
 291                         close(cooperative_pipe_fp);
 292                         towatch = g_io_channel_new_file(fifopath, "r+", NULL);
 293                         g_io_add_watch(towatch, G_IO_IN, (GIOFunc)remote_msg, NULL);
 294                 } else {
 295                         cfg.cooperative_alone = FALSE;
 296                 }
 297         }
 298
 299         g_free(fifopath);
 300 }
 301
 302 void
 303 changed_download_progress(GObject *obj, GParamSpec *pspec, gpointer data)
 304 {
 305         WebKitDownload *download = WEBKIT_DOWNLOAD(obj);
 306         WebKitURIResponse *resp;
 307         GtkToolItem *tb = GTK_TOOL_ITEM(data);
 308         gdouble p, size_mb;
 309         const gchar *uri;
 310         gchar *t, *filename, *base;
 311
 312         p = webkit_download_get_estimated_progress(download);
 313         p = p > 1 ? 1 : p;
 314         p = p < 0 ? 0 : p;
 315         p *= 100;
 316         resp = webkit_download_get_response(download);
 317         size_mb = webkit_uri_response_get_content_length(resp) / 1e6;
 318
 319         uri = webkit_download_get_destination(download);
 320         filename = g_filename_from_uri(uri, NULL, NULL);
 321         if (filename == NULL) {
 322                 /* This really should not happen because WebKit uses that URI to
 323                  * write to a file... */
 324                 fprintf(stderr, __NAME__": Could not construct file name from URI!\n");
 325                 t = g_strdup_printf("%s (%.0f%% of %.1f MB)",
 326                                                                                                 webkit_uri_response_get_uri(resp), p, size_mb);
 327         } else {
 328                 base = g_path_get_basename(filename);
 329                 t = g_strdup_printf("%s (%.0f%% of %.1f MB)", base, p, size_mb);
 330                 g_free(filename);
 331                 g_free(base);
 332         }
 333         gtk_tool_button_set_label(GTK_TOOL_BUTTON(tb), t);
 334         g_free(t);
 335 }
 336
 337 void
 338 changed_load_progress(GObject *obj, GParamSpec *pspec, gpointer data)
 339 {
 340         struct Client *c = (struct Client *)data;
 341         gdouble p;
 342         gchar *grab_feeds =
 343                 "a = document.querySelectorAll('"
 344                 "    html > head > link[rel=\"alternate\"][href][type=\"application/atom+xml\"],"
 345                 "    html > head > link[rel=\"alternate\"][href][type=\"application/rss+xml\"]"
 346                 "');"
 347                 "if (a.length == 0)"
 348                 "    null;"
 349                 "else"
 350                 "{"
 351                 "    out = '';"
 352                 "    for (i = 0; i < a.length; i++)"
 353                 "    {"
 354                 "        url = encodeURIComponent(a[i].href);"
 355                 "        if ('title' in a[i] && a[i].title != '')"
 356                 "            title = encodeURIComponent(a[i].title);"
 357                 "        else"
 358                 "            title = url;"
 359                 "        out += '<li><a href=\"' + url + '\">' + title + '</a></li>';"
 360                 "    }"
 361                 "    out;"
 362                 "}";
 363
 364
 365         p = webkit_web_view_get_estimated_load_progress(WEBKIT_WEB_VIEW(c->web_view));
 366         if (p == 1) {
 367                 p = 0;
 368
 369                 /* The page has loaded fully. We now run the short JavaScript
 370                  * snippet above that operates on the DOM. It tries to grab all
 371                  * occurences of <link rel="alternate" ...>, i.e. RSS/Atom feed
 372                  * references. */
 373                 webkit_web_view_run_javascript(WEBKIT_WEB_VIEW(c->web_view),
 374                                                                                                                                          grab_feeds, NULL,
 375                                                                                                                                          grab_feeds_finished, c);
 376
 377                 run_user_scripts(WEBKIT_WEB_VIEW(c->web_view));
 378         }
 379         gtk_entry_set_progress_fraction(GTK_ENTRY(c->location), p);
 380 }
 381
 382 void
 383 changed_favicon(GObject *obj, GParamSpec *pspec, gpointer data)
 384 {
 385         struct Client *c = (struct Client *)data;
 386         cairo_surface_t *f;
 387         int w, h, w_should, h_should;
 388         GdkPixbuf *pb, *pb_scaled;
 389
 390         f = webkit_web_view_get_favicon(WEBKIT_WEB_VIEW(c->web_view));
 391         if (f == NULL) {
 392                 gtk_image_set_from_icon_name(GTK_IMAGE(c->tabicon), "text-html",
 393                                                                                                                                  GTK_ICON_SIZE_SMALL_TOOLBAR);
 394         } else {
 395                 w = cairo_image_surface_get_width(f);
 396                 h = cairo_image_surface_get_height(f);
 397                 pb = gdk_pixbuf_get_from_surface(f, 0, 0, w, h);
 398                 if (pb != NULL) {
 399                         w_should = 16 * gtk_widget_get_scale_factor(c->tabicon);
 400                         h_should = 16 * gtk_widget_get_scale_factor(c->tabicon);
 401                         pb_scaled = gdk_pixbuf_scale_simple(pb, w_should, h_should,
 402                                                                                                                                                                         GDK_INTERP_BILINEAR);
 403                         gtk_image_set_from_pixbuf(GTK_IMAGE(c->tabicon), pb_scaled);
 404
 405                         g_object_unref(pb_scaled);
 406                         g_object_unref(pb);
 407                 }
 408         }
 409 }
 410
 411 void
 412 changed_title(GObject *obj, GParamSpec *pspec, gpointer data)
 413 {
 414         const gchar *t, *u;
 415         struct Client *c = (struct Client *)data;
 416
 417         u = webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view));
 418         t = webkit_web_view_get_title(WEBKIT_WEB_VIEW(c->web_view));
 419
 420         u = u == NULL ? __NAME__ : u;
 421         u = u[0] == 0 ? __NAME__ : u;
 422
 423         t = t == NULL ? u : t;
 424         t = t[0] == 0 ? u : t;
 425
 426         gtk_label_set_text(GTK_LABEL(c->tablabel), t);
 427         gtk_widget_set_tooltip_text(c->tablabel, t);
 428         mainwindow_title(gtk_notebook_get_current_page(GTK_NOTEBOOK(mw.notebook)));
 429 }
 430
 431 void
 432 changed_uri(GObject *obj, GParamSpec *pspec, gpointer data)
 433 {
 434         const gchar *t;
 435         struct Client *c = (struct Client *)data;
 436         FILE *fp;
 437
 438         t = webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view));
 439
 440         /* When a web process crashes, we get a "notify::uri" signal, but we
 441          * can no longer read a meaningful URI. It's just an empty string
 442          * now. Not updating the location bar in this scenario is important,
 443          * because we would override the "WEB PROCESS CRASHED" message. */
 444         if (t != NULL && strlen(t) > 0) {
 445                 gtk_entry_set_text(GTK_ENTRY(c->location), t);
 446
 447                 if (cfg.history_file != NULL) {
 448                         fp = fopen(cfg.history_file, "a");
 449                         if (fp != NULL) {
 450                                 fprintf(fp, "%s\n", t);
 451                                 fclose(fp);
 452                         } else {
 453                                 perror(__NAME__": Error opening history file");
 454                         }
 455                 }
 456         }
 457 }
 458
 459 gboolean
 460 crashed_web_view(WebKitWebView *web_view, gpointer data)
 461 {
 462         gchar *t;
 463         struct Client *c = (struct Client *)data;
 464
 465         t = g_strdup_printf("WEB PROCESS CRASHED: %s",
 466                                                                                         webkit_web_view_get_uri(WEBKIT_WEB_VIEW(web_view)));
 467         gtk_entry_set_text(GTK_ENTRY(c->location), t);
 468         g_free(t);
 469
 470         return TRUE;
 471 }
 472
 473 gboolean
 474 decide_policy(WebKitWebView *web_view, WebKitPolicyDecision *decision,
 475                                                         WebKitPolicyDecisionType type, gpointer data)
 476 {
 477         WebKitResponsePolicyDecision *r;
 478
 479         switch (type) {
 480         case WEBKIT_POLICY_DECISION_TYPE_RESPONSE:
 481                 r = WEBKIT_RESPONSE_POLICY_DECISION(decision);
 482                 if (!webkit_response_policy_decision_is_mime_type_supported(r))
 483                         webkit_policy_decision_download(decision);
 484                 else
 485                         webkit_policy_decision_use(decision);
 486                 break;
 487         default:
 488                 /* Use whatever default there is. */
 489                 return FALSE;
 490         }
 491         return TRUE;
 492 }
 493
 494 void
 495 download_handle_finished(WebKitDownload *download, gpointer data)
 496 {
 497         downloads--;
 498 }
 499
 500 void
 501 download_handle_start(WebKitWebView *web_view, WebKitDownload *download,
 502                                                                                         gpointer data)
 503 {
 504         g_signal_connect(G_OBJECT(download), "decide-destination",
 505                                                                          G_CALLBACK(download_handle), data);
 506 }
 507
 508 gboolean
 509 download_handle(WebKitDownload *download, gchar *suggested_filename, gpointer data)
 510 {
 511         gchar *sug_clean, *path, *path2 = NULL, *uri;
 512         GtkToolItem *tb;
 513         int suffix = 1;
 514         size_t i;
 515
 516         sug_clean = g_strdup(suggested_filename);
 517         for (i = 0; i < strlen(sug_clean); i++)
 518                 if (sug_clean[i] == G_DIR_SEPARATOR)
 519                         sug_clean[i] = '_';
 520
 521         path = g_build_filename(cfg.download_dir, sug_clean, NULL);
 522         path2 = g_strdup(path);
 523         while (g_file_test(path2, G_FILE_TEST_EXISTS) && suffix < 1000) {
 524                 g_free(path2);
 525
 526                 path2 = g_strdup_printf("%s.%d", path, suffix);
 527                 suffix++;
 528         }
 529
 530         if (suffix == 1000) {
 531                 fprintf(stderr, __NAME__": Suffix reached limit for download.\n");
 532                 webkit_download_cancel(download);
 533         } else {
 534                 uri = g_filename_to_uri(path2, NULL, NULL);
 535                 webkit_download_set_destination(download, uri);
 536                 g_free(uri);
 537
 538                 tb = gtk_tool_button_new(NULL, NULL);
 539                 gtk_tool_button_set_icon_name(GTK_TOOL_BUTTON(tb), "gtk-delete");
 540                 gtk_tool_button_set_label(GTK_TOOL_BUTTON(tb), sug_clean);
 541                 gtk_toolbar_insert(GTK_TOOLBAR(dm.toolbar), tb, 0);
 542                 gtk_widget_show_all(dm.win);
 543
 544                 g_signal_connect(G_OBJECT(download), "notify::estimated-progress",
 545                                                                                  G_CALLBACK(changed_download_progress), tb);
 546
 547                 downloads++;
 548                 g_signal_connect(G_OBJECT(download), "finished",
 549                                                                                  G_CALLBACK(download_handle_finished), NULL);
 550
 551                 g_object_ref(download);
 552                 g_signal_connect(G_OBJECT(tb), "clicked",
 553                                                                                  G_CALLBACK(downloadmanager_cancel), download);
 554         }
 555
 556         g_free(sug_clean);
 557         g_free(path);
 558         g_free(path2);
 559
 560         /* Propagate -- to whom it may concern. */
 561         return FALSE;
 562 }
 563
 564 void
 565 downloadmanager_cancel(GtkToolButton *tb, gpointer data)
 566 {
 567         WebKitDownload *download = WEBKIT_DOWNLOAD(data);
 568
 569         webkit_download_cancel(download);
 570         g_object_unref(download);
 571
 572         gtk_widget_destroy(GTK_WIDGET(tb));
 573 }
 574
 575 gboolean
 576 downloadmanager_delete(GtkWidget *obj, gpointer data)
 577 {
 578         if (!quit_if_nothing_active())
 579                 gtk_widget_hide(dm.win);
 580
 581         return TRUE;
 582 }
 583
 584 void
 585 downloadmanager_setup(void)
 586 {
 587         dm.win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
 588         gtk_window_set_type_hint(GTK_WINDOW(dm.win), GDK_WINDOW_TYPE_HINT_DIALOG);
 589         gtk_window_set_default_size(GTK_WINDOW(dm.win), 500, 250);
 590         gtk_window_set_title(GTK_WINDOW(dm.win), __NAME__" - Download Manager");
 591         g_signal_connect(G_OBJECT(dm.win), "delete-event",
 592                                                                          G_CALLBACK(downloadmanager_delete), NULL);
 593         g_signal_connect(G_OBJECT(dm.win), "key-press-event",
 594                                                                          G_CALLBACK(key_downloadmanager), NULL);
 595
 596         dm.toolbar = gtk_toolbar_new();
 597         gtk_orientable_set_orientation(GTK_ORIENTABLE(dm.toolbar),
 598                                                                                                                                  GTK_ORIENTATION_VERTICAL);
 599         gtk_toolbar_set_style(GTK_TOOLBAR(dm.toolbar), GTK_TOOLBAR_BOTH_HORIZ);
 600         gtk_toolbar_set_show_arrow(GTK_TOOLBAR(dm.toolbar), FALSE);
 601
 602         dm.scroll = gtk_scrolled_window_new(NULL, NULL);
 603         gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(dm.scroll),
 604                                                                                                                                  GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
 605         gtk_container_add(GTK_CONTAINER(dm.scroll), dm.toolbar);
 606
 607         gtk_container_add(GTK_CONTAINER(dm.win), dm.scroll);
 608 }
 609
 610 gchar *
 611 ensure_uri_scheme(const gchar *t)
 612 {
 613         gchar *f, *fabs;
 614
 615         f = g_ascii_strdown(t, -1);
 616         if (!g_str_has_prefix(f, "http:") &&
 617                         !g_str_has_prefix(f, "https:") &&
 618                         !g_str_has_prefix(f, "file:") &&
 619                         !g_str_has_prefix(f, "about:") &&
 620                         !g_str_has_prefix(f, "data:") &&
 621                         !g_str_has_prefix(f, "webkit:")) {
 622                 g_free(f);
 623                 fabs = realpath(t, NULL);
 624                 if (fabs != NULL) {
 625                         f = g_strdup_printf("file://%s", fabs);
 626                         free(fabs);
 627                 } else {
 628                         f = g_strdup_printf("http://%s", t);
 629                 }
 630                 return f;
 631         } else
 632                 return g_strdup(t);
 633 }
 634
 635 void
 636 get_config(void)
 637 {
 638         cfg.accepted_language[0] = NULL;
 639         cfg.accepted_language[1] = NULL;
 640         cfg.cooperative_alone = TRUE;
 641         cfg.cooperative_alone = TRUE;
 642         cfg.fifo_suffix = "main";
 643
 644         const char *e = g_getenv(__NAME_UPPERCASE__"_FIFO_SUFFIX");
 645         if (e != NULL)
 646                 cfg.fifo_suffix = g_strdup(e);
 647
 648         char *xdg_down = getenv("XDG_DOWNLOAD_DIR");
 649         cfg.download_dir = (xdg_down) ? xdg_down : "/var/tmp";
 650
 651         config = get_ini();
 652         cfg.accepted_language[0] = g_key_file_get_string(config, "browser",
 653                                                                                                                                                                                          "accepted_language", NULL);
 654         cfg.history_file = g_key_file_get_string(config, "browser", "history_file",
 655                                                                                                                                                                          NULL);
 656         cfg.home_uri = g_key_file_get_string(config, "browser", "homepage", NULL);
 657         cfg.enable_console_to_stdout = g_key_file_get_boolean(config, "browser",
 658                                                                                                                                                                                                                 "console_to_stdout", NULL);
 659         cfg.user_agent = g_key_file_get_string(config, "browser", "user_agent", NULL);
 660         cfg.javascript_disabled = g_key_file_get_boolean(config, "browser",
 661                                                                                                                                                                                          "javascript_disabled", NULL);
 662         char *input_cookie_policy = g_key_file_get_string(config, "browser",
 663                                                                                                                                                                                                                 "cookie_policy", NULL);
 664         cfg.cookie_policy = WEBKIT_COOKIE_POLICY_ACCEPT_NO_THIRD_PARTY;
 665         if (input_cookie_policy) {
 666                 if (strcmp(input_cookie_policy, "all") == 0) {
 667                         cfg.cookie_policy = WEBKIT_COOKIE_POLICY_ACCEPT_ALWAYS;
 668                 } else if (strcmp(input_cookie_policy, "none") == 0) {
 669                         cfg.cookie_policy = WEBKIT_COOKIE_POLICY_ACCEPT_NEVER;
 670                 }
 671         }
 672
 673         cfg.tab_width_chars = g_key_file_get_integer(config, "ui", "tab_width", NULL);
 674         cfg.tab_width_chars = (cfg.tab_width_chars) ? cfg.tab_width_chars : 20;
 675         cfg.global_zoom = g_key_file_get_double(config, "ui", "zoom_level", NULL);
 676         cfg.global_zoom = (cfg.global_zoom) ? cfg.global_zoom : 1.0;
 677         cfg.search_engine = g_key_file_get_string(config, "ui", "search_engine", NULL);
 678         cfg.search_engine = (cfg.search_engine) ? cfg.search_engine :
 679                 "https://duckduckgo.com?q=";
 680 }
 681
 682 void
 683 grab_feeds_finished(GObject *object, GAsyncResult *result, gpointer data)
 684 {
 685         struct Client *c = (struct Client *)data;
 686         WebKitJavascriptResult *js_result;
 687         JSCValue *value;
 688         JSCException *exception;
 689         GError *err = NULL;
 690         gchar *str_value;
 691
 692         g_free(c->feed_html);
 693         c->feed_html = NULL;
 694
 695         /* This was taken almost verbatim from the example in WebKit's
 696          * documentation:
 697          *
 698          * https://webkitgtk.org/reference/webkit2gtk/stable/WebKitWebView.html#webkit-web-view-run-javascript-finish */
 699
 700         js_result = webkit_web_view_run_javascript_finish(WEBKIT_WEB_VIEW(object),
 701                                                                                                                                                                                                                 result, &err);
 702         if (!js_result) {
 703                 fprintf(stderr, __NAME__": Error running javascript: %s\n", err->message);
 704                 g_error_free(err);
 705                 return;
 706         }
 707
 708         value = webkit_javascript_result_get_js_value(js_result);
 709         if (jsc_value_is_string(value)) {
 710                 str_value = jsc_value_to_string(value);
 711                 exception = jsc_context_get_exception(jsc_value_get_context(value));
 712                 if (exception != NULL) {
 713                         fprintf(stderr, __NAME__": Error running javascript: %s\n",
 714                                                         jsc_exception_get_message(exception));
 715                 } else {
 716                         c->feed_html = str_value;
 717                 }
 718
 719                 gtk_entry_set_icon_from_icon_name(GTK_ENTRY(c->location),
 720                                                                                                                                                         GTK_ENTRY_ICON_PRIMARY,
 721                                                                                                                                                         "application-rss+xml-symbolic");
 722                 gtk_entry_set_icon_activatable(GTK_ENTRY(c->location),
 723                                                                                                                                          GTK_ENTRY_ICON_PRIMARY,
 724                                                                                                                                          TRUE);
 725         } else {
 726                 gtk_entry_set_icon_from_icon_name(GTK_ENTRY(c->location),
 727                                                                                                                                                         GTK_ENTRY_ICON_PRIMARY,
 728                                                                                                                                                         NULL);
 729         }
 730
 731         webkit_javascript_result_unref(js_result);
 732 }
 733
 734 void
 735 hover_web_view(WebKitWebView *web_view, WebKitHitTestResult *ht, guint modifiers,
 736                                                          gpointer data)
 737 {
 738         struct Client *c = (struct Client *)data;
 739         const char *to_show;
 740
 741         g_free(c->hover_uri);
 742
 743         if (webkit_hit_test_result_context_is_link(ht)) {
 744                 to_show = webkit_hit_test_result_get_link_uri(ht);
 745                 c->hover_uri = g_strdup(to_show);
 746         } else {
 747                 to_show = webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view));
 748                 c->hover_uri = NULL;
 749         }
 750
 751         if (!gtk_widget_is_focus(c->location))
 752                 gtk_entry_set_text(GTK_ENTRY(c->location), to_show);
 753 }
 754
 755 void
 756 icon_location(GtkEntry *entry, GtkEntryIconPosition icon_pos, GdkEvent *event,
 757                                                         gpointer data)
 758 {
 759         struct Client *c = (struct Client *)data;
 760         gchar *d;
 761         gchar *data_template =
 762                 "data:text/html,"
 763                 "<!DOCTYPE html>"
 764                 "<html>"
 765                 "    <head>"
 766                 "        <meta charset=\"UTF-8\">"
 767                 "        <title>Feeds</title>"
 768                 "    </head>"
 769                 "    <body>"
 770                 "        <p>Feeds found on this page:</p>"
 771                 "        <ul>"
 772                 "        %s"
 773                 "        </ul>"
 774                 "    </body>"
 775                 "</html>";
 776
 777         if (c->feed_html != NULL)
 778                 {
 779                         /* What we're actually trying to do is show a simple HTML page
 780                          * that lists all the feeds on the current page. The function
 781                          * webkit_web_view_load_html() looks like the proper way to do
 782                          * that. Sad thing is, it doesn't create a history entry, but
 783                          * instead simply replaces the content of the current page. This
 784                          * is not what we want.
 785                          *
 786                          * RFC 2397 [0] defines the data URI scheme [1]. We abuse this
 787                          * mechanism to show my custom HTML snippet *and* create a
 788                          * history entry.
 789                          *
 790                          * [0]: https://tools.ietf.org/html/rfc2397
 791                          * [1]: https://en.wikipedia.org/wiki/Data_URI_scheme */
 792                         d = g_strdup_printf(data_template, c->feed_html);
 793                         webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), d);
 794                         g_free(d);
 795                 }
 796 }
 797
 798 void
 799 init_default_web_context(void)
 800 {
 801         gchar *p;
 802         WebKitWebContext *wc;
 803         WebKitCookieManager *cm;
 804
 805         wc = webkit_web_context_get_default();
 806
 807         p = g_build_filename(g_get_user_config_dir(), __NAME__, "adblock", NULL);
 808         webkit_web_context_set_sandbox_enabled(wc, TRUE);
 809         webkit_web_context_add_path_to_sandbox(wc, p, TRUE);
 810         g_free(p);
 811
 812         webkit_web_context_set_process_model(wc,
 813                                                                                                                                                          WEBKIT_PROCESS_MODEL_MULTIPLE_SECONDARY_PROCESSES);
 814
 815         p = g_build_filename(g_get_user_config_dir(), __NAME__, "web_extensions",
 816                                                                                          NULL);
 817         webkit_web_context_set_web_extensions_directory(wc, p);
 818         g_free(p);
 819
 820         if (cfg.accepted_language[0] != NULL)
 821                 webkit_web_context_set_preferred_languages(wc, cfg.accepted_language);
 822
 823         g_signal_connect(G_OBJECT(wc), "download-started",
 824                                                                          G_CALLBACK(download_handle_start), NULL);
 825
 826         trust_user_certs(wc);
 827
 828         cm = webkit_web_context_get_cookie_manager(wc);
 829         webkit_cookie_manager_set_accept_policy(cm, cfg.cookie_policy);
 830
 831         webkit_web_context_set_favicon_database_directory(wc, NULL);
 832         webkit_cookie_manager_set_persistent_storage(cm,
 833                                                                                                                                                                                          g_build_filename("/",
 834                                                                                                                                                                                                                                                                 g_get_user_cache_dir(),
 835                                                                                                                                                                                                                                                                 __NAME__,
 836                                                                                                                                                                                                                                                                 "cookies.db",
 837                                                                                                                                                                                                                                                                 NULL),
 838                                                                                                                                                                                          WEBKIT_COOKIE_PERSISTENT_STORAGE_SQLITE);
 839 }
 840
 841 void
 842 search(gpointer data, gint direction)
 843 {
 844         struct Client *c = (struct Client *)data;
 845         WebKitWebView *web_view = WEBKIT_WEB_VIEW(c->web_view);
 846         WebKitFindController *fc = webkit_web_view_get_find_controller(web_view);
 847
 848         if (search_text == NULL)
 849                 return;
 850
 851         switch (direction) {
 852         case 0:
 853                 webkit_find_controller_search(fc, search_text,
 854                                                                                                                                         WEBKIT_FIND_OPTIONS_CASE_INSENSITIVE |
 855                                                                                                                                         WEBKIT_FIND_OPTIONS_WRAP_AROUND,
 856                                                                                                                                         G_MAXUINT);
 857                 break;
 858         case 1:
 859                 webkit_find_controller_search_next(fc);
 860                 break;
 861         case -1:
 862                 webkit_find_controller_search_previous(fc);
 863                 break;
 864         case 2:
 865                 webkit_find_controller_search_finish(fc);
 866                 break;
 867         }
 868 }
 869
 870 void
 871 search_init(struct Client *c, int direction)
 872 {
 873         gtk_widget_grab_focus(c->location);
 874         const gchar *contents = gtk_entry_get_text(GTK_ENTRY(c->location));
 875         if (strcspn(contents, "s/")) {
 876                 gtk_entry_set_text(GTK_ENTRY(c->location), "s/");
 877         }
 878         gtk_editable_set_position(GTK_EDITABLE(c->location), -1);
 879         search(c, 0);
 880         search(c, -1);
 881         search(c, direction);
 882 }
 883
 884 int
 885 def_key(char *key, unsigned int def)
 886 {
 887         char *conf = g_key_file_get_string(config, "keybindings", key, NULL);
 888         return (conf) ? gdk_keyval_from_name((conf) ? conf : NULL) : def;
 889 }
 890
 891 gboolean
 892 key_common(GtkWidget *widget, GdkEvent *event, gpointer data)
 893 {
 894         struct Client *c = (struct Client *)data;
 895         gdouble now;
 896         gchar *f;
 897
 898         if (event->type == GDK_KEY_PRESS) {
 899                 if (((GdkEventKey *)event)->state & GDK_CONTROL_MASK) {
 900                         WebKitSettings *settings = webkit_web_view_get_settings(WEBKIT_WEB_VIEW(c->web_view));
 901                         gboolean js = webkit_settings_get_enable_javascript(settings);
 902                         int key = ((GdkEventKey *)event)->keyval;
 903                         if (def_key("close_tab", GDK_KEY_q) == key) {
 904                                 client_destroy(NULL, c);
 905                                 return TRUE;
 906                         } else if (def_key("new_tab", GDK_KEY_w) == key) {
 907                                 f = ensure_uri_scheme(cfg.home_uri);
 908                                 client_new(f, NULL, TRUE, TRUE);
 909                                 g_free(f);
 910                                 return TRUE;
 911                         } else if (def_key("reload", GDK_KEY_e) == key) {
 912                                 webkit_web_view_reload_bypass_cache(WEBKIT_WEB_VIEW(c->web_view));
 913                                 return TRUE;
 914                         } else if (def_key("download_manager", GDK_KEY_y) == key) {
 915                                 gtk_widget_show_all(dm.win);
 916                                 return TRUE;
 917                         } else if (def_key("web_search", GDK_KEY_d) == key) {
 918                                 gtk_widget_grab_focus(c->location);
 919                                 gtk_entry_set_text(GTK_ENTRY(c->location), "w/");
 920                                 gtk_editable_set_position(GTK_EDITABLE(c->location), -1);
 921                                 return TRUE;
 922                         } else if (def_key("search_forwards", GDK_KEY_s) == key) {
 923                                 search_init(c, 1);
 924                                 return TRUE;
 925                         } else if (def_key("search_backwards", GDK_KEY_r) == key) {
 926                                 search_init(c, -1);
 927                                 return TRUE;
 928                         } else if (def_key("location", GDK_KEY_t) == key) {
 929                                 gtk_widget_grab_focus(c->location);
 930                                 const char *uri = webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view));
 931                                 const char *goal = (uri) ? uri : "https://";
 932                                 gtk_entry_set_text(GTK_ENTRY(c->location), goal);
 933                                 gtk_editable_set_position(GTK_EDITABLE(c->location), -1);
 934                                 return TRUE;
 935                         } else if (def_key("previous_tab", GDK_KEY_u) == key) {
 936                                 gtk_notebook_prev_page(GTK_NOTEBOOK(mw.notebook));
 937                                 return TRUE;
 938                         } else if (def_key("next_tab", GDK_KEY_i) == key) {
 939                                 gtk_notebook_next_page(GTK_NOTEBOOK(mw.notebook));
 940                                 return TRUE;
 941                         } else if (def_key("history_back", GDK_KEY_h) == key) {
 942                                 webkit_web_view_go_back(WEBKIT_WEB_VIEW(c->web_view));
 943                                 return TRUE;
 944                         } else if (def_key("history_forwards", GDK_KEY_l) == key) {
 945                                 webkit_web_view_go_forward(WEBKIT_WEB_VIEW(c->web_view));
 946                                 return TRUE;
 947                         } else if (def_key("scroll_down", GDK_KEY_j) == key) {
 948                                 for (int i = 0; i < 2; i++) {
 949                                         event->key.keyval = GDK_KEY_Down;
 950                                         gdk_event_put(event);
 951                                 }
 952                                 return TRUE;
 953                         } else if (def_key("scroll_up", GDK_KEY_k) == key) {
 954                                 event->key.keyval = GDK_KEY_Up;
 955                                 gdk_event_put(event);
 956                                 return TRUE;
 957                         } else if (def_key("scroll_page_down", GDK_KEY_f) == key) {
 958                                 event->key.keyval = GDK_KEY_Page_Down;
 959                                 gdk_event_put(event);
 960                                 return TRUE;
 961                         } else if (def_key("scroll_page_up", GDK_KEY_b) == key) {
 962                                 event->key.keyval = GDK_KEY_Page_Up;
 963                                 gdk_event_put(event);
 964                                 return TRUE;
 965                         } else if (def_key("toggle_js", GDK_KEY_o) == key) {
 966                                 webkit_settings_set_enable_javascript(settings, (js) ? FALSE : TRUE);
 967                                 webkit_web_view_set_settings(WEBKIT_WEB_VIEW(c->web_view), settings);
 968                                 return TRUE;
 969                         } else if (def_key("quit", GDK_KEY_g) == key) {
 970                                 search(c, 2);
 971                                 gtk_widget_grab_focus(c->web_view);
 972                                 gtk_entry_set_text(GTK_ENTRY(c->location),
 973                                                                                                          webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view)));
 974                                 webkit_web_view_run_javascript(WEBKIT_WEB_VIEW(c->web_view),
 975                                                                                                                                                          "window.getSelection().removeAllRanges();"
 976                                                                                                                                                          "document.activeElement.blur();",
 977                                                                                                                                                          NULL, NULL, c);
 978                                 return TRUE;
 979                         } else if (def_key("zoom_in", GDK_KEY_equal) == key) {
 980                                 now = webkit_web_view_get_zoom_level(WEBKIT_WEB_VIEW(c->web_view));
 981                                 webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), now + 0.1);
 982                                 return TRUE;
 983                         } else if (def_key("zoom_out", GDK_KEY_minus) == key) {
 984                                 now = webkit_web_view_get_zoom_level(WEBKIT_WEB_VIEW(c->web_view));
 985                                 webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), now - 0.1);
 986                                 return TRUE;
 987                         } else if (def_key("zoom_reset", GDK_KEY_0) == key) {
 988                                 webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view),
 989                                                                                                                                                          cfg.global_zoom);
 990                                 return TRUE;
 991                         }
 992                 }
 993         }
 994         return FALSE;
 995 }
 996
 997 gboolean
 998 key_downloadmanager(GtkWidget *widget, GdkEvent *event, gpointer data)
 999 {
1000         if (event->type == GDK_KEY_PRESS) {
1001                 if (((GdkEventKey *)event)->state & GDK_CONTROL_MASK) {
1002                         int key = ((GdkEventKey *)event)->keyval;
1003                         if ((def_key("close_tab", GDK_KEY_q) == key) ||
1004                                         (def_key("download_manager", GDK_KEY_y) == key)) {
1005                                 downloadmanager_delete(dm.win, NULL);
1006                                 return TRUE;
1007                         }
1008                 }
1009         }
1010
1011         return FALSE;
1012 }
1013
1014 gboolean
1015 key_location(GtkWidget *widget, GdkEvent *event, gpointer data)
1016 {
1017         struct Client *c = (struct Client *)data;
1018         const gchar *t;
1019         gchar *f;
1020
1021         if (key_common(widget, event, data))
1022                 return TRUE;
1023
1024         if (event->type == GDK_KEY_PRESS) {
1025                 int key = ((GdkEventKey *)event)->keyval;
1026                 if ((GDK_KEY_KP_Enter == key) || (GDK_KEY_Return == key)) {
1027                         gtk_widget_grab_focus(c->web_view);
1028                         t = gtk_entry_get_text(GTK_ENTRY(c->location));
1029                         if (t != NULL && t[0] == 's' && t[1] == '/') {
1030                                 if (search_text != NULL)
1031                                         g_free(search_text);
1032                                 search_text = g_strdup(t + 2);
1033                                 search(c, 0);
1034                         } else if (t != NULL && t[0] == 'w' && t[1] == '/') {
1035                                 const char *engine = cfg.search_engine;
1036                                 int len = strlen(engine) + strlen(t) - 2;
1037                                 char *f = (char *) malloc(len);
1038                                 snprintf(f, len + 1, "%s%s", engine, t + 2);
1039                                 webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
1040                                 free(f);
1041                         } else {
1042                                 f = ensure_uri_scheme(t);
1043                                 webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
1044                                 g_free(f);
1045                         }
1046                         return TRUE;
1047                 } else if ((GDK_KEY_Escape == key) ||
1048                                                          (def_key("quit", GDK_KEY_g) == key)) {
1049                         t = webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view));
1050                         gtk_entry_set_text(GTK_ENTRY(c->location),
1051                                                                                                  (t == NULL ? __NAME__ : t));
1052                         return TRUE;
1053                 }
1054         }
1055         return FALSE;
1056 }
1057
1058 gboolean
1059 key_tablabel(GtkWidget *widget, GdkEvent *event, gpointer data)
1060 {
1061         GdkScrollDirection direction;
1062
1063         if (event->type == GDK_BUTTON_RELEASE) {
1064                 switch (((GdkEventButton *)event)->button) {
1065                 case 2:
1066                         client_destroy(NULL, data);
1067                         return TRUE;
1068                 }
1069         } else if (event->type == GDK_SCROLL) {
1070                 gdk_event_get_scroll_direction(event, &direction);
1071                 switch (direction) {
1072                 case GDK_SCROLL_UP:
1073                         gtk_notebook_prev_page(GTK_NOTEBOOK(mw.notebook));
1074                         break;
1075                 case GDK_SCROLL_DOWN:
1076                         gtk_notebook_next_page(GTK_NOTEBOOK(mw.notebook));
1077                         break;
1078                 default:
1079                         break;
1080                 }
1081                 return TRUE;
1082         }
1083         return FALSE;
1084 }
1085
1086 gboolean
1087 key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
1088 {
1089         struct Client *c = (struct Client *)data;
1090         gdouble dx, dy;
1091         gfloat z;
1092
1093         if (key_common(widget, event, data))
1094                 return TRUE;
1095
1096         if (event->type == GDK_KEY_PRESS) {
1097                 if (((GdkEventKey *)event)->keyval == GDK_KEY_Escape) {
1098                         webkit_web_view_stop_loading(WEBKIT_WEB_VIEW(c->web_view));
1099                         gtk_entry_set_progress_fraction(GTK_ENTRY(c->location), 0);
1100                 }
1101         } else if (event->type == GDK_BUTTON_RELEASE) {
1102                 switch (((GdkEventButton *)event)->button) {
1103                 case 2:
1104                         if (c->hover_uri != NULL) {
1105                                 client_new(c->hover_uri, NULL, TRUE, FALSE);
1106                                 return TRUE;
1107                         }
1108                         break;
1109                 case 8:
1110                         webkit_web_view_go_back(WEBKIT_WEB_VIEW(c->web_view));
1111                         return TRUE;
1112                 case 9:
1113                         webkit_web_view_go_forward(WEBKIT_WEB_VIEW(c->web_view));
1114                         return TRUE;
1115                 }
1116         } else if (event->type == GDK_SCROLL) {
1117                 if (((GdkEventScroll *)event)->state & GDK_CONTROL_MASK ||
1118                                 ((GdkEventScroll *)event)->state & GDK_CONTROL_MASK) {
1119                         gdk_event_get_scroll_deltas(event, &dx, &dy);
1120                         z = webkit_web_view_get_zoom_level(WEBKIT_WEB_VIEW(c->web_view));
1121                         z += -dy * 0.1;
1122                         z = dx != 0 ? cfg.global_zoom : z;
1123                         webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), z);
1124                         return TRUE;
1125                 }
1126         }
1127
1128         return FALSE;
1129 }
1130
1131 void
1132 mainwindow_setup(void)
1133 {
1134         mw.win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
1135         gtk_window_set_default_size(GTK_WINDOW(mw.win), 800, 600);
1136         g_signal_connect(G_OBJECT(mw.win), "destroy", gtk_main_quit, NULL);
1137         gtk_window_set_title(GTK_WINDOW(mw.win), __NAME__);
1138
1139         mw.notebook = gtk_notebook_new();
1140         gtk_notebook_set_scrollable(GTK_NOTEBOOK(mw.notebook), TRUE);
1141         gtk_container_add(GTK_CONTAINER(mw.win), mw.notebook);
1142         g_signal_connect(G_OBJECT(mw.notebook), "switch-page",
1143                                                                          G_CALLBACK(notebook_switch_page), NULL);
1144 }
1145
1146 void
1147 mainwindow_title(gint idx)
1148 {
1149         GtkWidget *child, *widg, *tablabel;
1150         const gchar *text;
1151
1152         child = gtk_notebook_get_nth_page(GTK_NOTEBOOK(mw.notebook), idx);
1153         if (child == NULL)
1154                 return;
1155
1156         widg = gtk_notebook_get_tab_label(GTK_NOTEBOOK(mw.notebook), child);
1157         tablabel = (GtkWidget *)g_object_get_data(G_OBJECT(widg), "lariza-tab-label");
1158         text = gtk_label_get_text(GTK_LABEL(tablabel));
1159         gtk_window_set_title(GTK_WINDOW(mw.win), text);
1160 }
1161
1162 void
1163 notebook_switch_page(GtkNotebook *nb, GtkWidget *p, guint idx, gpointer data)
1164 {
1165         mainwindow_title(idx);
1166 }
1167
1168 gboolean
1169 quit_if_nothing_active(void)
1170 {
1171         if (clients == 0) {
1172                 if (downloads == 0) {
1173                         gtk_main_quit();
1174                         return TRUE;
1175                 } else {
1176                         gtk_widget_show_all(dm.win);
1177                 }
1178         }
1179
1180         return FALSE;
1181 }
1182
1183 gboolean
1184 remote_msg(GIOChannel *channel, GIOCondition condition, gpointer data)
1185 {
1186         gchar *uri = NULL;
1187
1188         g_io_channel_read_line(channel, &uri, NULL, NULL, NULL);
1189         if (uri) {
1190                 g_strstrip(uri);
1191                 client_new(uri, NULL, TRUE, TRUE);
1192                 g_free(uri);
1193         }
1194         return TRUE;
1195 }
1196
1197 void
1198 run_user_scripts(WebKitWebView *web_view)
1199 {
1200         gchar *base = NULL, *path = NULL, *contents = NULL;
1201         const gchar *entry = NULL;
1202         GDir *scriptdir = NULL;
1203
1204         base = g_build_filename(g_get_user_config_dir(), __NAME__, "user-scripts",
1205                                                                                                         NULL);
1206         scriptdir = g_dir_open(base, 0, NULL);
1207         if (scriptdir != NULL) {
1208                 while ((entry = g_dir_read_name(scriptdir)) != NULL) {
1209                         path = g_build_filename(base, entry, NULL);
1210                         char *jscmd = malloc(strlen(path) + 36);
1211                         sprintf(jscmd, "console.log(\"Running userscript %s\");", path);
1212                         if (g_str_has_suffix(path, ".js")) {
1213                                 if (g_file_get_contents(path, &contents, NULL, NULL)) {
1214                                         webkit_web_view_run_javascript(web_view, jscmd, NULL, NULL, NULL);
1215                                         webkit_web_view_run_javascript(web_view, contents, NULL, NULL, NULL);
1216                                         g_free(contents);
1217                                         g_free(jscmd);
1218                                 }
1219                         }
1220                         g_free(path);
1221                 }
1222                 g_dir_close(scriptdir);
1223         }
1224
1225         g_free(base);
1226 }
1227
1228 void
1229 show_web_view(WebKitWebView *web_view, gpointer data)
1230 {
1231         struct Client *c = (struct Client *)data;
1232         gint idx;
1233
1234         (void)web_view;
1235
1236         gtk_widget_show_all(mw.win);
1237
1238         if (c->focus_new_tab) {
1239                 idx = gtk_notebook_page_num(GTK_NOTEBOOK(mw.notebook), c->vbox);
1240                 if (idx != -1)
1241                         gtk_notebook_set_current_page(GTK_NOTEBOOK(mw.notebook), idx);
1242
1243                 gtk_widget_grab_focus(c->web_view);
1244         }
1245 }
1246
1247 void
1248 trust_user_certs(WebKitWebContext *wc)
1249 {
1250         GTlsCertificate *cert;
1251         const gchar *basedir, *file, *absfile;
1252         GDir *dir;
1253
1254         basedir = g_build_filename(g_get_user_config_dir(), __NAME__, "certs", NULL);
1255         dir = g_dir_open(basedir, 0, NULL);
1256         if (dir != NULL) {
1257                 file = g_dir_read_name(dir);
1258                 while (file != NULL) {
1259                         absfile = g_build_filename(g_get_user_config_dir(), __NAME__, "certs",
1260
1261                                                                                                                                  file, NULL);
1262                         cert = g_tls_certificate_new_from_file(absfile, NULL);
1263                         if (cert == NULL)
1264                                 fprintf(stderr, __NAME__": Could not load trusted cert '%s'\n", file);
1265                         else
1266                                 webkit_web_context_allow_tls_certificate_for_host(wc, cert, file);
1267                         file = g_dir_read_name(dir);
1268                 }
1269                 g_dir_close(dir);
1270         }
1271 }
1272
1273 GKeyFile *
1274 get_ini(void)
1275 {
1276         GKeyFileFlags flags = G_KEY_FILE_NONE;
1277         g_autoptr(GError) error = NULL;
1278         config = g_key_file_new();
1279
1280         // Load user config
1281         if (!g_key_file_load_from_file(config,
1282                                                                                                                                  g_build_filename(g_get_user_config_dir(),
1283                                                                                                                                                                                                         __NAME__, "lariza.ini",
1284                                                                                                                                                                                                         NULL), flags, &error)) {
1285                 // Load global config
1286                 if (!g_key_file_load_from_file(config, "/etc/lariza.ini", flags,
1287                                                                                                                                          &error)) {
1288                         fprintf(stderr, "Could not load lariza.ini: %s", error->message);
1289                 }
1290         }
1291         return config;
1292 }
1293
1294 int
1295 main(int argc, char **argv)
1296 {
1297         int opt, i;
1298
1299         gtk_init(&argc, &argv);
1300         get_config();
1301
1302         while ((opt = getopt(argc, argv, "C")) != -1) {
1303                 switch (opt) {
1304                 case 'C':
1305                         cfg.cooperative_instances = FALSE;
1306                         break;
1307                 default:
1308                         fprintf(stderr, "Usage: "__NAME__" [OPTION]... [URI]...\n");
1309                         exit(EXIT_FAILURE);
1310                 }
1311         }
1312
1313         if (cfg.cooperative_instances)
1314                 cooperation_setup();
1315
1316         if (!cfg.cooperative_instances || cfg.cooperative_alone)
1317                 init_default_web_context();
1318
1319         downloadmanager_setup();
1320         mainwindow_setup();
1321
1322         if (optind >= argc) {
1323                 client_new(cfg.home_uri, NULL, TRUE, TRUE);
1324         } else {
1325                 for (i = optind; i < argc; i++)
1326                         client_new(argv[i], NULL, TRUE, TRUE);
1327         }
1328
1329         if (!cfg.cooperative_instances || cfg.cooperative_alone)
1330                 gtk_main();
1331
1332         exit(EXIT_SUCCESS);
1333 }