Add an environment variable to let us exclude crate IDs from the recent downloads list

Author: carols10cents

src/config.rs

@@ -25,6 +25,7 @@ pub struct Server {
     pub blocked_traffic: Vec<(String, Vec<String>)>,
     pub max_allowed_page_offset: u32,
     pub page_offset_ua_blocklist: Vec<String>,
+    pub excluded_crate_ids: Vec<i32>,
     pub domain_name: String,
     pub allowed_origins: Vec<String>,
     pub downloads_persist_interval_ms: usize,
@@ -84,6 +85,15 @@ impl Default for Server {
             Some(s) => s.split(',').map(String::from).collect(),
         };
         let base = Base::from_environment();
+        let excluded_crate_ids: Vec<i32> = match env_optional::<String>("EXCLUDED_CRATE_IDS") {
+            None => vec![],
+            Some(s) if s.is_empty() => vec![],
+            Some(s) => s
+                .split(',')
+                .map(|n| n.parse())
+                .collect::<Result<Vec<_>, _>>()
+                .unwrap_or_default(),
+        };
         Server {
             db: DatabasePools::full_from_environment(&base),
             base,
@@ -97,6 +107,7 @@ impl Default for Server {
             blocked_traffic: blocked_traffic(),
             max_allowed_page_offset: env_optional("WEB_MAX_ALLOWED_PAGE_OFFSET").unwrap_or(200),
             page_offset_ua_blocklist,
+            excluded_crate_ids,
             domain_name: domain_name(),
             allowed_origins,
             downloads_persist_interval_ms: dotenv::var("DOWNLOADS_PERSIST_INTERVAL_MS")

src/controllers/krate/metadata.rs

@@ -24,6 +24,9 @@ use crate::models::krate::ALL_COLUMNS;
 /// Handles the `GET /summary` route.
 pub fn summary(req: &mut dyn RequestExt) -> EndpointResult {
     use crate::schema::crates::dsl::*;
+    use diesel::dsl::any;
+
+    let config = &req.app().config;
 
     let conn = req.db_read_only()?;
     let num_crates: i64 = crates.count().get_result(&*conn)?;
@@ -77,8 +80,14 @@ pub fn summary(req: &mut dyn RequestExt) -> EndpointResult {
         .limit(10)
         .load(&*conn)?;
 
-    let most_recently_downloaded = crates
+    let mut most_recently_downloaded_query = crates
         .inner_join(recent_crate_downloads::table)
+        .into_boxed();
+    if !config.excluded_crate_ids.is_empty() {
+        most_recently_downloaded_query = most_recently_downloaded_query
+            .filter(recent_crate_downloads::crate_id.ne(any(&config.excluded_crate_ids)));
+    }
+    let most_recently_downloaded = most_recently_downloaded_query
         .then_order_by(recent_crate_downloads::downloads.desc())
         .select(selection)
         .limit(10)

src/tests/util/test_app.rs

@@ -300,6 +300,7 @@ fn simple_config() -> config::Server {
         blocked_traffic: Default::default(),
         max_allowed_page_offset: 200,
         page_offset_ua_blocklist: vec![],
+        excluded_crate_ids: vec![],
         domain_name: "crates.io".into(),
         allowed_origins: Vec::new(),
         downloads_persist_interval_ms: 1000,