1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
/*
* 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 <http://www.gnu.org/licenses/>.
*/
#include "view-source.h"
#include <string.h>
#include <glib.h>
#include <webkit2/webkit2.h>
#include "application.h"
#include "string-utils.h"
typedef struct {
WebKitURISchemeRequest *request;
gulong handler_id;
} DataSignalData;
static const gchar *document =
"<!doctype html>\n"
"<html dir=\"%s\">\n"
"<head>\n"
"<meta charset=\"utf-8\">\n"
"<title>Source of %s</title>\n"
"</head>\n"
"<body>\n"
"<pre>\n"
"<code>%s</code>\n"
"</pre>\n"
"</body>\n"
"</html>\n";
static void
respond(WebKitURISchemeRequest *request, 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",
"TODO", 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, length);
g_signal_handler_disconnect(web_view, data_signal_data->handler_id);
}
void
mq_view_source_request(WebKitURISchemeRequest *request,
MqApplication *application)
{
const gchar *path;
gint64 origin_tab_id;
MqTabPage *origin_tab;
MqWebView *origin_web_view;
guchar *data;
gsize length;
DataSignalData *data_signal_data;
path = webkit_uri_scheme_request_get_path(request);
/* TODO: Handle "view-source:URI" URIs. */
g_assert(g_str_has_prefix(path, "origin-tab="));
/* Parse origin tab ID out of path. */
path += strlen("origin-tab=");
origin_tab_id = atoi64(path);
/* Look up origin tab. */
origin_tab = mq_application_get_tab(application, origin_tab_id);
origin_web_view = mq_tab_page_get_web_view(origin_tab);
/* Get origin tab's data. */
data = mq_web_view_get_data(origin_web_view, &length);
if (data) {
respond(request, data, length);
} else {
data_signal_data = g_new(DataSignalData, 1);
data_signal_data->request = request;
data_signal_data->handler_id = g_signal_connect(origin_web_view,
"notify::data", G_CALLBACK(data_cb), data_signal_data);
}
}
|
