/* * Copyright (c) 2022 mini-rose * * Permission is hereby granted, free of charge, to any person * obtaining a copy of this software and any associated documentation * files to modify, copy, merge, publish, distribute and/or * sublicense copies of this sotware for their own use. * This code does not come with any warranty. * */ #include #include // necessary for free, malloc. #include #include #include "config.h" #include "plugins/libre_redirect/libre_redirect.h" #include "plugins/readability/readability.h" #include "plugins/style/style.h" // #include "plugins/stand_in/stand_in.h" int LIBRE_REDIRECT_ENABLED = true; int READABILITY_ENABLED = true; int CUSTOM_STYLE_ENABLED = true; int CUSTOM_USER_AGENT = true; // to enable plugins, // 1. Enable them: // - uncomment their #include statement // - set their variable to true // - in build.sh, uncomment: REQS= #./plugins/*/*.c // 2. Remove stand_in code; // - Comment out #include "plugins/stand_in/stand_in.h" line, or edit it // together with stand_in.c so as to not include the plugin functions. // - In build.sh, comment out REQS=./plugins/stand_in/stand_in.c #define CACHE \ "base-cache-directory", CACHE_DIR, "base-data-directory", CACHE_DIR, \ "disk-cache-directory", CACHE_DIR, "dom-cache-directory", CACHE_DIR, \ "hsts-cache-directory", CACHE_DIR, "indexeddb-directory", CACHE_DIR, \ "itp-directory", CACHE_DIR, "local-storage-directory", CACHE_DIR, \ "offline-application-cache-directory", CACHE_DIR, \ "service-worker-registrations-directory", CACHE_DIR enum { _SEARCH, _FIND, _HIDDEN }; static int entry_mode; static GtkWindow* window; static GtkHeaderBar* bar; static GtkEntryBuffer* search_buf; static GtkEntry* search; WebKitWebView* webview_new() { char* style; WebKitSettings* settings; WebKitWebContext* web_context; WebKitCookieManager* cookiemanager; WebKitUserContentManager* contentmanager; settings = webkit_settings_new_with_settings(WEBKIT, NULL); if (CUSTOM_USER_AGENT) { webkit_settings_set_user_agent( settings, "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, " "like Gecko) Chrome/110.0.0.0 Safari/537.36"); // See: for some common user agents } web_context = webkit_web_context_new_with_website_data_manager( webkit_website_data_manager_new(CACHE, NULL)); contentmanager = webkit_user_content_manager_new(); cookiemanager = webkit_web_context_get_cookie_manager(web_context); webkit_cookie_manager_set_persistent_storage( cookiemanager, CACHE_DIR "/cookies.sqlite", WEBKIT_COOKIE_PERSISTENT_STORAGE_SQLITE); webkit_cookie_manager_set_accept_policy(cookiemanager, WEBKIT_COOKIE_POLICY_ACCEPT_ALWAYS); webkit_web_context_set_process_model( web_context, WEBKIT_PROCESS_MODEL_MULTIPLE_SECONDARY_PROCESSES); if (g_file_get_contents("~/.config/rose/style.css", &style, NULL, NULL)) webkit_user_content_manager_add_style_sheet( contentmanager, webkit_user_style_sheet_new(style, WEBKIT_USER_CONTENT_INJECT_ALL_FRAMES, WEBKIT_USER_STYLE_LEVEL_USER, NULL, NULL)); return g_object_new(WEBKIT_TYPE_WEB_VIEW, "settings", settings, "web-context", web_context, "user-content-manager", contentmanager, NULL); } WebKitWebView* notebook_get_webview(GtkNotebook* notebook) { return WEBKIT_WEB_VIEW(gtk_notebook_get_nth_page( notebook, gtk_notebook_get_current_page(notebook))); } void load_uri(WebKitWebView* view, const char* uri) { if (g_str_has_prefix(uri, "http://") || g_str_has_prefix(uri, "https://") || g_str_has_prefix(uri, "file://") || g_str_has_prefix(uri, "about:")) { webkit_web_view_load_uri(view, uri); } else { // webkit_web_view_load_uri(view, uri); char tmp[strlen(uri) + strlen(SEARCH)]; snprintf(tmp, sizeof(tmp), SEARCH, uri); webkit_web_view_load_uri(view, tmp); } } void redirect_if_annoying(WebKitWebView* view, const char* uri) { int l = LIBRE_N + strlen(uri) + 1; char uri_filtered[l]; str_init(uri_filtered, l); int check = libre_redirect(uri, uri_filtered); if (check == 2) { webkit_web_view_load_uri(view, uri_filtered); } } void load_changed(WebKitWebView* self, WebKitLoadEvent load_event, GtkNotebook* notebook) { switch (load_event) { /* see */ case WEBKIT_LOAD_STARTED: if (CUSTOM_STYLE_ENABLED) { char* style_js = malloc(STYLE_N + 1); read_style_js(style_js); webkit_web_view_run_javascript(notebook_get_webview(notebook), style_js, NULL, NULL, NULL); free(style_js); } if (LIBRE_REDIRECT_ENABLED) { redirect_if_annoying(self, webkit_web_view_get_uri(self)); } break; case WEBKIT_LOAD_REDIRECTED: if (LIBRE_REDIRECT_ENABLED) { redirect_if_annoying(self, webkit_web_view_get_uri(self)); } break; case WEBKIT_LOAD_COMMITTED: if (LIBRE_REDIRECT_ENABLED) { redirect_if_annoying(self, webkit_web_view_get_uri(self)); } if (CUSTOM_STYLE_ENABLED) { char* style_js = malloc(STYLE_N + 1); read_style_js(style_js); webkit_web_view_run_javascript(notebook_get_webview(notebook), style_js, NULL, NULL, NULL); free(style_js); } break; case WEBKIT_LOAD_FINISHED: { /* Add gtk tab title */ const char* webpage_title = webkit_web_view_get_title(self); const int max_length = 25; char tab_title[max_length + 1]; if (webpage_title != NULL) { for (int i = 0; i < (max_length); i++) { tab_title[i] = webpage_title[i]; if (webpage_title[i] == '\0') { break; } } tab_title[max_length] = '\0'; } gtk_notebook_set_tab_label_text(notebook, GTK_WIDGET(self), webpage_title == NULL ? "—" : tab_title); // gtk_widget_hide(GTK_WIDGET(bar)); } } } void notebook_append(GtkNotebook* notebook, const char* uri); /* notebook_append calls handle_create, but handle_create also calls * notebook_append. Therefore we need to declare notebook_append, so that * handle_create_new_tab knows its type. */ GtkWidget* handle_create_new_tab(WebKitWebView* self, WebKitNavigationAction* navigation_action, GtkNotebook* notebook) { WebKitURIRequest* uri_request = webkit_navigation_action_get_request(navigation_action); const char* uri = webkit_uri_request_get_uri(uri_request); printf("Creating new window: %s\n", uri); notebook_append(notebook, uri); gtk_notebook_set_show_tabs(notebook, true); return NULL; /* WebKitGTK documentation recommends returning the new webview. * I imagine that this might allow e.g., to go back in a new tab * or generally to keep track of history. * However, this would require either modifying notebook_append * or duplicating its contents, for unclear gain. */ } void notebook_append(GtkNotebook* notebook, const char* uri) { GdkScreen* screen = gtk_window_get_screen(GTK_WINDOW(window)); GdkVisual* rgba_visual = gdk_screen_get_rgba_visual(screen); GdkRGBA rgba; gdk_rgba_parse(&rgba, BG_COLOR); WebKitWebView* view = webview_new(); gtk_widget_set_visual(GTK_WIDGET(window), rgba_visual); g_signal_connect(view, "load_changed", G_CALLBACK(load_changed), notebook); g_signal_connect(view, "create", G_CALLBACK(handle_create_new_tab), notebook); int n = gtk_notebook_append_page(notebook, GTK_WIDGET(view), NULL); gtk_notebook_set_tab_reorderable(notebook, GTK_WIDGET(view), true); gtk_widget_show_all(GTK_WIDGET(window)); gtk_widget_hide(GTK_WIDGET(bar)); webkit_web_view_set_background_color(view, &rgba); load_uri(view, (uri) ? uri : HOME); gtk_notebook_set_current_page(notebook, n); gtk_notebook_set_tab_label_text(notebook, GTK_WIDGET(view), "-"); webkit_web_view_set_zoom_level(view, ZOOM); } void show_bar(GtkNotebook* notebook) { if (entry_mode == _SEARCH) { const char* url = webkit_web_view_get_uri(notebook_get_webview(notebook)); gtk_entry_set_placeholder_text(search, "Search"); gtk_entry_buffer_set_text(search_buf, url, strlen(url)); gtk_widget_show(GTK_WIDGET(bar)); gtk_window_set_focus(window, GTK_WIDGET(search)); } else if (entry_mode == _HIDDEN) { gtk_widget_hide(GTK_WIDGET(bar)); } else { const char* search_text = webkit_find_controller_get_search_text( webkit_web_view_get_find_controller(notebook_get_webview(notebook))); if (search_text != NULL) gtk_entry_buffer_set_text(search_buf, search_text, strlen(search_text)); gtk_entry_set_placeholder_text(search, "Find"); gtk_widget_show(GTK_WIDGET(bar)); gtk_window_set_focus(window, GTK_WIDGET(search)); } } int handle_key(func id, GtkNotebook* notebook) { static double zoom = ZOOM; static bool is_fullscreen = 0; switch (id) { case goback: webkit_web_view_go_back(notebook_get_webview(notebook)); break; case goforward: webkit_web_view_go_forward(notebook_get_webview(notebook)); break; case refresh: webkit_web_view_reload(notebook_get_webview(notebook)); break; case refresh_force: webkit_web_view_reload_bypass_cache(notebook_get_webview(notebook)); break; case back_to_home: load_uri(notebook_get_webview(notebook), HOME); break; case zoomin: webkit_web_view_set_zoom_level(notebook_get_webview(notebook), (zoom += ZOOM_VAL)); break; case zoomout: webkit_web_view_set_zoom_level(notebook_get_webview(notebook), (zoom -= ZOOM_VAL)); break; case zoom_reset: webkit_web_view_set_zoom_level(notebook_get_webview(notebook), (zoom = ZOOM)); break; case prev_tab: if (gtk_notebook_get_current_page(notebook) == 0) { gtk_notebook_set_current_page(notebook, gtk_notebook_get_n_pages(notebook) - 1); } else { gtk_notebook_prev_page(notebook); } break; case next_tab: if (gtk_notebook_get_current_page(notebook) == gtk_notebook_get_n_pages(notebook) - 1) { notebook_append(notebook, NULL); gtk_notebook_set_show_tabs(notebook, true); webkit_web_view_set_zoom_level(notebook_get_webview(notebook), zoom); } else { gtk_notebook_next_page(notebook); } break; case close_tab: gtk_notebook_remove_page(notebook, gtk_notebook_get_current_page(notebook)); switch (gtk_notebook_get_n_pages(notebook)) { case 0: exit(0); break; case 1: gtk_notebook_set_show_tabs(notebook, false); break; } break; case toggle_fullscreen: if (is_fullscreen) gtk_window_unfullscreen(window); else gtk_window_fullscreen(window); is_fullscreen = !is_fullscreen; break; case show_searchbar: entry_mode = _SEARCH; show_bar(notebook); break; case show_finder: entry_mode = _FIND; show_bar(notebook); break; case finder_next: webkit_find_controller_search_next( webkit_web_view_get_find_controller(notebook_get_webview(notebook))); break; case finder_prev: webkit_find_controller_search_previous( webkit_web_view_get_find_controller(notebook_get_webview(notebook))); break; case new_tab: notebook_append(notebook, NULL); gtk_notebook_set_show_tabs(notebook, true); entry_mode = _SEARCH; show_bar(notebook); break; case hide_bar: entry_mode = _HIDDEN; show_bar(notebook); break; case prettify: { if (READABILITY_ENABLED) { char* readability_js = malloc(READABILITY_N + 1); read_readability_js(readability_js); webkit_web_view_run_javascript(notebook_get_webview(notebook), readability_js, NULL, NULL, NULL); free(readability_js); } break; } } return 1; } int keypress(void* self, GdkEvent* e, GtkNotebook* notebook) { (void)self; for (int i = 0; i < sizeof(keys) / sizeof(keys[0]); i++) if (e->key.keyval == keys[i].key && e->key.state == keys[i].mod) return handle_key(keys[i].id, notebook); return 0; } void search_activate(GtkEntry* self, GtkNotebook* notebook) { if (entry_mode == _SEARCH) load_uri(notebook_get_webview(notebook), gtk_entry_buffer_get_text(search_buf)); else if (entry_mode == _FIND) webkit_find_controller_search( webkit_web_view_get_find_controller(notebook_get_webview(notebook)), gtk_entry_buffer_get_text(search_buf), WEBKIT_FIND_OPTIONS_CASE_INSENSITIVE | WEBKIT_FIND_OPTIONS_WRAP_AROUND, G_MAXUINT); gtk_widget_hide(GTK_WIDGET(bar)); } void window_init(GtkNotebook* notebook) { GtkCssProvider* css = gtk_css_provider_new(); gtk_css_provider_load_from_path(css, "/usr/share/themes/rose/style.css", NULL); gtk_style_context_add_provider_for_screen(gdk_screen_get_default(), GTK_STYLE_PROVIDER(css), 800); gtk_entry_buffer_new("", 0); gtk_entry_set_alignment(search, 0.48); gtk_widget_set_size_request(GTK_WIDGET(search), 1200, -1); gtk_header_bar_set_custom_title(bar, GTK_WIDGET(search)); gtk_window_set_titlebar(window, GTK_WIDGET(bar)); g_signal_connect(search, "activate", G_CALLBACK(search_activate), notebook); g_signal_connect(window, "key-press-event", G_CALLBACK(keypress), notebook); g_signal_connect(window, "destroy", G_CALLBACK(exit), notebook); } void notebook_init(GtkNotebook* notebook, const char* uri) { gtk_notebook_set_show_border(notebook, false); gtk_notebook_set_show_tabs(notebook, false); notebook_append(notebook, uri); } void setup(GtkNotebook* notebook, int argc, char** argv) { // Define GTK entities window = GTK_WINDOW(gtk_window_new(0)); notebook = GTK_NOTEBOOK(gtk_notebook_new()); bar = GTK_HEADER_BAR(gtk_header_bar_new()); search_buf = GTK_ENTRY_BUFFER(gtk_entry_buffer_new("", 0)); search = GTK_ENTRY(gtk_entry_new_with_buffer(search_buf)); gtk_window_set_default_size(window, WIDTH, HEIGHT); window_init(notebook); // Initialize with first uri char* first_uri = argc > 1 ? argv[1] : NULL; notebook_init(notebook, first_uri); g_object_set(gtk_settings_get_default(), GTK, NULL); // More GTK stuff gtk_container_add(GTK_CONTAINER(window), GTK_WIDGET(notebook)); gtk_widget_show_all(GTK_WIDGET(window)); gtk_widget_hide(GTK_WIDGET(bar)); // Deal with more uris, if this is necessary. if (argc > 2) { gtk_notebook_set_show_tabs(notebook, true); for (int i = 2; i < argc; i++) { notebook_append(notebook, argv[i]); } } } int main(int argc, char** argv) { GtkNotebook* notebook; gtk_init(NULL, NULL); setup(notebook, argc, argv); gtk_main(); // this point is never reached, since gtk_main(); never exits. }