use std::io::Write;
use std::rc::Rc;
use crate::build::Val;
use crate::convert::traits::{ConvertResult, Converter};
use crate::error::{BuildError, ErrorType};
pub struct FlagConverter {
sep: &'static str,
}
impl FlagConverter {
pub fn new() -> Self {
FlagConverter { sep: "." }
}
pub fn with_sep(mut self, sep: &'static str) -> Self {
self.sep = sep;
self
}
fn write_flag_name(&self, pfx: &str, name: &str, w: &mut dyn Write) -> ConvertResult {
if name.chars().count() > 1 || pfx.chars().count() > 0 {
write!(w, "--{}{} ", pfx, name)?;
} else {
write!(w, "-{} ", name)?;
}
return Ok(());
}
fn write_list_flag(
&self,
pfx: &str,
name: &str,
def: &Vec<Rc<Val>>,
w: &mut dyn Write,
) -> ConvertResult {
for v in def.iter() {
let vref = v.as_ref();
if vref.is_list() || vref.is_tuple() {
eprintln!(
"Skipping non primitive val in list for flag {}{}",
pfx, name
);
} else {
self.write_flag_name(pfx, name, w)?;
self.write_simple_value(vref, w)?;
}
}
return Ok(());
}
fn write_simple_value(&self, v: &Val, w: &mut dyn Write) -> ConvertResult {
match v {
&Val::Empty => {
return Ok(());
}
&Val::Boolean(b) => {
write!(w, "{} ", if b { "true" } else { "false" })?;
}
&Val::Float(ref f) => {
write!(w, "{} ", f)?;
}
&Val::Int(ref i) => {
write!(w, "{} ", i)?;
}
&Val::Str(ref s) => {
write!(w, "'{}' ", s)?;
}
&Val::List(_) | &Val::Tuple(_) | &Val::Env(_) => {
eprintln!("Skipping {}...", v.type_name());
}
}
Ok(())
}
fn write(&self, pfx: &str, flds: &Vec<(Rc<str>, Rc<Val>)>, w: &mut dyn Write) -> ConvertResult {
for &(ref name, ref val) in flds.iter() {
if let &Val::Empty = val.as_ref() {
self.write_flag_name(pfx, name, w)?;
continue;
}
match val.as_ref() {
&Val::Tuple(_) | &Val::Env(_) => {
eprintln!("Skipping {} in flag output tuple.", val.type_name());
}
&Val::List(ref def) => {
self.write_list_flag(pfx, name, def, w)?;
}
&Val::Boolean(_) | &Val::Empty | &Val::Float(_) | &Val::Int(_) | &Val::Str(_) => {
self.write_flag_name(pfx, name, w)?;
self.write_simple_value(val, w)?;
}
}
}
Ok(())
}
}
impl Converter for FlagConverter {
fn convert(&self, v: Rc<Val>, mut w: &mut dyn Write) -> ConvertResult {
if let &Val::Tuple(ref flds) = v.as_ref() {
self.write("", flds, &mut w)
} else {
return Err(Box::new(BuildError::new(
"Flag outputs must be a tuple",
ErrorType::ConvertError,
)));
}
}
fn file_ext(&self) -> String {
String::from("txt")
}
fn description(&self) -> String {
"Convert ucg Vals into command line flags.".to_string()
}
#[allow(unused_must_use)]
fn help(&self) -> String {
include_str!("flags_help.txt").to_string()
}
}