123 lines
3.0 KiB
Rust
123 lines
3.0 KiB
Rust
use std::env;
|
|
use std::collections::HashMap;
|
|
use std::path::PathBuf;
|
|
use std::fs;
|
|
|
|
use lazy_static::lazy_static;
|
|
|
|
lazy_static!{
|
|
static ref LOGLEVEL: LogLevel = LogLevel::from_env();
|
|
static ref LANG: String = lang_from_env();
|
|
static ref TRANSLATIONS: HashMap<String, String> = load_translations();
|
|
}
|
|
|
|
fn load_translations() -> HashMap<String, String> {
|
|
let folder = env::var("I18N").unwrap_or("./i18n/".to_string());
|
|
let mut path = PathBuf::from(folder);
|
|
if !path.is_dir() {
|
|
panic!("Could not find translations in {:?}", path);
|
|
}
|
|
path.push(format!("{}.json", *LANG));
|
|
|
|
if !path.is_file() {
|
|
panic!("Could not find translation file: {:?}", path);
|
|
}
|
|
|
|
match fs::read_to_string(&path) {
|
|
Ok(content) => {
|
|
//let trans: HashMap<String, String> = serde_json::from_str(&content).expect("Could not load translations");
|
|
//return trans
|
|
match serde_json::from_str(&content) {
|
|
Ok(trans) => trans,
|
|
Err(e) => panic!("JSON ERROR: {}", e)
|
|
}
|
|
},
|
|
Err(e) => {
|
|
panic!("IO ERROR: {}", e);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn trans(key: &str) -> String {
|
|
match TRANSLATIONS.get(key) {
|
|
Some(t) => t.to_string(),
|
|
None => {
|
|
panic!("Unknown translation string in lang {}: {}", *LANG, key);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn lang_from_env() -> String {
|
|
let lang = env::var("LANG").expect("$LANG not set in environment. Your machine is misconfigured!");
|
|
lang[0..2].to_string()
|
|
}
|
|
|
|
|
|
struct LogLevel {
|
|
info: bool,
|
|
debug: bool,
|
|
error: bool
|
|
}
|
|
|
|
impl LogLevel {
|
|
fn from_env() -> LogLevel {
|
|
// Default values
|
|
let error = true;
|
|
let mut info = true;
|
|
let mut debug = false;
|
|
|
|
let env_log = env::var("LOG").unwrap_or("info".to_string());
|
|
match env_log.to_lowercase().as_str() {
|
|
"info" => {},
|
|
"debug" => { debug = true; },
|
|
"error" => { info = false; }
|
|
_ => {
|
|
// This happens before loglevel initialization
|
|
// so we can't use warn function
|
|
eprintln!("$LOG level is incorrect: {} (can be: debug, info, error", env_log);
|
|
}
|
|
}
|
|
|
|
return LogLevel {
|
|
info,
|
|
debug,
|
|
error
|
|
}
|
|
}
|
|
}
|
|
|
|
fn expand(msg: &str, vars: Option<&HashMap<String,&str>>) -> String {
|
|
let mut trans_string = trans(msg);
|
|
if vars.is_some() {
|
|
for (key, val) in vars.unwrap() {
|
|
trans_string = trans_string.replace(key, val);
|
|
}
|
|
}
|
|
trans_string
|
|
}
|
|
|
|
pub fn info(msg: &str, vars: Option<&HashMap<String,&str>>) {
|
|
if LOGLEVEL.info {
|
|
let trans_string = expand(msg, vars);
|
|
println!("[git-build] {}", trans_string);
|
|
}
|
|
}
|
|
|
|
pub fn error(msg: &str) {
|
|
if LOGLEVEL.error {
|
|
eprintln!("ERROR: {}", msg);
|
|
}
|
|
}
|
|
|
|
pub fn warn(msg: &str) {
|
|
if LOGLEVEL.error {
|
|
eprintln!("WARNING: {}", msg);
|
|
}
|
|
}
|
|
|
|
pub fn debug(msg: &str) {
|
|
if LOGLEVEL.debug {
|
|
println!("DEBUG: {}", msg);
|
|
}
|
|
}
|