models/version: Replace derive_builder with bon

Author: Turbo87

src/controllers/krate/publish.rs

@@ -360,19 +360,18 @@ pub async fn publish(app: AppState, req: BytesRequest) -> AppResult<Json<GoodCra
 
             // Persist the new version of this crate
             let version = NewVersion::builder(krate.id, &version_string)
-                .features(&features)?
-                .license(license.as_deref())
+                .features(serde_json::to_value(&features)?)
+                .maybe_license(license.as_deref())
                 // Downcast is okay because the file length must be less than the max upload size
                 // to get here, and max upload sizes are way less than i32 max
                 .size(content_length as i32)
                 .published_by(user.id)
                 .checksum(&hex_cksum)
-                .links(package.links.as_deref())
-                .rust_version(rust_version.as_deref())
+                .maybe_links(package.links.as_deref())
+                .maybe_rust_version(rust_version.as_deref())
                 .has_lib(tarball_info.manifest.lib.is_some())
                 .bin_names(bin_names.as_slice())
                 .build()
-                .map_err(|error| internal(error.to_string()))?
                 .save(conn, &verified_email_address)?;
 
             insert_version_owner_action(

src/models/version.rs

@@ -1,8 +1,8 @@
 use std::collections::BTreeMap;
 
+use bon::Builder;
 use chrono::NaiveDateTime;
 use crates_io_index::features::FeaturesMap;
-use derive_builder::Builder;
 use diesel::prelude::*;
 use serde::Deserialize;
 
@@ -80,58 +80,26 @@ impl Version {
 #[derive(Insertable, Debug, Builder)]
 #[diesel(table_name = versions, check_for_backend(diesel::pg::Pg))]
 pub struct NewVersion<'a> {
+    #[builder(start_fn)]
     crate_id: i32,
+    #[builder(start_fn)]
     num: &'a str,
-    #[builder(default)]
     created_at: Option<&'a NaiveDateTime>,
-    #[builder(default, setter(strip_option))]
     yanked: Option<bool>,
-    #[builder(
-        default = "serde_json::Value::Object(Default::default())",
-        setter(custom)
-    )]
+    #[builder(default = serde_json::Value::Object(Default::default()))]
     features: serde_json::Value,
-    #[builder(default)]
     license: Option<&'a str>,
-    #[builder(default, setter(name = "size"))]
+    #[builder(default, name = "size")]
     crate_size: i32,
     published_by: i32,
     checksum: &'a str,
-    #[builder(default)]
     links: Option<&'a str>,
-    #[builder(default)]
     rust_version: Option<&'a str>,
-    #[builder(default, setter(strip_option))]
     pub has_lib: Option<bool>,
-    #[builder(default, setter(strip_option))]
     pub bin_names: Option<&'a [&'a str]>,
 }
 
-impl NewVersionBuilder<'_> {
-    pub fn features(
-        &mut self,
-        features: &BTreeMap<String, Vec<String>>,
-    ) -> serde_json::Result<&mut Self> {
-        self.features = Some(serde_json::to_value(features)?);
-        Ok(self)
-    }
-
-    /// Set the `checksum` field to a basic dummy value.
-    pub fn dummy_checksum(&mut self) -> &mut Self {
-        const DUMMY_CHECKSUM: &str =
-            "0000000000000000000000000000000000000000000000000000000000000000";
-
-        self.checksum(DUMMY_CHECKSUM)
-    }
-}
-
 impl NewVersion<'_> {
-    pub fn builder(crate_id: i32, version: &str) -> NewVersionBuilder<'_> {
-        let mut builder = NewVersionBuilder::default();
-        builder.crate_id(crate_id).num(version);
-        builder
-    }
-
     pub fn save(&self, conn: &mut impl Conn, published_by_email: &str) -> AppResult<Version> {
         use diesel::dsl::exists;
         use diesel::{insert_into, select};

src/tests/builders/version.rs

@@ -5,7 +5,6 @@ use crate::{
 };
 use std::collections::BTreeMap;
 
-use crate::util::errors::internal;
 use chrono::NaiveDateTime;
 use diesel::prelude::*;
 
@@ -102,17 +101,16 @@ impl VersionBuilder {
         let version = self.num.to_string();
 
         let new_version = NewVersion::builder(crate_id, &version)
-            .features(&self.features)?
-            .license(self.license.as_deref())
+            .features(serde_json::to_value(&self.features)?)
+            .maybe_license(self.license.as_deref())
             .size(self.size)
             .published_by(published_by)
             .checksum(&self.checksum)
-            .links(self.links.as_deref())
-            .rust_version(self.rust_version.as_deref())
+            .maybe_links(self.links.as_deref())
+            .maybe_rust_version(self.rust_version.as_deref())
             .yanked(self.yanked)
-            .created_at(self.created_at.as_ref())
-            .build()
-            .map_err(|error| internal(error.to_string()))?;
+            .maybe_created_at(self.created_at.as_ref())
+            .build();
 
         let vers = new_version.save(connection, "[email protected]")?;
 

src/typosquat/test_util.rs

@@ -53,9 +53,8 @@ pub mod faker {
 
         let version = NewVersion::builder(krate.id, "1.0.0")
             .published_by(user.id)
-            .dummy_checksum()
+            .checksum("0000000000000000000000000000000000000000000000000000000000000000")
             .build()
-            .unwrap()
             .save(conn, "[email protected]")
             .unwrap();
 

src/worker/jobs/downloads/update_metadata.rs

@@ -126,9 +126,8 @@ mod tests {
 
         let version = NewVersion::builder(krate.id, "1.0.0")
             .published_by(user_id)
-            .dummy_checksum()
-            .build()
-            .unwrap();
+            .checksum("0000000000000000000000000000000000000000000000000000000000000000")
+            .build();
 
         let version = version.save(conn, "[email protected]").unwrap();
         (krate, version)