465 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			465 lines
		
	
	
		
			15 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| #include <stdbool.h>
 | |
| #include <stdlib.h> // necessary for free, malloc.
 | |
| #include <string.h>
 | |
| #include <webkit2/webkit2.h>
 | |
| 
 | |
| #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;
 | |
| //   - Add an #include "plugins/stand_in/stand_in.h" line, and modify
 | |
| //   stand_in.c so as to include stand-in functions for the excluded plugin
 | |
| //   - In the make file, include the stand_in.c file and exclude the
 | |
| //   relevant plugin
 | |
| 
 | |
| #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: <https://www.useragents.me/> 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 <https://webkitgtk.org/reference/webkit2gtk/2.5.1/WebKitWebView.html>
 | |
|      */
 | |
|     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), SEARCH_BAR_SIZE, -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.
 | |
| }
 |