Update

Author: DevHyperCoder

src/bookmark_query.rs

@@ -1,9 +1,20 @@
 use serde::{Deserialize, Serialize};
 use structopt::StructOpt;
+
 #[derive(Serialize, Deserialize, Debug, Clone, StructOpt)]
 pub struct BookmarkQuery {
     #[structopt(long, short)]
     pub id: Option<u32>,
     #[structopt(long, short)]
     pub name: Option<String>,
 }
+
+#[derive(Serialize, Deserialize, Debug, Clone, StructOpt)]
+pub struct BookmarkUpdateQuery {
+    #[structopt(long, short)]
+    pub id: u32,
+    #[structopt(long, short)]
+    pub name: Option<String>,
+    #[structopt(long, short)]
+    pub link: Option<String>,
+}

src/commands.rs

@@ -1,4 +1,9 @@
-use crate::{bookmark::Bookmark, bookmark_query::BookmarkQuery, data::Data, error::Result};
+use crate::{
+    bookmark::Bookmark,
+    bookmark_query::{BookmarkQuery, BookmarkUpdateQuery},
+    data::Data,
+    error::Result,
+};
 
 /// Insert commands
 /// Adds the bookmark to the data list and increments the last id
@@ -53,3 +58,22 @@ pub fn remove(data: &mut Data, query: BookmarkQuery) -> Result<Vec<Bookmark>> {
     data.write_to_file()?;
     Ok(removed)
 }
+
+pub fn update(data: &mut Data, query: BookmarkUpdateQuery) -> Result<()> {
+    data.bookmarks
+        .iter_mut()
+        .map(|e| {
+            if e.id != query.id {
+                return;
+            }
+
+            let name = query.name.as_ref().unwrap_or(&e.name);
+            let link = query.link.as_ref().unwrap_or(&e.link);
+
+            e.name = name.to_string();
+            e.link = link.to_string();
+        })
+        .for_each(drop);
+
+    data.write_to_file()
+}

src/config.rs

@@ -1,6 +1,6 @@
 use structopt::StructOpt;
 
-use crate::bookmark_query::BookmarkQuery;
+use crate::bookmark_query::{BookmarkQuery, BookmarkUpdateQuery};
 
 #[derive(StructOpt, Debug)]
 pub struct Config {
@@ -30,4 +30,8 @@ pub enum SubOpt {
         #[structopt(flatten)]
         query: BookmarkQuery,
     },
+    Update {
+        #[structopt(flatten)]
+        query: BookmarkUpdateQuery,
+    },
 }

src/lib.rs

@@ -6,7 +6,7 @@ pub mod data;
 pub mod error;
 pub mod parser;
 
-use commands::{insert, list, remove};
+use commands::{insert, list, remove, update};
 use config::{Config, SubOpt};
 use data::read_data_file;
 
@@ -37,6 +37,7 @@ pub fn run() -> Result<()> {
                 i.colored_fmt()
             }
         }
+        SubOpt::Update { query } => update(&mut data, query)?,
     }
     Ok(())
 }