diesel_derives/deprecated/
belongs_to.rs

Help
1use syn::parse::{ParseStream, Result};
2use syn::token::Comma;
3use syn::{parenthesized, Ident, LitStr};
4
5use crate::deprecated::utils::parse_eq_and_lit_str;
6use crate::parsers::BelongsTo;
7use crate::util::BELONGS_TO_NOTE;
8
9pub fn parse_belongs_to(name: Ident, input: ParseStream) -> Result<BelongsTo> {
10    if input.is_empty() {
11        return Err(syn::Error::new(
12            name.span(),
13            format!(
14                "unexpected end of input, expected parentheses\n\
15                 help: The correct format looks like `#[diesel({})]`",
16                BELONGS_TO_NOTE
17            ),
18        ));
19    }
20
21    let content;
22    parenthesized!(content in input);
23
24    let parent = if content.peek(Ident) {
25        let name: Ident = content.parse()?;
26
27        if name == "parent" {
28            let lit_str = parse_eq_and_lit_str(name, &content, BELONGS_TO_NOTE)?;
29            lit_str.parse()?
30        } else {
31            LitStr::new(&name.to_string(), name.span()).parse()?
32        }
33    } else {
34        content.parse()?
35    };
36
37    let mut foreign_key = None;
38
39    if content.peek(Comma) {
40        content.parse::<Comma>()?;
41
42        let name: Ident = content.parse()?;
43
44        if name != "foreign_key" {
45            return Err(syn::Error::new(name.span(), "expected `foreign_key`"));
46        }
47
48        let lit_str = parse_eq_and_lit_str(name, &content, BELONGS_TO_NOTE)?;
49        foreign_key = Some(lit_str.parse()?);
50    }
51
52    Ok(BelongsTo {
53        parent,
54        foreign_key,
55    })
56}
diesel_derives/deprecated/belongs_to.rs

diesel_derives/deprecated/
belongs_to.rs
