Make pedantic clippy happy

1 files changed, 67 insertions, 72 deletions

diff --git a/src/command/import.rs b/src/command/import.rs
index a593ef61..77db1c84 100644
--- a/src/command/import.rs
+++ b/src/command/import.rs
@@ -5,13 +5,13 @@ use eyre::{eyre, Result};
 use home::home_dir;
 use structopt::StructOpt;
 
-use crate::local::database::{Database, SqliteDatabase};
+use crate::local::database::{Database, Sqlite};
 use crate::local::history::History;
-use crate::local::import::ImportZsh;
+use crate::local::import::Zsh;
 use indicatif::ProgressBar;
 
 #[derive(StructOpt)]
-pub enum ImportCmd {
+pub enum Cmd {
     #[structopt(
         about="import history for the current shell",
         aliases=&["a", "au", "aut"],
@@ -25,94 +25,89 @@ pub enum ImportCmd {
     Zsh,
 }
 
-impl ImportCmd {
-    fn import_zsh(&self, db: &mut SqliteDatabase) -> Result<()> {
-        // oh-my-zsh sets HISTFILE=~/.zhistory
-        // zsh has no default value for this var, but uses ~/.zhistory.
-        // we could maybe be smarter about this in the future :)
-
-        let histpath = env::var("HISTFILE");
+impl Cmd {
+    pub fn run(&self, db: &mut Sqlite) -> Result<()> {
+        println!("        A'Tuin       ");
+        println!("=====================");
+        println!("          \u{1f30d}         ");
+        println!("       \u{1f418}\u{1f418}\u{1f418}\u{1f418}      ");
+        println!("          \u{1f422}         ");
+        println!("=====================");
+        println!("Importing history...");
 
-        let histpath = match histpath {
-            Ok(p) => PathBuf::from(p),
-            Err(_) => {
-                let mut home = home_dir().unwrap();
-                home.push(".zhistory");
+        match self {
+            Self::Auto => {
+                let shell = env::var("SHELL").unwrap_or_else(|_| String::from("NO_SHELL"));
 
-                home
+                if shell.as_str() == "/bin/zsh" {
+                    println!("Detected ZSH");
+                    import_zsh(db)
+                } else {
+                    println!("cannot import {} history", shell);
+                    Ok(())
+                }
             }
-        };
 
-        if !histpath.exists() {
-            return Err(eyre!(
-                "Could not find history file at {}, try setting $HISTFILE",
-                histpath.to_str().unwrap()
-            ));
+            Self::Zsh => import_zsh(db),
         }
+    }
+}
 
-        let zsh = ImportZsh::new(histpath.to_str().unwrap())?;
+fn import_zsh(db: &mut Sqlite) -> Result<()> {
+    // oh-my-zsh sets HISTFILE=~/.zhistory
+    // zsh has no default value for this var, but uses ~/.zhistory.
+    // we could maybe be smarter about this in the future :)
 
-        let progress = ProgressBar::new(zsh.loc);
+    let histpath = env::var("HISTFILE");
 
-        let buf_size = 100;
-        let mut buf = Vec::<History>::with_capacity(buf_size);
+    let histpath = if let Ok(p) = histpath {
+        PathBuf::from(p)
+    } else {
+        let mut home = home_dir().unwrap();
+        home.push(".zhistory");
 
-        for i in zsh {
-            match i {
-                Ok(h) => {
-                    buf.push(h);
-                }
-                Err(e) => {
-                    error!("{}", e);
-                    continue;
-                }
-            }
+        home
+    };
+
+    if !histpath.exists() {
+        return Err(eyre!(
+            "Could not find history file at {}, try setting $HISTFILE",
+            histpath.to_str().unwrap()
+        ));
+    }
+
+    let zsh = Zsh::new(histpath.to_str().unwrap())?;
+
+    let progress = ProgressBar::new(zsh.loc);
 
-            if buf.len() == buf_size {
-                db.save_bulk(&buf)?;
-                progress.inc(buf.len() as u64);
+    let buf_size = 100;
+    let mut buf = Vec::<History>::with_capacity(buf_size);
 
-                buf = Vec::<History>::with_capacity(buf_size);
+    for i in zsh {
+        match i {
+            Ok(h) => {
+                buf.push(h);
+            }
+            Err(e) => {
+                error!("{}", e);
+                continue;
             }
         }
 
-        if buf.len() > 0 {
+        if buf.len() == buf_size {
             db.save_bulk(&buf)?;
             progress.inc(buf.len() as u64);
-        }
 
-        progress.finish_with_message("Imported history!");
-
-        Ok(())
+            buf = Vec::<History>::with_capacity(buf_size);
+        }
     }
 
-    pub fn run(&self, db: &mut SqliteDatabase) -> Result<()> {
-        println!("        A'Tuin       ");
-        println!("=====================");
-        println!("          🌍         ");
-        println!("       🐘🐘🐘🐘      ");
-        println!("          🐢         ");
-        println!("=====================");
-        println!("Importing history...");
-
-        match self {
-            ImportCmd::Auto => {
-                let shell = env::var("SHELL").unwrap_or(String::from("NO_SHELL"));
-
-                match shell.as_str() {
-                    "/bin/zsh" => {
-                        println!("Detected ZSH");
-                        self.import_zsh(db)
-                    }
+    if !buf.is_empty() {
+        db.save_bulk(&buf)?;
+        progress.inc(buf.len() as u64);
+    }
 
-                    _ => {
-                        println!("cannot import {} history", shell);
-                        Ok(())
-                    }
-                }
-            }
+    progress.finish_with_message("Imported history!");
 
-            ImportCmd::Zsh => self.import_zsh(db),
-        }
-    }
+    Ok(())
 }