model/invitation: Simplify CrateOwnerInvitation::create() arguments

Passing in three `i32` arguments seems a bit error-prone...

Author: Turbo87

src/models.rs

@@ -1,6 +1,8 @@
 pub use self::action::{NewVersionOwnerAction, VersionAction, VersionOwnerAction};
 pub use self::category::{Category, CrateCategory, NewCategory};
-pub use self::crate_owner_invitation::{CrateOwnerInvitation, NewCrateOwnerInvitationOutcome};
+pub use self::crate_owner_invitation::{
+    CrateOwnerInvitation, NewCrateOwnerInvitation, NewCrateOwnerInvitationOutcome,
+};
 pub use self::default_versions::{update_default_version, verify_default_version};
 pub use self::deleted_crate::NewDeletedCrate;
 pub use self::dependency::{Dependency, DependencyKind, ReverseDependency};

src/models/crate_owner_invitation.rs

@@ -38,9 +38,7 @@ pub struct CrateOwnerInvitation {
 
 impl CrateOwnerInvitation {
     pub async fn create(
-        invited_user_id: i32,
-        invited_by_user_id: i32,
-        crate_id: i32,
+        invite: &NewCrateOwnerInvitation,
         conn: &mut AsyncPgConnection,
         config: &config::Server,
     ) -> QueryResult<NewCrateOwnerInvitationOutcome> {
@@ -52,7 +50,7 @@ impl CrateOwnerInvitation {
                 // This does a SELECT FOR UPDATE + DELETE instead of a DELETE with a WHERE clause to
                 // use the model's `is_expired` method, centralizing our expiration checking logic.
                 let existing: Option<CrateOwnerInvitation> = crate_owner_invitations::table
-                    .find((invited_user_id, crate_id))
+                    .find((invite.invited_user_id, invite.crate_id))
                     .for_update()
                     .first(conn)
                     .await
@@ -70,11 +68,7 @@ impl CrateOwnerInvitation {
         .await?;
 
         let res: Option<CrateOwnerInvitation> = diesel::insert_into(crate_owner_invitations::table)
-            .values(&NewCrateOwnerInvitation {
-                invited_user_id,
-                invited_by_user_id,
-                crate_id,
-            })
+            .values(invite)
             // The ON CONFLICT DO NOTHING clause results in not creating the invite if another one
             // already exists. This does not cause problems with expired invitation as those are
             // deleted before doing this INSERT.

src/models/krate.rs

@@ -13,8 +13,8 @@ use crate::controllers::helpers::pagination::*;
 use crate::models::helpers::with_count::*;
 use crate::models::version::TopVersions;
 use crate::models::{
-    CrateOwner, CrateOwnerInvitation, NewCrateOwnerInvitationOutcome, Owner, OwnerKind,
-    ReverseDependency, User, Version,
+    CrateOwner, CrateOwnerInvitation, NewCrateOwnerInvitation, NewCrateOwnerInvitationOutcome,
+    Owner, OwnerKind, ReverseDependency, User, Version,
 };
 use crate::schema::*;
 use crate::util::errors::{bad_request, version_not_found, AppResult};
@@ -401,10 +401,15 @@ impl Crate {
         match owner {
             // Users are invited and must accept before being added
             Owner::User(user) => {
-                let creation_ret =
-                    CrateOwnerInvitation::create(user.id, req_user.id, self.id, conn, &app.config)
-                        .await
-                        .map_err(BoxedAppError::from)?;
+                let invite = NewCrateOwnerInvitation {
+                    invited_user_id: user.id,
+                    invited_by_user_id: req_user.id,
+                    crate_id: self.id,
+                };
+
+                let creation_ret = CrateOwnerInvitation::create(&invite, conn, &app.config)
+                    .await
+                    .map_err(BoxedAppError::from)?;
 
                 match creation_ret {
                     NewCrateOwnerInvitationOutcome::InviteCreated { plaintext_token } => {