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

use std::collections::{BTreeMap, BTreeSet};

use crate::generator::ast::{
  DiscriminatedEnumDef, EnumDef, ResponseEnumDef, RustPrimitive, RustType, TypeAliasDef, VariantContent,
};

pub(crate) struct DependencyGraph {
  dependencies: BTreeMap<String, BTreeSet<String>>,
}

impl DependencyGraph {
  pub(crate) fn build(types: &[RustType]) -> Self {
    let mut dependencies = BTreeMap::new();

    for rust_type in types {
      let type_name = rust_type.type_name().to_string();
      let deps = Self::extract_dependencies(rust_type);
      dependencies
        .entry(type_name)
        .and_modify(|existing: &mut BTreeSet<String>| existing.extend(deps.clone()))
        .or_insert(deps);
    }

    Self { dependencies }
  }

  pub(crate) fn get_dependencies(&self, type_name: &str) -> Option<&BTreeSet<String>> {
    self.dependencies.get(type_name)
  }

  fn extract_dependencies(rust_type: &RustType) -> BTreeSet<String> {
    let mut deps = BTreeSet::new();

    match rust_type {
      RustType::Struct(def) => {
        for field in &def.fields {
          Self::extract_from_type_ref(&field.rust_type, &mut deps);
        }
      }
      RustType::Enum(def) => Self::extract_from_enum(def, &mut deps),
      RustType::TypeAlias(def) => Self::extract_from_type_alias(def, &mut deps),
      RustType::DiscriminatedEnum(def) => Self::extract_from_discriminated_enum(def, &mut deps),
      RustType::ResponseEnum(def) => Self::extract_from_response_enum(def, &mut deps),
    }

    deps
  }

  fn extract_from_enum(def: &EnumDef, deps: &mut BTreeSet<String>) {
    for variant in &def.variants {
      if let VariantContent::Tuple(types) = &variant.content {
        for type_ref in types {
          Self::extract_from_type_ref(type_ref, deps);
        }
      }
    }
  }

  fn extract_from_type_alias(def: &TypeAliasDef, deps: &mut BTreeSet<String>) {
    Self::extract_from_type_ref(&def.target, deps);
  }

  fn extract_from_discriminated_enum(def: &DiscriminatedEnumDef, deps: &mut BTreeSet<String>) {
    for variant in &def.variants {
      deps.insert(variant.type_name.clone());
    }
    if let Some(fallback) = &def.fallback {
      deps.insert(fallback.type_name.clone());
    }
  }

  fn extract_from_response_enum(def: &ResponseEnumDef, deps: &mut BTreeSet<String>) {
    for variant in &def.variants {
      if let Some(type_ref) = &variant.schema_type {
        Self::extract_from_type_ref(type_ref, deps);
      }
    }
  }

  fn extract_from_type_ref(type_ref: &crate::generator::ast::TypeRef, deps: &mut BTreeSet<String>) {
    if let RustPrimitive::Custom(name) = &type_ref.base_type {
      deps.insert(name.clone());
    }
  }
}