lifegpc/pixiv_downloader
1
0
Fork 0
mirror of https://github.com/lifegpc/pixiv_downloader.git synced 2026-06-06 05:49:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
lifegpc
2022-09-27 04:40:02 +00:00
committed by GitHub
parent 5b4ba8d80a
commit b98ac08e4b
4 changed files with 143 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
doc/api/auth.zh_CN.md
View File

@@ -12,3 +12,8 @@
* 方法: `GET` 或 `POST`
* RESTful: `PUT /api/auth/user` 或 `PUT /auth/user`
* 鉴权: 一般需要（如服务器状态内的`has_root_user`为`false`则无需鉴权）
## 获取Token
* 路径: `/api/auth/token/add`、 `/auth/token/add`
* 方法: `GET` 或 `POST`
* RESTful: `PUT /api/auth/token` 或 `PUT /auth/token`
* 鉴权：不需要

2
src/server/auth/mod.rs
View File

@@ -1,9 +1,11 @@
pub mod pubkey;
pub mod status;
pub mod token;
pub mod user;
pub use pubkey::{AuthPubkeyContext, AuthPubkeyRoute, RSAKey};
pub use status::{AuthStatusContext, AuthStatusRoute};
pub use token::{AuthTokenContext, AuthTokenRoute};
pub use user::{AuthUserContext, AuthUserRoute};
const PASSWORD_SALT: [u8; 64] = [

135
src/server/auth/token.rs Normal file
View File

@@ -0,0 +1,135 @@
use super::super::preclude::*;
use crate::ext::json::ToJson2;
/// Action to perform about token
pub enum AuthTokenAction {
/// Add a new token
Add,
}
pub struct AuthTokenContext {
ctx: Arc<ServerContext>,
action: Option<AuthTokenAction>,
is_restful: bool,
}
impl AuthTokenContext {
pub fn new(ctx: Arc<ServerContext>, action: Option<AuthTokenAction>, is_restful: bool) -> Self {
Self {
ctx,
action,
is_restful,
}
}
async fn handle(&self, mut req: Request<Body>) -> JSONResult {
Ok(json::object! {})
}
}
#[async_trait]
impl ResponseJsonFor<Body> for AuthTokenContext {
async fn response_json(
&self,
req: Request<Body>,
) -> Result<Response<JsonValue>, PixivDownloaderError> {
let builder = if self.is_restful {
filter_http_methods!(
req,
json::object! {},
true,
self.ctx,
allow_headers = [CONTENT_TYPE, X_SIGN, X_TOKEN_ID],
OPTIONS,
PUT,
);
builder
} else {
filter_http_methods!(
req,
json::object! {},
true,
self.ctx,
allow_headers = [CONTENT_TYPE, X_SIGN, X_TOKEN_ID],
GET,
OPTIONS,
POST,
);
builder
};
let re = self.handle(req).await;
let builder = match &re {
Ok(_) => builder,
Err(err) => {
if err.code <= -400 && err.code >= -600 {
builder.status((-err.code) as u16)
} else if err.code < 0 {
builder.status(500)
} else if err.code > 0 {
builder.status(400)
} else {
builder
}
}
};
Ok(builder.body(re.to_json2())?)
}
}
pub struct AuthTokenRoute {
regex: Regex,
}
impl AuthTokenRoute {
pub fn new() -> Self {
Self {
regex: Regex::new(r"^(/+api)?/+auth/+token(/+add)?$").unwrap(),
}
}
}
impl MatchRoute<Body, Body> for AuthTokenRoute {
fn match_route(
&self,
ctx: &Arc<ServerContext>,
req: &http::Request<Body>,
) -> Option<Box<ResponseForType>> {
let path = req.uri().path();
let pat = self.regex.captures(path);
match pat {
Some(cap) => {
if req.method() == Method::OPTIONS {
return Some(Box::new(AuthTokenContext::new(
Arc::clone(ctx),
None,
cap.get(2).is_none(),
)));
}
let cap2 = cap.get(2);
let is_restful = cap2.is_none();
let action = match cap2 {
Some(m) => {
let m = m.as_str().trim_start_matches("/");
match m {
"add" => Some(AuthTokenAction::Add),
_ => return None,
}
}
None => {
if req.method() == Method::PUT {
Some(AuthTokenAction::Add)
} else {
None
}
}
};
Some(Box::new(AuthTokenContext::new(
Arc::clone(ctx),
action,
is_restful,
)))
}
None => None,
}
}
}

1
src/server/route.rs
View File

@@ -21,6 +21,7 @@ impl ServerRoutes {
routes.push(Box::new(AuthStatusRoute::new()));
routes.push(Box::new(AuthUserRoute::new()));
routes.push(Box::new(AuthPubkeyRoute::new()));
routes.push(Box::new(AuthTokenRoute::new()));
Self { routes }
}