Switch to excluding by crate name instead

Author: carols10cents

src/config.rs

@@ -25,7 +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 excluded_crate_names: Vec<String>,
     pub domain_name: String,
     pub allowed_origins: Vec<String>,
     pub downloads_persist_interval_ms: usize,
@@ -85,14 +85,10 @@ 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") {
+        let excluded_crate_names = match env_optional::<String>("EXCLUDED_CRATE_NAMES") {
             None => vec![],
             Some(s) if s.is_empty() => vec![],
-            Some(s) => s
-                .split(',')
-                .map(|n| n.parse())
-                .collect::<Result<Vec<_>, _>>()
-                .unwrap_or_default(),
+            Some(s) => s.split(',').map(String::from).collect(),
         };
         Server {
             db: DatabasePools::full_from_environment(&base),
@@ -107,7 +103,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,
+            excluded_crate_names,
             domain_name: domain_name(),
             allowed_origins,
             downloads_persist_interval_ms: dotenv::var("DOWNLOADS_PERSIST_INTERVAL_MS")

src/controllers/krate/metadata.rs

@@ -24,7 +24,7 @@ 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;
+    use diesel::dsl::all;
 
     let config = &req.app().config;
 
@@ -83,9 +83,9 @@ pub fn summary(req: &mut dyn RequestExt) -> EndpointResult {
     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)));
+    if !config.excluded_crate_names.is_empty() {
+        most_recently_downloaded_query =
+            most_recently_downloaded_query.filter(name.ne(all(&config.excluded_crate_names)));
     }
     let most_recently_downloaded = most_recently_downloaded_query
         .then_order_by(recent_crate_downloads::downloads.desc())

src/tests/krate/summary.rs

@@ -107,3 +107,42 @@ fn summary_new_crates() {
 
     assert_eq!(json.new_crates.len(), 5);
 }
+
+#[test]
+fn excluded_crate_id() {
+    let (app, anon, user) = TestApp::init()
+        .with_config(|config| {
+            config.excluded_crate_names = vec![
+                "most_recent_downloads".into(),
+                // make sure no error occurs with a crate name that doesn't exist and that the name
+                // matches are exact, not substrings
+                "downloads".into(),
+            ];
+        })
+        .with_user();
+    let user = user.as_model();
+    app.db(|conn| {
+        CrateBuilder::new("some_downloads", user.id)
+            .version(VersionBuilder::new("0.1.0"))
+            .description("description")
+            .keyword("popular")
+            .category("cat1")
+            .downloads(20)
+            .recent_downloads(10)
+            .expect_build(conn);
+
+        CrateBuilder::new("most_recent_downloads", user.id)
+            .version(VersionBuilder::new("0.2.0"))
+            .keyword("popular")
+            .category("cat1")
+            .downloads(5000)
+            .recent_downloads(50)
+            .expect_build(conn);
+    });
+
+    let json: SummaryResponse = anon.get("/api/v1/summary").good();
+
+    assert_eq!(json.most_recently_downloaded.len(), 1);
+    assert_eq!(json.most_recently_downloaded[0].name, "some_downloads");
+    assert_eq!(json.most_recently_downloaded[0].recent_downloads, Some(10));
+}

src/tests/util/test_app.rs

@@ -300,7 +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![],
+        excluded_crate_names: vec![],
         domain_name: "crates.io".into(),
         allowed_origins: Vec::new(),
         downloads_persist_interval_ms: 1000,