magmaus3/continuwuity
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

move attribute argument extractor to utils

Browse source 
Signed-off-by: Jason Volk <jason@zemos.net>
This commit is contained in:
Jason Volk 2024-11-23 23:25:14 +00:00
parent f30b08f015
commit fd4c447a2d
2 changed files with 39 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

47
src/macros/config.rs
View file

@ -1,9 +1,4 @@
use std::{
collections::{HashMap, HashSet},
fmt::Write as _,
fs::OpenOptions,
io::Write as _,
};
use std::{collections::HashSet, fmt::Write as _, fs::OpenOptions, io::Write as _};
use proc_macro::TokenStream;
use proc_macro2::Span;
@ -13,7 +8,10 @@ use syn::{
ItemStruct, Lit, Meta, MetaList, MetaNameValue, Type, TypePath,
};
use crate::{utils::is_cargo_build, Result};
use crate::{
utils::{get_simple_settings, is_cargo_build},
Result,
};
const UNDOCUMENTED: &str = "# This item is undocumented. Please contribute documentation for it.";
@ -29,7 +27,7 @@ pub(super) fn example_generator(input: ItemStruct, args: &[Meta]) -> Result<Toke
#[allow(clippy::needless_pass_by_value)]
#[allow(unused_variables)]
fn generate_example(input: &ItemStruct, args: &[Meta]) -> Result<()> {
let settings = get_settings(args);
let settings = get_simple_settings(args);
let filename = settings
.get("filename")
@ -120,39 +118,6 @@ fn generate_example(input: &ItemStruct, args: &[Meta]) -> Result<()> {
Ok(())
}
fn get_settings(args: &[Meta]) -> HashMap<String, String> {
let mut map = HashMap::new();
for arg in args {
let Meta::NameValue(MetaNameValue {
path,
value,
..
}) = arg
else {
continue;
};
let Expr::Lit(
ExprLit {
lit: Lit::Str(str),
..
},
..,
) = value
else {
continue;
};
let Some(key) = path.segments.iter().next().map(|s| s.ident.clone()) else {
continue;
};
map.insert(key.to_string(), str.value());
}
map
}
fn get_default(field: &Field) -> Option<String> {
for attr in &field.attrs {
let Meta::List(MetaList {

34
src/macros/utils.rs
View file

@ -1,7 +1,39 @@
use syn::{parse_str, Expr, Generics, Lit, Meta};
use std::collections::HashMap;
use syn::{parse_str, Expr, ExprLit, Generics, Lit, Meta, MetaNameValue};
use crate::Result;
pub(crate) fn get_simple_settings(args: &[Meta]) -> HashMap<String, String> {
args.iter().fold(HashMap::new(), |mut map, arg| {
let Meta::NameValue(MetaNameValue {
path,
value,
..
}) = arg
else {
return map;
};
let Expr::Lit(
ExprLit {
lit: Lit::Str(str),
..
},
..,
) = value
else {
return map;
};
if let Some(key) = path.segments.iter().next().map(|s| s.ident.clone()) {
map.insert(key.to_string(), str.value());
}
map
})
}
pub(crate) fn is_cargo_build() -> bool {
std::env::args()
.find(|flag| flag.starts_with("--emit"))