Allow overwriting signup data if a user signs up more than once with the same email address

Cargo.lock

@@ -1160,6 +1160,7 @@ dependencies = [
  "lsp",
  "nanoid",
  "parking_lot 0.11.2",
+ "pretty_assertions",
  "project",
  "prometheus",
  "rand 0.8.5",
@@ -1730,6 +1731,12 @@ dependencies = [
  "workspace",
 ]
 
+[[package]]
+name = "diff"
+version = "0.1.13"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "56254986775e3233ffa9c4d7d3faaf6d36a2c09d30b20687e9f88bc8bafc16c8"
+
 [[package]]
 name = "digest"
 version = "0.9.0"
@@ -4005,6 +4012,15 @@ dependencies = [
  "workspace",
 ]
 
+[[package]]
+name = "output_vt100"
+version = "0.1.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "628223faebab4e3e40667ee0b2336d34a5b960ff60ea743ddfdbcf7770bcfb66"
+dependencies = [
+ "winapi 0.3.9",
+]
+
 [[package]]
 name = "overload"
 version = "0.1.1"
@@ -4346,6 +4362,18 @@ version = "0.2.17"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "5b40af805b3121feab8a3c29f04d8ad262fa8e0561883e7653e024ae4479e6de"
 
+[[package]]
+name = "pretty_assertions"
+version = "1.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a25e9bcb20aa780fd0bb16b72403a9064d6b3f22f026946029acb941a50af755"
+dependencies = [
+ "ctor",
+ "diff",
+ "output_vt100",
+ "yansi",
+]
+
 [[package]]
 name = "proc-macro-crate"
 version = "0.1.5"
@@ -8065,6 +8093,12 @@ dependencies = [
  "linked-hash-map",
 ]
 
+[[package]]
+name = "yansi"
+version = "0.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "09041cd90cf85f7f8b2df60c646f853b7f535ce68f85244eb6731cf89fa498ec"
+
 [[package]]
 name = "zed"
 version = "0.67.0"

crates/collab/Cargo.toml

@@ -64,6 +64,7 @@ fs = { path = "../fs", features = ["test-support"] }
 git = { path = "../git", features = ["test-support"] }
 live_kit_client = { path = "../live_kit_client", features = ["test-support"] }
 lsp = { path = "../lsp", features = ["test-support"] }
+pretty_assertions = "1.3.0"
 project = { path = "../project", features = ["test-support"] }
 rpc = { path = "../rpc", features = ["test-support"] }
 settings = { path = "../settings", features = ["test-support"] }

crates/collab/src/db.rs

@@ -669,7 +669,15 @@ impl Database {
             })
             .on_conflict(
                 OnConflict::column(signup::Column::EmailAddress)
-                    .update_column(signup::Column::EmailAddress)
+                    .update_columns([
+                        signup::Column::PlatformMac,
+                        signup::Column::PlatformWindows,
+                        signup::Column::PlatformLinux,
+                        signup::Column::EditorFeatures,
+                        signup::Column::ProgrammingLanguages,
+                        signup::Column::DeviceId,
+                        signup::Column::AddedToMailingList,
+                    ])
                     .to_owned(),
             )
             .exec(&*tx)
@@ -679,6 +687,21 @@ impl Database {
         .await
     }
 
+    pub async fn get_signup(&self, email_address: &str) -> Result<signup::Model> {
+        self.transaction(|tx| async move {
+            let signup = signup::Entity::find()
+                .filter(signup::Column::EmailAddress.eq(email_address))
+                .one(&*tx)
+                .await?
+                .ok_or_else(|| {
+                    anyhow!("signup with email address {} doesn't exist", email_address)
+                })?;
+
+            Ok(signup)
+        })
+        .await
+    }
+
     pub async fn get_waitlist_summary(&self) -> Result<WaitlistSummary> {
         self.transaction(|tx| async move {
             let query = "

crates/collab/src/db/signup.rs

@@ -34,7 +34,7 @@ pub struct Invite {
     pub email_confirmation_code: String,
 }
 
-#[derive(Clone, Deserialize)]
+#[derive(Clone, Debug, Deserialize)]
 pub struct NewSignup {
     pub email_address: String,
     pub platform_mac: bool,

crates/collab/src/db/tests.rs

@@ -2,6 +2,9 @@ use super::*;
 use gpui::executor::{Background, Deterministic};
 use std::sync::Arc;
 
+#[cfg(test)]
+use pretty_assertions::{assert_eq, assert_ne};
+
 macro_rules! test_both_dbs {
     ($postgres_test_name:ident, $sqlite_test_name:ident, $db:ident, $body:block) => {
         #[gpui::test]
@@ -727,6 +730,78 @@ async fn test_invite_codes() {
     assert_eq!(invite_count, 1);
 }
 
+#[gpui::test]
+async fn test_multiple_signup_overwrite() {
+    let test_db = TestDb::postgres(build_background_executor());
+    let db = test_db.db();
+
+    let email_address = "user_1@example.com".to_string();
+
+    let signup = NewSignup {
+        email_address: email_address.clone(),
+        platform_mac: false,
+        platform_linux: true,
+        platform_windows: false,
+        editor_features: vec!["speed".into()],
+        programming_languages: vec!["rust".into(), "c".into()],
+        device_id: Some(format!("device_id")),
+        added_to_mailing_list: false,
+    };
+
+    db.create_signup(&signup).await.unwrap();
+
+    // TODO: Remove this method and just have create_signup return an instance?
+    let signup_from_db = db.get_signup(&signup.email_address).await.unwrap();
+
+    assert_eq!(
+        signup_from_db.clone(),
+        signup::Model {
+            email_address: signup.email_address,
+            platform_mac: signup.platform_mac,
+            platform_linux: signup.platform_linux,
+            platform_windows: signup.platform_windows,
+            editor_features: Some(signup.editor_features),
+            programming_languages: Some(signup.programming_languages),
+            added_to_mailing_list: signup.added_to_mailing_list,
+            ..signup_from_db
+        }
+    );
+
+    let signup_overwrite = NewSignup {
+        email_address,
+        platform_mac: true,
+        platform_linux: false,
+        platform_windows: true,
+        editor_features: vec!["git integration".into(), "clean design".into()],
+        programming_languages: vec!["d".into(), "elm".into()],
+        device_id: Some(format!("different_device_id")),
+        added_to_mailing_list: true,
+    };
+
+    db.create_signup(&signup_overwrite).await.unwrap();
+
+    let signup_overwrite_from_db = db
+        .get_signup(&signup_overwrite.email_address)
+        .await
+        .unwrap();
+
+    assert_eq!(
+        signup_overwrite_from_db.clone(),
+        signup::Model {
+            platform_mac: signup_overwrite.platform_mac,
+            platform_linux: signup_overwrite.platform_linux,
+            platform_windows: signup_overwrite.platform_windows,
+            editor_features: Some(signup_overwrite.editor_features),
+            programming_languages: Some(signup_overwrite.programming_languages),
+            device_id: signup_overwrite.device_id,
+            added_to_mailing_list: signup_overwrite.added_to_mailing_list,
+            // shouldn't overwrite their creation Datetime - user shouldn't lose their spot in line
+            created_at: signup_from_db.created_at,
+            ..signup_overwrite_from_db
+        }
+    );
+}
+
 #[gpui::test]
 async fn test_signups() {
     let test_db = TestDb::postgres(build_background_executor());