models/owner: Derive bon::Builder for CrateOwner struct

Author: Turbo87

src/controllers/krate/owners.rs

@@ -4,7 +4,7 @@ use crate::controllers::krate::CratePath;
 use crate::models::krate::OwnerRemoveError;
 use crate::models::{
     krate::NewOwnerInvite, token::EndpointScope, CrateOwner, NewCrateOwnerInvitation,
-    NewCrateOwnerInvitationOutcome, OwnerKind,
+    NewCrateOwnerInvitationOutcome,
 };
 use crate::models::{Crate, Owner, Rights, Team, User};
 use crate::util::errors::{bad_request, crate_not_found, custom, AppResult, BoxedAppError};
@@ -342,14 +342,14 @@ async fn add_team_owner(
 
     // Teams are added as owners immediately, since the above call ensures
     // the user is a team member.
+    let crate_owner = CrateOwner::builder()
+        .crate_id(krate.id)
+        .team_id(team.id)
+        .created_by(req_user.id)
+        .build();
+
     diesel::insert_into(crate_owners::table)
-        .values(&CrateOwner {
-            crate_id: krate.id,
-            owner_id: team.id,
-            created_by: req_user.id,
-            owner_kind: OwnerKind::Team,
-            email_notifications: true,
-        })
+        .values(&crate_owner)
         .on_conflict(crate_owners::table.primary_key())
         .do_update()
         .set(crate_owners::deleted.eq(false))

src/models/krate.rs

@@ -128,13 +128,11 @@ impl NewCrate<'_> {
                     .get_result(conn)
                     .await?;
 
-                let owner = CrateOwner {
-                    crate_id: krate.id,
-                    owner_id: user_id,
-                    created_by: user_id,
-                    owner_kind: OwnerKind::User,
-                    email_notifications: true,
-                };
+                let owner = CrateOwner::builder()
+                    .crate_id(krate.id)
+                    .user_id(user_id)
+                    .created_by(user_id)
+                    .build();
 
                 diesel::insert_into(crate_owners::table)
                     .values(&owner)

src/models/owner.rs

@@ -1,11 +1,13 @@
+use bon::Builder;
 use diesel::pg::Pg;
 use diesel::prelude::*;
 
+use self::crate_owner_builder::{SetOwnerId, SetOwnerKind};
 use crate::models::{Crate, CrateOwnerInvitation, Team, User};
 use crate::schema::crate_owners;
 use crates_io_diesel_helpers::pg_enum;
 
-#[derive(Insertable, Associations, Identifiable, Debug, Clone, Copy)]
+#[derive(Insertable, Associations, Identifiable, Debug, Clone, Copy, Builder)]
 #[diesel(
     table_name = crate_owners,
     check_for_backend(diesel::pg::Pg),
@@ -16,12 +18,33 @@ use crates_io_diesel_helpers::pg_enum;
 )]
 pub struct CrateOwner {
     pub crate_id: i32,
+    #[builder(setters(vis = "pub(self)"))]
     pub owner_id: i32,
     pub created_by: i32,
+    #[builder(setters(vis = "pub(self)"))]
     pub owner_kind: OwnerKind,
+    #[builder(default = true)]
     pub email_notifications: bool,
 }
 
