/* * view-source: URI scheme * * Copyright (C) 2017 Patrick McDermott * * This file is part of Marquee. * * Marquee is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * Marquee is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with Marquee. If not, see . */ #include "view-source.h" #include #include #include #include "../application.h" #include "../utils/resources.h" #include "../utils/string.h" typedef struct { WebKitURISchemeRequest *request; gchar *uri; gulong handler_id; } DataSignalData; static const gchar *document = "\n" "\n" "\n" "\n" "Source of %s\n" "\n" "\n" "\n" "\n" "
\n"
				""
					"%s\n"
			"
\n" "\n" "\n"; static void respond(WebKitURISchemeRequest *request, const gchar *uri, guchar *data, gsize length) { gchar *escaped_data; GInputStream *stream; escaped_data = g_markup_escape_text((gchar *) data, length); stream = g_memory_input_stream_new_from_data( g_strdup_printf(document, gtk_widget_get_default_direction() == GTK_TEXT_DIR_RTL ? "rtl" : "ltr", uri, escaped_data), -1, g_free); webkit_uri_scheme_request_finish(request, stream, -1, "text/html"); g_object_unref(stream); } static void data_cb(MqWebView *web_view, GParamSpec G_GNUC_UNUSED *param_spec, DataSignalData *data_signal_data) { guchar *data; gsize length; data = mq_web_view_get_data(web_view, &length); respond(data_signal_data->request, data_signal_data->uri, data, length); g_signal_handler_disconnect(web_view, data_signal_data->handler_id); g_free(data_signal_data->uri); g_free(data_signal_data); } void mq_view_source_request(WebKitURISchemeRequest *request, MqApplication *application) { const gchar *path; gchar *query_str; GHashTable *query; gint64 origin_tab_id; gchar *uri; MqTabPage *origin_tab; MqWebView *origin_web_view; guchar *data; gsize length; DataSignalData *data_signal_data; path = webkit_uri_scheme_request_get_path(request); if (g_str_has_prefix(path, "resources/")) { mq_resource_response(application, path + sizeof("resources"), request); return; } query_str = g_strdup(path); query = mq_parse_query_string(query_str); /* TODO: Handle missing tab ID ("origin-tab=0"). */ origin_tab_id = mq_atoi64(g_hash_table_lookup(query, "origin-tab")); uri = g_strdup(g_hash_table_lookup(query, "uri")); g_hash_table_unref(query); g_free(query_str); origin_tab = mq_application_get_tab(application, origin_tab_id); if (origin_tab) { origin_web_view = mq_tab_page_get_web_view(origin_tab); data = mq_web_view_get_data(origin_web_view, &length); if (data) { respond(request, uri, data, length); } else { data_signal_data = g_new(DataSignalData, 1); data_signal_data->request = request; data_signal_data->uri = uri; data_signal_data->handler_id = g_signal_connect( origin_web_view, "notify::data", G_CALLBACK(data_cb), data_signal_data); } } else { /* TODO: Handling reloads by newly requesting the resource * should obviate this. */ respond(request, uri, (guchar *) g_strdup("Viewed tab no longer open"), -1); } }