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
|
#include <jsc/jsc.h>
#include <webkit2/webkit-web-extension.h>
struct script_info {
gchar *str;
gsize len;
gchar *src;
};
static GArray *script_infos;
static void add_file(const gchar *dirname, const gchar *filename)
{
gchar *path = g_build_filename(dirname, filename, NULL);
struct script_info s;
if (g_file_get_contents(path, &s.str, &s.len, NULL)) {
s.src = g_filename_to_uri(path, NULL, NULL);
g_array_append_val(script_infos, s);
}
g_free(path);
}
static void add_path(const gchar *path)
{
gchar *dirname = g_build_filename(path, "badwolf", "userscripts", NULL);
GDir *dir = g_dir_open(dirname, 0, NULL);
if (dir != NULL) {
const gchar *filename;
while ((filename = g_dir_read_name(dir)) != NULL)
add_file(dirname, filename);
g_dir_close(dir);
}
g_free(dirname);
}
static void on_document_loaded(WebKitWebPage *page, gpointer user_data)
{
(void) user_data;
JSCContext *ctx = webkit_frame_get_js_context(webkit_web_page_get_main_frame(page));
for (gsize i = 0; i < script_infos->len; i++) {
const struct script_info *s = &g_array_index(script_infos, struct script_info, i);
g_object_unref(jsc_context_evaluate_with_source_uri(ctx, s->str, s->len, s->src, 0));
}
}
static void on_page_created(WebKitWebExtension *extension, WebKitWebPage *page, gpointer user_data)
{
(void) extension;
(void) user_data;
g_signal_connect(page, "document-loaded", G_CALLBACK(on_document_loaded), NULL);
}
G_MODULE_EXPORT void webkit_web_extension_initialize(WebKitWebExtension *extension)
{
script_infos = g_array_new(FALSE, FALSE, sizeof(struct script_info));
const gchar * const *paths = g_get_system_data_dirs();
while (*paths != NULL)
add_path(*(paths++));
add_path(g_get_user_data_dir());
g_signal_connect(extension, "page-created", G_CALLBACK(on_page_created), NULL);
}
|
