oas3-gen 0.25.1

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
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 proc_macro2::TokenStream;
use quote::{ToTokens, quote};

use super::Visibility;
use crate::generator::{
  ast::{ClientRootNode, GlobalLintsNode},
  codegen::generate_source,
};

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ModFileKind {
  Client,
  Server,
}

impl ModFileKind {
  const fn secondary_module_name(self) -> &'static str {
    match self {
      Self::Client => "client",
      Self::Server => "server",
    }
  }
}

#[derive(Debug, Clone)]
pub struct ModFileFragment {
  metadata: ClientRootNode,
  visibility: Visibility,
  kind: ModFileKind,
  source_path: String,
  gen_version: String,
}

impl ModFileFragment {
  pub fn new(
    metadata: ClientRootNode,
    visibility: Visibility,
    kind: ModFileKind,
    source_path: String,
    gen_version: String,
  ) -> Self {
    Self {
      metadata,
      visibility,
      kind,
      source_path,
      gen_version,
    }
  }

  pub fn for_client(
    metadata: ClientRootNode,
    visibility: Visibility,
    source_path: String,
    gen_version: String,
  ) -> Self {
    Self::new(metadata, visibility, ModFileKind::Client, source_path, gen_version)
  }

  pub fn for_server(
    metadata: ClientRootNode,
    visibility: Visibility,
    source_path: String,
    gen_version: String,
  ) -> Self {
    Self::new(metadata, visibility, ModFileKind::Server, source_path, gen_version)
  }

  pub fn generate(&self) -> anyhow::Result<String> {
    let lint_config = GlobalLintsNode::default();
    generate_source(
      &self.to_token_stream(),
      &self.metadata,
      Some(&lint_config),
      &self.source_path,
      &self.gen_version,
    )
  }
}

impl ToTokens for ModFileFragment {
  fn to_tokens(&self, tokens: &mut TokenStream) {
    let vis = &self.visibility;
    let secondary_mod = syn::Ident::new(self.kind.secondary_module_name(), proc_macro2::Span::call_site());

    tokens.extend(quote! {
      mod types;
      mod #secondary_mod;

      #vis use types::*;
      #vis use #secondary_mod::*;
    });
  }
}