Create MissingBlobs struct

Author: joshchoo

src/lib.rs

@@ -4,29 +4,35 @@ use std::collections::HashMap;
 use std::fs;
 use std::path::{Path, PathBuf};
 
-pub struct Config {
-    pub recursive: bool,
+pub struct MissingBlobs {
+    recursive: bool,
 }
 
-pub fn run(paths: &[&str], config: Config) {
-    let file_paths: Vec<PathBuf> = if config.recursive {
-        find_files_recursively(&paths)
-    } else {
-        find_files(&paths)
-    };
-
-    let blob_paths: Vec<&PathBuf> = file_paths
-        .iter()
-        .filter(|path| match path.extension() {
-            // Assume that valid blobs have ".so" extension.
-            Some(ext) => ext == "so",
-            None => false,
-        })
-        .collect();
+impl MissingBlobs {
+    pub fn new(recursive: bool) -> Self {
+        Self { recursive }
+    }
 
-    let blobs_to_dependencies = get_dependencies(&blob_paths);
-    let missing_blobs = identify_missing(&blobs_to_dependencies);
-    display_missing_blobs(&missing_blobs);
+    pub fn run(&self, paths: &[&str]) {
+        let file_paths: Vec<PathBuf> = if self.recursive {
+            find_files_recursively(&paths)
+        } else {
+            find_files(&paths)
+        };
+
+        let blob_paths: Vec<&PathBuf> = file_paths
+            .iter()
+            .filter(|path| match path.extension() {
+                // Assume that valid blobs have ".so" extension.
+                Some(ext) => ext == "so",
+                None => false,
+            })
+            .collect();
+
+        let blobs_to_dependencies = get_dependencies(&blob_paths);
+        let missing_blobs = identify_missing(&blobs_to_dependencies);
+        display_missing_blobs(&missing_blobs);
+    }
 }
 
 fn find_files(paths: &[&str]) -> Vec<PathBuf> {

src/main.rs

@@ -1,4 +1,4 @@
-use aosp_missing_blobs::{run, Config};
+use aosp_missing_blobs::MissingBlobs;
 use clap::{crate_description, crate_name, crate_version, App, Arg};
 
 fn main() {
@@ -20,11 +20,7 @@ fn main() {
         .get_matches();
 
     let paths = matches.values_of("PATHS").unwrap().collect::<Vec<_>>();
+    let recursive = matches.is_present("recursive");
 
-    run(
-        &paths,
-        Config {
-            recursive: matches.is_present("recursive"),
-        },
-    );
+    MissingBlobs::new(recursive).run(&paths);
 }