diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.rs | 148 | ||||
| -rw-r--r-- | src/plugin_page.js | 82 |
2 files changed, 0 insertions, 230 deletions
diff --git a/src/main.rs b/src/main.rs deleted file mode 100644 index 36143ccc..00000000 --- a/src/main.rs +++ /dev/null @@ -1,148 +0,0 @@ -use std::fmt::{Debug, Display, Formatter}; -use std::fs; -use hyper::{Client, Uri}; -use hyper::body::Buf; -use serde::{ Serialize, Deserialize }; -use serde_json; -use tungstenite::Message; - -type TokioResult<T> = std::result::Result<T, Box<dyn std::error::Error + Send + Sync>>; - -enum AppError { - ContentNotFound -} - -impl Debug for AppError { - fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result { - let message = match self { - AppError::ContentNotFound => "Content not found" - }; - - write!(f, "{}", message) - } -} - -impl Display for AppError { - fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result { - write!(f, "{}", self) - } -} - -unsafe impl Send for AppError { } -unsafe impl Sync for AppError { } - -impl std::error::Error for AppError { } - -#[allow(non_snake_case)] -#[allow(dead_code)] -#[derive(Deserialize)] -struct WebContent { - description: String, - devtoolsFrontendUrl: String, - id: String, - title: String, - r#type: String, - url: String, - webSocketDebuggerUrl: String -} - -#[allow(non_snake_case)] -#[allow(dead_code)] -#[derive(Serialize)] -struct DebuggerCommandParams { - expression: String, - userGesture: bool -} - -#[allow(non_snake_case)] -#[allow(dead_code)] -#[derive(Serialize)] -struct DebuggerCommand { - id: u32, - method: String, - params: DebuggerCommandParams -} - -/// Downloads all content from a website -async fn get_web_content(url: Uri) -> TokioResult<Vec<WebContent>> { - let client = Client::new(); - let response = client.get(url).await?; - let body = hyper::body::aggregate(response).await?; - - let data = String::from(std::str::from_utf8(body.chunk())?); - - Ok(serde_json::from_str(data.as_str())?) -} - -/// Loads plugins -fn load_plugins() -> String { - let paths = fs::read_dir("./plugins"); - if let Ok(paths) = paths { - - let mut result = String::new(); - - for entry in paths { - if let Ok(entry) = entry { - if let Ok(file_type) = entry.file_type() { - if file_type.is_file() { - if let Ok(content) = fs::read_to_string(entry.path()) { - result.push_str(format!("plugins.push(new {});", content).as_str()); - } - } - } - } - } - - result - } else { - String::from("") - } -} - -#[tokio::main] -async fn main() -> TokioResult<()> { - // If CEF Debugging is enabled, it will be accessible through port 8080 - let url = "http://127.0.0.1:8080/json".parse::<hyper::Uri>().unwrap(); - - // Load all available tabs that can be debugged - let contents = get_web_content(url).await?; - - // Find QuickAccess tab (sidebar menu) and get the debugger websocket interface url - let mut quick_access_debug_url: Option<String> = None; - for content in &contents { - if content.title == "QuickAccess" { - quick_access_debug_url = Some(content.webSocketDebuggerUrl.clone()); - } - } - - if let Some(url) = quick_access_debug_url { - - // Connect to debugger websocket - let (mut socket, _) = tungstenite::connect(url)?; - - // Create a inject command to send to the debugger - let command = DebuggerCommand { - id: 1, - method: String::from("Runtime.evaluate"), - params: DebuggerCommandParams { - expression: String::from(include_str!("plugin_page.js").replace("{{ PLUGINS }}", load_plugins().as_str())), - userGesture: true - } - }; - - // Send command to debugger - socket.write_message(Message::Text(serde_json::to_string(&command)?))?; - - // Print response - let response = socket.read_message()?; - println!("{}", response); - - socket.close(None)?; - - } else { - return Err(AppError::ContentNotFound.into()); - } - - - Ok(()) -} diff --git a/src/plugin_page.js b/src/plugin_page.js deleted file mode 100644 index 537f53a3..00000000 --- a/src/plugin_page.js +++ /dev/null @@ -1,82 +0,0 @@ -(function () { - let plugins = []; - - {{ PLUGINS }} - - const PLUGIN_ICON = ` - <svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" fill="currentColor" class="bi bi-plugin" viewBox="0 0 16 16"> - <path fill-rule="evenodd" d="M1 8a7 7 0 1 1 2.898 5.673c-.167-.121-.216-.406-.002-.62l1.8-1.8a3.5 3.5 0 0 0 - 4.572-.328l1.414-1.415a.5.5 0 0 0 0-.707l-.707-.707 1.559-1.563a.5.5 0 1 0-.708-.706l-1.559 1.562-1.414-1.414 - 1.56-1.562a.5.5 0 1 0-.707-.706l-1.56 1.56-.707-.706a.5.5 0 0 0-.707 0L5.318 5.975a3.5 3.5 0 0 0-.328 - 4.571l-1.8 1.8c-.58.58-.62 1.6.121 2.137A8 8 0 1 0 0 8a.5.5 0 0 0 1 0Z"/> - </svg> - `; - - function createTitle(text) { - return `<div class="quickaccessmenu_Title_34nl5">${text}</div>`; - } - - function createTabGroupPanel(content) { - return `<div class="quickaccessmenu_TabGroupPanel_1QO7b Panel Focusable">${content}</div>`; - } - - function createPanelSelection(content) { - return `<div class="quickaccesscontrols_PanelSection_Ob5uo">${content}</div>`; - } - - function createPanelSelectionRow(content) { - return `<div class="quickaccesscontrols_PanelSectionRow_26R5w">${content}</div>`; - } - - function createButton(text, id) { - return ` - <div class="basicdialog_Field_ugL9c basicdialog_WithChildrenBelow_1RjOd basicdialog_InlineWrapShiftsChildrenBelow_3a6QZ basicdialog_ExtraPaddingOnChildrenBelow_2-owv basicdialog_StandardPadding_1HrfN basicdialog_HighlightOnFocus_1xh2W Panel Focusable" style="--indent-level:0;"> - <div class="basicdialog_FieldChildren_279n8"> - <button id="${id}" type="button" tabindex="0" class="DialogButton _DialogLayout Secondary basicdialog_Button_1Ievp Focusable">${text}</button> - </div> - </div> - `; - } - - function createPluginList() { - let pages = document.getElementsByClassName("quickaccessmenu_AllTabContents_2yKG4 quickaccessmenu_Down_3rR0o")[0]; - let pluginPage = pages.children[pages.children.length - 1]; - - pluginPage.innerHTML = createTitle("Plugins"); - - let buttons = ""; - for (let i = 0; i < plugins.length; i++) { - buttons += createPanelSelectionRow(createButton(plugins[i].getName(), "plugin_btn_" + i)) - } - - pluginPage.innerHTML += createTabGroupPanel(createPanelSelection(buttons)); - - for (let i = 0; i < plugins.length; i++) { - document.getElementById("plugin_btn_" + i).onclick = (function(plugin, pluginPage) { - return function() { - pluginPage.innerHTML = createButton("Back", "plugin_back") + createTitle(plugin.getName()) + createTabGroupPanel(plugin.getPageContent()); - plugin.runCode(); - - document.getElementById("plugin_back").onclick = (e) => { - createPluginList(); - }; - }; - }(plugins[i], pluginPage)) - } - - } - - function inject() { - let tabs = document.getElementsByClassName("quickaccessmenu_TabContentColumn_2z5NL Panel Focusable")[0]; - tabs.children[tabs.children.length - 1].innerHTML = PLUGIN_ICON; - - createPluginList(); - } - - let injector = setInterval(function () { - if (document.hasFocus()) { - inject(); - clearInterval(injector); - } - }, 100); -})();
\ No newline at end of file |