+impl<S: crate_owner_builder::State> CrateOwnerBuilder<S> {
+    pub fn team_id(self, team_id: i32) -> CrateOwnerBuilder<SetOwnerId<SetOwnerKind<S>>>
+    where
+        S::OwnerId: crate_owner_builder::IsUnset,
+        S::OwnerKind: crate_owner_builder::IsUnset,
+    {
+        self.owner_kind(OwnerKind::Team).owner_id(team_id)
+    }
+
+    pub fn user_id(self, user_id: i32) -> CrateOwnerBuilder<SetOwnerId<SetOwnerKind<S>>>
+    where
+        S::OwnerId: crate_owner_builder::IsUnset,
+        S::OwnerKind: crate_owner_builder::IsUnset,
+    {
+        self.owner_kind(OwnerKind::User).owner_id(user_id)
+    }
+}
+
 type BoxedQuery<'a> = crate_owners::BoxedQuery<'a, Pg, crate_owners::SqlType>;
 
 impl CrateOwner {
@@ -35,13 +58,11 @@ impl CrateOwner {
     }
 
     pub fn from_invite(invite: &CrateOwnerInvitation) -> Self {
-        Self {
-            crate_id: invite.crate_id,
-            owner_id: invite.invited_user_id,
-            created_by: invite.invited_by_user_id,
-            owner_kind: OwnerKind::User,
-            email_notifications: true,
-        }
+        CrateOwner::builder()
+            .crate_id(invite.crate_id)
+            .user_id(invite.invited_user_id)
+            .created_by(invite.invited_by_user_id)
+            .build()
     }
 }
 

src/tests/issues/issue2736.rs

@@ -1,4 +1,4 @@
-use crate::models::{CrateOwner, OwnerKind};
+use crate::models::CrateOwner;
 use crate::tests::builders::CrateBuilder;
 use crate::tests::util::{RequestHelper, TestApp};
 use crates_io_database::schema::{crate_owners, users};
@@ -23,14 +23,14 @@ async fn test_issue_2736() -> anyhow::Result<()> {
         .expect_build(&mut conn)
         .await;
 
+    let owner = CrateOwner::builder()
+        .crate_id(krate.id)
+        .user_id(foo1.as_model().id)
+        .created_by(someone_else.as_model().id)
+        .build();
+
     diesel::insert_into(crate_owners::table)
-        .values(CrateOwner {
-            crate_id: krate.id,
-            owner_id: foo1.as_model().id,
-            created_by: someone_else.as_model().id,
-            owner_kind: OwnerKind::User,
-            email_notifications: true,
-        })
+        .values(owner)
         .execute(&mut conn)
         .await?;
 

src/tests/mod.rs

@@ -1,4 +1,4 @@
-use crate::models::{Crate, CrateOwner, NewCategory, NewTeam, NewUser, OwnerKind, Team, User};
+use crate::models::{Crate, CrateOwner, NewCategory, NewTeam, NewUser, Team, User};
 use crate::schema::crate_owners;
 use crate::tests::util::{RequestHelper, TestApp};
 use crate::views::{
@@ -114,13 +114,11 @@ pub async fn add_team_to_crate(
     u: &User,
     conn: &mut AsyncPgConnection,
 ) -> QueryResult<()> {
-    let crate_owner = CrateOwner {
-        crate_id: krate.id,
-        owner_id: t.id,
-        created_by: u.id,
-        owner_kind: OwnerKind::Team,
-        email_notifications: true,
-    };
+    let crate_owner = CrateOwner::builder()
+        .crate_id(krate.id)
+        .team_id(t.id)
+        .created_by(u.id)
+        .build();
 
     diesel::insert_into(crate_owners::table)
         .values(&crate_owner)

src/tests/routes/crates/owners/remove.rs

@@ -1,4 +1,4 @@
-use crate::models::{CrateOwner, OwnerKind};
+use crate::models::CrateOwner;
 use crate::tests::builders::CrateBuilder;
 use crate::tests::util::{RequestHelper, TestApp};
 use crates_io_database::schema::crate_owners;
@@ -93,14 +93,14 @@ async fn test_remove_uppercase_user() {
         .expect_build(&mut conn)
         .await;
 
+    let owner = CrateOwner::builder()
+        .crate_id(krate.id)
+        .user_id(user2.as_model().id)
+        .created_by(cookie.as_model().id)
+        .build();
+
     diesel::insert_into(crate_owners::table)
-        .values(CrateOwner {
-            crate_id: krate.id,
-            owner_id: user2.as_model().id,
-            created_by: cookie.as_model().id,
-            owner_kind: OwnerKind::User,
-            email_notifications: true,
-        })
+        .values(owner)
         .execute(&mut conn)
         .await
         .unwrap();