mirror of
https://github.com/wisdgod/cursor-api.git
synced 2025-12-24 13:38:01 +08:00
46 lines
1.4 KiB
Rust
46 lines
1.4 KiB
Rust
use crate::{
|
|
app::{constant::AUTHORIZATION_BEARER_PREFIX, lazy::AUTH_TOKEN},
|
|
common::model::error::ChatError,
|
|
};
|
|
use axum::{
|
|
Json,
|
|
body::Body,
|
|
http::{Request, StatusCode, header::AUTHORIZATION},
|
|
middleware::Next,
|
|
response::{IntoResponse, Response},
|
|
};
|
|
|
|
// 管理员认证中间件函数
|
|
pub async fn admin_auth_middleware(request: Request<Body>, next: Next) -> Response {
|
|
let auth_header = request
|
|
.headers()
|
|
.get(AUTHORIZATION)
|
|
.and_then(|h| h.to_str().ok())
|
|
.and_then(|h| h.strip_prefix(AUTHORIZATION_BEARER_PREFIX));
|
|
|
|
match auth_header {
|
|
Some(token) if token == AUTH_TOKEN.as_str() => next.run(request).await,
|
|
_ => (
|
|
StatusCode::UNAUTHORIZED,
|
|
Json(ChatError::Unauthorized.to_json()),
|
|
)
|
|
.into_response(),
|
|
}
|
|
}
|
|
|
|
// 旧的认证中间件函数，保留向后兼容性
|
|
// pub async fn auth_middleware(request: Request<Body>, next: Next) -> Result<Response, StatusCode> {
|
|
// let auth_header = request
|
|
// .headers()
|
|
// .get(AUTHORIZATION)
|
|
// .and_then(|h| h.to_str().ok())
|
|
// .and_then(|h| h.strip_prefix(AUTHORIZATION_BEARER_PREFIX))
|
|
// .ok_or(StatusCode::UNAUTHORIZED)?;
|
|
|
|
// if auth_header != AUTH_TOKEN.as_str() {
|
|
// return Err(StatusCode::UNAUTHORIZED);
|
|
// }
|
|
|
|
// Ok(next.run(request).await)
|
|
// }
|