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 buckets {
pub const LIST: &str = "bucket:list";
pub const READ: &str = "bucket:read";
pub const CREATE: &str = "bucket:create";
pub const UPDATE: &str = "bucket:update";
pub const DELETE: &str = "bucket:delete";
pub const RESTORE: &str = "bucket:restore";
pub const TRASH: &str = "bucket:trash";
pub const EMPTY_TRASH: &str = "bucket:empty_trash";
pub const BULK_CREATE: &str = "bucket:bulk_create";
pub const UPSERT: &str = "bucket:upsert";
pub const TRANSITION_LOCK: &str = "bucket:transition:lock";
pub const TRANSITION_UNLOCK: &str = "bucket:transition:unlock";
pub const TRANSITION_ARCHIVE: &str = "bucket:transition:archive";
pub const TRANSITION_RESTORE: &str = "bucket:transition:restore";
pub const TRANSITION_DELETE: &str = "bucket:transition:delete";
pub fn all() -> Vec<&'static str> {
vec![
LIST,
READ,
CREATE,
UPDATE,
DELETE,
RESTORE,
TRASH,
EMPTY_TRASH,
BULK_CREATE,
UPSERT,
TRANSITION_LOCK,
TRANSITION_UNLOCK,
TRANSITION_ARCHIVE,
TRANSITION_RESTORE,
TRANSITION_DELETE,
]
}
}
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_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::LIST) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_read_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::READ) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_create_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::CREATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_update_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::UPDATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_delete_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::DELETE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_restore_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::RESTORE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_trash_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_empty_trash_bucket(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, buckets::EMPTY_TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}