mps struct begin

src/main.rs

@@ -27,13 +27,13 @@ enum Commands {
     /// Lists all ids stored in db
     List,
 
-    /// Show content of item
+    /// Show content of an item
     Show {
         #[arg(value_name="item_id")]
         id: String
     },
     
-    /// Adds new item with unique id to the db
+    /// Adds new item with unique id to the storage
     Add {
         #[arg(value_name="item_id")]
         id: String
@@ -68,23 +68,16 @@ fn get_prompt(question: &str) -> io::Result<PROMPT> {
 
 }
 
-// TODO: change password functionality
-// TODO: return storage inited
-fn login() -> io::Result<String> {
-    print!("Enter passphrase for storage: ");
-    io::stdout().flush()?;
-    // TODO: check in db
-    // TODO: return error if db is not inited
-    let password = rpassword::read_password()?;
-    if password != "pass" {
-        return Err(io::Error::new(io::ErrorKind::InvalidData, "Wrong passphrase"));
-    }
-    print!("\x1B[1A\x1B[2K");  // Move cursor up one line and clear that line
-    io::stdout().flush()?;     // Ensure the changes are reflected immediately
-    Ok(String::from(password))
+struct MPS {
+    storage: Option<Storage>
 }
 
-fn init() -> io::Result<()> {
+impl MPS {
+    pub fn new() -> MPS {
+        MPS { storage: None }
+    }
+
+    fn init() -> io::Result<()> {
         if Storage::is_inited() {
             return Err(io::Error::new(io::ErrorKind::AlreadyExists, "Reinitialization attempted"));
         }
@@ -99,26 +92,45 @@ fn init() -> io::Result<()> {
         }
         Storage::init(ps)?;
         Ok(())
-}
+    }
 
-fn list() -> io:: Result<()> {
+    // TODO: change password functionality
+    // TODO: set self.storage inited
+    pub fn login(&mut self) -> io::Result<String> {
+        // TODO: do nothing if already storate inited
+        print!("Enter passphrase for storage: ");
+        io::stdout().flush()?;
+        // TODO: check in db
+        // TODO: return error if db is not inited
+        let password = rpassword::read_password()?;
+        if password != "pass" {
+            return Err(io::Error::new(io::ErrorKind::InvalidData, "Wrong passphrase"));
+        }
+        print!("\x1B[1A\x1B[2K");  // Move cursor up one line and clear that line
+        io::stdout().flush()?;     // Ensure the changes are reflected immediately
+        Ok(String::from(password))
+    }
+
+    fn list(&mut self) -> io:: Result<()> {
         // TODO: get storage from login
-    let passphrase = login()?;
+        let passphrase = self.login()?;
+        // TODO: use self.storage
         let st = Storage::from_db(passphrase)?;
         for id in st.ids() {
             println!("{}", id);
         }
         Ok(())
-}
+    }
 
-fn add(id: &String) -> io::Result<()> {
-    let passphrase = login()?;
+    fn add(&mut self, id: &String) -> io::Result<()> {
+        let passphrase = self.login()?;
+        // TODO: use self.storage
         let mut st = Storage::from_db(passphrase)?;
         if st.contains(id) {
             let question = format!("Item [{}] exist. Do you want to edit it instead?", id);
             match get_prompt(&question)? {
                 PROMPT::YES => {
-                edit(id)?;
+                    self.edit(id)?;
                     return Ok(());
                 },
                 PROMPT::NO => return Ok(()),
@@ -130,17 +142,18 @@ fn add(id: &String) -> io::Result<()> {
         st.dump()?;
 
         Ok(())
-}
+    }
 
-fn edit(id: &String) -> io::Result<()> {
+    fn edit(&mut self, id: &String) -> io::Result<()> {
         // TODO: implement
-    let passphrase = login()?;
+        let passphrase = self.login()?;
+        // TODO: use self.storage
         let mut st = Storage::from_db(passphrase)?;
         if !st.contains(id) {
             let question = format!("Item [{}] exist. Do you want to add it instead?", id);
             match get_prompt(&question)? {
                 PROMPT::YES => {
-                add(id)?;
+                    self.add(id)?;
                     return Ok(());
                 }
                 PROMPT::NO => return Ok(())
@@ -154,11 +167,12 @@ fn edit(id: &String) -> io::Result<()> {
         st.dump()?;
 
         Ok(())
-} 
+    }
 
-fn show(id: &String) -> io::Result<()> {
+    fn show(&mut self, id: &String) -> io::Result<()> {
         // TODO: get storage from login
-    let passphrase = login()?;
+        let passphrase = self.login()?;
+        // TODO: use self.storage
         let st = Storage::from_db(passphrase)?;
         if !st.contains(id) {
             return Err(io::Error::new(
@@ -168,39 +182,42 @@ fn show(id: &String) -> io::Result<()> {
         }
 
         let item = st.get(id);
-    //editor::open_to_show(&item.content)?;
-    println!("---------");
-    println!("{}", item.content);
-    println!("---------");
+        editor::open_to_show(&item.content)?;
         Ok(())
+    }
 }
 
 fn run_command() -> io::Result<()> {
     let cli = Cli::parse();
     match &cli.command {
         Some(Commands::Init) => {
-            init()?;
+            MPS::init()?;
         }
         Some(Commands::List) => {
-            list()?;
+            let mut mps = MPS::new();
+            mps.list()?;
         }
         Some(Commands::Show { id }) => {
-            show(id)?;
+            let mut mps = MPS::new();
+            mps.show(id)?;
         }
         Some(Commands::Add { id }) => {
-            add(id)?;
+            let mut mps = MPS::new();
+            mps.add(id)?;
         }
         Some(Commands::Edit { id }) => {
-            edit(id)?
+            let mut mps = MPS::new();
+            mps.edit(id)?
         }
         None => {
             if !Storage::is_inited() {
                 match get_prompt("Do you want to init your storage?")? {
-                    PROMPT::YES => init()?,
+                    PROMPT::YES => MPS::init()?,
                     PROMPT::NO => Storage::print_init_hint(),
                 }
             } else {
-                list()?
+                let mut mps = MPS::new();
+                mps.list()?
             }
         }
     }