Allow import and use common insert method

src/main.rs

@@ -152,6 +152,14 @@ fn main() -> Result<(), Box<dyn Error>> {
                         .action(ArgAction::SetTrue),
                 ),
         )
+        .subcommand(
+            Command::new("import").arg_required_else_help(true).arg(
+                Arg::new("files")
+                    .help("import document files")
+                    .required(true)
+                    .action(ArgAction::Append),
+            ),
+        )
         .subcommand(
             Command::new("index")
                 .about("Reindex the database")
@@ -327,7 +335,7 @@ fn main() -> Result<(), Box<dyn Error>> {
             attachments: Vec::new(),
         };
 
-        snip::insert_snip(&conn, &s)?;
+        s.insert(&conn)?;
         s.index(&conn)?;
         if sub_matches.get_flag("verbose") {
             print!("{}", s.text);
@@ -492,6 +500,34 @@ fn main() -> Result<(), Box<dyn Error>> {
         }
     }
 
+    // IMPORT
+    if let Some(("import", sub_matches)) = matches.subcommand() {
+        if let Some(files) = sub_matches.get_many::<String>("files") {
+            let mut errors = false;
+            // Note: attachments are ignored and not imported
+            for file in files.into_iter() {
+                print!("importing {:?}...", file);
+
+                let mut s = snip::from_file(file)?;
+                if snip::get_from_uuid(&conn, &s.uuid).is_ok() {
+                    println!("refusing duplicate insert {}", s.uuid);
+                    errors = true;
+                    // proceed but notify of errors after iterations
+                    continue;
+                }
+
+                // check for existing id to avoid duplicates
+                s.insert(&conn)?;
+                // always index after import
+                s.index(&conn)?;
+                println!("success");
+            }
+            if errors {
+                eprintln!("WARNING: Some documents were not imported.");
+            }
+        }
+    }
+
     // INDEX
     if let Some(("index", _)) = matches.subcommand() {
         // creation is conditional on non-existence

src/snip/doc.rs

@@ -141,6 +141,20 @@ impl Snip {
         Ok(())
     }
 
+    /// Inserts a new document to the database
+    pub fn insert(&self, conn: &Connection) -> Result<(), Box<dyn Error>> {
+        let mut stmt =
+            conn.prepare("INSERT INTO snip(uuid, timestamp, name, data) VALUES (?1, ?2, ?3, ?4)")?;
+        stmt.execute([
+            self.uuid.to_string(),
+            self.timestamp.to_rfc3339(),
+            self.name.clone(),
+            self.text.clone(),
+        ])?;
+
+        Ok(())
+    }
+
     pub fn print(&self) {
         println!(
             "uuid: {}\nname: {}\ntimestamp: {}\n----",
@@ -475,20 +489,6 @@ pub fn index_all_items(conn: &Connection) -> Result<(), Box<dyn Error>> {
     Ok(())
 }
 
-/// Adds a new document to the database
-pub fn insert_snip(conn: &Connection, s: &Snip) -> Result<(), Box<dyn Error>> {
-    let mut stmt =
-        conn.prepare("INSERT INTO snip(uuid, timestamp, name, data) VALUES (?1, ?2, ?3, ?4)")?;
-    stmt.execute([
-        s.uuid.to_string(),
-        s.timestamp.to_rfc3339(),
-        s.name.clone(),
-        s.text.clone(),
-    ])?;
-
-    Ok(())
-}
-
 fn parse_text(data: &str) -> Result<String, Box<dyn Error>> {
     let lines: Vec<&str> = data.split('\n').collect();
 
@@ -803,7 +803,7 @@ mod tests {
     }
 
     #[test]
-    fn test_insert_snip() -> Result<(), Box<dyn Error>> {
+    fn test_insert_new() -> Result<(), Box<dyn Error>> {
         let conn = prepare_database().expect("preparing in-memory database");
         let id = Uuid::new_v4();
 
@@ -815,7 +815,7 @@ mod tests {
             analysis: SnipAnalysis { words: Vec::new() },
             attachments: Vec::new(),
         };
-        insert_snip(&conn, &s)?;
+        s.insert(&conn)?;
 
         // verify
         let mut stmt = conn.prepare("SELECT uuid FROM snip WHERE uuid = ?")?;
@@ -878,6 +878,29 @@ mod tests {
         Ok(())
     }
 
+    #[test]
+    fn test_insert() -> Result<(), Box<dyn Error>> {
+        let conn = prepare_database()?;
+        let id = Uuid::try_parse(ID_STR)?;
+
+        let s = snip::get_from_uuid(&conn, &id)?;
+        // first remove from database
+        remove_snip(&conn, id)?;
+
+        // verify removal
+        if snip::get_from_uuid(&conn, &id).is_ok() {
+            panic!("expected missing document, got document with id {}", id);
+        }
+
+        // insert and verify presence
+        s.insert(&conn)?;
+        if snip::get_from_uuid(&conn, &id).is_err() {
+            panic!("expected document, got Err searching for id {}", id);
+        }
+
+        Ok(())
+    }
+
     #[test]
     fn test_update() -> Result<(), Box<dyn Error>> {
         let conn = prepare_database()?;