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 stored_files {
pub const LIST: &str = "stored_file:list";
pub const READ: &str = "stored_file:read";
pub const CREATE: &str = "stored_file:create";
pub const UPDATE: &str = "stored_file:update";
pub const DELETE: &str = "stored_file:delete";
pub const RESTORE: &str = "stored_file:restore";
pub const TRASH: &str = "stored_file:trash";
pub const EMPTY_TRASH: &str = "stored_file:empty_trash";
pub const BULK_CREATE: &str = "stored_file:bulk_create";
pub const UPSERT: &str = "stored_file:upsert";
pub const TRANSITION_COMPLETE_UPLOAD: &str = "stored_file:transition:complete_upload";
pub const TRANSITION_MARK_SAFE: &str = "stored_file:transition:mark_safe";
pub const TRANSITION_QUARANTINE: &str = "stored_file:transition:quarantine";
pub const TRANSITION_MARK_SAFE_BY_ADMIN: &str = "stored_file:transition:mark_safe_by_admin";
pub const TRANSITION_SOFT_DELETE: &str = "stored_file:transition:soft_delete";
pub const TRANSITION_RESTORE: &str = "stored_file:transition:restore";
pub const TRANSITION_PURGE: &str = "stored_file:transition:purge";
pub fn all() -> Vec<&'static str> {
vec![
LIST,
READ,
CREATE,
UPDATE,
DELETE,
RESTORE,
TRASH,
EMPTY_TRASH,
BULK_CREATE,
UPSERT,
TRANSITION_COMPLETE_UPLOAD,
TRANSITION_MARK_SAFE,
TRANSITION_QUARANTINE,
TRANSITION_MARK_SAFE_BY_ADMIN,
TRANSITION_SOFT_DELETE,
TRANSITION_RESTORE,
TRANSITION_PURGE,
]
}
}
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_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::LIST) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_read_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::READ) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_create_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::CREATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_update_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::UPDATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_delete_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::DELETE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_restore_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::RESTORE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_trash_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_empty_trash_stored_file(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, stored_files::EMPTY_TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}