factorio-exporter 0.1.0

A library to export prototype definitions from Factorio (http://www.factorio.com)
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
55

use std::{fs, path::PathBuf};

use clap::Parser;
use factorio_exporter::{load_api, ApiGenerator, Result};
use rust_format::{Config, Edition, Formatter, PostProcess, RustFmt};

/// Generates Rust struct definitions that can be deserialized from Factorio
/// rulesets.
#[derive(Parser, Debug)]
#[command(version)]
struct Args {
    /// Directory where Factorio is installed. This needs to be the full
    /// version. Neither the demo nor the headless version are sufficient
    #[arg(long)]
    factorio_dir: PathBuf,

    /// Output path for the generated struct definitions. Will use STDOUT if
    /// omitted
    #[arg(long, short)]
    output: Option<PathBuf>,

    /// Whether to format the generated code with `rustfmt`
    #[arg(long, short)]
    format: bool,
}

pub fn main() -> Result<()> {
    tracing_subscriber::fmt::init();

    let args = Args::parse();

    let api = load_api(&args.factorio_dir)?;
    let mut code = ApiGenerator::new(&api).generate_api();

    if args.format {
        code = RustFmt::from_config(
            Config::new_str()
                .edition(Edition::Rust2021)
                .post_proc(PostProcess::ReplaceMarkers)
                .option("normalize_doc_attributes", "true")
                .option("wrap_comments", "true")
                .option("comment_width", "80"),
        )
        .format_str(&code)
        .unwrap();
    }

    if let Some(path) = args.output {
        fs::write(path, code)?;
    } else {
        println!("{}", code);
    }

    Ok(())
}