path: root/rose.c

/*
 * 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 <string.h>
#include <stdbool.h>
#include <webkit2/webkit2.h>

#include "config.h"
#include "plugins/libre_redirect/libre_redirect.h"

#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);
	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);
}

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 {
		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:
                                redirect_if_annoying(self, webkit_web_view_get_uri(self));
                                break;
                case WEBKIT_LOAD_REDIRECTED:
                                redirect_if_annoying(self, webkit_web_view_get_uri(self));
                                break;
                case WEBKIT_LOAD_COMMITTED:
                                redirect_if_annoying(self, webkit_web_view_get_uri(self));
                                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);
}

WebKitWebView *notebook_get_webview(GtkNotebook *notebook)
{
	return WEBKIT_WEB_VIEW(
	    gtk_notebook_get_nth_page(notebook, gtk_notebook_get_current_page(notebook)));
}

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);
		} else {
			gtk_notebook_next_page(notebook);
		}
		break;

	case close_tab:
		gtk_notebook_remove_p