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-07-08 04:43:29 +00:00
committed by GitHub
parent 23ca8010e0
commit 8542a68176
3 changed files with 53 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
proc_macros/proc_macros.rs
View File

@@ -209,3 +209,52 @@ pub fn fanbox_api_quick_test(item: TokenStream) -> TokenStream {
};
stream.into()
}
struct FilterHttpMethods {
pub req: Ident,
pub typ: Expr,
pub methods: Vec<Ident>,
}
impl Parse for FilterHttpMethods {
fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
let req = Ident::parse(input)?;
token::Comma::parse(input)?;
let typ = Expr::parse(input)?;
let mut methods = Vec::new();
loop {
if input.cursor().eof() {
break;
}
token::Comma::parse(input)?;
let method = Ident::parse(input)?;
methods.push(method);
}
Ok(Self { req, typ, methods })
}
}
/// Filter http methods.
///
/// `request, 405 body, [method [, method [, method ...]]]`
#[proc_macro]
pub fn filter_http_methods(item: TokenStream) -> TokenStream {
let FilterHttpMethods { req, typ, methods } = parse_macro_input!(item as FilterHttpMethods);
let mut header_value = Vec::new();
let mut streams = Vec::new();
for method in methods {
header_value.push(method.to_string());
streams.push(quote!(&hyper::Method::#method => {}));
}
let allow_header = header_value.join(", ");
let allow_header = LitStr::new(allow_header.as_str(), req.span());
let stream = quote! {
match #req.method() {
#(#streams)*
_ => {
return Ok(hyper::Response::builder().status(405).header("Allow", #allow_header).body(#typ).unwrap())
}
}
};
stream.into()
}

1
src/server/service.rs
View File

@@ -32,6 +32,7 @@ impl Service<Request<Body>> for PixivDownloaderSvc {
}
fn call(&mut self, req: Request<Body>) -> Self::Future {
println!("{} {}", req.method(), req.uri());
match self.routes.match_route(&req) {
Some(route) => Box::pin(async move {
match route.response(req).await {

4
src/server/version.rs
View File

@@ -6,6 +6,7 @@ use hyper::Body;
use hyper::Request;
use hyper::Response;
use json::JsonValue;
use proc_macros::filter_http_methods;
use regex::Regex;
pub struct VersionContext {
@@ -22,8 +23,9 @@ impl VersionContext {
impl ResponseJsonFor<Body> for VersionContext {
async fn response_json(
&self,
_req: Request<Body>,
req: Request<Body>,
) -> Result<Response<JsonValue>, PixivDownloaderError> {
filter_http_methods!(req, json::object! {}, GET, OPTIONS, POST);
Ok(Response::new(json::object! {"version": [0, 0, 1, 0]}))
}
}