athene_macro 0.1.2

Macro generation for athene
Documentation

example

Useage -- Future

use athene::handler::Handler;
use athene::prelude::*;
use std::future::Future;
use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize, Default)]
pub struct User {
    pub name: String,
    pub age: i32,
}

impl Handler for User {
    type Responder = Builder;
    type Future = Box<dyn Future<Output = Self::Responder> + Unpin + Send>;

    fn handle(&self, req: Request) -> Self::Future {
        Box::new(Box::pin(async move { 
            let uri = req.uri().path();
            let method = req.method();
            let peer_addr = req.peer_addr();

            let builder = Builder::new();
            builder.text(format!("uri : {}, method: {}, peer_addr: {}",uri,method,peer_addr))
         }))
    }
}

impl User {
    pub fn hello(_: Request) -> impl Future<Output = impl Responder> {
        async { "hello" }
    }

    pub fn get_user(mut req: Request) -> impl Future<Output = impl Responder> {
        async move {
            let user = req.parse::<User>().await?;
            Ok::<_, Error>(json!(user))
        }
    }

    pub fn basic_router(r: Router) -> Router {
        r.get("/hello", Self::hello)
        .post("/user", Self::default())
        .post("/get_user", Self::get_user)
    }
}

#[tokio::main]
pub async fn main() -> Result<()> {
    let app = athene::new()
        .router(|r| {
            r.get("/",|_: Request| async {
                "Hello World"
            })   
        })
        .router(User::basic_router)
        .build();
    
    app.listen("127.0.0.1:7878").await
}

Useage MacroController

use athene::prelude::*;
use serde::{Deserialize,Serialize};

use validator::Validate;
#[derive(Serialize, Deserialize,Validate,Default)]
pub struct UserController {
    #[validate(email)]
    pub username: String,
    #[validate(range(min = 18, max = 20))]
    pub age: u16,
}

// http://127.0.0.1:7878/api/v1/user
#[controller(prefix = "api", version = 1, name = "user")]
impl UserController {

    #[get("/*/**")]
    pub async fn match_any_route(&self, req: Request) -> impl Responder {
        let uri_path = req.uri().path().to_string();
        let method = req.method().to_string();
        (200, format!("uri : {}, method: {}", uri_path, method))
    }

    #[delete("/{username}/{age}")]
    pub async fn delete_by_param(&self, username: String, age: Option<u16>) -> impl Responder {
        (200, format!("username is : {}, and age is : {:?}", username, age))
    }

    #[get("/get_query_1")]
    pub async fn get_query_1(&self, username: String, age: u16) -> impl Responder {
        (200,json!(&Self {
                username,
                age,
            }))
    }

    #[get("/get_query_2")]
    //#[validator(exclude("user"))] // The user parameter will not be validated
    pub async fn get_query_2(&self, request: Request) -> impl Responder {
        let user = request.query::<Self>()?;
        Ok::<_, Error>((200, json!(&user)))
    }

    // Context-Type : application/json  Or application/x-www-form-urlencoded
    // Context-Type : application/msgpack Or application/cbor
    #[post("/parse_body")]
    #[get("/parse_body")]
    //#[validator] // User will be validated
    async fn parse_body(&self, mut req: Request) -> impl Responder {
        let user = req.parse::<Self>().await?;
        Ok::<_, Error>((
            200,
            format!("username = {} and age = {}", user.username, user.age),
        ))
    }

    // Context-Type : application/multipart-formdata
    #[post("/files")]
    async fn files(&self, mut req: Request) -> impl Responder {
        let files = req.files("files").await?;
        let fist_file_name = files[0].name().unwrap();
        let second_file_name = files[1].name().unwrap();
        Ok::<_, Error>((
            200,
            format!(
                "fist {}, second {}",
                fist_file_name, second_file_name
            ),
        ))
    }

    // Content-Disposition: application/octet-stream
    #[get("/download")]
    pub async fn download(&self,_req: Request) -> impl Responder {
        let mut res = status!(hyper::StatusCode::OK);
        res.write_file("templates/author.txt", DispositionType::Attachment)?;
        Ok::<_, Error>(res)
    }
}

#[tokio::main]
pub async fn main() -> Result<()> {

    let app = athene::new()
        .router(|r|r.controller(UserController::default()));

    let app = app.build();

    app.listen("127.0.0.1:7878").await
}

Useage BasicController

use athene::prelude::*;
use serde::{Deserialize, Serialize};

#[derive(Default, Deserialize, Serialize)]
pub struct AtheneController {
    pub label: String,
    pub keyword: String,
}

impl Controller for AtheneController {

