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

use std::io::Read;
use std::io::Write;
use std::process::{self};

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

pub struct RustfmtStdioOptions {
    pub check: bool,
    pub preserve_config: bool,
    pub additional_args: Vec<String>,
}

pub fn rustfmt_stdio(options: RustfmtStdioOptions) -> anyhow::Result<()> {
    let cargo_toml = CargoToml::read_from(&std::env::current_dir()?)?;
    let rustfmt_toml = RustfmtToml::read_from(&std::env::current_dir()?)?;

    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::inherit());
    cmd.stderr(process::Stdio::inherit());

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

    let input = {
        let mut stdin = std::io::stdin();
        let mut buf = vec![];
        stdin.read_to_end(&mut buf)?;
        String::from_utf8(buf)?
    };

    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 status = child.wait()?;

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