oas3-gen 0.24.0

A rust type generator for OpenAPI v3.1.x specification.
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

use inflections::Inflect as _;
use quote::{ToTokens, quote};

use crate::generator::{
  ast::{
    RegexKey,
    tokens::{ConstToken, HeaderNameToken},
  },
  naming::identifiers::sanitize,
};

impl From<&RegexKey> for ConstToken {
  fn from(key: &RegexKey) -> Self {
    let joined = key
      .parts()
      .iter()
      .map(|part| sanitize(part))
      .collect::<Vec<_>>()
      .join("_");

    let mut ident = joined.to_constant_case();

    if ident.starts_with(|c: char| c.is_ascii_digit()) {
      ident.insert(0, '_');
    }

    ConstToken::new(format!("REGEX_{ident}"))
  }
}

#[derive(Debug, Clone, PartialEq, Eq, PartialOrd, Ord, Hash, bon::Builder)]
pub struct HttpHeaderRef {
  pub const_token: ConstToken,
  pub header_name: HeaderNameToken,
}

impl From<&str> for HttpHeaderRef {
  fn from(s: &str) -> Self {
    Self {
      const_token: ConstToken::from_raw(s),
      header_name: HeaderNameToken::from_raw(s),
    }
  }
}

impl ToTokens for HttpHeaderRef {
  fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
    let const_token = &self.const_token;
    let header_name = &self.header_name;

    let header = quote! {
      pub const #const_token: http::HeaderName = http::HeaderName::from_static(#header_name);
    };

    tokens.extend(header);
  }
}