z_mirror/fmm

git clone https://git.raiguard.me/z_mirror/fmm

raiguardRead mod-settings.dat (causing Rust ICE)216c7c72021-12-21

#![feature(iter_intersperse)]

use std::fs;
use std::path::PathBuf;

use anyhow::Result;
use console::style;
use reqwest::blocking::Client;
use semver::Version;
use structopt::StructOpt;

mod config;
mod dependency;
mod directory;
mod download;
mod read;
mod sync;
mod types;

use config::*;
use directory::*;
use types::*;

#[derive(Clone, StructOpt)]
#[structopt(name = "fmm", about = "Manage your Factorio mods.")]
pub struct App {
    /// The path to the configuration file
    #[structopt(long)]
    config: Option<PathBuf>,
    /// Disables all mods in the directory
    #[structopt(short = "o", long)]
    disable_all: bool,
    /// Disables the given mods. Mods are formatted as `Name`
    #[structopt(short, long)]
    disable: Vec<ModIdent>,
    /// Enables all mods in the directory
    #[structopt(short = "a", long)]
    enable_all: bool,
    /// Enables the mods in the given set
    #[structopt(short = "E", long)]
    enable_set: Option<String>,
    /// Enables the given mods. Mods are formatted as `Name` or `Name@Version`
    #[structopt(short, long)]
    enable: Vec<ModIdent>,
    /// The game directory to manipulate. Optional if a configuration file is in use
    #[structopt(long)]
    game_dir: Option<PathBuf>,
    /// Lists all mods in the directory
    #[structopt(short, long)]
    list: bool,
    /// The mods directory to manipulate. Optional if a configuration file is in use
    #[structopt(long)]
    mods_dir: Option<PathBuf>,
    /// Removes the given mods from the mods directory. Mods are formatted as `Name` or `Name@Version`
    #[structopt(short, long)]
    remove: Vec<ModIdent>,
    /// A path to a save file to sync with
    #[structopt(short, long)]
    sync: Option<PathBuf>,
}

// DESIGN NOTES:
// - Get a list of all mods + versions in the folder _without_ reading the ZIP files (use filenames)
// - Only read ZIPs if we need to get dependencies or other info
// - Cache will only be used once we have advanced features that would benefit from it

fn main() -> Result<()> {
    let mut app = App::from_args();
    let client = Client::new();

    let config = Config::new(&app)?;

    let mut directory = Directory::new(config.mods_dir.clone())?;

    // List mods
    if app.list {
        let mut lines: Vec<String> = directory
            .mods
            .iter()
            .flat_map(|(mod_name, mod_versions)| {
                mod_versions
                    .iter()
                    .map(|version| format!("{} v{}", mod_name, version.version))
                    .collect::<Vec<String>>()
            })
            .collect();

        lines.sort();

        for line in lines {
            println!("{}", line)
        }
    }

    // Combine enable commands
    let mut combined_enable = vec![];
    // Enable set
    if let Some(set_name) = app.enable_set {
        if let Some(sets) = config.sets.as_ref() {
            if let Some(set) = sets.get(&set_name) {
                combined_enable.append(&mut set.to_vec());
            }
        }
    }
    // Sync with save
    if let Some(sync_path) = app.sync {
        let save_file = sync::SaveFile::from(sync_path)?;

        let mut mods = save_file.mods.to_vec();

        if config.sync_latest_versions {
            for mod_ident in mods.iter_mut() {
                mod_ident.version_req = None;
            }
        }

        combined_enable.append(&mut mods);
    }
    // Manually enable
    combined_enable.append(&mut app.enable.to_vec());
    app.enable = combined_enable;

    // Remove specified mods
    for mod_ident in app.remove {
        if mod_ident.name != "base" {
            directory.remove(&mod_ident);
        }
    }

    // Disable all mods
    if app.disable_all {
        directory.disable_all();
    }

    // Disable specified mods
    for mod_ident in app.disable {
        directory.disable(&mod_ident);
    }

    // Enable all mods
    if app.enable_all {
        directory.enable_all();
    }

    // Enable and/or download specified mods
    let mut cycle_orders: Vec<ManageOrder> = app
        .enable
        .iter()
        .map(|mod_ident| ManageOrder::Enable(mod_ident.clone()))
        .collect();

    while !cycle_orders.is_empty() {
        cycle_orders = cycle_orders
            .iter_mut()
            .filter(|order| order.get_name() != "base")
            .filter_map(|order| match order {
                ManageOrder::Download(mod_ident) => {
                    if config.auto_download {
                        if download::download_mod(mod_ident, &mut directory, &config, &client)
                            .ok()?
                        {
                            Some(vec![ManageOrder::Enable(mod_ident.clone())])
                        } else {
                            None
                        }
                    } else {
                        println!("{} {}", style("Did not download").red(), mod_ident.name);
                        None
                    }
                }
                ManageOrder::Enable(mod_ident) => directory.enable(mod_ident),
            })
            .flatten()
            .collect();
    }

    // Write mod-list.json
    fs::write(
        &directory.mod_list_path,
        serde_json::to_string_pretty(&ModListJson {
            mods: directory.mod_list,
        })?,
    )?;

    Ok(())
}

trait HasVersion {
    fn get_version(&self) -> &Version;
}

fn get_mod_version<'a, T>(list: &'a [T], mod_ident: &ModIdent) -> Option<&'a T>
where
    T: HasVersion,
{
    if let Some(version_req) = &mod_ident.version_req {
        list.iter()
            .rev()
            .find(|entry| version_req.matches(entry.get_version()))
    } else {
        list.last()
    }
}