Created a route to check what emails are already connected to an iden…

…tity (#477)

* Created a route to check what emails are already connected to an identity

* Delete .vscode/settings.json

Signed-off-by: Nathaniel Richards <46858886+NateTheDev1@users.noreply.github.com>

* Fixing errors

* Fixed trivial errors

---------

Signed-off-by: Nathaniel Richards <46858886+NateTheDev1@users.noreply.github.com>

crates/client-api/src/lib.rs

@@ -111,6 +111,7 @@ pub trait ControlStateReadAccess {
 
     // Identities
     fn get_identities_for_email(&self, email: &str) -> spacetimedb::control_db::Result<Vec<IdentityEmail>>;
+    fn get_emails_for_identity(&self, identity: &Identity) -> spacetimedb::control_db::Result<Vec<IdentityEmail>>;
     fn get_recovery_codes(&self, email: &str) -> spacetimedb::control_db::Result<Vec<RecoveryCode>>;
 
     // Energy
@@ -213,6 +214,9 @@ impl<T: ControlStateReadAccess + ?Sized> ControlStateReadAccess for ArcEnv<T> {
     fn get_identities_for_email(&self, email: &str) -> spacetimedb::control_db::Result<Vec<IdentityEmail>> {
         self.0.get_identities_for_email(email)
     }
+    fn get_emails_for_identity(&self, identity: &Identity) -> spacetimedb::control_db::Result<Vec<IdentityEmail>> {
+        self.0.get_emails_for_identity(identity)
+    }
     fn get_recovery_codes(&self, email: &str) -> spacetimedb::control_db::Result<Vec<RecoveryCode>> {
         self.0.get_recovery_codes(email)
     }
@@ -366,6 +370,9 @@ impl<T: ControlStateReadAccess + ?Sized> ControlStateReadAccess for Arc<T> {
     fn get_identities_for_email(&self, email: &str) -> spacetimedb::control_db::Result<Vec<IdentityEmail>> {
         (**self).get_identities_for_email(email)
     }
+    fn get_emails_for_identity(&self, identity: &Identity) -> spacetimedb::control_db::Result<Vec<IdentityEmail>> {
+        (**self).get_emails_for_identity(identity)
+    }
     fn get_recovery_codes(&self, email: &str) -> spacetimedb::control_db::Result<Vec<RecoveryCode>> {
         (**self).get_recovery_codes(email)
     }

crates/client-api/src/routes/identity.rs

@@ -4,11 +4,12 @@ use http::StatusCode;
 use serde::{Deserialize, Serialize};
 
 use spacetimedb::auth::identity::encode_token_with_expiry;
+use spacetimedb::messages::control_db::IdentityEmail;
 use spacetimedb_lib::de::serde::DeserializeWrapper;
 use spacetimedb_lib::{Address, Identity};
 
 use crate::auth::{SpacetimeAuth, SpacetimeAuthHeader};
-use crate::{log_and_500, ControlStateDelegate, ControlStateWriteAccess, NodeDelegate};
+use crate::{log_and_500, ControlStateDelegate, ControlStateReadAccess, ControlStateWriteAccess, NodeDelegate};
 
 #[derive(Deserialize)]
 pub struct CreateIdentityQueryParams {
@@ -130,6 +131,28 @@ pub async fn set_email<S: ControlStateWriteAccess>(
     Ok(())
 }
 
+pub async fn check_email<S: ControlStateReadAccess>(
+    State(ctx): State<S>,
+    Path(SetEmailParams { identity }): Path<SetEmailParams>,
+    auth: SpacetimeAuthHeader,
+) -> axum::response::Result<impl IntoResponse> {
+    let identity = identity.into();
+    let auth = auth.get().ok_or(StatusCode::BAD_REQUEST)?;
+
+    if auth.identity != identity {
+        return Err(StatusCode::UNAUTHORIZED.into());
+    }
+
+    let emails = ctx
+        .get_emails_for_identity(&identity)
+        .map_err(log_and_500)?
+        .into_iter()
+        .map(|IdentityEmail { email, .. }| email)
+        .collect::<Vec<_>>();
+
+    Ok(axum::Json(emails))
+}
+
 #[derive(Deserialize)]
 pub struct GetDatabasesParams {
     identity: IdentityForUrl,
@@ -216,5 +239,6 @@ where
         .route("/websocket_token", post(create_websocket_token::<S>))
         .route("/:identity/verify", get(validate_token))
         .route("/:identity/set-email", post(set_email::<S>))
+        .route("/:identity/emails", get(check_email::<S>))
         .route("/:identity/databases", get(get_databases::<S>))
 }

crates/core/src/control_db.rs

@@ -291,6 +291,19 @@ impl ControlDb {
         Ok(result)
     }
 
+    pub fn get_emails_for_identity(&self, identity: &Identity) -> Result<Vec<IdentityEmail>> {
+        let mut result = Vec::<IdentityEmail>::new();
+        let tree = self.db.open_tree("email")?;
+        for i in tree.iter() {
+            let (_, value) = i?;
+            let iemail: IdentityEmail = bsatn::from_slice(&value)?;
+            if &iemail.identity == identity {
+                result.push(iemail);
+            }
+        }
+        Ok(result)
+    }
+
     pub fn get_databases(&self) -> Result<Vec<Database>> {
         let tree = self.db.open_tree("database")?;
         let mut databases = Vec::new();

crates/standalone/src/lib.rs

@@ -255,6 +255,10 @@ impl spacetimedb_client_api::ControlStateReadAccess for StandaloneEnv {
         self.control_db.get_identities_for_email(email)
     }
 
+    fn get_emails_for_identity(&self, identity: &Identity) -> spacetimedb::control_db::Result<Vec<IdentityEmail>> {
+        self.control_db.get_emails_for_identity(identity)
+    }
+
     fn get_recovery_codes(&self, email: &str) -> spacetimedb::control_db::Result<Vec<RecoveryCode>> {
         self.control_db.spacetime_get_recovery_codes(email)
     }