    const BASE_PATH: &'static str = "/api/v1/athene";

    fn method(&self) -> Vec<ControllerMethod<Self>>
    where
        Self: Sized,
    {
        ControllerBuilder::new()

        .post("/add", Self::add)
        .delete("/{label}/{keyword}", Self::delete)
        .put("/update", Self::update)
        .get("/get", Self::get)
        .build()
    }
}

impl AtheneController {

    // http://127.0.0.1:7878/api/v1/athene/add
    pub async fn add(&self, mut req: Request) -> impl Responder {
        let obj = req.parse::<Self>().await?;
        Ok::<_, Error>((200, json!(obj)))
    }

    // http://127.0.0.1:7878/api/v1/athene/
    pub async fn delete(&self, mut req: Request) -> impl Responder {
        let lable = req.param("label");
        let keyword = req.param("keyword");

        (200,format!("lable = {},keyword = {}",lable,keyword))
    }

    // http://127.0.0.1:7878/api/v1/athene/update
    async fn update(&self, mut req: Request) -> impl Responder {
        let obj = req.parse::<Self>().await?;
        Ok::<_, Error>((200, json!(obj)))
    }

    // http://127.0.0.1:7878/api/v1/athene/get
    async fn get(&self, req: Request) -> impl Responder {
        #[derive(Deserialize, Serialize)]
        struct QueryParam<'a> {
            label: &'a str,
            keyword: &'a str,
        }

        let arg = req.query::<QueryParam>()?;
        let res = json!(&arg);
        Ok::<_, Error>(res)
    }
}

#[tokio::main]
async fn main() -> Result<(), Error> {
    let app = athene::new()
    .router(|r|r.controller(AtheneController::default()))
    .build();
         
    app.listen("127.0.0.1:7878").await
}

Useage Handler

use athene::prelude::*;
use headers::{authorization::Bearer, Authorization};
use serde::{Deserialize, Serialize};
use tracing::info;

#[derive(Serialize, Deserialize)]
pub struct User {
    pub username: String,
    pub age: u16,
}

pub async fn user_params(mut req: Request) -> impl Responder {
    let username = req.param("username");
    let age = req.param("age");
    let age = age.parse::<u16>().unwrap();
    (200, json!(&User { username, age }))
}

pub async fn login(mut req: Request) -> impl Responder {
    if let Ok(user) = req.parse::<User>().await {
        (
            200,
            format!("username: {} , age: {} ", user.username, user.age),
        )
    } else {
        (400, String::from("Bad user format"))
    }
}

pub async fn sign_up(mut req: Request) -> impl Responder {
    let user = req.parse::<User>().await?;
    Ok::<_, Error>((200, json!(user)))
}

pub async fn read_body(mut req: Request) -> impl Responder {
    let body = req.body_bytes().await?;
    Ok::<_, Error>((200, body))
}

pub fn user_router(r: Router) -> Router {
    r.get("/{username}/{age}", user_params)
        .post("/login", login)
        .post("/sign_up", sign_up)
        .put("/read_body", read_body)
}

//   ============================ Middleware Fuction ============================
pub async fn log_middleware(ctx: Context, next: &'static dyn Next) -> Result<Context, Error> {
    info!(
        "new request on path: {}",
        ctx.state.request_unchecked().uri().path()
    );

    let ctx = next.next(ctx).await?;

    info!(
        "new response with status: {}",
        ctx.state.response_unchecked().status()
    );
    Ok(ctx)
}

// ================================ Middleware  ======================
struct ApiKeyMiddleware {
    api_key: String,
}

#[middleware]
impl ApiKeyMiddleware {
    async fn next(&self, ctx: Context, next: &dyn Next) -> Result<Context, Error> {
        if let Some(bearer) = ctx
            .state
            .request_unchecked()
            .header::<Authorization<Bearer>>()
        {
            let token = bearer.0.token();
            if token == self.api_key {
                info!(
                    "Handler {} will be used",
                    ctx.metadata.name.unwrap_or("unknown")
                );
                next.next(ctx).await
            } else {
                info!("Invalid token");
                Ok(ctx)
            }
        } else {
            info!("Not Authenticated, ");
            Ok(ctx)
        }
    }
}

#[tokio::main]
pub async fn main() -> Result<()> {
    let app = athene::new()
        .router(user_router)
        .middleware(|m| {
            m.apply(log_middleware, vec!["/"], None).apply(
                ApiKeyMiddleware {
                    api_key: "athene".to_string(),
                },
                vec!["/login", "/sign_up"],
                vec!["/read_body"],
            )
        })
        .build();
    app.listen("127.0.0.1:7878").await
}