prettyplease-cli 0.1.0

Unofficial and basic CLI wrapper around the `::prettyplease` library.
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
78
79
80
81
82
83
84
85
86
87
88
89
90

#![doc = include_str!("../README.md")]
#![forbid(unsafe_code)]

#[allow(unused_imports)]
use ::{
    anyhow::{
        anyhow, bail, Error, Result,
    },
    std::{
        fs,
        io::{Read as _, Write as _},
        mem,
        ops::Not as _,
        path::{Path, PathBuf},
    },
};

/// Current workaround to write "comments" that are part of the AST and thus,
/// preserved. Important when testing this tool in a boostrapping fashion.
#[doc(hidden)]
macro_rules! rem {( $($_:tt)* ) => ()}

fn main ()
  -> Result<()>
{
    let ref mut args = ::std::env::args().peekable();
    let ref binary_name = args.next().expect("Missing argv[0]");
    let check_only = match args.peek().map(String::as_str) {
        | Some("--help" | "-h") | None => help(binary_name),
        | Some("--check") => {
            drop(args.next());
            true
        },
        | Some(_thing_else) => false,
    };

    let mut found_diff = false;
    rem! {"TODO: walkdir + ideally in parallel."}
    for ref fname in args.map(PathBuf::from) {
        let ref contents = fs::read_to_string(fname)?;
        let ref ast_contents = ::syn::parse_file(contents)?;
        let ref formatted_code = ::prettyplease::unparse(ast_contents);
        if contents != formatted_code {
            if check_only {
                found_diff = true;
                eprintln!(
                    "\n\n\t>>> Ill-formatted file: `{}` <<<\n",
                    fname.display(),
                );
                #[cfg(feature = "check-shows-diffs")]
                eprintln!(
                    "{}",
                    ::prettydiff::diff_lines(contents, formatted_code),
                );
            } else {
                fs::write(fname, formatted_code)?;
            }
        }
    }
    if found_diff && check_only {
        bail!("Found ill-formatted code.");
    }
    Ok(())
}

#[doc(hidden)]
fn help (binary_name: &'_ str)
  -> !
{
    eprintln! {"\
Unofficial CLI wrapper around `::prettyplease::unparse` to format files and \
trim comments.

This modifies them IN PLACE, unless the `--check` flag is passed.

USAGE:
    {binary_name} [OPTION] <files…>

OPTIONS:
        --check     Runs in 'check' mode. Exits with 0 if input is formatted
                    correctly. Exits with 1 if formatting is required, printing
                    a diff unless the `check-shows-diffs` Cargo feature were
                    disabled at compile-time.
    -h, --help      Shows this very message.

See https://crates.io/crates/prettyplease for more info about the formatting \
itself.\
    "}
    ::std::process::exit(-1);
}