Compare commits

...

2 commits

Author SHA1 Message Date
3f6e996c64 added ANSI formatting for better output readability 2023-08-26 00:00:55 -04:00
4f61060109 moved from tera to upon 2023-08-25 23:48:43 -04:00
6 changed files with 69 additions and 43 deletions

View file

@ -1,15 +1,15 @@
[package] [package]
name = "oink" name = "oink"
version = "0.0.2" version = "0.1.1"
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
[dependencies] [dependencies]
#copy_dir = "0.1.3"
pico-args = "0.5.0" pico-args = "0.5.0"
tera = "1.19.0" termion = "2.0.1"
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,6 @@ 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 - [termion](https://crates.io/crates/termion) — ANSI formatting
- [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,27 @@
//! 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 termion::{
color::{ self, Fg },
style::{
Bold as BOLD,
Italic as ITALIC,
Reset as RESET
}
};
use toml::{ map::Map, Value }; use toml::{ map::Map, Value };
use tera::{ Context, Tera }; use upon::Engine;
use crate::config::Config;
static SUCCESS: Fg<color::Green> = Fg(color::Green);
static WARNING: Fg<color::Yellow> = Fg(color::Yellow);
static FAILURE: Fg<color::Red> = Fg(color::Red);
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();
@ -22,12 +35,12 @@ pub fn apply(targets: &Vec<Map<String, Value>>) {
if path.is_none() { if path.is_none() {
if i_name.is_some() { if i_name.is_some() {
let name = i_name.unwrap().as_str().unwrap(); let name = i_name.unwrap().as_str().unwrap();
println!(" \"{name}\" is missing its path property; skipping"); println!(" {WARNING}\"{name}\" is missing its path property; skipping{RESET}");
} else { println!(" skipping empty target"); } } else { println!(" {WARNING}skipping empty target{RESET}"); }
continue; continue;
} }
if i_name.is_none() { if i_name.is_none() {
println!(" target missing name; skipping"); println!(" {WARNING}target missing name; skipping{RESET}");
continue; continue;
} }
@ -42,20 +55,28 @@ pub fn apply(targets: &Vec<Map<String, Value>>) {
// copy and print // copy and print
let result = fs::copy(source, destination); let result = fs::copy(source, destination);
if result.is_err() { println!(" failed to copy!"); } if result.is_err() {
else { println!(" completed successfully"); } println!(" {BOLD}{FAILURE}failed to copy{RESET}");
}
else {
println!(" {BOLD}{SUCCESS}completed{RESET}");
}
} }
} }
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");
// handle empty names gracefully // handle empty names gracefully
if i_name.is_none() { if i_name.is_none() {
println!(" target missing name; skipping"); println!(" {WARNING}target missing name; skipping{RESET}");
continue; continue;
} }
@ -63,14 +84,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!(" {ITALIC}compiling{RESET}");
// 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!(" {BOLD}{FAILURE}failed to compile template:{RESET}\n {FAILURE}{error}\n {BOLD}skipping{RESET}");
continue;
}
// render
println!(" {ITALIC}rendering{RESET}");
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!(" {BOLD}{FAILURE}failed to render template:{RESET}\n {FAILURE}{error}\n {BOLD}skipping{RESET}");
println!(" failed to render template:\n {message}");
continue; continue;
} }
@ -80,14 +111,14 @@ pub fn build(targets: &Vec<Map<String, Value>>, tera: &mut Tera, context: &Conte
let path = Path::new(&destination); let path = Path::new(&destination);
let i_file = File::create(path); let i_file = File::create(path);
if i_file.is_err() { if i_file.is_err() {
println!(" failed to create destination file at {path:?}"); println!(" {BOLD}{FAILURE}failed to create destination file at {path:?}{RESET}");
continue; continue;
} }
let mut file = i_file.unwrap(); let mut file = i_file.unwrap();
// 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!(" {FAILURE}failed to write to destination file at {path:?}{RESET}"); }
else { println!(" completed successfully"); } else { println!(" {BOLD}{SUCCESS}completed{RESET}"); }
} }
} }