axum-openapi3-derive 0.2.3

Facilitate the generation of OpenAPI 3.0 documentation for axum applications - derive macro
Documentation 
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

use syn::{spanned::Spanned, FnArg, Signature};

use crate::util::recursive_type;

#[derive(Debug)]
pub enum HandlerArgument {
    RequestBody(String),
    Query(String),
    State(String),
    Path(String),
}

pub fn parse_handler_arguments(sig: &Signature) -> Result<Vec<HandlerArgument>, syn::Error> {
    let mut ret = vec![];
    for input in &sig.inputs {
        match input {
            FnArg::Typed(pat_type) => {
                let ty = &pat_type.ty;

                match ty.as_ref() {
                    syn::Type::Path(path) => {
                        let mut v = vec![];
                        recursive_type(path, &mut v);

                        let extractor = v.remove(0);

                        match extractor.as_str() {
                            "Json" => {
                                let l = v.len();

                                ret.push(HandlerArgument::RequestBody(
                                    v.join("<") + ">".repeat(l - 1).as_str(),
                                ));
                            }
                            "Query" => {
                                let l = v.len();

                                ret.push(HandlerArgument::Query(
                                    v.join("<") + ">".repeat(l - 1).as_str(),
                                ));
                            }
                            "State" => {
                                let l = v.len();

                                ret.push(HandlerArgument::State(
                                    v.join("<") + ">".repeat(l - 1).as_str(),
                                ));
                            }
                            "Path" => {
                                let l = v.len();

                                ret.push(HandlerArgument::Path(
                                    v.join("<") + ">".repeat(l - 1).as_str(),
                                ));
                            }
                            _ => continue,
                        };
                    }
                    _ => {
                        return Err(syn::Error::new(ty.span(), "Expected Type::Path type"));
                    }
                }
            }
            _ => {
                return Err(syn::Error::new(
                    input.span(),
                    "Self argument is unsupported",
                ));
            }
        };
    }

    Ok(ret)
}

pub fn parse_handler_ret_type(sig: &Signature) -> Result<Option<String>, syn::Error> {
    match &sig.output {
        syn::ReturnType::Default => Err(syn::Error::new(sig.output.span(), "Expected return type")),
        syn::ReturnType::Type(_, ty) => match ty.as_ref() {
            syn::Type::Path(path) => {
                let mut v = vec![];
                recursive_type(path, &mut v);

                if v.remove(0) != "Json" {
                    Ok(None)
                } else {
                    let l = v.len();
                    Ok(Some(v.join("<") + ">".repeat(l - 1).as_str()))
                }
            }
            _ => Ok(None),
        },
    }
}