Make Proxy Load Bootstrap

Rust/MelonProxy/Cargo.toml

@@ -13,8 +13,9 @@ readme = "../README.md"
 thiserror = "1.0.48"
 melon_proxy-sys = { path = "../MelonProxy-sys" }
 utf16string = "0.2.0"
-windows = { version = "0.51.1", features = ["Win32_Foundation", "Win32_System", "Win32_System_LibraryLoader"] }
+windows = { version = "0.52.0", features = ["Win32_Foundation", "Win32_System", "Win32_System_LibraryLoader"] }
 libloading = "0.8.0"
+msgbox = "0.7.0"
 
 #[target.'cfg(windows)'.dependencies]
 

Rust/MelonProxy/build.rs

@@ -1,4 +1,4 @@
-#[cfg(msvc)]
+#[cfg(windows)]
 fn main() {
     println!("cargo:warning=Linking Exports File..");
     use std::path::Path;
@@ -10,7 +10,7 @@ fn main() {
     );
 }
 
-#[cfg(not(msvc))]
+#[cfg(not(windows))]
 fn main() {
 
 }

Rust/MelonProxy/src/core.rs

@@ -0,0 +1,40 @@
+use std::{error::Error, sync::{LazyLock, Mutex}, path::PathBuf};
+
+use libloading::Library;
+
+use crate::utils::files;
+
+pub static BOOTSTRAP: LazyLock<Mutex<Option<Library>>> = LazyLock::new(|| Mutex::new(None));
+
+pub fn init() -> Result<(), Box<dyn Error>> {
+    let args: Vec<String> = std::env::args().collect();
+
+    //TODO: Support UTF-16 (it will suck)
+    let mut base_dir = std::env::current_dir()?;
+    let mut no_mods = false;
+
+    for arg in args.iter() {
+        if arg.starts_with("--melonloader.basedir") {
+            let a: Vec<&str> = arg.split("=").collect();
+            base_dir = PathBuf::from(a[1]);
+        }
+
+        if arg.contains("--no-mods") {
+            no_mods = true;
+        }
+    }
+
+    //return Ok, and silently stop loading MelonLoader, if the user has specified to not load mods,
+    //or if the game is not a Unity game
+    if no_mods {
+        return Ok(());
+    }
+
+    let bootstrap_path = files::get_bootstrap_path(&base_dir)?;
+
+    unsafe {
+        *BOOTSTRAP.try_lock()? = Some(Library::new(&bootstrap_path)?);
+    }
+
+    Ok(())
+}

Rust/MelonProxy/src/lib.rs

@@ -20,8 +20,12 @@ use melon_proxy_sys::proxy;
 pub use windows::Win32::Foundation::HINSTANCE;
 
 pub mod proxy;
+pub mod utils;
+pub mod core;
 
 #[proxy]
 fn main() {
-    println!("hello world");
+    core::init().unwrap_or_else(|e| {
+        internal_failure!("Failed to initialize MelonLoader: {}", e);
+    });
 }

Rust/MelonProxy/src/utils/assert.rs

@@ -0,0 +1,23 @@
+//! lets us throw an internal failure
+
+/// Throws an internal failure with the given message
+/// 
+/// This creates a message box, and then panics.
+/// It uses the same syntax as _format!_
+/// 
+/// # Examples
+/// 
+/// ```
+/// # use utils::assert;
+/// assert::internal_failure!("This is an internal failure");
+/// ```
+#[macro_export]
+macro_rules! internal_failure {
+    ($($arg:tt)*) => {{
+        let msg = &format_args!($($arg)*).to_string();
+
+        std::println!("{}", msg);
+        let _ = msgbox::create("Internal Failure", msg, msgbox::IconType::Error);
+        std::process::exit(1);
+    }};
+}

Rust/MelonProxy/src/utils/errors.rs

@@ -0,0 +1,9 @@
+use std::path::PathBuf;
+
+use thiserror::Error;
+
+#[derive(Debug, Error)]
+pub enum ProxyError {
+    #[error("failed to find Bootstrap at \"{0}\" please make sure you have installed MelonLoader correctly")]
+    BootstrapNotFound(PathBuf),
+}

Rust/MelonProxy/src/utils/files.rs

@@ -0,0 +1,20 @@
+use std::{env::consts::DLL_EXTENSION, path::PathBuf};
+
+use super::errors::ProxyError;
+
+/// search for Bootstrap in the given path
+pub fn get_bootstrap_path(base_path: &PathBuf) -> Result<PathBuf, ProxyError> {
+    let bootstrap_names = ["melon_bootstrap", "libmelon_bootstrap"]; //by convention, on unix, the library is prefixed with "lib"
+
+    let path = base_path.join("MelonLoader").join("Dependencies");
+
+    for name in bootstrap_names.iter() {
+        let bootstrap_path = path.join(name).with_extension(DLL_EXTENSION);
+
+        if bootstrap_path.exists() {
+            return Ok(bootstrap_path);
+        }
+    }
+
+    Err(ProxyError::BootstrapNotFound(base_path.to_owned()))
+}

Rust/MelonProxy/src/utils/mod.rs

@@ -0,0 +1,3 @@
+pub mod assert;
+pub mod errors;
+pub mod files;