kaizenkodo/HEL
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Differentiate in printer between error print and output print.

Browse Source
This commit is contained in:
Oleksandr Kozachuk
2022-12-18 13:21:09 +01:00
parent 1234c79db1
commit 602cac9500
1 changed files with 70 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/repl.rs
+70 -59
View File
@@ -34,6 +34,7 @@ pub struct LKEval<'a> {
#[derive(Debug, PartialEq)] #[derive(Debug, PartialEq)]
pub struct LKPrint { pub struct LKPrint {
out: Vec<String>, out: Vec<String>,
err: Vec<String>,
quit: bool, quit: bool,
state: Rc<RefCell<LK>>, state: Rc<RefCell<LK>>,
} }
@@ -161,7 +162,7 @@ impl<'a> LKEval<'a> {
} }
} }
fn cmd_enc(&self, out: Option<&mut Vec<String>>, name: &String) -> Option<String> { fn cmd_enc(&self, out: Option<&mut Vec<String>>, err: Option<&mut Vec<String>>, name: &String) -> Option<String> {
let root_folder = Rc::new("/".to_string()); let root_folder = Rc::new("/".to_string());
let pass = if name == "/" && self.state.borrow().secrets.contains_key(&root_folder) { let pass = if name == "/" && self.state.borrow().secrets.contains_key(&root_folder) {
self.state.borrow().secrets.get(&root_folder).unwrap().to_string() self.state.borrow().secrets.get(&root_folder).unwrap().to_string()
@@ -169,7 +170,7 @@ impl<'a> LKEval<'a> {
let pwd = match self.get_password(name) { let pwd = match self.get_password(name) {
Some(p) => p.clone(), Some(p) => p.clone(),
None => { None => {
if out.is_some() { out.unwrap().push(format!("error: name {} not found", name)) }; if err.is_some() { err.unwrap().push(format!("error: name {} not found", name)) };
return None; return None;
} }
}; };
@@ -180,7 +181,7 @@ impl<'a> LKEval<'a> {
match self.read_master(pwd.clone(), true) { match self.read_master(pwd.clone(), true) {
Some(sec) => pwd.borrow().encode(sec.as_str()), Some(sec) => pwd.borrow().encode(sec.as_str()),
None => { None => {
if out.is_some() { out.unwrap().push(format!("error: master for {} not found", pwd.borrow().name)) }; if err.is_some() { err.unwrap().push(format!("error: master for {} not found", pwd.borrow().name)) };
return None; return None;
} }
} }
@@ -190,40 +191,45 @@ impl<'a> LKEval<'a> {
Some(pass) Some(pass)
} }
fn cmd_pb(&self, out: &mut Vec<String>, command: &String) { fn cmd_pb(&self, out: &mut Vec<String>, err: &mut Vec<String>, command: &String) {
match command_parser::cmd(command) { match command_parser::cmd(command) {
Ok(cmd) => { Ok(cmd) => {
let print = LKEval::new(cmd, self.state.clone(), prompt_password).eval(); let print = LKEval::new(cmd, self.state.clone(), prompt_password).eval();
let data = print.out.join("\n"); let data = print.out.join("\n");
let (copy_command, copy_cmd_args) = get_copy_command_from_env(); for line in print.err {
match call_cmd_with_input(&copy_command, &copy_cmd_args, &data) { err.push(line)
Ok(s) if s.len() > 0 => { }
out.push(format!("Copied output with the command {}, and got following output:", copy_command)); if data.len() > 0 {
out.push(s.trim().to_string()); let (copy_command, copy_cmd_args) = get_copy_command_from_env();
} match call_cmd_with_input(&copy_command, &copy_cmd_args, &data) {
Ok(_) => out.push(format!("Copied output with command {}", copy_command)), Ok(s) if s.len() > 0 => {
Err(e) => out.push(format!("error: failed to copy: {}", e.to_string())), out.push(format!("Copied output with the command {}, and got following output:", copy_command));
}; out.push(s.trim().to_string());
}
Ok(_) => out.push(format!("Copied output with command {}", copy_command)),
Err(e) => err.push(format!("error: failed to copy: {}", e.to_string())),
};
}
} }
Err(e) => out.push(format!("error: faild to parse command {}: {}", command, e.to_string())), Err(e) => err.push(format!("error: faild to parse command {}: {}", command, e.to_string())),
}; };
} }
fn cmd_source(&self, out: &mut Vec<String>, source: &String) { fn cmd_source(&self, out: &mut Vec<String>, err: &mut Vec<String>, source: &String) {
let script = if source.trim().ends_with("|") { let script = if source.trim().ends_with("|") {
let (cmd, args) = match get_cmd_args_from_command(source.trim().trim_end_matches('|')) { let (cmd, args) = match get_cmd_args_from_command(source.trim().trim_end_matches('|')) {
Ok(c) => c, Ok(c) => c,
Err(e) => { out.push(format!("error: failed to parse command {:?}: {}", source, e.to_string())); return; }, Err(e) => { err.push(format!("error: failed to parse command {:?}: {}", source, e.to_string())); return; },
}; };
match call_cmd_with_input(&cmd, &args, "") { match call_cmd_with_input(&cmd, &args, "") {
Ok(o) => o, Ok(o) => o,
Err(e) => { out.push(format!("error: failed to execute command {}: {}", cmd, e.to_string())); return; }, Err(e) => { err.push(format!("error: failed to execute command {}: {}", cmd, e.to_string())); return; },
} }
} else { } else {
let script = shellexpand::full(source).unwrap().into_owned(); let script = shellexpand::full(source).unwrap().into_owned();
match std::fs::read_to_string(script) { match std::fs::read_to_string(script) {
Ok(script) => script, Ok(script) => script,
Err(err) => { out.push(format!("error: failed to read file {}: {}", source, err.to_string())); return; } Err(e) => { err.push(format!("error: failed to read file {}: {}", source, e.to_string())); return; }
} }
}; };
match command_parser::script(&script) { match command_parser::script(&script) {
@@ -232,15 +238,15 @@ impl<'a> LKEval<'a> {
LKEval::new(cmd, self.state.clone(), prompt_password).eval().print(); LKEval::new(cmd, self.state.clone(), prompt_password).eval().print();
} }
} }
Err(err) => { out.push(format!("error: {}", err.to_string())); return; } Err(e) => { err.push(format!("error: {}", e.to_string())); return; }
}; };
} }
fn cmd_ls(&self, out: &mut Vec<String>, filter: String) { fn cmd_ls(&self, out: &mut Vec<String>, err: &mut Vec<String>, filter: String) {
let re = match Regex::new(&filter) { let re = match Regex::new(&filter) {
Ok(re) => re, Ok(re) => re,
Err(e) => { Err(e) => {
out.push(format!("error: failed to parse re: {:?}", e)); err.push(format!("error: failed to parse re: {:?}", e));
return; return;
} }
}; };
@@ -265,8 +271,8 @@ impl<'a> LKEval<'a> {
} }
} }
fn cmd_correct(&self, out: &mut Vec<String>, name: &String, correct: bool) { fn cmd_correct(&self, out: &mut Vec<String>, err: &mut Vec<String>, name: &String, correct: bool) {
let pwd = match self.cmd_enc(None, &name) { Some(v) => v, None => return }; let pwd = match self.cmd_enc(None, None, &name) { Some(v) => v, None => return };
fn load_lines() -> std::io::Result<HashSet<String>> { fn load_lines() -> std::io::Result<HashSet<String>> {
let file = fs::File::open(CORRECT_FILE.to_str().unwrap())?; let file = fs::File::open(CORRECT_FILE.to_str().unwrap())?;
let reader = BufReader::new(file); let reader = BufReader::new(file);
@@ -300,23 +306,24 @@ impl<'a> LKEval<'a> {
} }
match save_lines(&data) { match save_lines(&data) {
Ok(()) => out.push(format!("Hash of the password {} {}", if correct { "remembered to" } else { "removed from" },CORRECT_FILE.to_str().unwrap())), Ok(()) => out.push(format!("Hash of the password {} {}", if correct { "remembered to" } else { "removed from" },CORRECT_FILE.to_str().unwrap())),
Err(err) => out.push(format!("error: failed to write: {}", err.to_string())), Err(e) => err.push(format!("error: failed to write: {}", e.to_string())),
}; };
} }
pub fn eval(&self) -> LKPrint { pub fn eval(&self) -> LKPrint {
let mut out: Vec<String> = vec![]; let mut out: Vec<String> = vec![];
let mut err: Vec<String> = vec![];
let mut quit: bool = false; let mut quit: bool = false;
match &self.cmd { match &self.cmd {
Command::Quit => { Command::Quit => {
out.push("Bye!".to_string()); err.push("Bye!".to_string());
quit = true; quit = true;
} }
Command::Ls(filter) => self.cmd_ls(&mut out, filter.to_string()), Command::Ls(filter) => self.cmd_ls(&mut out, &mut err, filter.to_string()),
Command::Add(name) => { Command::Add(name) => {
if self.state.borrow().db.get(&name.borrow().name).is_some() { if self.state.borrow().db.get(&name.borrow().name).is_some() {
out.push("error: password already exist".to_string()); err.push("error: password already exist".to_string());
} else { } else {
self.state.borrow_mut().db.insert(name.borrow().name.clone(), name.clone()); self.state.borrow_mut().db.insert(name.borrow().name.clone(), name.clone());
self.state.borrow().fix_hierarchy(); self.state.borrow().fix_hierarchy();
@@ -329,18 +336,18 @@ impl<'a> LKEval<'a> {
None => None, None => None,
} }
} }
None => out.push("error: password not found".to_string()), None => err.push("error: password not found".to_string()),
}, },
Command::Rm(name) => match self.get_password(name) { Command::Rm(name) => match self.get_password(name) {
Some(pwd) => { Some(pwd) => {
self.state.borrow_mut().db.remove(&pwd.borrow().name); self.state.borrow_mut().db.remove(&pwd.borrow().name);
out.push(format!("removed {}", pwd.borrow().name)); out.push(format!("removed {}", pwd.borrow().name));
} }
None => out.push("error: password not found".to_string()), None => err.push("error: password not found".to_string()),
}, },
Command::Enc(name) => { self.cmd_enc(Some(&mut out), name); }, Command::Enc(name) => { self.cmd_enc(Some(&mut out), Some(&mut err), name); },
Command::PasteBuffer(command) => self.cmd_pb(&mut out, command), Command::PasteBuffer(command) => self.cmd_pb(&mut out, &mut err, command),
Command::Source(script) => self.cmd_source(&mut out, script), Command::Source(script) => self.cmd_source(&mut out, &mut err, script),
Command::Pass(name) => match self.get_password(name) { Command::Pass(name) => match self.get_password(name) {
Some(p) => { Some(p) => {
self.state.borrow_mut().secrets.insert( self.state.borrow_mut().secrets.insert(
@@ -355,16 +362,16 @@ impl<'a> LKEval<'a> {
.secrets .secrets
.insert(Rc::new("/".to_string()), (self.read_password)("Master: ".to_string()).unwrap()); .insert(Rc::new("/".to_string()), (self.read_password)("Master: ".to_string()).unwrap());
} else { } else {
out.push(format!("error: password with name {} not found", name)); err.push(format!("error: password with name {} not found", name));
} }
} }
}, },
Command::UnPass(name) => match self.state.borrow_mut().secrets.remove(name) { Command::UnPass(name) => match self.state.borrow_mut().secrets.remove(name) {
Some(_) => out.push(format!("Removed saved password for {}", name)), Some(_) => out.push(format!("Removed saved password for {}", name)),
None => out.push(format!("error: saved password for {} not found", name)), None => err.push(format!("error: saved password for {} not found", name)),
} }
Command::Correct(name) => self.cmd_correct(&mut out, name, true), Command::Correct(name) => self.cmd_correct(&mut out, &mut err, name, true),
Command::Uncorrect(name) => self.cmd_correct(&mut out, name, false), Command::Uncorrect(name) => self.cmd_correct(&mut out, &mut err, name, false),
Command::Noop => (), Command::Noop => (),
Command::Help => { Command::Help => {
out.push("HELP".to_string()); out.push("HELP".to_string());
@@ -379,29 +386,32 @@ impl<'a> LKEval<'a> {
pwd.borrow_mut().parent = Some(fld.clone()); pwd.borrow_mut().parent = Some(fld.clone());
fix_password_recursion(pwd.clone()); fix_password_recursion(pwd.clone());
} }
None => out.push(format!("error: folder {} not found", folder)), None => err.push(format!("error: folder {} not found", folder)),
} }
} }
} }
None => out.push(format!("error: password with name {} not found", name)), None => err.push(format!("error: password with name {} not found", name)),
}, },
Command::Error(err) => match err { Command::Error(error) => match error {
LKErr::ParseError(e) => out.push(e.to_string()), LKErr::ParseError(e) => err.push(e.to_string()),
LKErr::ReadError(e) => out.push(e.to_string()), LKErr::ReadError(e) => err.push(e.to_string()),
LKErr::Error(e) => out.push(format!("error: {}", e.to_string())), LKErr::Error(e) => err.push(format!("error: {}", e.to_string())),
}, },
} }
LKPrint::new(out, quit, self.state.clone()) LKPrint::new(out, err, quit, self.state.clone())
} }
} }
impl LKPrint { impl LKPrint {
pub fn new(out: Vec<String>, quit: bool, state: Rc<RefCell<LK>>) -> Self { pub fn new(out: Vec<String>, err: Vec<String>, quit: bool, state: Rc<RefCell<LK>>) -> Self {
Self { out, quit, state } Self { out, err, quit, state }
} }
pub fn print(&mut self) -> bool { pub fn print(&mut self) -> bool {
for line in &self.err {
eprintln!("{}", line);
}
for line in &self.out { for line in &self.out {
println!("{}", line); println!("{}", line);
} }
@@ -432,7 +442,7 @@ mod tests {
let lk = Rc::new(RefCell::new(LK::new())); let lk = Rc::new(RefCell::new(LK::new()));
assert_eq!( assert_eq!(
LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(), LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(),
LKPrint::new(vec![], false, lk.clone()) LKPrint::new(vec![], vec![], false, lk.clone())
); );
let pwd1 = Rc::new(RefCell::new(Password { let pwd1 = Rc::new(RefCell::new(Password {
name: Rc::new("t1".to_string()), name: Rc::new("t1".to_string()),
@@ -451,11 +461,11 @@ mod tests {
}); });
assert_eq!( assert_eq!(
LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(), LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(),
LKPrint::new(vec![" 1 t1 R 99 2022-12-30 comment".to_string()], false, lk.clone()) LKPrint::new(vec![" 1 t1 R 99 2022-12-30 comment".to_string()], vec![], false, lk.clone())
); );
assert_eq!( assert_eq!(
LKEval::news(Command::Quit, lk.clone()).eval(), LKEval::news(Command::Quit, lk.clone()).eval(),
LKPrint::new(vec!["Bye!".to_string()], true, lk.clone()) LKPrint::new(vec![], vec!["Bye!".to_string()], true, lk.clone())
); );
let pwd2 = Rc::new(RefCell::new(Password { let pwd2 = Rc::new(RefCell::new(Password {
name: Rc::new("t2".to_string()), name: Rc::new("t2".to_string()),
@@ -477,17 +487,18 @@ mod tests {
LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(), LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(),
LKPrint::new( LKPrint::new(
vec![" 1 t1 R 99 2022-12-30 comment".to_string(), " 2 t2 R 99 2022-12-31 bli blup".to_string()], vec![" 1 t1 R 99 2022-12-30 comment".to_string(), " 2 t2 R 99 2022-12-31 bli blup".to_string()],
vec![],
false, false,
lk.clone() lk.clone()
) )
); );
assert_eq!( assert_eq!(
LKEval::news(Command::Rm("2".to_string()), lk.clone()).eval(), LKEval::news(Command::Rm("2".to_string()), lk.clone()).eval(),
LKPrint::new(vec!["removed t2".to_string()], false, lk.clone()) LKPrint::new(vec!["removed t2".to_string()], vec![], false, lk.clone())
); );
assert_eq!( assert_eq!(
LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(), LKEval::news(Command::Ls(".".to_string()), lk.clone()).eval(),
LKPrint::new(vec![" 1 t1 R 99 2022-12-30 comment".to_string()], false, lk.clone()) LKPrint::new(vec![" 1 t1 R 99 2022-12-30 comment".to_string()], vec![], false, lk.clone())
); );
} }
@@ -521,16 +532,16 @@ mod tests {
NaiveDate::from_ymd_opt(2022, 12, 30).unwrap(), NaiveDate::from_ymd_opt(2022, 12, 30).unwrap(),
None, None,
))); )));
assert_eq!(LKEval::news(Command::Add(t1.clone()), lk.clone()).eval(), LKPrint::new(vec![], false, lk.clone())); assert_eq!(LKEval::news(Command::Add(t1.clone()), lk.clone()).eval(), LKPrint::new(vec![], vec![], false, lk.clone()));
assert_eq!(LKEval::news(Command::Add(t2.clone()), lk.clone()).eval(), LKPrint::new(vec![], false, lk.clone())); assert_eq!(LKEval::news(Command::Add(t2.clone()), lk.clone()).eval(), LKPrint::new(vec![], vec![], false, lk.clone()));
assert_eq!(LKEval::news(Command::Add(t3.clone()), lk.clone()).eval(), LKPrint::new(vec![], false, lk.clone())); assert_eq!(LKEval::news(Command::Add(t3.clone()), lk.clone()).eval(), LKPrint::new(vec![], vec![], false, lk.clone()));
assert_eq!( assert_eq!(
LKEval::news(Command::Mv("t3".to_string(), "t2".to_string()), lk.clone()).eval(), LKEval::news(Command::Mv("t3".to_string(), "t2".to_string()), lk.clone()).eval(),
LKPrint::new(vec![], false, lk.clone()) LKPrint::new(vec![], vec![], false, lk.clone())
); );
assert_eq!( assert_eq!(
LKEval::news(Command::Mv("t2".to_string(), "t1".to_string()), lk.clone()).eval(), LKEval::news(Command::Mv("t2".to_string(), "t1".to_string()), lk.clone()).eval(),
LKPrint::new(vec![], false, lk.clone()) LKPrint::new(vec![], vec![], false, lk.clone())
); );
assert_eq!( assert_eq!(
LKEval::new(Command::Enc("t3".to_string()), lk.clone(), |p| if p == "NULL" { LKEval::new(Command::Enc("t3".to_string()), lk.clone(), |p| if p == "NULL" {
@@ -539,7 +550,7 @@ mod tests {
Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test")) Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test"))
}) })
.eval(), .eval(),
LKPrint::new(vec!["error: master for t3 not found".to_string()], false, lk.clone()) LKPrint::new(vec![], vec!["error: master for t3 not found".to_string()], false, lk.clone())
); );
assert_eq!( assert_eq!(
LKEval::new(Command::Enc("t3".to_string()), lk.clone(), |p| if p == "Master: " { LKEval::new(Command::Enc("t3".to_string()), lk.clone(), |p| if p == "Master: " {
@@ -548,7 +559,7 @@ mod tests {
Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test")) Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test"))
}) })
.eval(), .eval(),
LKPrint::new(vec!["san bud most noon jaw cash".to_string()], false, lk.clone()) LKPrint::new(vec!["san bud most noon jaw cash".to_string()], vec![], false, lk.clone())
); );
assert_eq!( assert_eq!(
LKEval::new(Command::Enc("t2".to_string()), lk.clone(), |p| if p == "NULL" { LKEval::new(Command::Enc("t2".to_string()), lk.clone(), |p| if p == "NULL" {
@@ -557,7 +568,7 @@ mod tests {
Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test")) Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test"))
}) })
.eval(), .eval(),
LKPrint::new(vec!["alga barn wise tim skin mock".to_string()], false, lk.clone()) LKPrint::new(vec!["alga barn wise tim skin mock".to_string()], vec![], false, lk.clone())
); );
assert_eq!( assert_eq!(
LKEval::new(Command::Enc("t1".to_string()), lk.clone(), |p| if p == "NULL" { LKEval::new(Command::Enc("t1".to_string()), lk.clone(), |p| if p == "NULL" {
@@ -566,7 +577,7 @@ mod tests {
Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test")) Err(std::io::Error::new(std::io::ErrorKind::NotFound, "test"))
}) })
.eval(), .eval(),
LKPrint::new(vec!["lime rudy jay my kong tack".to_string()], false, lk.clone()) LKPrint::new(vec!["lime rudy jay my kong tack".to_string()], vec![], false, lk.clone())
); );
} }
} }