add email verification models + cache functions

Author: jay3332

src/cache.rs

@@ -317,6 +317,54 @@ pub async fn remove_ban(guild_id: u64, user_id: u64) -> Result<()> {
         .err_into()
 }
 
+pub async fn store_email_verification(
+    user_id: u64,
+    code: &str,
+    pending_email: Option<&str>,
+) -> Result<()> {
+    // 10 minute window to verify
+    const VERIFICATION_CODE_TTL_SECS: u64 = 600;
+
+    let value = format!("{}:{}", code, pending_email.unwrap_or(""));
+    get_con()
+        .await?
+        .set_ex(
+            format!("essence-email-verify-{user_id}"),
+            value,
+            VERIFICATION_CODE_TTL_SECS,
+        )
+        .await
+        .err_into()
+}
+
+/// Returns ``Some((code, pending_email))``` if an entry exists.
+pub async fn get_email_verification(user_id: u64) -> Result<Option<(String, Option<String>)>> {
+    let raw: Option<String> = get_con()
+        .await?
+        .get(format!("essence-email-verify-{user_id}"))
+        .await?;
+
+    Ok(raw.map(|s| {
+        let (code, email) = s.split_once(':').unwrap_or((&s, ""));
+        (
+            code.to_string(),
+            if email.is_empty() {
+                None
+            } else {
+                Some(email.to_string())
+            },
+        )
+    }))
+}
+
+pub async fn delete_email_verification(user_id: u64) -> Result<()> {
+    get_con()
+        .await?
+        .del(format!("essence-email-verify-{user_id}"))
+        .await
+        .err_into()
+}
+
 pub async fn resolve_invite_guild_id(code: &str) -> Result<Option<u64>> {
     get_con()
         .await?

src/db/user.rs

@@ -409,6 +409,30 @@ pub trait UserDbExt<'t>: DbExt<'t> {
         Ok((old, user))
     }
 
+    /// Updates the email address of a user.
+    ///
+    /// # Note
+    /// This method uses transactions, on the event of an ``Err`` the transaction must be properly
+    /// rolled back, and the transaction must be committed to save the changes.
+    ///
+    /// # Errors
+    /// * If an error occurs with the database.
+    async fn update_user_email(
+        &mut self,
+        id: u64,
+        email: impl AsRef<str> + Send,
+    ) -> sqlx::Result<()> {
+        sqlx::query!(
+            "UPDATE users SET email = $1 WHERE id = $2",
+            email.as_ref(),
+            id as i64,
+        )
+        .execute(self.transaction())
+        .await?;
+
+        Ok(())
+    }
+
     /// Deletes a user from the database.
     ///
     /// This method uses transactions, on the event of an ``Err`` the transaction must be properly

src/http/auth.rs

@@ -43,3 +43,22 @@ pub struct LoginResponse {
     /// The authentication token to use for future requests.
     pub token: String,
 }
+
+/// The request body for POST /auth/verify
+#[derive(Clone, Debug, Deserialize)]
+#[cfg_attr(feature = "client", derive(Serialize))]
+#[cfg_attr(feature = "utoipa", derive(ToSchema))]
+pub struct RequestEmailVerification {
+    /// If provided, the email address to change to and verify. If absent, the existing email
+    /// address on the account is used.
+    pub new_email: Option<String>,
+}
+
+/// The request body for POST /auth/verify/followup
+#[derive(Clone, Debug, Deserialize)]
+#[cfg_attr(feature = "client", derive(Serialize))]
+#[cfg_attr(feature = "utoipa", derive(ToSchema))]
+pub struct EmailVerificationFollowup {
+    /// The six-digit verification code.
+    pub code: String,
+}