webserver-rs 0.1.1

Feature

An out-of-the-box web server that is encapsulated based on salvo.

Use DataBase (sea-orm)

1. Enable one of the following features, depending on what you need

• mysql
• sqlite
• postgres

2. Run the following command if you want to use database

sea-orm-cli generate entity -o src/model

3. Import the generated model in your main.rs file

mod model;

Use Http3

Enable http3 feature

Example:

use salvo::{jwt_auth::HeaderFinder, prelude::*};
use serde::{Deserialize, Serialize};
use serde_json::json;
use tokio::io::AsyncReadExt;
use webserver_rs::{
    assets::MemoryStream, authorization, build_cros, config::Config, expire_time, html_err,
    json_err, serve, AnyResult, FromConfigFile,
};

use webserver_rs::web_core::authorization::gen_token;

#[derive(Debug, Serialize, Deserialize)]
struct JwtClaim {
    username: String,
    exp: i64,
}

#[handler]
pub fn hello(req: &mut Request, res: &mut Response) -> AnyResult<()> {
    let c = req.query::<String>("id");
    println!("{c:?}");
    let p = req.query::<String>("id").ok_or(html_err!(
        400,
        json!({
            "status":"fail",
            "msg":"id not found"
        })
        .to_string()
    ))?;
    res.render(Text::Plain(format!("hello, {p}")));
    Ok(())
}

#[handler]
pub fn login(depot: &mut Depot, res: &mut Response) -> AnyResult<()> {
    let config = depot.obtain::<Config>().map_err(|_e| {
        crate::json_err!(400,{
            "status":"fail",
            "msg":"config not found"
        })
    })?;
    let token = gen_token(
        config.secret_key.clone(),
        JwtClaim {
            exp: crate::expire_time!(Days(1)),
            username: "a".to_string(),
        },
    )
    .map_err(|e| crate::json_err!(400, {"err":e.to_string()}))?;
    res.render(Text::Plain(token));
    Ok(())
}

#[handler]
pub async fn image(res: &mut Response) -> AnyResult<()> {
    let mut file = tokio::fs::File::open("./test.png")
        .await
        .map_err(|_| crate::html_err!(404, "".to_string()))?;
    let mut s = Vec::new();
    file.read_to_end(&mut s).await.unwrap();
    res.stream(MemoryStream::new(s, 200));
    Ok(())
}
#[handler]
pub async fn text_json(res: &mut Response) -> AnyResult<()> {
    res.render(Text::Json(
        json!({
            "title":1
        })
        .to_string(),
    ));
    Ok(())
}

mod ab {
    use salvo::handler;

    #[handler]
    pub fn show() -> &'static str {
        "abc"
    }
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    let config = Config::from_config_file("./config.toml").expect("config file not found");
    let jwt = authorization::gen_jwt_auth::<JwtClaim>(
        config.secret_key.clone(),
        vec![Box::new(HeaderFinder::new())],
    );
    let serve_route = webserver_rs::define_route![
		// http://localhost:8080/hello
        webserver_rs::create_router!(hello, get, post)
            .hoop(jwt)
            .hoop(authorization::AuthGuard::new(|_e| html_err!(String::from(
                "unauthorized"
            )))),
		// http://localhost:8080/user/login
        webserver_rs::create_router!("user", login, get), 
		// http://localhost:8080/a/b/show
        webserver_rs::create_router!("a/b", ab::show, get, post), 
		// http://localhost:8080/test_json
        webserver_rs::create_router!(text_json, get, post).hoop(build_cros("*")), 
    ];
    serve(config, serve_route).await?;
    Ok(())
}