Merge pull request #78 from catap/only-the-best

Introduce `--only-the-best` option

Author: westandskif

src/config.rs

@@ -251,6 +251,10 @@ pub struct Config {
     )]
     pub top_mirrors_number_to_retest: usize,
 
+    /// Only one, the best mirror
+    #[arg(env = "RATE_MIRRORS_ONLY_THE_BEST", long)]
+    pub only_the_best: bool,
+
     /// Filename to save the output to in case of success
     #[arg(env = "RATE_MIRRORS_SAVE", long = "save", verbatim_doc_comment)]
     pub save_to_file: Option<String>,

src/main.rs

@@ -93,6 +93,7 @@ impl<'a, T: LogFormatter> OutputSink<'a, T> {
 
 fn main() -> Result<(), AppError> {
     let config = Arc::new(Config::parse());
+    let only_the_best = config.only_the_best;
     if !config.allow_root && Uid::effective().is_root() {
         return Err(AppError::Root);
     }
@@ -157,8 +158,12 @@ fn main() -> Result<(), AppError> {
 
         output.display_comment(format!("FINISHED AT: {}", Local::now()));
 
-        for result in results.into_iter() {
-            output.display_mirror(&result.item);
+        if only_the_best {
+                output.display_mirror(&results.first().unwrap().item);
+        } else {
+            for result in results.into_iter() {
+                output.display_mirror(&result.item);
+            }
         }
     }