genie 0.5.0

Libraries for reading/writing Age of Empires II data files
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use genie::scx::{convert::AutoToWK, VersionBundle};
use genie::Scenario;
use std::{fs::File, path::PathBuf};
use structopt::StructOpt;

/// Convert Age of Empires scenario files between versions.
#[derive(Debug, StructOpt)]
struct Cli {
    /// Input scenario file.
    #[structopt(parse(from_os_str))]
    input: PathBuf,
    /// Output scenario file.
    #[structopt(parse(from_os_str))]
    output: PathBuf,
    /// Scenario version to output: 'aoe', 'ror', 'aoc', 'hd', 'wk'
    ///
    /// When setting the version to 'wk', HD edition and AoC scenarios will automatically be
    /// converted (swapping out unit types and terrains).
    version: Option<String>,
}

fn main() -> anyhow::Result<()> {
    let Cli {
        input,
        output,
        version,
    } = Cli::from_args();
    let version_arg = version;
    let version = match version_arg.as_ref().map(|s| &**s) {
        Some("aoe") => VersionBundle::aoe(),
        Some("ror") => VersionBundle::ror(),
        Some("aoc") => VersionBundle::aoc(),
        Some("hd") => VersionBundle::hd_edition(),
        Some("wk") => VersionBundle::userpatch_15(),
        Some(name) => panic!("unknown version {}", name),
        _ => VersionBundle::aoc(),
    };

    let instream = File::open(input)?;
    let mut scen = Scenario::read_from(instream)?;

    if version_arg == Some("wk".to_string()) {
        println!("Applying WololoKingdoms conversion...");
        let converter = AutoToWK::default();
        converter.convert(&mut scen)?;
    }

    let outstream = File::create(output)?;
    scen.write_to_version(outstream, &version)?;

    println!("Conversion complete!");

    Ok(())
}