fix: firefox scaffolding and custom profiles dir

ext/mv2/after.js

@@ -1,4 +1,3 @@
-
 class FlooedApi {
   ws = null;
   version = null;

src/compat/firefox.rs

@@ -0,0 +1,7 @@
+use webui_rs::webui::bindgen::*;
+
+pub unsafe fn move_firefox_extension() {
+  // Take the extension file (should be beside the binary)
+  let ext = std::env::current_dir().unwrap().join("ext").join("mv2.xpi");
+  // get the firefox profile path
+}

src/compat/mod.rs

@@ -0,0 +1 @@
+pub mod firefox;

src/main.rs

@@ -3,20 +3,49 @@
   windows_subsystem = "windows"
 )]
 
+mod compat;
 mod config;
 mod util;
 
+use std::fs;
+
 use config::get_config;
-use util::logger;
+use util::{logger, paths::get_profile_dir};
 use util::ws::WsConnector;
-use webui_rs::webui::{wait, WebUIBrowser, Window};
+//use webui_rs::webui::bindgen::webui_get_best_browser;
+use webui_rs::webui::{bindgen::webui_set_profile, wait, WebUIBrowser, Window};
 
 const VERSION: &str = env!("CARGO_PKG_VERSION");
 
 fn main() {
   logger::init(true);
   let config = get_config();
   let mut win = Window::new();
+  let browser = WebUIBrowser::ChromiumBased;
+  //let browser = unsafe { WebUIBrowser::from_usize(webui_get_best_browser(win.id)) };
+
+  // Ensure profile dir exists
+  let profile_dir = get_profile_dir(browser.to_usize());
+
+  if fs::metadata(&profile_dir).is_err() {
+    println!("Creating profile dir: {:?}", profile_dir);
+    fs::create_dir_all(&profile_dir).expect("Failed to create profile dir");
+  }
+
+  logger::log(format!("Profile dir: {:?}", profile_dir));
+
+  // Set the profile dir
+  unsafe {
+    let path_cstr = std::ffi::CString::new(
+      profile_dir.to_str().expect("Failed to convert profile dir to string"),
+    ).expect("Failed to convert profile dir to CString");
+
+    webui_set_profile(
+      win.id,
+      "Flooed".as_ptr() as *const i8,
+      path_cstr.as_ptr() as *const i8,
+    );
+  }
 
   let client = match config.client_type.unwrap_or("default".to_string()).as_str() {
     "default" => "https://discord.com/app",
@@ -33,11 +62,16 @@ fn main() {
 
   // Get current working dir
   let cwd = std::env::current_dir().unwrap();
-  // Append ./ext
-  let ext = cwd.join("ext").join("mv2");
 
-  // This should load our extension
-  win.add_extension(ext.to_str().unwrap());
+  if browser == WebUIBrowser::Firefox {
+    unsafe {
+      compat::firefox::move_firefox_extension();
+    }
+  } else {
+    // Append ./ext
+    let ext = cwd.join("ext").join("mv2");
+    win.add_extension(ext.to_str().unwrap());
+  }
 
   // Start the websocket connector
   let mut ws = WsConnector::new();
@@ -47,7 +81,7 @@ fn main() {
   ws.start();
 
   // Start the browser
-  win.show_browser(client, WebUIBrowser::ChromiumBased);
+  win.show_browser(client, browser);
 
   wait();
 }

src/util/paths.rs

@@ -1,5 +1,7 @@
 use std::{fs, path::PathBuf};
 
+use webui_rs::webui::WebUIBrowser;
+
 use crate::util::logger::log;
 
 pub fn get_config_dir() -> PathBuf {
@@ -129,6 +131,24 @@ pub fn get_theme_dir() -> std::path::PathBuf {
   theme_dir
 }
 
+pub fn get_profile_dir(browser: usize) -> PathBuf {
+  let current_exe = std::env::current_exe().unwrap_or_default();
+  let local_config_dir = current_exe.parent().unwrap().join("config.json");
+
+  if fs::metadata(local_config_dir).is_ok() {
+    // This is a portable install, so we can use the local injection dir
+    return current_exe.parent().unwrap().join("profile");
+  }
+
+  #[cfg(target_os = "windows")]
+  let appdata = dirs::data_dir().unwrap_or_default();
+
+  #[cfg(not(target_os = "windows"))]
+  let appdata = dirs::config_dir().unwrap_or_default();
+
+  appdata.join("flooed").join("profile").join(format!("{}", browser))
+}
+
 pub fn custom_detectables_path() -> PathBuf {
   let current_exe = std::env::current_exe().unwrap_or_default();
   let local_config_dir = current_exe.parent().unwrap().join("config.json");