berkus
/
actix-web
mirror of https://github.com/fafhrd91/actix-web
1
0
Fork 0
Code Issues Releases Wiki Activity

Handle multi-line doc comments. 

These are converted into multiple attributes, and so we need to store
them all.
This commit is contained in:
Erik Johnston 2021-02-24 09:59:03 +00:00
parent cb1a712296
commit 257a7d3307
1 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
actix-web-codegen/src/route.rs
View File

@ -177,8 +177,8 @@ pub struct Route {
ast: syn::ItemFn, ast: syn::ItemFn,
resource_type: ResourceType, resource_type: ResourceType,
/// The doc comment attribute to copy to generated struct, if any. /// The doc comment attributes to copy to generated struct, if any.
doc_attribute: Option<syn::Attribute>, doc_attributes: Vec<syn::Attribute>,
} }
fn guess_resource_type(typ: &syn::Type) -> ResourceType { fn guess_resource_type(typ: &syn::Type) -> ResourceType {
@ -226,11 +226,14 @@ impl Route {
// Try and pull out the doc comments so that we can reapply them to the // Try and pull out the doc comments so that we can reapply them to the
// generated struct. // generated struct.
let mut doc_attribute = None; //
// Note that multi line doc comments are converted to multiple doc
// attributes.
let mut doc_attribute = Vec::new();
for attr in &ast.attrs { for attr in &ast.attrs {
if let Some(ident) = attr.path.get_ident() { if let Some(ident) = attr.path.get_ident() {
if ident == "doc" { if ident == "doc" {
doc_attribute = Some(attr.clone()); doc_attribute.push(attr.clone());
} }
} }
} }
@ -262,7 +265,7 @@ impl Route {
args, args,
ast, ast,
resource_type, resource_type,
doc_attribute, doc_attributes: doc_attribute,
}) })
} }
} }
@ -280,7 +283,7 @@ impl ToTokens for Route {
methods, methods,
}, },
resource_type, resource_type,
doc_attribute, doc_attributes: doc_attribute,
} = self; } = self;
let resource_name = name.to_string(); let resource_name = name.to_string();
let method_guards = { let method_guards = {
@ -302,8 +305,13 @@ impl ToTokens for Route {
} }
}; };
let doc_comment_token_stream: TokenStream2 = doc_attribute
.iter()
.map(ToTokens::to_token_stream)
.collect();
let stream = quote! { let stream = quote! {
#doc_attribute #doc_comment_token_stream
#[allow(non_camel_case_types, missing_docs)] #[allow(non_camel_case_types, missing_docs)]
pub struct #name; pub struct #name;