Add an environment variable output target.

This commit is contained in:
Jeremy Wall 2018-02-12 22:47:42 -06:00
parent 30c9bc64b2
commit 5a6737d858
4 changed files with 101 additions and 4 deletions

1
examples/env.txt Normal file
View File

@ -0,0 +1 @@
dbconn_list=db_conn1='db1.prod.net:3306/testdb' db_conn2='db2.prod.net:3306/testdb' tmpldir='./templates'

86
src/convert/env.rs Normal file
View File

@ -0,0 +1,86 @@
// Copyright 2017 Jeremy Wall <jeremy@marzhillstudios.com>
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//! An environment variable converter.
use std::rc::Rc;
use std::io::Write;
use std::io::Result;
use ast;
use build::Val;
use convert::traits::Converter;
/// EnvConverter implements the conversion logic for converting a Val into a set of environment variables.
pub struct EnvConverter {}
impl EnvConverter {
pub fn new() -> Self {
EnvConverter {}
}
fn convert_tuple(
&self,
flds: &Vec<(ast::Positioned<String>, Rc<Val>)>,
w: &mut Write,
) -> Result<()> {
for &(ref name, ref val) in flds.iter() {
// TODO(jwall): What if the value is a tuple?
if val.is_tuple() {
eprintln!("Skipping embedded tuple...");
return Ok(());
}
try!(write!(w, "{}=", name.val));
try!(self.write(&val, w));
}
Ok(())
}
fn convert_list(&self, _items: &Vec<Rc<Val>>, _w: &mut Write) -> Result<()> {
// FIXME(jwall): Fill this in?
// TODO(jwall)
eprintln!("Skipping List...");
Ok(())
}
fn write(&self, v: &Val, w: &mut Write) -> Result<()> {
match v {
&Val::Float(ref f) => {
try!(write!(w, "{} ", f));
}
&Val::Int(ref i) => {
try!(write!(w, "{} ", i));
}
&Val::String(ref s) => {
try!(write!(w, "'{}' ", s));
}
&Val::List(ref items) => {
try!(self.convert_list(items, w));
}
&Val::Tuple(ref flds) => {
try!(self.convert_tuple(flds, w));
}
&Val::Macro(ref _def) => {
// This is ignored
eprintln!("Skipping macro...");
}
}
Ok(())
}
}
impl Converter for EnvConverter {
fn convert(&self, v: Rc<Val>, mut w: Box<Write>) -> Result<()> {
self.write(&v, &mut w)
}
}

View File

@ -12,7 +12,7 @@
// See the License for the specific language governing permissions and // See the License for the specific language governing permissions and
// limitations under the License. // limitations under the License.
//! Flags contains code for converting a UCG Val into the command line flag output target. //! Contains code for converting a UCG Val into the command line flag output target.
use std::rc::Rc; use std::rc::Rc;
use std::io::Write; use std::io::Write;
use std::io::Result; use std::io::Result;

View File

@ -15,6 +15,7 @@
//! The conversion stage of the ucg compiler. //! The conversion stage of the ucg compiler.
pub mod flags; pub mod flags;
pub mod json; pub mod json;
pub mod env;
pub mod traits; pub mod traits;
use std::io; use std::io;
@ -30,15 +31,24 @@ pub struct ConverterRunner {
impl ConverterRunner { impl ConverterRunner {
/// new creates a new ConverterRunner with a converter for the provided output target. /// new creates a new ConverterRunner with a converter for the provided output target.
/// ///
/// * flags /// * flags
/// * json /// * json
pub fn new(typ: &str) -> Result<Self, String> { pub fn new(typ: &str) -> Result<Self, String> {
if typ == "flags" { if typ == "flags" {
return Ok(ConverterRunner { converter: Box::new(flags::FlagConverter::new()) }); return Ok(ConverterRunner {
converter: Box::new(flags::FlagConverter::new()),
});
} }
if typ == "json" { if typ == "json" {
return Ok(ConverterRunner { converter: Box::new(json::JsonConverter::new()) }); return Ok(ConverterRunner {
converter: Box::new(json::JsonConverter::new()),
});
}
if typ == "env" {
return Ok(ConverterRunner {
converter: Box::new(env::EnvConverter::new()),
});
} }
return Err(format!("Unknown Target output type: {}", typ)); return Err(format!("Unknown Target output type: {}", typ));
} }