mirror of
https://github.com/zaphar/ucg.git
synced 2025-07-22 18:19:54 -04:00
Add an environment variable output target.
This commit is contained in:
parent
30c9bc64b2
commit
5a6737d858
1
examples/env.txt
Normal file
1
examples/env.txt
Normal 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
86
src/convert/env.rs
Normal 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)
|
||||||
|
}
|
||||||
|
}
|
@ -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;
|
||||||
|
@ -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;
|
||||||
@ -35,10 +36,19 @@ impl ConverterRunner {
|
|||||||
/// * 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));
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user