git.delta.rocks / jrsonnet / refs/heads / master

1use std::{2	fs,3	io::{self, Write as _},4	path::PathBuf,5	process,6};78use clap::Parser;9use jrsonnet_formatter::{FormatOptions, format};1011#[derive(Parser)]12#[allow(clippy::struct_excessive_bools)]13struct Opts {14	/// Treat input as code, reformat it instead of reading file.15	#[clap(long, short = 'e')]16	exec: bool,17	/// Path to be reformatted if `--exec` if unset, otherwise code itself.18	input: String,19	/// Replace code with formatted in-place, instead of printing it to stdout.20	/// Only applicable if `--exec` is unset.21	#[clap(long, short = 'i')]22	in_place: bool,2324	/// Exit with error if formatted does not match input25	#[arg(long)]26	test: bool,27	/// Number of spaces to indent with28	#[arg(long, default_value = "4")]29	indent: u8,30	/// Force hard tab for indentation31	#[arg(long, default_value = "true")]32	use_tabs: bool,33	/// Max formatted source width34	#[arg(long, default_value = "100")]35	max_width: u32,3637	/// Debug option: how many times to call reformatting in case of unstable dprint output resolution.38	///39	/// 0 for not retrying to reformat.40	#[arg(long, default_value = "0")]41	conv_limit: usize,42}4344#[derive(thiserror::Error, Debug)]45enum Error {46	#[error("--in-place is incompatible with --exec")]47	InPlaceExec,48	#[error("io: {0}")]49	Io(#[from] io::Error),50	#[error("persist: {0}")]51	Persist(#[from] tempfile::PersistError),52	#[error("parsing failed, refusing to reformat corrupted input")]53	Parse,54}5556fn main_result() -> Result<(), Error> {57	let opts = Opts::parse();58	let input = if opts.exec {59		if opts.in_place {60			return Err(Error::InPlaceExec);61		}62		opts.input.clone()63	} else {64		fs::read_to_string(&opts.input)?65	};6667	let mut iteration = 0;68	let mut formatted = input.clone();69	let mut convergence_tmp;70	// https://github.com/dprint/dprint/pull/42371	loop {72		let reformatted = match format(73			&formatted,74			&FormatOptions {75				indent: opts.indent,76				use_tabs: opts.use_tabs,77				max_width: opts.max_width,78			},79		) {80			Ok(v) => v,81			Err(e) => {82				let snippet = e.build();83				let ansi = hi_doc::source_to_ansi(&snippet);84				eprintln!("{ansi}");85				return Err(Error::Parse);86			}87		};88		convergence_tmp = reformatted.trim().to_owned();89		if formatted == convergence_tmp {90			break;91		}92		formatted = convergence_tmp;93		if opts.conv_limit == 0 {94			break;95		}96		iteration += 1;97		assert!(iteration <= opts.conv_limit, "formatting not converged");98	}99	formatted.push('\n');100	if opts.test && formatted != input {101		process::exit(1);102	}103	if opts.in_place {104		let path = PathBuf::from(opts.input);105		let mut temp = tempfile::NamedTempFile::new_in(path.parent().expect(106			"not failed during read, this path is not a directory, and there is a parent",107		))?;108		temp.write_all(formatted.as_bytes())?;109		temp.flush()?;110		temp.persist(&path)?;111	} else {112		print!("{formatted}");113	}114	Ok(())115}116117fn main() {118	if let Err(e) = main_result() {119		eprintln!("{e}");120		process::exit(1);121	}122}