mod cache;
mod editor;
mod os_command;
mod parser;
mod prelude;
mod programs;
mod utils;
use std::{fs::OpenOptions, path::Path};
use cache::CacheManager;
use nom::error::ErrorKind;
use os_command::exec_command;
pub use parser::{parse_command, CacheCommand};
pub use prelude::*;
use rustyline::error::ReadlineError;
use crate::utils::clear_terminal;
const CACHE_COMMAND_DOC: &[(&[&str], &str)] = CacheCommand::doc();
const VERSION: &str = env!("CARGO_PKG_VERSION");
const PKG_NAME: &str = env!("CARGO_PKG_NAME");
lazy_static::lazy_static! {
static ref CONFIG_FILE_PATH: PathBuf = {
let mut conf_dir = dirs::config_dir().expect("conf dir not found");
conf_dir.push(".karsher.conf.json");
conf_dir
};
}
fn main() -> anyhow::Result<()> {
let mut cache_manager = {
let f = OpenOptions::new()
.read(true)
.write(true)
.create(true)
.open(CONFIG_FILE_PATH.as_path())
.expect("cannot open config");
let reader = BufReader::new(f);
let cache: Option<CacheManager> = serde_json::from_reader(reader).ok();
if let Some(manager) = cache {
manager
} else {
CacheManager::default()
}
};
let mut current_cache = {
cache_manager
.get_default_cache()
.as_ref()
.map_or("DEFAULT".into(), |v| v.clone())
};
clear_terminal();
println!("{PKG_NAME} v{VERSION}\n");
let mut rl = editor::build_editor();
loop {
let readline = editor::read_line(&mut rl, ¤t_cache);
match readline {
Ok(line) => {
rl.add_history_entry(line.as_str());
if process_repl(&line).is_err() {
process_command(
&mut cache_manager,
&mut current_cache,
&line,
)?;
}
}
Err(ReadlineError::Interrupted) | Err(ReadlineError::Eof) => {
break;
}
Err(err) => {
eprintln!("Error: {:?}", err);
std::process::exit(1);
}
}
}
if let Ok(json) = serde_json::to_string_pretty(&cache_manager) {
if std::fs::write(CONFIG_FILE_PATH.as_path(), json).is_err() {
eprintln!("could not write to target conf file. gomenasai");
}
} else {
eprintln!("could not acquire lock or could not serialize to json. sorry! bye.");
}
editor::save_history(&mut rl)?;
println!(
"{}",
colors::Style::new().bold().fg(colors::LightBlue).paint("BYE")
);
Ok(())
}
fn process_command(
cache_manager: &mut CacheManager,
current_cache: &mut String,
line: &str,
) -> anyhow::Result<()> {
match parse_command(line) {
Ok((_, command)) => match command {
CacheCommand::Add { aliases, value } => {
if CACHE_COMMAND_DOC
.iter()
.flat_map(|c| c.0.iter().map(|comm| comm.to_uppercase()))
.any(|c| aliases.iter().any(|al| al.to_uppercase() == c))
{
eprintln!("You cannot use a reserved command name as an alias. check help for list of reserved names.");
} else {
let cache = cache_manager.get_mut_or_insert(current_cache);
let key = cache.insert(aliases, value);
println!(
"added {} with hash key {}",
colors::Yellow.paint(value),
colors::Red.paint(key.to_string())
);
}
}
CacheCommand::Remove(key) => {
let cache = cache_manager.get_mut_or_insert(current_cache);
if let Some(v) = cache.remove(key) {
println!(
"removed {} with hash key {}",
colors::Yellow.paint(v),
colors::Red.paint(key.to_string())
);
} else {
println!(
"key {key} not found in current cache {current_cache}"
);
}
}
CacheCommand::Get(key) => {
let cache = cache_manager.get_mut_or_insert(current_cache);
if let Some(value) = cache.get(key) {
println!("found '{}'", colors::Yellow.paint(value));
} else {
println!("{key} not found");
}
}
CacheCommand::Exec { key, args } => {
let cache = cache_manager.get_mut_or_insert(current_cache);
if let Some(value) = cache.get(key) {
let _ = exec_command(value, &args)
.map_err(|e| anyhow::Error::msg(e.to_string()))?;
} else if !key.trim().is_empty() {
println!("{key} not found");
}
}
CacheCommand::Using(key) => {
current_cache.clear();
current_cache.push_str(key);
cache_manager.set_default_cache(current_cache);
let _ = cache_manager.get_mut_or_insert(current_cache);
println!("current cache: {}", colors::LightCyan.paint(key));
}
CacheCommand::ListCache => {
println!(
">> [ {} ]",
cache_manager
.get_cache_names()
.iter()
.map(|c| colors::Red.bold().paint(*c).to_string())
.collect::<Vec<_>>()
.join(", ")
);
}
CacheCommand::CurrentCache => {
println!(
">> {}",
colors::LightBlue.bold().paint(current_cache.to_string())
);
}
CacheCommand::Concat(key) if key == current_cache => {
eprintln!("You cannot merge a cache with itself!")
}
CacheCommand::Concat(key) => {
if let Some((current, cache)) =
cache_manager.get_mut_pair(current_cache, key)
{
current.concat(cache);
println!(
"cache {} has been merged with cache {}.",
colors::Red.bold().paint(¤t_cache.to_string()),
colors::Yellow.bold().paint(key)
);
} else {
eprintln!("something went wrong!");
}
}
CacheCommand::Dump(key) => {
if let Some(key) = key {
if let Some(cache) = cache_manager.get(key) {
let cache = serde_json::to_string_pretty(&cache)?;
println!("{cache}")
} else {
eprintln!("cache {key} not found");
}
} else {
let caches = serde_json::to_string_pretty(&*cache_manager)?;
println!("{caches}")
}
}
CacheCommand::RemoveCache(key) => {
if let Some(cache_name) = key {
if cache_name != current_cache {
println!(
"remove {cache_name}: {}",
cache_manager.remove_cache(cache_name).is_some()
);
} else {
println!(
"clear all values from {current_cache}: {}",
cache_manager.clear_values(current_cache)
);
}
}
}
CacheCommand::List => {
if let Some(cache) = cache_manager.get(current_cache) {
for (key, value, aliases) in cache.list() {
println!(
">> Key: {}, Aliases: {} => Value: '{}'",
colors::Red.paint(key.to_string()),
aliases
.iter()
.map(|a| colors::Yellow.paint(*a).to_string())
.collect::<Vec<_>>()
.join(","),
colors::LightCyan.paint(value)
);
}
}
}
CacheCommand::Cd(path) => {
if Path::new(path).exists() {
std::env::set_current_dir(path)?;
println!(
">> working directory {}",
colors::LightMagenta.paint(path)
);
} else {
eprintln!("path {} doesn't exist", colors::Red.paint(path));
}
}
CacheCommand::Help => {
for doc in CACHE_COMMAND_DOC {
let (command, doc) = doc;
println!(
">> {} : {}",
command
.iter()
.map(|c| colors::Yellow.paint(*c).to_string())
.collect::<Vec<_>>()
.join("/"),
colors::LightBlue.paint(*doc)
);
}
}
CacheCommand::Clear => {
clear_terminal();
}
},
Err(e) => match e {
nom::Err::Failure(failure) if failure.code == ErrorKind::Verify => {
eprintln!(
"invalid command: {}",
colors::Red.paint(failure.to_string())
)
}
nom::Err::Error(err) if err.input.trim().is_empty() => {}
_ => eprintln!(
"error parsing command: {}",
colors::Red.paint(e.to_string())
),
},
}
Ok(())
}
fn process_repl(line: &str) -> anyhow::Result<()> {
let calc = crate::programs::compute(line)?;
println!("{calc}");
Ok(())
}