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

Update

Browse Source
This commit is contained in:
lifegpc
2022-07-13 04:32:12 +00:00
committed by GitHub
parent 6496197024
commit 16105daa66
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
proc_macros/proc_macros.rs
View File

@@ -287,6 +287,9 @@ impl Parse for FilterHttpMethods {
break;
}
token::Comma::parse(input)?;
if input.cursor().eof() {
break;
}
let method = Ident::parse(input)?;
if method.to_string() == "cors_methods" {
cors_methods.replace(Vec::new());
@@ -359,10 +362,12 @@ pub fn filter_http_methods(item: TokenStream) -> TokenStream {
let mut header_value = Vec::new();
let mut streams = Vec::new();
let mut enable_options = false;
let mut options_method = None;
for method in methods {
header_value.push(method.to_string());
if method == "OPTIONS" && handle_options.value() {
enable_options = true;
options_method = Some(method);
} else {
streams.push(quote!(&hyper::Method::#method => {}));
}
@@ -401,7 +406,8 @@ pub fn filter_http_methods(item: TokenStream) -> TokenStream {
}
None => quote!(),
};
streams.push(quote!(&hyper::Method::OPTIONS => {
let options_method = options_method.unwrap();
streams.push(quote!(&hyper::Method::#options_method => {
let builder = hyper::Response::builder();
let headers = #req.headers();
let origin = match headers.get(hyper::header::ORIGIN) {