cargo-xfmt 0.0.16

Safely use nightly fmt options without using unstable Rust
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

use std::fs;
use std::io::Write;
use std::path::PathBuf;
use std::process::{self};

use crate::platform::cargo_toml::CargoToml;
use crate::platform::path_ext::path_to_absolute;
use crate::platform::rustfmt_toml::RustfmtToml;

pub struct RustfmtOptions {
    pub check: bool,
    pub files: Vec<PathBuf>,
    pub preserve_config: bool,
    pub additional_args: Vec<String>,
}

pub fn rustfmt(options: RustfmtOptions) -> anyhow::Result<()> {
    for file in options.files.clone() {
        rustfmt_one(file, &options)?;
    }
    Ok(())
}

fn rustfmt_one(
    file: PathBuf,
    options: &RustfmtOptions,
) -> anyhow::Result<()> {
    let cargo_toml = CargoToml::read_from(&file)?;
    let rustfmt_toml = RustfmtToml::read_from(&file)?;

    let mut cmd = process::Command::new("rustfmt");

    if options.check {
        cmd.arg("--check");
    }

    if let Some(edition) = cargo_toml.edition {
        cmd.arg("--edition");
        cmd.arg(edition.clone());
    }

    for (key, value) in rustfmt_toml.iter() {
        cmd.arg("--config");
        cmd.arg(format!("{}={}", key, value));
    }

    for arg in &options.additional_args {
        cmd.arg(arg);
    }

    cmd.stdin(process::Stdio::piped());
    cmd.stdout(process::Stdio::piped());
    cmd.stderr(process::Stdio::inherit());

    if !options.preserve_config {
        rustfmt_toml.hide()?;
    }
    let mut child = cmd.spawn()?;

    let input_path = path_to_absolute(&file)?;
    let input = std::fs::read_to_string(&input_path)?;

    let Some(mut stdin) = child.stdin.take() else {
        return Err(anyhow::anyhow!("Unable to get stdin for rustfmt"));
    };

    stdin.write_all(input.as_bytes())?;
    drop(stdin);

    let result = child.wait_with_output()?;
    if result.status.success() {
        fs::write(&input_path, result.stdout)?;
    }

    rustfmt_toml.unhide()?;
    std::process::exit(result.status.code().unwrap_or(1))
}