Shield Axum: Add subprovider and user routes (#18)

packages/integrations/shield-axum/src/error.rs

@@ -8,7 +8,7 @@ pub struct RouteError(ShieldError);
 
 impl IntoResponse for RouteError {
     fn into_response(self) -> Response {
-        (StatusCode::INTERNAL_SERVER_ERROR, self.0.to_string()).into_response()
+        (StatusCode::INTERNAL_SERVER_ERROR, format!("{:?}", self.0)).into_response()
     }
 }
 

packages/integrations/shield-axum/src/router.rs

@@ -4,10 +4,11 @@ use axum::{
 };
 use shield::User;
 
-use crate::routes::{sign_in, sign_in_callback, sign_out};
+use crate::routes::{sign_in, sign_in_callback, sign_out, subproviders, user};
 
 pub fn auth_router<U: User + Clone + 'static, S: Clone + Send + Sync + 'static>() -> Router<S> {
     Router::new()
+        .route("/subproviders", get(subproviders::<U>))
         .route("/sign-in/:provider_id", post(sign_in::<U>))
         .route("/sign-in/:provider_id/:subprovider_id", post(sign_in::<U>))
         .route("/sign-in/callback/:provider_id", get(sign_in_callback::<U>))
@@ -16,4 +17,5 @@ pub fn auth_router<U: User + Clone + 'static, S: Clone + Send + Sync + 'static>(
             get(sign_in_callback::<U>),
         )
         .route("/sign-out", post(sign_out::<U>))
+        .route("/user", get(user::<U>))
 }

packages/integrations/shield-axum/src/routes.rs

@@ -1,7 +1,11 @@
 mod sign_in;
 mod sign_in_callback;
 mod sign_out;
+mod subproviders;
+mod user;
 
 pub use sign_in::*;
 pub use sign_in_callback::*;
 pub use sign_out::*;
+pub use subproviders::*;
+pub use user::*;

packages/integrations/shield-axum/src/routes/subproviders.rs

@@ -0,0 +1,10 @@
+use axum::Json;
+use shield::{SubproviderVisualisation, User};
+
+use crate::{error::RouteError, extract::ExtractShield};
+
+pub async fn subproviders<U: User>(
+    ExtractShield(shield): ExtractShield<U>,
+) -> Result<Json<Vec<SubproviderVisualisation>>, RouteError> {
+    Ok(Json(shield.subprovider_visualisations().await?))
+}

packages/integrations/shield-axum/src/routes/user.rs

@@ -0,0 +1,26 @@
+use axum::{
+    http::StatusCode,
+    response::{IntoResponse, Response},
+    Json,
+};
+use serde_json::json;
+use shield::User;
+
+use crate::extract::ExtractUser;
+
+pub async fn user<U: User>(ExtractUser(user): ExtractUser<U>) -> Response {
+    // TODO: Send JSON error using some util.
+    match user {
+        Some(user) => {
+            // TODO: Include email addresses.
+            // let email_addresses = user.email_addresses().await;
+
+            Json(json!({
+                "id": user.id(),
+                "name": user.name(),
+            }))
+            .into_response()
+        }
+        None => (StatusCode::UNAUTHORIZED, "Unauthorized").into_response(),
+    }
+}