Kiyomichi Kosaka
42 lines
1.2 KiB
Rust
42 lines
1.2 KiB
Rust
use std::{cell::RefCell, rc::Rc};
|
|
use std::collections::HashMap;
|
|
use regex::{Regex, Captures};
|
|
use crate::password::{Password, fix_password_recursion};
|
|
|
|
#[derive(PartialEq, Debug)]
|
|
pub struct LK {
|
|
pub db: HashMap<Rc<String>, Rc<RefCell<Password>>>,
|
|
}
|
|
|
|
impl LK {
|
|
pub fn new() -> Self {
|
|
Self { db: HashMap::new() }
|
|
}
|
|
|
|
pub fn fix_hierarchy(&self) {
|
|
lazy_static! {
|
|
static ref RE: Regex = Regex::new(r"\s*\^([!-~]+)").unwrap();
|
|
}
|
|
for (_, name) in &self.db {
|
|
if name.borrow().comment.is_some() {
|
|
let mut folder: Option<String> = None;
|
|
let prev_comment = name.borrow().comment.as_ref().unwrap().clone();
|
|
let comment = RE.replace(prev_comment.as_str(), |c: &Captures| { folder = Some(c[1].to_string()); "" });
|
|
if folder.is_some() {
|
|
let folder_name = folder.unwrap();
|
|
for (_, entry) in &self.db {
|
|
if *entry.borrow().name == *folder_name {
|
|
let mut tmp = name.borrow_mut();
|
|
tmp.parent = Some(entry.clone());
|
|
if comment.len() == 0 { tmp.comment = None }
|
|
else { tmp.comment = Some(comment.to_string()) }
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
fix_password_recursion(name.clone());
|
|
}
|
|
}
|
|
}
|