1pub mod location;23use clap::Clap;4use jsonnet_evaluator::{EvaluationState, LocError, StackTrace, Val};5use location::{offset_to_location, CodeLocation};6use std::env::current_dir;7use std::str::FromStr;89enum Format {10 None,11 Json,12 Yaml,13}1415impl FromStr for Format {16 type Err = &'static str;17 fn from_str(s: &str) -> Result<Self, Self::Err> {18 Ok(match s {19 "none" => Format::None,20 "json" => Format::Json,21 "yaml" => Format::Yaml,22 _ => return Err("no such format"),23 })24 }25}2627#[derive(PartialEq)]28enum TraceFormat {29 CppJsonnet,30 GoJsonnet,31 Custom,32}33impl FromStr for TraceFormat {34 type Err = &'static str;35 fn from_str(s: &str) -> Result<Self, Self::Err> {36 Ok(match s {37 "cpp" => TraceFormat::CppJsonnet,38 "go" => TraceFormat::GoJsonnet,39 "default" => TraceFormat::Custom,40 _ => return Err("no such format"),41 })42 }43}4445#[derive(Clap)]46#[clap(version = "0.1.0", author = "Lach <iam@lach.pw>")]47struct Opts {48 #[clap(long, about = "Disable global std variable")]49 no_stdlib: bool,50 #[clap(long, about = "Add external string")]51 ext_str: Option<Vec<String>>,52 #[clap(long, about = "Add external string from code")]53 ext_code: Option<Vec<String>>,54 #[clap(long, about = "Add TLA")]55 tla_str: Option<Vec<String>>,56 #[clap(long, about = "Add TLA from code")]57 tla_code: Option<Vec<String>>,58 #[clap(long, short = "f", default_value = "json", possible_values = &["none", "json", "yaml"], about = "Output format, wraps resulting value to corresponding std.manifest call")]59 format: Format,60 #[clap(long, default_value = "default", possible_values = &["cpp", "go", "default"], about = "Emulated needed stacktrace display")]61 trace_format: TraceFormat,6263 #[clap(64 long,65 short = "s",66 default_value = "200",67 about = "Number of allowed stack frames"68 )]69 max_stack: usize,70 #[clap(71 long,72 short = "t",73 default_value = "20",74 about = "Max length of stack trace before cropping"75 )]76 max_trace: usize,7778 #[clap(about = "File to compile", index = 1)]79 input: String,80}8182fn main() {83 let opts: Opts = Opts::parse();84 let evaluator = jsonnet_evaluator::EvaluationState::default();85 if !opts.no_stdlib {86 evaluator.add_stdlib();87 }88 let mut input = current_dir().unwrap();89 input.push(opts.input.clone());90 evaluator91 .add_file(92 input.clone(),93 String::from_utf8(std::fs::read(opts.input.clone()).unwrap()).unwrap(),94 )95 .unwrap();96 let result = evaluator.evaluate_file(&input);97 match result {98 Ok(v) => {99 let v = match opts.format {100 Format::Json => {101 if opts.no_stdlib {102 evaluator.add_stdlib();103 }104 evaluator.add_global("__tmp__to_json__".to_owned(), v);105 let v = evaluator106 .parse_evaluate_raw("std.manifestJsonEx(__tmp__to_json__, \" \")");107 match v {108 Ok(v) => v,109 Err(err) => {110 print_error(&err, evaluator, &opts);111 std::process::exit(2);112 }113 }114 }115 Format::Yaml => {116 if opts.no_stdlib {117 evaluator.add_stdlib();118 }119 evaluator.add_global("__tmp__to_yaml__".to_owned(), v);120 let v = evaluator121 .parse_evaluate_raw("std.manifestYamlDoc(__tmp__to_yaml__, \" \")");122 match v {123 Ok(v) => v,124 Err(err) => {125 print_error(&err, evaluator, &opts);126 std::process::exit(2);127 }128 }129 }130 _ => v,131 };132 match v {133 Val::Str(s) => println!("{}", s),134 Val::Num(n) => println!("{}", n),135 _v => eprintln!(136 "jsonnet output is not a string.\nDid you forgot to set --format, or wrap your data with std.manifestJson?"137 ),138 }139 }140 Err(err) => {141 print_error(&err, evaluator, &opts);142 }143 }144}145146fn print_error(err: &LocError, evaluator: EvaluationState, opts: &Opts) {147 println!("Error: {:?}", err.0);148 print_trace(&(err.1), evaluator, &opts);149}150151fn print_trace(trace: &StackTrace, evaluator: EvaluationState, opts: &Opts) {152 use annotate_snippets::{153 display_list::{DisplayList, FormatOptions},154 snippet::{Annotation, AnnotationType, Slice, Snippet, SourceAnnotation},155 };156 for item in trace.0.iter() {157 let desc = &item.1;158 if (item.0).1.is_none() {159 continue;160 }161 let source = (item.0).1.clone().unwrap();162 let code = evaluator.get_source(&source.0);163 if code.is_none() {164 continue;165 }166 let code = code.unwrap();167 let start_end = offset_to_location(&code, &[source.1, source.2]);168 if opts.trace_format == TraceFormat::Custom {169 let source_fragment: String = code170 .chars()171 .skip(start_end[0].line_start_offset)172 .take(start_end[1].line_end_offset - start_end[0].line_start_offset)173 .collect();174 let snippet = Snippet {175 opt: FormatOptions {176 color: true,177 ..Default::default()178 },179 title: Some(Annotation {180 label: Some(&item.1),181 id: None,182 annotation_type: AnnotationType::Error,183 }),184 footer: vec![],185 slices: vec![Slice {186 source: &source_fragment,187 line_start: start_end[0].line,188 origin: Some(&source.0.to_str().unwrap()),189 fold: false,190 annotations: vec![SourceAnnotation {191 label: desc,192 annotation_type: AnnotationType::Error,193 range: (194 source.1 - start_end[0].line_start_offset,195 source.2 - start_end[0].line_start_offset,196 ),197 }],198 }],199 };200201 let dl = DisplayList::from(snippet);202 println!("{}", dl);203 } else {204 print_jsonnet_pair(205 source.0.to_str().unwrap(),206 &start_end[0],207 &start_end[1],208 opts.trace_format == TraceFormat::GoJsonnet,209 );210 }211 }212}213214fn print_jsonnet_pair(file: &str, start: &CodeLocation, end: &CodeLocation, is_go: bool) {215 if is_go {216 print!(" ");217 } else {218 print!(" ");219 }220 print!("{}:", file);221 if start.line == end.line {222 223 if start.column == end.column || !is_go && start.column + 1 == end.column {224 println!("{}:{}", start.line, end.column)225 } else {226 println!("{}:{}-{}", start.line, start.column, end.column);227 }228 } else {229 println!(230 "({}:{})-({}:{})",231 start.line, end.column, start.line, end.column232 );233 }234}