initial commit

src/scanner.rs

@@ -0,0 +1,66 @@
+use std::path::PathBuf;
+
+use crate::{cli::App, database::File};
+use crate::database;
+use anyhow::Result;
+use glob::glob;
+use std::{fs, io};
+
+pub fn duplicates(app_opts: App) -> Result<Vec<File>> {
+    index_files(scan(app_opts)?);
+    database::duplicate_hashes()
+}
+
+fn scan(app_opts: App) -> Result<Vec<String>> {
+    let directory: String = app_opts
+        .dir
+        .unwrap_or(std::env::current_dir()?)
+        .as_os_str()
+        .to_str()
+        .unwrap()
+        .to_string();
+
+    let glob_patterns: Vec<PathBuf> = app_opts
+        .filetypes
+        .unwrap_or(String::from("*"))
+        .split(",")
+        .map(|filetype| format!("*.{}", filetype))
+        .map(|filetype| {
+            vec![directory.clone(), String::from("**"), filetype]
+                .iter()
+                .collect()
+        })
+        .collect();
+
+    let files: Vec<String> = glob_patterns
+        .into_iter()
+        .map(|glob_pattern| glob(&glob_pattern.as_os_str().to_str().unwrap()))
+        .map(|glob_result| glob_result.unwrap())
+        .flat_map(|file_vec| {
+            file_vec
+                .map(|x| x.unwrap().as_os_str().to_str().unwrap().to_string())
+                .collect::<Vec<String>>()
+        })
+        .collect();
+
+    Ok(files)
+}
+
+fn index_files(files: Vec<String>) {
+    files
+        .into_iter()
+        .map(|file| {
+            let hash = hash_file(&file).unwrap();
+            database::File { path: file, hash }
+        })
+        .for_each(|file| {
+            database::put(file).unwrap();
+        });
+}
+
+pub fn hash_file(filepath: &str) -> Result<String> {
+    let file = fs::read(filepath)?;
+    let hash = sha256::digest(&*file);
+
+    Ok(hash)
+}