moved from tera to upon

This commit is contained in:
Valerie Wolfe 2023-08-25 23:48:43 -04:00
parent 3941e77f6a
commit 4f61060109
6 changed files with 43 additions and 34 deletions

View file

@ -1,6 +1,6 @@
[package] [package]
name = "oink" name = "oink"
version = "0.0.2" version = "0.1.0"
edition = "2021" edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
@ -8,8 +8,8 @@ edition = "2021"
[dependencies] [dependencies]
#copy_dir = "0.1.3" #copy_dir = "0.1.3"
pico-args = "0.5.0" pico-args = "0.5.0"
tera = "1.19.0"
toml = "0.7.6" toml = "0.7.6"
upon = "0.7.1"
[profile.release] [profile.release]
opt-level = "s" opt-level = "s"

View file

@ -41,6 +41,5 @@ path = "/home/test/.config/oink/oink.toml"
## Libraries ## Libraries
- [pico-args](https://crates.io/crates/pico-args) — argument parsing - [pico-args](https://crates.io/crates/pico-args) — argument parsing
- [tera](https://crates.io/crates/tera) — template engine
- [toml](https://crates.io/crates/toml) — configuration parsing - [toml](https://crates.io/crates/toml) — configuration parsing
- [upon](https://crates.io/crates/upon) — template engine

View file

@ -1,4 +0,0 @@
[toolchain]
channel = "nightly"

View file

@ -1,7 +1,7 @@
//! configuration struct and implementations //! configuration struct and implementations
use std::{ use std::{
collections::HashMap, collections::{ BTreeMap, HashMap },
env::var, env::var,
fs::{ create_dir, read_to_string, File }, fs::{ create_dir, read_to_string, File },
io::Error, io::Error,
@ -9,7 +9,7 @@ use std::{
process::exit process::exit
}; };
use tera::Context; use upon::Value as ContextValue;
use toml::{ use toml::{
map::Map, map::Map,
Value Value
@ -17,6 +17,8 @@ use toml::{
use crate::error; use crate::error;
pub type Context = BTreeMap<String, ContextValue>;
/// configuration struct /// configuration struct
pub struct Config { pub struct Config {
pub dir: String, pub dir: String,
@ -61,17 +63,17 @@ impl Config {
if vars.is_some() { if vars.is_some() {
let vars = vars.unwrap().as_table().unwrap(); let vars = vars.unwrap().as_table().unwrap();
for (key, value) in vars.iter() { for (key, value) in vars.iter() {
output.insert(key, value.as_str().unwrap()); output.insert(key.to_owned(), value.as_str().into());
} }
} }
let colors = self.inner.get("colors"); let colors = self.inner.get("colors");
if colors.is_some() { if colors.is_some() {
let colors = colors.unwrap().as_table().unwrap(); let colors = colors.unwrap().as_table().unwrap();
let mut map = HashMap::<&str, &str>::new(); let mut map = Context::new();
for (key, value) in colors.iter() { for (key, value) in colors.iter() {
map.insert(key, value.as_str().unwrap()); map.insert(key.to_owned(), value.as_str().unwrap().into());
} }
output.insert("colors", &map); output.insert("colors".to_owned(), map.into());
} }
output output

View file

@ -1,12 +1,11 @@
#![feature(error_in_core)]
use std::process::exit; use std::process::exit;
use pico_args::Arguments; use pico_args::Arguments;
use tera::Tera;
mod config; mod config;
mod error; mod error;
mod operation; mod operation;
use crate::config::Config; use crate::config::Config;
fn main() { fn main() {
@ -20,10 +19,8 @@ fn main() {
// init configuration // init configuration
let config = Config::new(); let config = Config::new();
// tera init // build template dir
let context = config.context(); let template_dir = format!("{}/templates/", &(config.dir));
let template_dir = format!("{}/templates/*", &(config.dir));
let mut tera = Tera::new(&template_dir).unwrap();
let targets = config.targets(); let targets = config.targets();
if targets.len() == 0 { error::no_targets(); } if targets.len() == 0 { error::no_targets(); }
@ -35,11 +32,11 @@ fn main() {
}, },
Some("build") Some("build")
=> { => {
operation::build(&targets, &mut tera, &context); operation::build(&targets, template_dir, &config);
}, },
Some("full") Some("full")
=> { => {
operation::build(&targets, &mut tera, &context); operation::build(&targets, template_dir, &config);
operation::apply(&targets); operation::apply(&targets);
}, },
_ _

View file

@ -1,14 +1,15 @@
//! higher-level operation functions //! higher-level operation functions
use std::{ use std::{
env::var, env::var,
fs::{ self, File }, fs::{self, read_to_string, File },
io::Write, io::Write,
path::Path path::{ Path, PathBuf }
}; };
use core::error::Error;
use toml::{ map::Map, Value }; use toml::{ map::Map, Value };
use tera::{ Context, Tera }; use upon::Engine;
use crate::config::Config;
pub fn apply(targets: &Vec<Map<String, Value>>) { pub fn apply(targets: &Vec<Map<String, Value>>) {
let home = var("HOME").unwrap(); let home = var("HOME").unwrap();
@ -47,9 +48,13 @@ pub fn apply(targets: &Vec<Map<String, Value>>) {
} }
} }
pub fn build(targets: &Vec<Map<String, Value>>, tera: &mut Tera, context: &Context) { pub fn build(targets: &Vec<Map<String, Value>>, template_dir: String, config: &Config) {
let home = var("HOME").unwrap(); let home = var("HOME").unwrap();
println!("running build:"); println!("running build:");
let engine = Engine::new();
let context = config.context();
for target in targets { for target in targets {
// get name property // get name property
let i_name = target.get("name"); let i_name = target.get("name");
@ -63,14 +68,24 @@ pub fn build(targets: &Vec<Map<String, Value>>, tera: &mut Tera, context: &Conte
let name = i_name.unwrap().as_str().unwrap(); let name = i_name.unwrap().as_str().unwrap();
println!(" building \"{name}\":"); println!(" building \"{name}\":");
// render template // compile
let render = tera.render(name, context); println!(" compiling");
// handle rendering errors gracefully let mut path = PathBuf::from(&template_dir);
path.push(name);
let content = read_to_string(path).unwrap();
let template = engine.compile(&content);
if template.is_err() {
let error = template.err().unwrap();
println!(" failed to compile template:\n {error}\n skipping");
continue;
}
// render
println!(" rendering");
let render = template.unwrap().render(&context).to_string();
if render.is_err() { if render.is_err() {
let error = render.err().unwrap(); let error = render.err().unwrap();
let message = error.source().unwrap(); println!(" failed to render template:\n {error}\n skipping");
println!(" failed to render template:\n {message}");
continue; continue;
} }
@ -87,7 +102,7 @@ pub fn build(targets: &Vec<Map<String, Value>>, tera: &mut Tera, context: &Conte
// write to destination file // write to destination file
let written = write!(&mut file, "{output}"); let written = write!(&mut file, "{output}");
if written.is_err() { println!(" failed to write to destination file at {path:?}"); } if written.is_err() { println!(" failed to write to destination file at {path:?}"); }
else { println!(" completed successfully"); } else { println!(" completed"); }
} }
} }