overlays! simple

Cargo.lock

@@ -1166,7 +1166,7 @@ checksum = "81e544489bf3d8ef66c953931f56617f423cd4b5494be343d9b9d3dda037b9a3"
 
 [[package]]
 name = "untitled-minecraft-launcher"
-version = "0.1.0"
+version = "0.4.0"
 dependencies = [
  "anyhow",
  "clap",

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "untitled-minecraft-launcher"
-version = "0.1.0"
+version = "0.4.0"
 edition = "2024"
 
 [dependencies]

src/auth.rs

@@ -129,7 +129,7 @@ fn poll(agent: &Agent, device: &Device) -> anyhow::Result<(String, String)> {
 
         if let Some(token) = body.access_token {
             let refresh = body.refresh_token.ok_or_else(|| {
-                anyhow::anyhow!("no refresh token — was offline_access in the scope?")
+                anyhow::anyhow!("no refresh token, was offline_access in the scope?")
             })?;
             return Ok((token, refresh));
         }
@@ -176,10 +176,10 @@ fn xsts(agent: &Agent, xbl_token: &str) -> anyhow::Result<XblResponse> {
         anyhow::bail!(
             "{}",
             match e.xerr {
-                2148916233 => "no Xbox profile — sign in at minecraft.net once".into(),
+                2148916233 => "no Xbox profile, sign in at minecraft.net once".into(),
                 2148916235 => "Xbox Live unavailable in this region".into(),
                 2148916237 => "account needs adult verification".into(),
-                2148916238 => "child account — must be added to a Family".into(),
+                2148916238 => "child account, must be added to a Family".into(),
                 other => format!("XSTS failed, XErr {other}"),
             }
         );

src/download.rs

@@ -10,7 +10,9 @@ pub fn download(url: &str, path: &Path) -> anyhow::Result<()> {
     if let Some(parent) = path.parent() {
         create_dir_all(parent)?;
     }
-    let mut res = get(url).call()?;
+    let mut res = get(url)
+        .header("User-Agent", crate::util::USER_AGENT)
+        .call()?;
     let mut file = File::create(path)?;
     copy(&mut res.body_mut().as_reader(), &mut file)?;
     Ok(())

src/main.rs

@@ -8,15 +8,16 @@ mod fabric;
 mod instance;
 mod launch;
 mod meta;
+mod modrinth;
 mod mrpack;
+mod overlay;
 mod prism;
 mod util;
 use clap::{Parser, Subcommand};
 use launch::run;
 use meta::fetch_version;
 use std::path::{Path, PathBuf};
-
-use crate::util::{is_valid_name, open_path};
+use util::{is_valid_name, open_path};
 
 /// Untitled Minecraft Launcher
 #[derive(Parser)]
@@ -50,7 +51,12 @@ enum ModVerb {
         item: String,
     },
 }
-
+#[derive(Subcommand)]
+enum OverlayCmd {
+    /// Apply an overlay
+    Apply { overlay: String, instance: String },
+    // add List, Create, Add, Remove, Unapply, Strip
+}
 #[derive(Subcommand)]
 enum Command {
     /// Launch an instance
@@ -68,6 +74,11 @@ enum Command {
         #[command(subcommand)]
         action: InstanceAction,
     },
+    // Manage overlays
+    Overlays {
+        #[command(subcommand)]
+        cmd: OverlayCmd,
+    },
     /// Manage instances
     Instances {
         #[command(subcommand)]
@@ -78,6 +89,7 @@ enum Command {
 }
 #[derive(Subcommand)]
 enum InstanceAction {
+    /// Manage instance mods
     Mods {
         #[command(subcommand)]
         verb: ModVerb,
@@ -142,8 +154,8 @@ fn data_dir() -> anyhow::Result<PathBuf> {
 fn main() -> anyhow::Result<()> {
     let uml_dir = data_dir()?;
     let instances_dir = uml_dir.join("instances");
+    let overlays_dir = uml_dir.join("overlays");
     let shared_dir = uml_dir.join("shared");
-
     let cli = Cli::parse();
     match cli.command {
         Command::Folder { name } => {
@@ -288,6 +300,26 @@ fn main() -> anyhow::Result<()> {
                 },
             }
         }
+        Command::Overlays { cmd } => match cmd {
+            OverlayCmd::Apply { overlay, instance } => {
+                if !is_valid_name(&instance) {
+                    anyhow::bail!("invalid instance name: {instance:?}");
+                }
+                if !is_valid_name(&overlay) {
+                    anyhow::bail!("invalid overlay name: {overlay:?}");
+                }
+                let mods_dir = instances_dir.join(&instance).join("mods");
+                let (cfg, _) = instance::load(&instances_dir, &instance)?;
+                let loader = cfg
+                    .loader
+                    .as_ref()
+                    .map(|l| l.kind.as_str())
+                    .ok_or_else(|| {
+                        anyhow::anyhow!("instance has no loader, overlays need a modded instance")
+                    })?;
+                overlay::apply(&overlays_dir, &overlay, &mods_dir, loader, &cfg.version)?;
+            }
+        },
     }
     Ok(())
 }

src/modrinth.rs

@@ -0,0 +1,62 @@
+use std::path::Path;
+
+use crate::{
+    download::{self, verify_sha512},
+    util::Hashes,
+};
+use serde::Deserialize;
+
+const USER_AGENT: &str = "owenthepuppy/uml/0.3.0";
+
+#[derive(Deserialize)]
+pub struct Version {
+    pub files: Vec<VersionFile>,
+}
+#[derive(Deserialize, Clone)]
+pub struct VersionFile {
+    pub url: String,
+    pub filename: String,
+    pub hashes: Hashes,
+    pub primary: bool,
+}
+pub fn get_versions(slug: &str, loader: &str, game_version: &str) -> anyhow::Result<Vec<Version>> {
+    let versions = match ureq::get(format!(
+        "https://api.modrinth.com/v2/project/{slug}/version"
+    ))
+    .header("User-Agent", USER_AGENT)
+    .query("loaders", format!("[\"{loader}\"]"))
+    .query("game_versions", format!("[\"{game_version}\"]"))
+    .call()
+    {
+        Ok(mut resp) => resp.body_mut().read_json()?,
+        Err(ureq::Error::StatusCode(404)) => {
+            anyhow::bail!("mod not found: {slug}");
+        }
+        Err(e) => return Err(e.into()),
+    };
+    Ok(versions)
+}
+pub fn get_version(slug: &str, loader: &str, game_version: &str) -> anyhow::Result<VersionFile> {
+    let mod_versions = get_versions(slug, loader, game_version)?;
+    let latest_version = mod_versions
+        .first()
+        .ok_or_else(|| anyhow::anyhow!("no compatible version for {slug}"))?;
+    let file = latest_version
+        .files
+        .iter()
+        .find(|f| f.primary)
+        .or_else(|| latest_version.files.first())
+        .ok_or_else(|| anyhow::anyhow!("version has no files"))?;
+    Ok(file.clone())
+}
+pub fn download_mod(file: &VersionFile, mods_dir: &Path) -> anyhow::Result<()> {
+    let dest = mods_dir.join(&file.filename);
+    download::download(&file.url, &dest)?;
+    if !verify_sha512(&dest, &file.hashes.sha512)? {
+        anyhow::bail!(
+            "hash mismatch after download: {:?}. This file is NOT DELETED AND WILL LOAD ON NEXT LAUNCH. DO NOT LAUNCH MINECRAFT, until you are SURE this is safe.",
+            dest
+        );
+    }
+    Ok(())
+}

src/mrpack.rs

@@ -1,7 +1,7 @@
 use crate::{
     download::{download, verify_sha512},
     instance::{Loader, create, load, save},
-    util::is_safe_relative_path,
+    util::{Hashes, is_safe_relative_path},
 };
 use indicatif::{ProgressBar, ProgressStyle};
 use serde::Deserialize;
@@ -26,10 +26,6 @@ struct PackFile {
     #[serde(default)]
     env: Option<Env>,
 }
-#[derive(Deserialize)]
-struct Hashes {
-    sha512: String,
-}
 
 #[derive(Deserialize)]
 struct Env {
@@ -157,7 +153,10 @@ pub fn import(
                 .ok_or_else(|| anyhow::anyhow!("no download url for {}", f.path))?;
             download(url, &dest)?;
             if !verify_sha512(&dest, &f.hashes.sha512)? {
-                anyhow::bail!("hash mismatch after download: {}", f.path);
+                anyhow::bail!(
+                    "hash mismatch after download: {}. This file is NOT DELETED AND WILL LOAD ON NEXT LAUNCH. DO NOT LAUNCH MINECRAFT, until you are SURE this is safe.",
+                    f.path
+                );
             }
         }
         pb.inc(1);

src/overlay.rs

@@ -0,0 +1,70 @@
+use std::path::Path;
+
+use serde::{Deserialize, Serialize};
+
+use crate::{content, modrinth};
+use indicatif::{ProgressBar, ProgressStyle};
+
+#[derive(Serialize, Deserialize)]
+pub struct Overlay {
+    pub name: String,
+    pub mods: Vec<OverlayMod>,
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct OverlayMod {
+    pub source: String,
+    pub id: String,
+    #[serde(default)]
+    pub pin: Option<String>, // NOT IMPLEMENTED YET
+}
+pub fn load(overlays_dir: &Path, name: &str) -> anyhow::Result<Overlay> {
+    let path = overlays_dir.join(format!("{name}.umloverlay"));
+    if !path.exists() {
+        anyhow::bail!("no overlay named {name}");
+    }
+    let json = std::fs::read_to_string(path)?;
+    Ok(serde_json::from_str(&json)?)
+}
+
+pub fn save(overlays_dir: &Path, overlay: &Overlay) -> anyhow::Result<()> {
+    let path = overlays_dir.join(format!("{}.umloverlay", overlay.name));
+    let json = serde_json::to_string_pretty(overlay)?;
+    std::fs::write(path, json)?;
+    Ok(())
+}
+pub fn apply(
+    overlays_dir: &Path,
+    name: &str,
+    mods_dir: &Path,
+    loader: &str,
+    game_version: &str,
+) -> anyhow::Result<()> {
+    let overlay = load(overlays_dir, name)?;
+    let existing = content::list(mods_dir)?;
+    let all_present = overlay
+        .mods
+        .iter()
+        .all(|m| existing.iter().any(|e| e.id == m.id));
+    if all_present {
+        println!(
+            "Overlay '{}' is already fully applied, nothing to do.",
+            overlay.name
+        );
+        return Ok(());
+    }
+    let pb = ProgressBar::new(overlay.mods.len() as u64);
+    pb.set_style(ProgressStyle::with_template("{bar:40} {pos}/{len} {msg}").unwrap());
+    for m in overlay.mods {
+        if existing.iter().any(|e| e.id == m.id) {
+            println!("Skipping {} (already installed).", m.id);
+            continue;
+        }
+        pb.println(format!("Downloading {}...", m.id)); // not println!
+        let file = modrinth::get_version(&m.id, loader, game_version)?;
+        modrinth::download_mod(&file, &mods_dir)?;
+        pb.inc(1);
+    }
+    pb.finish_with_message("done");
+    Ok(())
+}

src/util.rs

@@ -1,5 +1,9 @@
 use std::path::Path;
 
+use serde::Deserialize;
+
+pub const USER_AGENT: &str = concat!("owenthepuppy/uml/", env!("CARGO_PKG_VERSION"));
+
 pub fn is_valid_name(name: &str) -> bool {
     let mut components = Path::new(name).components();
     matches!(
@@ -17,3 +21,7 @@ pub fn open_path(path: &Path) -> anyhow::Result<()> {
     std::process::Command::new("xdg-open").arg(path).spawn()?;
     Ok(())
 }
+#[derive(Deserialize, Clone)]
+pub struct Hashes {
+    pub sha512: String,
+}