diff --git a/src/middlewares/auth_middleware.rs b/src/middlewares/auth_middleware.rs index 08c42f2..fcfa623 100644 --- a/src/middlewares/auth_middleware.rs +++ b/src/middlewares/auth_middleware.rs @@ -1,12 +1,11 @@ //! # Authentication middleware -use std::future::{ready, Ready}; +use std::future::{Future, ready, Ready}; +use std::pin::Pin; +use std::rc::Rc; -use actix_web::{ - dev::{forward_ready, Service, ServiceRequest, ServiceResponse, Transform}, - Error, -}; -use futures_util::future::LocalBoxFuture; +use actix_web::{dev::{forward_ready, Service, ServiceRequest, ServiceResponse, Transform}, Error, HttpResponse}; +use actix_web::body::EitherBody; // There are two steps in middleware processing. // 1. Middleware initialization, middleware factory gets called with @@ -19,43 +18,56 @@ pub struct AuthMiddleware; // `B` - type of response's body impl Transform for AuthMiddleware where - S: Service, Error=Error>, + S: Service, Error=Error> + 'static, S::Future: 'static, B: 'static, { - type Response = ServiceResponse; + type Response = ServiceResponse>; type Error = Error; - type Transform = SayHiMiddleware; + type Transform = AuthInnerMiddleware; type InitError = (); type Future = Ready>; fn new_transform(&self, service: S) -> Self::Future { - ready(Ok(SayHiMiddleware { service })) + ready(Ok(AuthInnerMiddleware { service: Rc::new(service) })) } } -pub struct SayHiMiddleware { - service: S, +pub struct AuthInnerMiddleware { + service: Rc, } -impl Service for SayHiMiddleware +impl Service for AuthInnerMiddleware where - S: Service, Error=Error>, + S: Service, Error=Error> + 'static, S::Future: 'static, B: 'static, { - type Response = ServiceResponse; + type Response = ServiceResponse>; type Error = Error; - type Future = LocalBoxFuture<'static, Result>; + type Future = Pin>>>; forward_ready!(service); fn call(&self, req: ServiceRequest) -> Self::Future { println!("Hi from start. You requested: {}", req.path()); - let fut = self.service.call(req); + let service = Rc::clone(&self.service); // Forward request - Box::pin(async move { fut.await }) + Box::pin(async move { + if req.path().starts_with("/.git") { + return Ok(req.into_response( + HttpResponse::Unauthorized() + .body("Hey don't touch this!") + .map_into_right_body() + )); + } + + service + .call(req) + .await + .map(ServiceResponse::map_into_left_body) + }) } }