1 files changed, 11 insertions, 15 deletions
diff --git a/src/media/mod.rs b/src/media/mod.rs
index 71f875e..47f456a 100644
--- a/src/media/mod.rs
+++ b/src/media/mod.rs
@@ -1,14 +1,9 @@
-use std::convert::TryFrom;
-
 use axum::{
-    extract::{Extension, Host, multipart::Multipart, Path},
-    response::{IntoResponse, Response},
-    headers::{Header, HeaderValue, IfNoneMatch, HeaderMapExt},
-    TypedHeader,
+    extract::{multipart::Multipart, FromRef, Host, Path, State}, headers::{HeaderMapExt, HeaderValue, IfNoneMatch}, response::{IntoResponse, Response}, TypedHeader
 };
 use kittybox_util::error::{MicropubError, ErrorType};
 use kittybox_indieauth::Scope;
-use crate::indieauth::{User, backend::AuthBackend};
+use crate::indieauth::{backend::AuthBackend, User};
 
 pub mod storage;
 use storage::{MediaStore, MediaStoreError, Metadata, ErrorKind};
@@ -25,7 +20,7 @@ impl From<MediaStoreError> for MicropubError {
 
 #[tracing::instrument(skip(blobstore))]
 pub(crate) async fn upload<S: MediaStore, A: AuthBackend>(
-    Extension(blobstore): Extension<S>,
+    State(blobstore): State<S>,
     user: User<A>,
     mut upload: Multipart
 ) -> Response {
@@ -70,7 +65,7 @@ pub(crate) async fn serve<S: MediaStore>(
     Host(host): Host,
     Path(path): Path<String>,
     if_none_match: Option<TypedHeader<IfNoneMatch>>,
-    Extension(blobstore): Extension<S>
+    State(blobstore): State<S>
 ) -> Response {
     use axum::http::StatusCode;
     tracing::debug!("Searching for file...");
@@ -131,11 +126,12 @@ pub(crate) async fn serve<S: MediaStore>(
     }
 }
 
-#[must_use]
-pub fn router<S: MediaStore, A: AuthBackend>(blobstore: S, auth: A) -> axum::Router {
+pub fn router<St, A, M>() -> axum::Router<St> where
+    A: AuthBackend + FromRef<St>,
+    M: MediaStore + FromRef<St>,
+    St: Clone + Send + Sync + 'static
+{
     axum::Router::new()
-        .route("/", axum::routing::post(upload::<S, A>))
-        .route("/uploads/*file", axum::routing::get(serve::<S>))
-        .layer(axum::Extension(blobstore))
-        .layer(axum::Extension(auth))
+        .route("/", axum::routing::post(upload::<M, A>))
+        .route("/uploads/*file", axum::routing::get(serve::<M>))
 }