1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
use syn::punctuated::Punctuated;
use syn::{Attribute, Error, Expr, ExprLit, Lit, Meta, Token};
use crate::case::{RenameRule, RENAME_RULES};
pub struct Overrides {
pub name: Option<String>,
pub rename_all: Option<RenameRule>,
pub transparent: bool,
pub allow_mismatch: bool,
}
impl Overrides {
pub fn extract(attrs: &[Attribute], container_attr: bool) -> Result<Overrides, Error> {
let mut overrides = Overrides {
name: None,
rename_all: None,
transparent: false,
allow_mismatch: false,
};
for attr in attrs {
if !attr.path().is_ident("postgres") {
continue;
}
let list = match &attr.meta {
Meta::List(ref list) => list,
bad => return Err(Error::new_spanned(bad, "expected a #[postgres(...)]")),
};
let nested = list.parse_args_with(Punctuated::<Meta, Token![,]>::parse_terminated)?;
for item in nested {
match item {
Meta::NameValue(meta) => {
let name_override = meta.path.is_ident("name");
let rename_all_override = meta.path.is_ident("rename_all");
if !container_attr && rename_all_override {
return Err(Error::new_spanned(
&meta.path,
"rename_all is a container attribute",
));
}
if !name_override && !rename_all_override {
return Err(Error::new_spanned(&meta.path, "unknown override"));
}
let value = match &meta.value {
Expr::Lit(ExprLit {
lit: Lit::Str(lit), ..
}) => lit.value(),
bad => {
return Err(Error::new_spanned(bad, "expected a string literal"))
}
};
if name_override {
overrides.name = Some(value);
} else if rename_all_override {
let rename_rule = RenameRule::from_str(&value).ok_or_else(|| {
Error::new_spanned(
&meta.value,
format!(
"invalid rename_all rule, expected one of: {}",
RENAME_RULES
.iter()
.map(|rule| format!("\"{rule}\""))
.collect::<Vec<_>>()
.join(", ")
),
)
})?;
overrides.rename_all = Some(rename_rule);
}
}
Meta::Path(path) => {
if path.is_ident("transparent") {
if overrides.allow_mismatch {
return Err(Error::new_spanned(
path,
"#[postgres(allow_mismatch)] is not allowed with #[postgres(transparent)]",
));
}
overrides.transparent = true;
} else if path.is_ident("allow_mismatch") {
if overrides.transparent {
return Err(Error::new_spanned(
path,
"#[postgres(transparent)] is not allowed with #[postgres(allow_mismatch)]",
));
}
overrides.allow_mismatch = true;
} else {
return Err(Error::new_spanned(path, "unknown override"));
}
}
bad => return Err(Error::new_spanned(bad, "unknown attribute")),
}
}
}
Ok(overrides)
}
}