brent/shoutdotdev
1
0
Fork 0
forked from 2sys/shoutdotdev
Code Pull requests Activity

generalize extractor impls for any Into<AppState>

Browse source
This commit is contained in:
Brent Schroeter 2025-03-25 11:20:24 -07:00
parent 377035ce88
commit b1e31b4026
3 changed files with 25 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
src/app_state.rs
View file

@ -68,20 +68,26 @@ pub type AppState = Arc<App>;
#[derive(Clone)] #[derive(Clone)]
pub struct ReqwestClient(pub reqwest::Client); pub struct ReqwestClient(pub reqwest::Client);
impl FromRef<AppState> for ReqwestClient { impl<S> FromRef<S> for ReqwestClient
fn from_ref(state: &AppState) -> Self { where
ReqwestClient(state.reqwest_client.clone()) S: Into<AppState> + Clone,
{
fn from_ref(state: &S) -> Self {
ReqwestClient(Into::<AppState>::into(state.clone()).reqwest_client.clone())
} }
} }
/// Extractor to automatically obtain a Deadpool database connection /// Extractor to automatically obtain a Deadpool database connection
pub struct DbConn(pub Connection); pub struct DbConn(pub Connection);
impl FromRequestParts<AppState> for DbConn { impl<S> FromRequestParts<S> for DbConn
where
S: Into<AppState> + Clone + Sync,
{
type Rejection = AppError; type Rejection = AppError;
async fn from_request_parts(_: &mut Parts, state: &AppState) -> Result<Self, Self::Rejection> { async fn from_request_parts(_: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
let conn = state.db_pool.get().await?; let conn = Into::<AppState>::into(state.clone()).db_pool.get().await?;
Ok(Self(conn)) Ok(Self(conn))
} }
} }

9
src/settings.rs
View file

@ -108,8 +108,11 @@ impl Settings {
} }
} }
impl FromRef<AppState> for Settings { impl<S> FromRef<S> for Settings
fn from_ref(state: &AppState) -> Self { where
state.settings.clone() S: Into<AppState> + Clone,
{
fn from_ref(state: &S) -> Self {
Into::<AppState>::into(state.clone()).settings.clone()
} }
} }

13
src/users.rs
View file

@ -53,15 +53,16 @@ impl User {
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
pub struct CurrentUser(pub User); pub struct CurrentUser(pub User);
impl FromRequestParts<AppState> for CurrentUser { impl<S> FromRequestParts<S> for CurrentUser
where
S: Into<AppState> + Clone + Sync,
{
type Rejection = AppError; type Rejection = AppError;
async fn from_request_parts( async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
parts: &mut Parts, let state: AppState = state.clone().into();
state: &AppState,
) -> Result<Self, <Self as FromRequestParts<AppState>>::Rejection> {
let auth_info = parts let auth_info = parts
.extract_with_state::<AuthInfo, AppState>(state) .extract_with_state::<AuthInfo, AppState>(&state)
.await .await
.map_err(|_| { .map_err(|_| {
AppError::auth_redirect_from_base_path(state.settings.base_path.clone()) AppError::auth_redirect_from_base_path(state.settings.base_path.clone())