G_MODULE_EXPORT void
webkit_web_extension_initialize(WebKitWebExtension *extension) {
LightDMGreeter *greeter = lightdm_greeter_new();
g_signal_connect(G_OBJECT(greeter),
"authentication-complete",
G_CALLBACK(authentication_complete_cb),
extension);
g_signal_connect(G_OBJECT(greeter),
"autologin-timer-expired",
G_CALLBACK(autologin_timer_expired_cb),
extension);
g_signal_connect(webkit_script_world_get_default(),
"window-object-cleared",
G_CALLBACK(window_object_cleared_callback),
greeter);
g_signal_connect(G_OBJECT(greeter), "show-prompt", G_CALLBACK(show_prompt_cb), extension);
g_signal_connect(G_OBJECT(greeter), "show-message", G_CALLBACK(show_message_cb), extension);
lightdm_greeter_connect_sync(greeter, NULL);
/* load greeter settings from config file */
keyfile = g_key_file_new();
g_key_file_load_from_file(keyfile,
CONFIG_DIR "/lightdm-webkit2-greeter.conf",
G_KEY_FILE_NONE, NULL);
}
//
// Extension initialization thing.
//
extern "C" G_MODULE_EXPORT void
webkit_web_extension_initialize (WebKitWebExtension *extension)
{
std::cout << "EXTENSION INITIALIZED\n";
g_signal_connect (webkit_script_world_get_default(),
"window-object-cleared",
G_CALLBACK (window_object_cleared_callback),
NULL);
}
/**
* Handle dbus method calls.
*/
static void dbus_handle_method_call(GDBusConnection *conn, const char *sender,
const char *object_path, const char *interface_name, const char *method,
GVariant *parameters, GDBusMethodInvocation *invocation, gpointer extension)
{
char *value;
guint64 pageid;
WebKitWebPage *page;
if (g_str_has_prefix(method, "EvalJs")) {
char *result = NULL;
gboolean success;
gboolean no_result;
JSValueRef ref = NULL;
JSGlobalContextRef jsContext;
g_variant_get(parameters, "(ts)", &pageid, &value);
page = get_web_page_or_return_dbus_error(invocation, WEBKIT_WEB_EXTENSION(extension), pageid);
if (!page) {
return;
}
no_result = !g_strcmp0(method, "EvalJsNoResult");
jsContext = webkit_frame_get_javascript_context_for_script_world(
webkit_web_page_get_main_frame(page),
webkit_script_world_get_default()
);
success = ext_util_js_eval(jsContext, value, &ref);
if (no_result) {
g_dbus_method_invocation_return_value(invocation, NULL);
} else {
result = ext_util_js_ref_to_string(jsContext, ref);
g_dbus_method_invocation_return_value(invocation, g_variant_new("(bs)", success, result));
g_free(result);
}
} else if (!g_strcmp0(method, "FocusInput")) {
g_variant_get(parameters, "(t)", &pageid);
page = get_web_page_or_return_dbus_error(invocation, WEBKIT_WEB_EXTENSION(extension), pageid);
if (!page) {
return;
}
ext_dom_focus_input(webkit_web_page_get_dom_document(page));
g_dbus_method_invocation_return_value(invocation, NULL);
} else if (!g_strcmp0(method, "SetHeaderSetting")) {
g_variant_get(parameters, "(s)", &value);
if (ext.headers) {
soup_header_free_param_list(ext.headers);
ext.headers = NULL;
}
ext.headers = soup_header_parse_param_list(value);
g_dbus_method_invocation_return_value(invocation, NULL);
}
}
extern "C" void webkit_web_extension_initialize_with_user_data(WebKitWebExtension* extension, GVariant* userData)
{
initializationUserData = userData;
g_signal_connect(extension, "page-created", G_CALLBACK(pageCreatedCallback), extension);
g_signal_connect(webkit_script_world_get_default(), "window-object-cleared", G_CALLBACK(windowObjectCleared), 0);
g_bus_own_name(
G_BUS_TYPE_SESSION,
"org.webkit.gtk.WebExtensionTest",
G_BUS_NAME_OWNER_FLAGS_NONE,
busAcquiredCallback,
0, 0,
g_object_ref(extension),
static_cast<GDestroyNotify>(g_object_unref));
}
static void methodCallCallback(GDBusConnection* connection, const char* sender, const char* objectPath, const char* interfaceName, const char* methodName, GVariant* parameters, GDBusMethodInvocation* invocation, gpointer userData)
{
if (g_strcmp0(interfaceName, "org.webkit.gtk.WebExtensionTest"))
return;
if (!g_strcmp0(methodName, "GetTitle")) {
uint64_t pageID;
g_variant_get(parameters, "(t)", &pageID);
WebKitWebPage* page = getWebPage(WEBKIT_WEB_EXTENSION(userData), pageID, invocation);
if (!page)
return;
WebKitDOMDocument* document = webkit_web_page_get_dom_document(page);
GUniquePtr<char> title(webkit_dom_document_get_title(document));
g_dbus_method_invocation_return_value(invocation, g_variant_new("(s)", title.get()));
} else if (!g_strcmp0(methodName, "RunJavaScriptInIsolatedWorld")) {
uint64_t pageID;
const char* script;
g_variant_get(parameters, "(t&s)", &pageID, &script);
WebKitWebPage* page = getWebPage(WEBKIT_WEB_EXTENSION(userData), pageID, invocation);
if (!page)
return;
GRefPtr<WebKitScriptWorld> world = adoptGRef(webkit_script_world_new());
g_assert(webkit_script_world_get_default() != world.get());
WebKitFrame* frame = webkit_web_page_get_main_frame(page);
JSGlobalContextRef jsContext = webkit_frame_get_javascript_context_for_script_world(frame, world.get());
JSRetainPtr<JSStringRef> jsScript(Adopt, JSStringCreateWithUTF8CString(script));
JSEvaluateScript(jsContext, jsScript.get(), 0, 0, 0, 0);
g_dbus_method_invocation_return_value(invocation, 0);
} else if (!g_strcmp0(methodName, "AbortProcess")) {
abort();
} else if (!g_strcmp0(methodName, "GetInitializationUserData")) {
g_assert(initializationUserData);
g_assert(g_variant_is_of_type(initializationUserData.get(), G_VARIANT_TYPE_STRING));
g_dbus_method_invocation_return_value(invocation, g_variant_new("(s)",
g_variant_get_string(initializationUserData.get(), nullptr)));
} else if (!g_strcmp0(methodName, "GetProcessIdentifier")) {
g_dbus_method_invocation_return_value(invocation,
g_variant_new("(u)", static_cast<guint32>(getCurrentProcessID())));
}
}
extern "C" void webkit_web_extension_initialize(WebKitWebExtension* extension)
{
g_signal_connect(webkit_script_world_get_default(), "window-object-cleared", G_CALLBACK(windowObjectClearedCallback), extension);
}
