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 upload_sessions {
pub const LIST: &str = "upload_session:list";
pub const READ: &str = "upload_session:read";
pub const CREATE: &str = "upload_session:create";
pub const UPDATE: &str = "upload_session:update";
pub const DELETE: &str = "upload_session:delete";
pub const RESTORE: &str = "upload_session:restore";
pub const TRASH: &str = "upload_session:trash";
pub const EMPTY_TRASH: &str = "upload_session:empty_trash";
pub const BULK_CREATE: &str = "upload_session:bulk_create";
pub const UPSERT: &str = "upload_session:upsert";
pub const TRANSITION_START_UPLOAD: &str = "upload_session:transition:start_upload";
pub const TRANSITION_ADD_PART: &str = "upload_session:transition:add_part";
pub const TRANSITION_COMPLETE: &str = "upload_session:transition:complete";
pub const TRANSITION_FINISH: &str = "upload_session:transition:finish";
pub const TRANSITION_FAIL: &str = "upload_session:transition:fail";
pub const TRANSITION_ABORT: &str = "upload_session:transition:abort";
pub const TRANSITION_EXPIRE: &str = "upload_session:transition:expire";
pub const TRANSITION_RETRY: &str = "upload_session:transition:retry";
pub fn all() -> Vec<&'static str> {
vec![
LIST,
READ,
CREATE,
UPDATE,
DELETE,
RESTORE,
TRASH,
EMPTY_TRASH,
BULK_CREATE,
UPSERT,
TRANSITION_START_UPLOAD,
TRANSITION_ADD_PART,
TRANSITION_COMPLETE,
TRANSITION_FINISH,
TRANSITION_FAIL,
TRANSITION_ABORT,
TRANSITION_EXPIRE,
TRANSITION_RETRY,
]
}
}
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_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::LIST) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_read_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::READ) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_create_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::CREATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_update_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::UPDATE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_delete_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::DELETE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_restore_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::RESTORE) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_trash_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}
pub async fn require_empty_trash_upload_session(
auth: axum::extract::Extension<AuthContext>,
request: Request,
next: Next,
) -> Result<Response, StatusCode> {
if has_permission(&auth, upload_sessions::EMPTY_TRASH) {
Ok(next.run(request).await)
} else {
Err(StatusCode::FORBIDDEN)
}
}