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 file_locks {
pub const LIST: &str = "file_lock:list";
pub const READ: &str = "file_lock:read";
pub const CREATE: &str = "file_lock:create";
pub const UPDATE: &str = "file_lock:update";
pub const DELETE: &str = "file_lock:delete";
pub const RESTORE: &str = "file_lock:restore";
pub const TRASH: &str = "file_lock:trash";
pub const EMPTY_TRASH: &str = "file_lock:empty_trash";
pub const BULK_CREATE: &str = "file_lock:bulk_create";
pub const UPSERT: &str = "file_lock:upsert";
pub const TRANSITION_EXPIRE: &str = "file_lock:transition:expire";
pub const TRANSITION_RELEASE: &str = "file_lock:transition:release";
pub const TRANSITION_REFRESH: &str = "file_lock:transition:refresh";
pub fn all() -> Vec<&'static str> {
vec![
LIST,
READ,
CREATE,
UPDATE,
DELETE,
RESTORE,
TRASH,
EMPTY_TRASH,
BULK_CREATE,
UPSERT,
TRANSITION_EXPIRE,
TRANSITION_RELEASE,
TRANSITION_REFRESH,
]
}
}
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_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::LIST) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_read_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::READ) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_create_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::CREATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_update_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::UPDATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_delete_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::DELETE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_restore_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::RESTORE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_trash_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_empty_trash_file_lock(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, file_locks::EMPTY_TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}