10 #include <gdk/gdkkeysyms.h>
12 #include <webkit/webkit.h>
15 static void adblock(WebKitWebView *, WebKitWebFrame *, WebKitWebResource *,
16 WebKitNetworkRequest *, WebKitNetworkResponse *, gpointer);
17 static void adblock_load(void);
18 static void client_destroy(GtkWidget *obj, gpointer data);
19 static gboolean client_destroy_request(WebKitWebView *, gpointer);
20 static WebKitWebView *client_new(const gchar *uri);
21 static WebKitWebView *client_new_request(WebKitWebView *, WebKitWebFrame *,
23 static void cooperation_setup(void);
24 static void changed_load_progress(GObject *obj, GParamSpec *pspec,
26 static void changed_title(GObject *, GParamSpec *, gpointer);
27 static void changed_uri(GObject *, GParamSpec *, gpointer);
28 static gboolean download_handle(WebKitWebView *, WebKitDownload *, gpointer);
29 static gboolean download_reset_indicator(gpointer);
30 static gboolean download_request(WebKitWebView *, WebKitWebFrame *,
31 WebKitNetworkRequest *, gchar *,
32 WebKitWebPolicyDecision *, gpointer);
33 static void downloadmanager_cancel(GtkToolButton *, gpointer data);
34 static void downloadmanager_progress(GObject *obj, GParamSpec *pspec,
36 static void downloadmanager_setup(void);
37 static gchar *ensure_url_scheme(const gchar *);
38 static void grab_environment_configuration(void);
39 static void hover_web_view(WebKitWebView *, gchar *, gchar *, gpointer);
40 static gboolean key_location(GtkWidget *, GdkEvent *, gpointer);
41 static gboolean key_web_view(GtkWidget *, GdkEvent *, gpointer);
42 static gboolean remote_msg(GIOChannel *, GIOCondition, gpointer);
43 static void search(gpointer, gint);
44 static void scroll(GtkAdjustment *, gint, gdouble);
45 static Window tabbed_launch(void);
46 static void usage(void);
61 struct DownloadManager
69 static gchar *accepted_language = "en-US";
70 static GSList *adblock_patterns = NULL;
71 static gint clients = 0;
72 static gboolean cooperative_alone = TRUE;
73 static gboolean cooperative_instances = TRUE;
74 static int cooperative_pipe_fp = 0;
75 static gchar *download_dir = "/tmp";
76 static gint downloads_indicated = 0;
77 static Window embed = 0;
78 static gchar *first_uri = NULL;
79 static gdouble global_zoom = 1.0;
80 static gboolean language_is_set = FALSE;
81 static gchar *search_text = NULL;
82 static gboolean show_all_requests = FALSE;
83 static gboolean tabbed_automagic = TRUE;
87 adblock(WebKitWebView *web_view, WebKitWebFrame *frame,
88 WebKitWebResource *resource, WebKitNetworkRequest *request,
89 WebKitNetworkResponse *response, gpointer data)
91 GSList *it = adblock_patterns;
100 uri = webkit_network_request_get_uri(request);
101 if (show_all_requests)
102 fprintf(stderr, " -> %s\n", uri);
106 if (g_regex_match((GRegex *)(it->data), uri, 0, NULL))
108 webkit_network_request_set_uri(request, "about:blank");
109 if (show_all_requests)
110 fprintf(stderr, " BLOCKED!\n");
113 it = g_slist_next(it);
122 GIOChannel *channel = NULL;
126 path = g_build_filename(g_get_user_config_dir(), __NAME__, "adblock.black",
128 channel = g_io_channel_new_file(path, "r", &err);
131 while (g_io_channel_read_line(channel, &buf, NULL, NULL, NULL)
132 == G_IO_STATUS_NORMAL)
135 re = g_regex_new(buf,
136 G_REGEX_CASELESS | G_REGEX_OPTIMIZE,
137 G_REGEX_MATCH_PARTIAL, &err);
140 fprintf(stderr, __NAME__": Could not compile regex: %s\n", buf);
144 adblock_patterns = g_slist_append(adblock_patterns, re);
149 g_io_channel_shutdown(channel, FALSE, NULL);
154 client_destroy(GtkWidget *obj, gpointer data)
156 struct Client *c = (struct Client *)data;
169 client_destroy_request(WebKitWebView *web_view, gpointer data)
171 struct Client *c = (struct Client *)data;
175 gtk_widget_destroy(c->win);
181 client_new(const gchar *uri)
186 if (uri != NULL && cooperative_instances && !cooperative_alone)
188 write(cooperative_pipe_fp, uri, strlen(uri));
189 write(cooperative_pipe_fp, "\n", 1);
193 c = malloc(sizeof(struct Client));
196 fprintf(stderr, __NAME__": fatal: malloc failed\n");
203 c->win = gtk_plug_new(embed);
204 if (!gtk_plug_get_embedded(GTK_PLUG(c->win)))
206 fprintf(stderr, __NAME__": Can't plug-in to XID %ld.\n", embed);
207 gtk_widget_destroy(c->win);
215 c->win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
216 gtk_window_set_wmclass(GTK_WINDOW(c->win), __NAME__, __NAME_CAPITALIZED__);
219 /* When using Gtk2, it only shows a white area when run in suckless'
220 * tabbed. It appears we need to set a default window size for this
221 * to work. This is not needed when using Gtk3. */
222 gtk_window_set_default_size(GTK_WINDOW(c->win), 1024, 768);
224 g_signal_connect(G_OBJECT(c->win), "destroy", G_CALLBACK(client_destroy), c);
225 gtk_window_set_title(GTK_WINDOW(c->win), __NAME__);
227 c->web_view = webkit_web_view_new();
229 /* XXX I really do want to enable this option. However, I get
230 * reproducable crashes with it enabled. I've seen bug reports from
231 * 2010 about this... WebKit crashes in libpixman, so maybe it's not
233 * Yeah, well. I'll turn it off for now. */
234 /*webkit_web_view_set_full_content_zoom(WEBKIT_WEB_VIEW(c->web_view), TRUE);*/
236 webkit_web_view_set_zoom_level(WEBKIT_WEB_VIEW(c->web_view), global_zoom);
237 g_signal_connect(G_OBJECT(c->web_view), "notify::title",
238 G_CALLBACK(changed_title), c);
239 g_signal_connect(G_OBJECT(c->web_view), "notify::uri",
240 G_CALLBACK(changed_uri), c);
241 g_signal_connect(G_OBJECT(c->web_view), "notify::progress",
242 G_CALLBACK(changed_load_progress), c);
243 g_signal_connect(G_OBJECT(c->web_view), "create-web-view",
244 G_CALLBACK(client_new_request), NULL);
245 g_signal_connect(G_OBJECT(c->web_view), "close-web-view",
246 G_CALLBACK(client_destroy_request), c);
247 g_signal_connect(G_OBJECT(c->web_view),
248 "mime-type-policy-decision-requested",
249 G_CALLBACK(download_request), NULL);
250 g_signal_connect(G_OBJECT(c->web_view), "download-requested",
251 G_CALLBACK(download_handle), c);
252 g_signal_connect(G_OBJECT(c->web_view), "key-press-event",
253 G_CALLBACK(key_web_view), c);
254 g_signal_connect(G_OBJECT(c->web_view), "button-press-event",
255 G_CALLBACK(key_web_view), c);
256 g_signal_connect(G_OBJECT(c->web_view), "hovering-over-link",
257 G_CALLBACK(hover_web_view), c);
258 g_signal_connect(G_OBJECT(c->web_view), "resource-request-starting",
259 G_CALLBACK(adblock), NULL);
261 if (!language_is_set)
263 g_object_set(webkit_get_default_session(), "accept-language",
264 accepted_language, NULL);
265 language_is_set = TRUE;
268 c->scroll = gtk_scrolled_window_new(NULL, NULL);
270 gtk_container_add(GTK_CONTAINER(c->scroll), c->web_view);
272 c->location = gtk_entry_new();
273 g_signal_connect(G_OBJECT(c->location), "key-press-event",
274 G_CALLBACK(key_location), c);
276 c->progress = gtk_progress_bar_new();
277 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(c->progress), 0);
279 c->status = gtk_progress_bar_new();
280 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(c->status), 0);
281 gtk_widget_set_size_request(c->status, 20, -1);
283 c->top_box = gtk_hbox_new(FALSE, 0);
284 gtk_box_pack_start(GTK_BOX(c->top_box), c->status, FALSE, FALSE, 2);
285 gtk_box_pack_start(GTK_BOX(c->top_box), c->location, TRUE, TRUE, 0);
286 gtk_box_pack_end(GTK_BOX(c->top_box), c->progress, FALSE, TRUE, 2);
288 c->vbox = gtk_vbox_new(FALSE, 0);
289 gtk_box_pack_start(GTK_BOX(c->vbox), c->top_box, FALSE, FALSE, 2);
290 gtk_container_add(GTK_CONTAINER(c->vbox), c->scroll);
292 gtk_container_add(GTK_CONTAINER(c->win), c->vbox);
294 gtk_widget_grab_focus(c->web_view);
295 gtk_widget_show_all(c->win);
299 f = ensure_url_scheme(uri);
300 if (show_all_requests)
301 fprintf(stderr, "====> %s\n", uri);
302 webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
308 return WEBKIT_WEB_VIEW(c->web_view);
312 client_new_request(WebKitWebView *web_view, WebKitWebFrame *frame, gpointer data)
318 return client_new(NULL);
322 cooperation_setup(void)
327 fifopath = g_build_filename(g_get_user_runtime_dir(), __NAME__".fifo", NULL);
329 if (!g_file_test(fifopath, G_FILE_TEST_EXISTS))
330 mkfifo(fifopath, 0600);
332 cooperative_pipe_fp = open(fifopath, O_WRONLY | O_NONBLOCK);
333 if (!cooperative_pipe_fp)
335 fprintf(stderr, __NAME__": Can't open FIFO at all.\n");
339 if (write(cooperative_pipe_fp, "", 0) == -1)
341 /* Could not do an empty write to the FIFO which means there's
342 * no one listening. */
343 close(cooperative_pipe_fp);
344 towatch = g_io_channel_new_file(fifopath, "r+", NULL);
345 g_io_add_watch(towatch, G_IO_IN, (GIOFunc)remote_msg, NULL);
348 cooperative_alone = FALSE;
355 changed_load_progress(GObject *obj, GParamSpec *pspec, gpointer data)
357 struct Client *c = (struct Client *)data;
363 p = webkit_web_view_get_progress(WEBKIT_WEB_VIEW(c->web_view));
364 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(c->progress), p);
368 changed_title(GObject *obj, GParamSpec *pspec, gpointer data)
371 struct Client *c = (struct Client *)data;
376 t = webkit_web_view_get_title(WEBKIT_WEB_VIEW(c->web_view));
377 gtk_window_set_title(GTK_WINDOW(c->win), (t == NULL ? __NAME__ : t));
381 changed_uri(GObject *obj, GParamSpec *pspec, gpointer data)
384 struct Client *c = (struct Client *)data;
389 t = webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view));
390 gtk_entry_set_text(GTK_ENTRY(c->location), (t == NULL ? __NAME__ : t));
394 download_handle(WebKitWebView *web_view, WebKitDownload *download, gpointer data)
396 struct Client *c = (struct Client *)data;
397 gchar *path, *path2 = NULL, *uri;
405 path = g_build_filename(download_dir,
406 webkit_download_get_suggested_filename(download),
408 path2 = g_strdup(path);
409 while (g_file_test(path2, G_FILE_TEST_EXISTS) && suffix < 1000)
413 path2 = g_strdup_printf("%s.%d", path, suffix);
419 fprintf(stderr, __NAME__": Suffix reached limit for download.\n");
424 uri = g_filename_to_uri(path2, NULL, NULL);
425 webkit_download_set_destination_uri(download, uri);
429 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(c->status), 1);
430 downloads_indicated++;
431 g_timeout_add(500, download_reset_indicator, c);
433 tb = gtk_tool_button_new_from_stock(GTK_STOCK_DELETE);
434 gtk_tool_button_set_label(GTK_TOOL_BUTTON(tb),
435 webkit_download_get_suggested_filename(download));
436 gtk_toolbar_insert(GTK_TOOLBAR(dm.toolbar), tb, 0);
437 gtk_widget_show_all(dm.toolbar);
439 g_signal_connect(G_OBJECT(download), "notify::progress",
440 G_CALLBACK(downloadmanager_progress), tb);
442 g_object_ref(download);
443 g_signal_connect(G_OBJECT(tb), "clicked",
444 G_CALLBACK(downloadmanager_cancel), download);
454 download_reset_indicator(gpointer data)
456 struct Client *c = (struct Client *)data;
458 downloads_indicated--;
459 if (downloads_indicated == 0)
460 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(c->status), 0);
466 download_request(WebKitWebView *web_view, WebKitWebFrame *frame,
467 WebKitNetworkRequest *request, gchar *mime_type,
468 WebKitWebPolicyDecision *policy_decision, gpointer data)
474 if (!webkit_web_view_can_show_mime_type(web_view, mime_type))
476 webkit_web_policy_decision_download(policy_decision);
483 downloadmanager_cancel(GtkToolButton *tb, gpointer data)
485 WebKitDownload *download = WEBKIT_DOWNLOAD(data);
487 webkit_download_cancel(download);
488 g_object_unref(download);
490 gtk_widget_destroy(GTK_WIDGET(tb));
494 downloadmanager_progress(GObject *obj, GParamSpec *pspec, gpointer data)
496 WebKitDownload *download = WEBKIT_DOWNLOAD(obj);
497 GtkToolItem *tb = GTK_TOOL_ITEM(data);
503 p = webkit_download_get_progress(download) * 100;
504 t = g_strdup_printf("%s (%.0f%%)",
505 webkit_download_get_suggested_filename(download),
507 gtk_tool_button_set_label(GTK_TOOL_BUTTON(tb), t);
512 downloadmanager_setup(void)
514 dm.win = gtk_window_new(GTK_WINDOW_TOPLEVEL);
515 gtk_window_set_type_hint(GTK_WINDOW(dm.win), GDK_WINDOW_TYPE_HINT_DIALOG);
516 gtk_window_set_default_size(GTK_WINDOW(dm.win), 500, 250);
517 gtk_window_set_title(GTK_WINDOW(dm.win), __NAME__" - Download Manager");
518 g_signal_connect(G_OBJECT(dm.win), "delete-event",
519 G_CALLBACK(gtk_widget_hide_on_delete), NULL);
521 dm.toolbar = gtk_toolbar_new();
522 gtk_orientable_set_orientation(GTK_ORIENTABLE(dm.toolbar),
523 GTK_ORIENTATION_VERTICAL);
524 gtk_toolbar_set_style(GTK_TOOLBAR(dm.toolbar), GTK_TOOLBAR_BOTH_HORIZ);
525 gtk_toolbar_set_show_arrow(GTK_TOOLBAR(dm.toolbar), FALSE);
527 dm.scroll = gtk_scrolled_window_new(NULL, NULL);
528 gtk_scrolled_window_set_policy(GTK_SCROLLED_WINDOW(dm.scroll),
529 GTK_POLICY_AUTOMATIC, GTK_POLICY_AUTOMATIC);
530 gtk_scrolled_window_add_with_viewport(GTK_SCROLLED_WINDOW(dm.scroll),
533 gtk_container_add(GTK_CONTAINER(dm.win), dm.scroll);
537 ensure_url_scheme(const gchar *t)
541 f = g_ascii_strdown(t, -1);
542 if (!g_str_has_prefix(f, "http:") &&
543 !g_str_has_prefix(f, "https:") &&
544 !g_str_has_prefix(f, "file:") &&
545 !g_str_has_prefix(f, "about:"))
548 f = g_strdup_printf("http://%s", t);
556 grab_environment_configuration(void)
560 e = g_getenv(__NAME_UPPERCASE__"_ACCEPTED_LANGUAGE");
562 accepted_language = g_strdup(e);
564 e = g_getenv(__NAME_UPPERCASE__"_DOWNLOAD_DIR");
566 download_dir = g_strdup(e);
568 e = g_getenv(__NAME_UPPERCASE__"_ZOOM");
570 global_zoom = atof(e);
574 hover_web_view(WebKitWebView *web_view, gchar *title, gchar *uri, gpointer data)
576 struct Client *c = (struct Client *)data;
581 if (!gtk_widget_is_focus(c->location))
584 gtk_entry_set_text(GTK_ENTRY(c->location),
585 webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view)));
587 gtk_entry_set_text(GTK_ENTRY(c->location), uri);
592 key_location(GtkWidget *widget, GdkEvent *event, gpointer data)
594 struct Client *c = (struct Client *)data;
600 if (event->type == GDK_KEY_PRESS)
602 switch (((GdkEventKey *)event)->keyval)
605 gtk_widget_grab_focus(c->web_view);
606 t = gtk_entry_get_text(GTK_ENTRY(c->location));
607 if (t != NULL && t[0] == '/')
609 if (search_text != NULL)
611 search_text = g_strdup(t + 1); /* XXX whacky */
616 f = ensure_url_scheme(t);
617 if (show_all_requests)
618 fprintf(stderr, "====> %s\n", f);
619 webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
624 t = webkit_web_view_get_uri(WEBKIT_WEB_VIEW(c->web_view));
625 gtk_entry_set_text(GTK_ENTRY(c->location),
626 (t == NULL ? __NAME__ : t));
629 if (((GdkEventKey *)event)->state & GDK_CONTROL_MASK)
631 gtk_widget_show_all(dm.win);
643 key_web_view(GtkWidget *widget, GdkEvent *event, gpointer data)
645 struct Client *c = (struct Client *)data;
646 WebKitHitTestResultContext ht_context;
647 WebKitHitTestResult *ht_result = NULL;
648 gchar *ht_uri = NULL, *f;
652 if (event->type == GDK_KEY_PRESS)
654 if (((GdkEventKey *)event)->state & GDK_CONTROL_MASK)
656 switch (((GdkEventKey *)event)->keyval)
659 gtk_widget_grab_focus(c->location);
662 scroll(gtk_scrolled_window_get_hadjustment(
663 GTK_SCROLLED_WINDOW(c->scroll)), 0, -1);
666 scroll(gtk_scrolled_window_get_vadjustment(
667 GTK_SCROLLED_WINDOW(c->scroll)), 0, 1);
670 scroll(gtk_scrolled_window_get_vadjustment(
671 GTK_SCROLLED_WINDOW(c->scroll)), 0, -1);
674 scroll(gtk_scrolled_window_get_hadjustment(
675 GTK_SCROLLED_WINDOW(c->scroll)), 0, 1);
678 scroll(gtk_scrolled_window_get_vadjustment(
679 GTK_SCROLLED_WINDOW(c->scroll)), 1, 0.5);
682 scroll(gtk_scrolled_window_get_vadjustment(
683 GTK_SCROLLED_WINDOW(c->scroll)), 1, -0.5);
692 f = ensure_url_scheme(first_uri);
693 if (show_all_requests)
694 fprintf(stderr, "====> %s\n", f);
695 webkit_web_view_load_uri(WEBKIT_WEB_VIEW(c->web_view), f);
699 gtk_widget_show_all(dm.win);
703 else if (((GdkEventKey *)event)->keyval == GDK_KEY_Escape)
705 webkit_web_view_stop_loading(WEBKIT_WEB_VIEW(c->web_view));
706 gtk_progress_bar_set_fraction(GTK_PROGRESS_BAR(c->progress), 0);
709 else if (event->type == GDK_BUTTON_PRESS)
711 switch (((GdkEventButton *)event)->button)
714 ht_result = webkit_web_view_get_hit_test_result(
715 WEBKIT_WEB_VIEW(c->web_view),
716 (GdkEventButton *)event);
717 g_object_get(ht_result, "context", &ht_context, NULL);
718 if (ht_context & WEBKIT_HIT_TEST_RESULT_CONTEXT_LINK)
720 g_object_get(ht_result, "link-uri", &ht_uri, NULL);
722 g_object_unref(ht_result);
725 g_object_unref(ht_result);
728 webkit_web_view_go_back(WEBKIT_WEB_VIEW(c->web_view));
731 webkit_web_view_go_forward(WEBKIT_WEB_VIEW(c->web_view));
740 remote_msg(GIOChannel *channel, GIOCondition condition, gpointer data)
747 g_io_channel_read_line(channel, &uri, NULL, NULL, NULL);
758 search(gpointer data, gint direction)
760 struct Client *c = (struct Client *)data;
762 if (search_text == NULL)
765 webkit_web_view_search_text(WEBKIT_WEB_VIEW(c->web_view), search_text,
766 FALSE, direction == 1, TRUE);
770 scroll(GtkAdjustment *a, gint step_type, gdouble factor)
772 gdouble new, lower, upper, step;
773 lower = gtk_adjustment_get_lower(a);
774 upper = gtk_adjustment_get_upper(a) - gtk_adjustment_get_page_size(a) + lower;
776 step = gtk_adjustment_get_step_increment(a);
778 step = gtk_adjustment_get_page_increment(a);
779 new = gtk_adjustment_get_value(a) + factor * step;
780 new = new < lower ? lower : new;
781 new = new > upper ? upper : new;
782 gtk_adjustment_set_value(a, new);
789 GIOChannel *tabbed_stdout_channel;
791 gchar *output = NULL;
792 char *argv[] = { "tabbed", "-c", "-d", "-p", "s1", "-n", __NAME__, NULL };
795 if (!g_spawn_async_with_pipes(NULL, argv, NULL, G_SPAWN_SEARCH_PATH, NULL,
796 NULL, NULL, NULL, &tabbed_stdout, NULL,
799 fprintf(stderr, __NAME__": Could not launch tabbed: %s\n", err->message);
804 tabbed_stdout_channel = g_io_channel_unix_new(tabbed_stdout);
805 g_io_channel_read_line(tabbed_stdout_channel, &output, NULL, NULL, NULL);
808 fprintf(stderr, __NAME__": Could not read XID from tabbed\n");
812 g_io_channel_shutdown(tabbed_stdout_channel, FALSE, NULL);
815 plug_into = strtol(output, NULL, 16);
823 fprintf(stderr, "Usage: "__NAME__" [OPTION]... <URI>...\n");
829 main(int argc, char **argv)
833 gtk_init(&argc, &argv);
835 grab_environment_configuration();
837 while ((opt = getopt(argc, argv, "e:rCT")) != -1)
842 embed = atol(optarg);
843 tabbed_automagic = FALSE;
846 show_all_requests = TRUE;
849 cooperative_instances = FALSE;
852 tabbed_automagic = FALSE;
864 downloadmanager_setup();
866 if (tabbed_automagic && !(cooperative_instances && !cooperative_alone))
867 embed = tabbed_launch();
869 first_uri = g_strdup(argv[optind]);
870 for (i = optind; i < argc; i++)
872 if (!cooperative_instances || cooperative_alone)