`d37d57f47273`

1 file changed

modifiedcmds/jsonnet/src/main.rsdiff before after both

1use clap::Clap;2use jsonnet_evaluator::Val;34#[derive(Clap)]5#[clap(version = "0.1.0", author = "Lach <iam@lach.pw>")]6struct Opts {7	#[clap(about = "File to compile")]8	input: String,9	#[clap(long, about = "Disable global std variable")]10	no_stdlib: bool,11}1213fn main() {14	let opts: Opts = Opts::parse();15	let evaluator = jsonnet_evaluator::EvaluationState::default();16	if !opts.no_stdlib {17		evaluator.add_stdlib();18	}19	evaluator20		.add_file(21			opts.input.clone(),22			String::from_utf8(std::fs::read(opts.input.clone()).unwrap()).unwrap(),23		)24		.unwrap();25	let result = evaluator.evaluate_file(&opts.input);26	match result {27		Ok(v) => match v {28			Val::Str(s) => println!("{}", s),29			Val::Num(n) => println!("{}", n),30			_v => eprintln!(31				"jsonnet output is not a string.\nDid you forgot to set --format, or wrap your data with std.manifestJson?"32			),33		},34		Err(err) => {35			println!("Error: {:?}", err.0);36			use annotate_snippets::{37				display_list::{DisplayList, FormatOptions},38				snippet::{Annotation, AnnotationType, Slice, Snippet, SourceAnnotation},39			};40			for item in (err.1).0.iter() {41				let desc = &item.1;42				if (item.0).1.is_none() {43					continue;44				}45				let source = (item.0).1.clone().unwrap();46				let code = evaluator.get_source(&source.0);47				let snippet = Snippet {48					opt: FormatOptions {49						color: true,50						..Default::default()51					},52					title: Some(Annotation {53						label: Some(&item.1),54						id: None,55						annotation_type: AnnotationType::Error,56					}),57					footer: vec![],58					slices: vec![Slice {59						source: &code,60						line_start: 1,61						origin: Some(&source.0),62						fold: true,63						annotations: vec![SourceAnnotation {64							label: desc,65							annotation_type: AnnotationType::Error,66							range: (source.1, source.2),67						}],68					}],69				};7071				let dl = DisplayList::from(snippet);72				println!("{}", dl);73			}74		}75	}76}

1use clap::Clap;2use jsonnet_evaluator::Val;3use std::str::FromStr;45enum Format {6	None,7	Json,8}910impl FromStr for Format {11	type Err = &'static str;12	fn from_str(s: &str) -> Result<Self, Self::Err> {13		Ok(match s {14			"none" => Format::None,15			"json" => Format::Json,16			_ => return Err("no such format"),17		})18	}19}2021#[derive(Clap)]22#[clap(version = "0.1.0", author = "Lach <iam@lach.pw>")]23struct Opts {24	#[clap(about = "File to compile")]25	input: String,26	#[clap(long, about = "Disable global std variable")]27	no_stdlib: bool,28	#[clap(long, short = "f", default_value = "none", possible_values = &["none", "json"])]29	format: Format,30}3132fn main() {33	let opts: Opts = Opts::parse();34	let evaluator = jsonnet_evaluator::EvaluationState::default();35	if !opts.no_stdlib {36		evaluator.add_stdlib();37	}38	evaluator39		.add_file(40			opts.input.clone(),41			String::from_utf8(std::fs::read(opts.input.clone()).unwrap()).unwrap(),42		)43		.unwrap();44	let result = evaluator.evaluate_file(&opts.input);45	match result {46		Ok(mut v) => {47			if let Format::Json = opts.format {48				if opts.no_stdlib {49					evaluator.add_stdlib();50				}51				evaluator.add_global("__tmp__to_json__".to_owned(), v);52				v = evaluator53					.parse_evaluate_raw("std.manifestJson(__tmp__to_json__)")54					.expect("json serialization");55			}56			match v {57			Val::Str(s) => println!("{}", s),58			Val::Num(n) => println!("{}", n),59			_v => eprintln!(60				"jsonnet output is not a string.\nDid you forgot to set --format, or wrap your data with std.manifestJson?"61			),62		}63		}64		Err(err) => {65			println!("Error: {:?}", err.0);66			use annotate_snippets::{67				display_list::{DisplayList, FormatOptions},68				snippet::{Annotation, AnnotationType, Slice, Snippet, SourceAnnotation},69			};70			for item in (err.1).0.iter() {71				let desc = &item.1;72				if (item.0).1.is_none() {73					continue;74				}75				let source = (item.0).1.clone().unwrap();76				let code = evaluator.get_source(&source.0);77				let snippet = Snippet {78					opt: FormatOptions {79						color: true,80						..Default::default()81					},82					title: Some(Annotation {83						label: Some(&item.1),84						id: None,85						annotation_type: AnnotationType::Error,86					}),87					footer: vec![],88					slices: vec![Slice {89						source: &code,90						line_start: 1,91						origin: Some(&source.0),92						fold: true,93						annotations: vec![SourceAnnotation {94							label: desc,95							annotation_type: AnnotationType::Error,96							range: (source.1, source.2),97						}],98					}],99				};100101				let dl = DisplayList::from(snippet);102				println!("{}", dl);103			}104		}105	}106}