use std::sync::Arc;
use super::AuthContextExt;
use backbone_auth::middleware::AuthContext;
use axum::{
extract::{Request, State},
http::StatusCode,
middleware::Next,
response::Response,
};
pub mod user_quotas {
pub const LIST: &str = "user_quota:list";
pub const READ: &str = "user_quota:read";
pub const CREATE: &str = "user_quota:create";
pub const UPDATE: &str = "user_quota:update";
pub const DELETE: &str = "user_quota:delete";
pub const RESTORE: &str = "user_quota:restore";
pub const TRASH: &str = "user_quota:trash";
pub const EMPTY_TRASH: &str = "user_quota:empty_trash";
pub const BULK_CREATE: &str = "user_quota:bulk_create";
pub const UPSERT: &str = "user_quota:upsert";
pub const TRANSITION_EXCEED: &str = "user_quota:transition:exceed";
pub const TRANSITION_RESTORE: &str = "user_quota:transition:restore";
pub fn all() -> Vec<&'static str> {
vec![
LIST,
READ,
CREATE,
UPDATE,
DELETE,
RESTORE,
TRASH,
EMPTY_TRASH,
BULK_CREATE,
UPSERT,
TRANSITION_EXCEED,
TRANSITION_RESTORE,
]
}
}
pub fn has_permission(auth: &AuthContext, permission: &str) -> bool {
auth.has_permission(permission)
}
pub fn has_role(auth: &AuthContext, role: &str) -> bool {
auth.has_role(role)
}
pub fn has_all_roles(auth: &AuthContext, roles: &[&str]) -> bool {
auth.has_all_roles(roles)
}
pub fn has_any_permission(auth: &AuthContext, permissions: &[&str]) -> bool {
permissions.iter().any(|p| auth.has_permission(p))
}
pub async fn require_list_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::LIST) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_read_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::READ) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_create_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::CREATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_update_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::UPDATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_delete_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::DELETE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_restore_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::RESTORE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_trash_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_empty_trash_user_quota(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, user_quotas::EMPTY_TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}