oas3-gen 0.26.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
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

use std::collections::HashSet;

use oas3::{
  Spec,
  spec::{Operation, Response},
};

use crate::{generator::naming::identifiers::to_rust_type_name, utils::SchemaRefName};

const SUCCESS_RESPONSE_PREFIX: char = '2';

pub struct ResponseTypes {
  pub success: Vec<String>,
  pub error: Vec<String>,
}

pub fn extract_response_type_name(spec: &Spec, operation: &Operation) -> Option<String> {
  find_primary_response(spec, operation)
    .and_then(|resp| extract_schema_name_from_response(&resp))
    .map(|s| to_rust_type_name(&s))
}

pub fn extract_all_response_content_types(spec: &Spec, operation: &Operation) -> Vec<String> {
  find_primary_response(spec, operation)
    .map(|resp| resp.content.keys().cloned().collect())
    .unwrap_or_default()
}

pub fn extract_all_response_types(spec: &Spec, operation: &Operation) -> ResponseTypes {
  let mut success_set = HashSet::new();
  let mut error_set = HashSet::new();

  let Some(responses) = operation.responses.as_ref() else {
    return ResponseTypes {
      success: vec![],
      error: vec![],
    };
  };

  for (code, resp_ref) in responses {
    if let Ok(resp) = resp_ref.resolve(spec)
      && let Some(schema_name) = extract_schema_name_from_response(&resp)
    {
      let rust_name = to_rust_type_name(&schema_name);
      if is_success_code(code) {
        success_set.insert(rust_name);
      } else if is_error_code(code) {
        error_set.insert(rust_name);
      }
    }
  }

  ResponseTypes {
    success: success_set.into_iter().collect(),
    error: error_set.into_iter().collect(),
  }
}

fn find_primary_response(spec: &Spec, operation: &Operation) -> Option<Response> {
  let responses = operation.responses.as_ref()?;
  responses
    .iter()
    .find(|(code, _)| code.starts_with(SUCCESS_RESPONSE_PREFIX))
    .or_else(|| responses.iter().next())
    .and_then(|(_, resp_ref)| resp_ref.resolve(spec).ok())
}

pub fn is_success_code(code: &str) -> bool {
  code.starts_with(SUCCESS_RESPONSE_PREFIX)
}

pub fn is_error_code(code: &str) -> bool {
  code.starts_with('4') || code.starts_with('5')
}

pub fn extract_schema_name_from_response(response: &Response) -> Option<String> {
  response
    .content
    .values()
    .next()?
    .schema
    .as_ref()
    .and_then(SchemaRefName::schema_ref_name)
}