diff --git a/src/lk.rs b/src/lk.rs index e1128ab..7db276b 100644 --- a/src/lk.rs +++ b/src/lk.rs @@ -1,11 +1,10 @@ -use crate::password::{fix_password_recursion, Password}; +use crate::password::{fix_password_recursion, NameRef, PasswordRef}; use regex::{Captures, Regex}; use std::collections::HashMap; -use std::{cell::RefCell, rc::Rc}; #[derive(PartialEq, Debug)] pub struct LK { - pub db: HashMap, Rc>>, + pub db: HashMap, } impl LK { diff --git a/src/parser.rs b/src/parser.rs index ebca622..8fb3806 100644 --- a/src/parser.rs +++ b/src/parser.rs @@ -14,25 +14,24 @@ peg::parser! { / error_cmd() / ls_cmd() / mv_cmd() + / comment_cmd() ) { c } pub rule name() -> Password = name:(jname() / pname() / mname() / sname()) { name } - + rule _() -> &'input str = s:$((" " / "\t" / "\r" / "\n")+) { s } - rule comment() -> &'input str = _ c:$([' '..='~']+) { c } - rule word() -> &'input str = n:$(['!'..='~']+) { n } + rule comment() -> String = _ c:$([' '..='~']+) { c.to_string() } + rule word() -> String = n:$(['!'..='~']+) { n.to_string() } rule num() -> u32 = n:$(['0'..='9']+) {? n.parse().or(Err("not a number")) } - rule pname() -> Password = &(word() _ word() _ num()? mode() _ num() _ date()) pr:word() _ pn:word() _ pl:num()? pm:mode() _ ps:num() _ pd:date() pc:comment()? { - Password { prefix: Some(pr.to_string()), length: pl, name: Rc::new(pn.to_string()), mode: pm, seq: ps, date: pd, parent: None, - comment: match pc { Some(s) => Some(s.to_string()), None => None } } } - rule jname() -> Password = &(word() _ num()? mode() _ num() _ date()) pn:word() _ pl:num()? pm:mode() _ ps:num() _ pd:date() pc:comment()? { - Password { prefix: None, length: pl, name: Rc::new(pn.to_string()), mode: pm, seq: ps, date: pd, parent: None, - comment: match pc { Some(s) => Some(s.to_string()), None => None } } } - rule mname() -> Password = &(word() _ word() _ num()? mode() _ date()) pr:word() _ pn:word() _ pl:num()? pm:mode() _ pd:date() pc:comment()? { - Password { prefix: Some(pr.to_string()), length: pl, name: Rc::new(pn.to_string()), mode: pm, seq: 99, date: pd, parent: None, - comment: match pc { Some(s) => Some(s.to_string()), None => None } } } - rule sname() -> Password = &(word() _ num()? mode() _ date()) pn:word() _ pl:num()? pm:mode() _ pd:date() pc:comment()? { - Password { prefix: None, length: pl, name: Rc::new(pn.to_string()), mode: pm, seq: 99, date: pd, parent: None, - comment: match pc { Some(s) => Some(s.to_string()), None => None } } } + + rule pname() -> Password = &(word() _ word() _ num()? mode() _ num() _ date()) pr:word() _ pn:word() _ pl:num()? pm:mode() _ ps:num() _ pd:date() pc:comment()? + { Password::new(Some(pr), pn, pl, pm, ps, pd, pc) } + rule jname() -> Password = &(word() _ num()? mode() _ num() _ date()) pn:word() _ pl:num()? pm:mode() _ ps:num() _ pd:date() pc:comment()? + { Password::new(None, pn, pl, pm, ps, pd, pc) } + rule mname() -> Password = &(word() _ word() _ num()? mode() _ date()) pr:word() _ pn:word() _ pl:num()? pm:mode() _ pd:date() pc:comment()? + { Password::new(Some(pr), pn, pl, pm, 99, pd, pc) } + rule sname() -> Password = &(word() _ num()? mode() _ date()) pn:word() _ pl:num()? pm:mode() _ pd:date() pc:comment()? + { Password::new(None, pn, pl, pm, 99, pd, pc) } + rule date() -> NaiveDate = y:$("-"? ['0'..='9']*<1,4>) "-" m:$(['0'..='9']*<1,2>) "-" d:$(['0'..='9']*<1,2>) {? let year: i32 = match y.parse() { Ok(n) => n, Err(_) => return Err("year") }; let month: u32 = match m.parse() { Ok(n) => n, Err(_) => return Err("month") }; @@ -65,7 +64,8 @@ peg::parser! { rule ls_cmd() -> Command<'input> = "ls" { Command::Ls } rule add_cmd() -> Command<'input> = "add" _ name:name() { Command::Add(Rc::new(RefCell::new(name))) } rule error_cmd() -> Command<'input> = "error" _ e:$(([' '..='~'])+) { Command::Error(LKErr::Error(e)) } - rule mv_cmd() -> Command<'input> = "mv" _ name:word() _ folder:word() { Command::Mv(name.to_string(), folder.to_string()) } + rule mv_cmd() -> Command<'input> = "mv" _ name:word() _ folder:word() { Command::Mv(name, folder) } + rule comment_cmd() -> Command<'input> = "comment" _ name:word() c:comment()? { Command::Comment(name, c) } } } diff --git a/src/password.rs b/src/password.rs index a80f961..ccac691 100644 --- a/src/password.rs +++ b/src/password.rs @@ -2,29 +2,47 @@ use crate::structs::Mode; use chrono::naive::NaiveDate; use std::{cell::RefCell, rc::Rc}; +pub type Name = String; +pub type NameRef = Rc; +pub type Prefix = Option; +pub type Comment = Option; +pub type PasswordRef = Rc>; +pub type Parent = Option; +pub type Length = Option; +pub type Seq = u32; +pub type Date = NaiveDate; + #[derive(PartialEq, Debug)] pub struct Password { - pub parent: Option>>, - pub prefix: Option, - pub name: Rc, - pub length: Option, + pub parent: Parent, + pub prefix: Prefix, + pub name: NameRef, + pub length: Length, pub mode: Mode, - pub seq: u32, - pub date: NaiveDate, - pub comment: Option, + pub seq: Seq, + pub date: Date, + pub comment: Comment, } impl Password { - pub fn new(prefix: Option, name: String, mode: Mode, date: NaiveDate) -> Password { + pub fn new( + prefix: Prefix, + name: Name, + length: Length, + mode: Mode, + seq: Seq, + date: Date, + comment: Comment, + ) -> Password { Password { prefix, + name: Rc::new(name), + length, mode, date, + comment, parent: None, - name: Rc::new(name), - length: None, - seq: 99, - comment: None, + seq, } } } @@ -95,8 +113,11 @@ mod tests { let p1 = Rc::new(RefCell::new(Password::new( None, "p1".to_string(), + None, Mode::Regular, + 99, NaiveDate::from_ymd_opt(2022, 12, 3).unwrap(), + None, ))); p1.borrow_mut().parent = Some(p1.clone()); @@ -106,29 +127,41 @@ mod tests { let p2 = Rc::new(RefCell::new(Password::new( None, "p2".to_string(), + None, Mode::Regular, + 99, NaiveDate::from_ymd_opt(2022, 12, 3).unwrap(), + None, ))); p2.borrow_mut().parent = Some(p1.clone()); let p3 = Rc::new(RefCell::new(Password::new( None, "p3".to_string(), + None, Mode::Regular, + 99, NaiveDate::from_ymd_opt(2022, 12, 3).unwrap(), + None, ))); p3.borrow_mut().parent = Some(p2.clone()); let p4 = Rc::new(RefCell::new(Password::new( None, "p4".to_string(), + None, Mode::Regular, + 99, NaiveDate::from_ymd_opt(2022, 12, 3).unwrap(), + None, ))); p4.borrow_mut().parent = Some(p3.clone()); let p5 = Rc::new(RefCell::new(Password::new( None, "p5".to_string(), + None, Mode::Regular, + 99, NaiveDate::from_ymd_opt(2022, 12, 3).unwrap(), + None, ))); p5.borrow_mut().parent = Some(p4.clone()); diff --git a/src/repl.rs b/src/repl.rs index 93a5f14..cf31bc7 100644 --- a/src/repl.rs +++ b/src/repl.rs @@ -105,6 +105,15 @@ impl<'a> LKEval<'a> { self.state.borrow().fix_hierarchy(); } } + Command::Comment(name, comment) => match self.state.borrow().db.get(name) { + Some(pwd) => { + pwd.borrow_mut().comment = match comment { + Some(c) => Some(c.to_string()), + None => None, + } + } + None => out.push("error: password not found".to_string()), + }, Command::Help => { out.push("HELP".to_string()); } diff --git a/src/structs.rs b/src/structs.rs index 5915da9..e0bc14b 100644 --- a/src/structs.rs +++ b/src/structs.rs @@ -1,5 +1,4 @@ -use crate::password::Password; -use std::{cell::RefCell, rc::Rc}; +use crate::password::{Comment, Name, PasswordRef}; #[derive(thiserror::Error, Debug, PartialEq)] pub enum LKErr<'a> { @@ -13,9 +12,10 @@ pub enum LKErr<'a> { #[derive(PartialEq, Debug)] pub enum Command<'a> { - Add(Rc>), + Add(PasswordRef), Ls, - Mv(String, String), + Mv(Name, Name), + Comment(Name, Comment), Error(LKErr<'a>), Help, Quit,