use axum::{
handler::Handler,
routing::{delete, get, on, post, MethodFilter, MethodRouter},
Router,
};
#[derive(Debug)]
#[must_use]
pub struct Resource<S = ()> {
pub(crate) name: String,
pub(crate) router: Router<S>,
}
impl<S> Resource<S>
where
S: Clone + Send + Sync + 'static,
{
pub fn named(resource_name: &str) -> Self {
Self {
name: resource_name.to_owned(),
router: Router::new(),
}
}
pub fn index<H, T>(self, handler: H) -> Self
where
H: Handler<T, S>,
T: 'static,
{
let path = self.index_create_path();
self.route(&path, get(handler))
}
pub fn create<H, T>(self, handler: H) -> Self
where
H: Handler<T, S>,
T: 'static,
{
let path = self.index_create_path();
self.route(&path, post(handler))
}
pub fn new<H, T>(self, handler: H) -> Self
where
H: Handler<T, S>,
T: 'static,
{
let path = format!("/{}/new", self.name);
self.route(&path, get(handler))
}
pub fn show<H, T>(self, handler: H) -> Self
where
H: Handler<T, S>,
T: 'static,
{
let path = self.show_update_destroy_path();
self.route(&path, get(handler))
}
pub fn edit<H, T>(self, handler: H) -> Self
where
H: Handler<T, S>,
T: 'static,
{
let path = format!("/{0}/:{0}_id/edit", self.name);
self.route(&path, get(handler))
}
pub fn update<H, T>(self, handler: H) -> Self
where
H: Handler<T, S>,
T: 'static,
{
let path = self.show_update_destroy_path();
self.route(
&path,
on(MethodFilter::PUT.or(MethodFilter::PATCH), handler),
)
}
pub fn destroy<H, T>(self, handler: H) -> Self
where
H: Handler<T, S>,
T: 'static,
{
let path = self.show_update_destroy_path();
self.route(&path, delete(handler))
}
fn index_create_path(&self) -> String {
format!("/{}", self.name)
}
fn show_update_destroy_path(&self) -> String {
format!("/{0}/:{0}_id", self.name)
}
fn route(mut self, path: &str, method_router: MethodRouter<S>) -> Self {
self.router = self.router.route(path, method_router);
self
}
}
impl<S> From<Resource<S>> for Router<S> {
fn from(resource: Resource<S>) -> Self {
resource.router
}
}
#[cfg(test)]
mod tests {
#[allow(unused_imports)]
use super::*;
use axum::{body::Body, extract::Path, http::Method};
use http::Request;
use http_body_util::BodyExt;
use tower::ServiceExt;
#[tokio::test]
async fn works() {
let users = Resource::named("users")
.index(|| async { "users#index" })
.create(|| async { "users#create" })
.new(|| async { "users#new" })
.show(|Path(id): Path<u64>| async move { format!("users#show id={id}") })
.edit(|Path(id): Path<u64>| async move { format!("users#edit id={id}") })
.update(|Path(id): Path<u64>| async move { format!("users#update id={id}") })
.destroy(|Path(id): Path<u64>| async move { format!("users#destroy id={id}") });
let app = Router::new().merge(users);
assert_eq!(call_route(&app, Method::GET, "/users").await, "users#index");
assert_eq!(
call_route(&app, Method::POST, "/users").await,
"users#create"
);
assert_eq!(
call_route(&app, Method::GET, "/users/new").await,
"users#new"
);
assert_eq!(
call_route(&app, Method::GET, "/users/1").await,
"users#show id=1"
);
assert_eq!(
call_route(&app, Method::GET, "/users/1/edit").await,
"users#edit id=1"
);
assert_eq!(
call_route(&app, Method::PATCH, "/users/1").await,
"users#update id=1"
);
assert_eq!(
call_route(&app, Method::PUT, "/users/1").await,
"users#update id=1"
);
assert_eq!(
call_route(&app, Method::DELETE, "/users/1").await,
"users#destroy id=1"
);
}
async fn call_route(app: &Router, method: Method, uri: &str) -> String {
let res = app
.clone()
.oneshot(
Request::builder()
.method(method)
.uri(uri)
.body(Body::empty())
.unwrap(),
)
.await
.unwrap();
let bytes = res.collect().await.unwrap().to_bytes();
String::from_utf8(bytes.to_vec()).unwrap()
}
}