rosenrot/webview.c

#include "webview.h"
#include "config.h"

struct _RoseWebView {
	WebKitWebView parent_instance;

	GCancellable *cancellable;
	RoseWebViewNavigationFlags navigation;

	char *address;
};

enum {
	LAST_PROP,
	PROP_NAVIGATION,
	PROP_ADDRESS
};

static GParamSpec *obj_properties[LAST_PROP];

G_DEFINE_TYPE(RoseWebView, rose_webview, WEBKIT_TYPE_WEB_VIEW)

static void update_navigation_flags(RoseWebView *webview)
{
	guint flags = 0;
	if (webkit_web_view_can_go_back(WEBKIT_WEB_VIEW(webview)))
		flags |= ROSE_WEBVIEW_NAV_BACK;

	if (webkit_web_view_can_go_forward(WEBKIT_WEB_VIEW(webview)))
		flags |= ROSE_WEBVIEW_FORWARD;

	if (webview->navigation != (RoseWebViewNavigationFlags) flags) {
		webview->navigation = (RoseWebViewNavigationFlags) flags;
		g_object_notify_by_pspec(G_OBJECT(webview), obj_properties[PROP_NAVIGATION]);
	}
}

static void rose_webview_constructed(GObject *object)
{
	RoseWebView *webview = ROSE_WEBVIEW(object);
	g_auto(GStrv) cors_allowlist = NULL;
	GtkStyleContext *context;
	GdkRGBA color;

	G_OBJECT_CLASS(rose_webview_parent_class)->constructed (object);

	g_signal_connect_swapped(webkit_web_view_get_back_forward_list(WEBKIT_WEB_VIEW(webview)),
                           "changed", G_CALLBACK(update_navigation_flags), webview);

	context = gtk_widget_get_style_context(GTK_WIDGET(webview));

	if (gtk_style_context_lookup_color(context, "theme_base_color", &color))
		webkit_web_view_set_background_color(WEBKIT_WEB_VIEW(webview), &color);

	cors_allowlist = g_new(char*, 2);
	cors_allowlist[0] = g_strdup("rose-resource://*/*");
	cors_allowlist[1] = NULL;

	webkit_web_view_set_cors_allowlist(WEBKIT_WEB_VIEW(webview), (const char* const *) cors_allowlist);
}

static void rose_webview_dispose(GObject *object)
{
	RoseWebView *webview = ROSE_WEBVIEW(object);
	WebKitUserContentManager *ucm = webkit_web_view_get_user_content_manager(WEBKIT_WEB_VIEW(webview));

	if (webview->cancellable) {
		g_cancellable_cancel(webview->cancellable);
		g_clear_object(&webview->cancellable);
	}

	G_OBJECT_CLASS(rose_webview_parent_class)->dispose (object);
}

static void rose_webview_finalize(GObject *object)
{
	RoseWebView *view = ROSE_WEBVIEW(object);

	g_free(view->address);

	G_OBJECT_CLASS(rose_webview_parent_class)->finalize (object);
}

static void rose_webview_class_init(RoseWebViewClass *class)
{
	GObjectClass *object_class = G_OBJECT_CLASS(class);
	WebKitWebViewClass *webkit_webview_class = WEBKIT_WEB_VIEW_CLASS(class);

	object_class->dispose = rose_webview_dispose;
	object_class->finalize = rose_webview_finalize;
	object_class->constructed = rose_webview_constructed;

	obj_properties[PROP_ADDRESS] =
    g_param_spec_string ("address",
                         "Address",
                         "address",
                         "",
                         G_PARAM_READABLE | G_PARAM_STATIC_STRINGS);
}

static void rose_webview_init(RoseWebView *webview)
{
	gtk_widget_set_overflow(GTK_WIDGET(webview), GTK_OVERFLOW_HIDDEN);
}

void rose_webview_load_url(WebKitWebView *webview, const char *url)
{
	webkit_web_view_load_uri(WEBKIT_WEB_VIEW(webview), url);
}

GtkWidget* rose_webview_new()
{
	char cookiefile[64];
	WebKitWebView *webview;
	WebKitCookieManager *cookiemanager;
	WebKitUserContentManager *contentmanager;

	WebKitSettings *settings = webkit_settings_new_with_settings(
		"auto-load-images", true,
		"enable-back-forward-navigation-gestures", true,
		"enable-developer-extras", true,
		"enable-media-stream", true,
		"hardware-acceleration-policy", WEBKIT_HARDWARE_ACCELERATION_POLICY_ALWAYS,
		"javascript-can-access-clipboard", true, NULL);

	WebKitWebContext *context = webkit_web_context_new_with_website_data_manager(
		webkit_website_data_manager_new(
		"base-cache-directory", cachedir,
		"base-data-directory", cachedir,
		NULL));

	webkit_settings_set_user_agent_with_application_details(
		settings, "Mini", "0.1");

	webkit_web_context_set_process_model(context, WEBKIT_PROCESS_MODEL_MULTIPLE_SECONDARY_PROCESSES);
	webkit_web_context_set_cache_model(context, WEBKIT_CACHE_MODEL_WEB_BROWSER);
	contentmanager = webkit_user_content_manager_new();
	cookiemanager = webkit_web_context_get_cookie_manager(context);

	strcpy(cookiefile, cachedir);
	strcat(cookiefile, "cookies");

	webkit_cookie_manager_set_persistent_storage(cookiemanager,
		cookiefile, WEBKIT_COOKIE_PERSISTENT_STORAGE_SQLITE);

	webkit_cookie_manager_set_accept_policy(cookiemanager, WEBKIT_COOKIE_POLICY_ACCEPT_ALWAYS);

	return g_object_new(
			WEBKIT_TYPE_WEB_VIEW,
			"settings", settings,
			"user-content-manager", contentmanager,
			"web-context", context, NULL);
}