Add ability to inject JavaScript

browser-client.cpp

@@ -677,6 +677,13 @@ void BrowserClient::OnLoadEnd(CefRefPtr<CefBrowser>, CefRefPtr<CefFrame> frame,
 
 		frame->ExecuteJavaScript(script, "", 0);
 	}
+
+	if (bs->javascript_active && !bs->javascript.empty()) {
+		blog(LOG_INFO,
+		     "obs-browser: the user has javascript enabled on source %s.",
+		     obs_source_get_name(bs->source));
+		frame->ExecuteJavaScript(bs->javascript, "", 0);
+	}
 }
 
 bool BrowserClient::OnConsoleMessage(CefRefPtr<CefBrowser>,

data/locale/en-US.ini

@@ -9,6 +9,9 @@ RefreshBrowserActive="Refresh browser when scene becomes active"
 RefreshNoCache="Refresh cache of current page"
 RestartCEF="Restart CEF"
 BrowserSource="Browser"
+JavaScript="JavaScript Code"
+JavaScriptActive="I understand the risks of injecting JavaScript into the website above."
+JavaScript.ToolTip="Do not copy random JavaScript code from strangers."
 CustomFrameRate="Use custom frame rate"
 RerouteAudio="Control audio via OBS"
 Inspect="Inspect"

obs-browser-plugin.cpp

@@ -133,6 +133,9 @@ static void browser_source_get_defaults(obs_data_t *settings)
 				 (int)DEFAULT_CONTROL_LEVEL);
 	obs_data_set_default_string(settings, "css", default_css);
 	obs_data_set_default_bool(settings, "reroute_audio", false);
+
+	obs_data_set_default_string(settings, "javascript", "");
+	obs_data_set_default_bool(settings, "javascript_active", false);
 }
 
 static bool is_local_file_modified(obs_properties_t *props, obs_property_t *,
@@ -210,6 +213,17 @@ static obs_properties_t *browser_source_get_properties(void *data)
 	obs_properties_add_bool(props, "restart_when_active",
 				obs_module_text("RefreshBrowserActive"));
 
+	obs_property_t *j = obs_properties_add_text(
+		props, "javascript", obs_module_text("JavaScript"),
+		OBS_TEXT_MULTILINE);
+	obs_property_text_set_monospace(j, true);
+
+	prop = obs_properties_add_bool(props, "javascript_active",
+				       obs_module_text("JavaScriptActive"));
+
+	obs_property_set_long_description(
+		j, obs_module_text("JavaScript.ToolTip"));
+
 	obs_property_t *controlLevel = obs_properties_add_list(
 		props, "webpage_control_level",
 		obs_module_text("WebpageControlLevel"), OBS_COMBO_TYPE_LIST,

obs-browser-source.cpp

@@ -519,6 +519,8 @@ void BrowserSource::Update(obs_data_t *settings)
 		ControlLevel n_webpage_control_level;
 		std::string n_url;
 		std::string n_css;
+		std::string n_javascript;
+		bool n_javascript_active;
 
 		n_is_local = obs_data_get_bool(settings, "is_local_file");
 		n_width = (int)obs_data_get_int(settings, "width");
@@ -534,6 +536,10 @@ void BrowserSource::Update(obs_data_t *settings)
 		n_webpage_control_level = static_cast<ControlLevel>(
 			obs_data_get_int(settings, "webpage_control_level"));
 
+		n_javascript = obs_data_get_string(settings, "javascript");
+		n_javascript_active =
+			obs_data_get_bool(settings, "javascript_active");
+
 		if (n_is_local && !n_url.empty()) {
 			n_url = CefURIEncode(n_url, false);
 
@@ -579,7 +585,9 @@ void BrowserSource::Update(obs_data_t *settings)
 		    n_fps == fps && n_shutdown == shutdown_on_invisible &&
 		    n_restart == restart && n_css == css && n_url == url &&
 		    n_reroute == reroute_audio &&
-		    n_webpage_control_level == webpage_control_level) {
+		    n_webpage_control_level == webpage_control_level &&
+		    n_javascript == javascript &&
+		    n_javascript_active == javascript_active) {
 
 			if (n_width == width && n_height == height)
 				return;
@@ -614,6 +622,9 @@ void BrowserSource::Update(obs_data_t *settings)
 		css = n_css;
 		url = n_url;
 
+		javascript = n_javascript;
+		javascript_active = n_javascript_active;
+
 		obs_source_set_audio_active(source, reroute_audio);
 	}
 

obs-browser-source.hpp

@@ -65,6 +65,8 @@ struct BrowserSource {
 
 	std::string url;
 	std::string css;
+	std::string javascript;
+	bool javascript_active = false;
 	gs_texture_t *texture = nullptr;
 	gs_texture_t *extra_texture = nullptr;
 	uint32_t last_cx = 0;