openapi_nexus/generators/rust/ureq/
codegen.rs

1//! Rust ureq code generator: IR → idiomatic Rust + ureq (synchronous).
2
3use std::error::Error;
4
5use heck::ToKebabCase as _;
6
7use super::runtime::runtime_files;
8use super::sigil_emit_api::{emit_method_body, ureq_backend_config};
9use crate::codegen::traits::code_generator::CodeGenerator;
10use crate::codegen::traits::file_writer::{FileInfo, FileWriter};
11use crate::codegen::{GeneratorType, Language};
12use crate::generators::rust::common::{
13    config::RustGeneratorConfig, emit_api, emit_models, project_files,
14};
15use crate::ir::types::{IrInfo, IrSpec};
16
17/// Rust ureq code generator (synchronous).
18#[derive(Debug, Clone)]
19pub struct RustUreqCodeGenerator {
20    config: RustGeneratorConfig,
21}
22
23impl RustUreqCodeGenerator {
24    pub fn new(config: toml::value::Table) -> Self {
25        Self {
26            config: RustGeneratorConfig::from(config),
27        }
28    }
29
30    fn crate_name(&self, info: &IrInfo) -> String {
31        self.config
32            .crate_name
33            .clone()
34            .unwrap_or_else(|| info.title.to_kebab_case())
35    }
36
37    fn generate_ir(&self, ir: &IrSpec) -> Result<Vec<FileInfo>, Box<dyn Error + Send + Sync>> {
38        let header = project_files::render_file_header(&ir.info);
39        let crate_name = self.crate_name(&ir.info);
40        let backend_config = ureq_backend_config();
41
42        let mut files = Vec::new();
43
44        files.extend(
45            emit_models::generate_model_files(ir, &header, &self.config).map_err(|msg| {
46                Box::<dyn Error + Send + Sync>::from(format!("emit_models: {msg}"))
47            })?,
48        );
49
50        let response_extra = self
51            .config
52            .extra_derives
53            .as_ref()
54            .and_then(|e| e.response_structs.as_ref());
55        files.extend(
56            emit_api::generate_api_files(
57                ir,
58                &header,
59                &backend_config,
60                response_extra,
61                &emit_method_body,
62            )
63            .map_err(|msg| Box::<dyn Error + Send + Sync>::from(format!("emit_api: {msg}")))?,
64        );
65
66        files.extend(runtime_files(&header));
67
68        files.push(cargo_toml_file(&crate_name, &ir.info, &self.config));
69        files.push(project_files::lib_rs_file(&header));
70        files.push(project_files::readme_file(&ir.info));
71
72        Ok(files)
73    }
74}
75
76impl CodeGenerator for RustUreqCodeGenerator {
77    fn language(&self) -> Language {
78        Language::Rust
79    }
80
81    fn generator_type(&self) -> GeneratorType {
82        GeneratorType::RustUreq
83    }
84
85    fn generate(&self, ir: &IrSpec) -> Result<Vec<FileInfo>, Box<dyn Error + Send + Sync>> {
86        self.generate_ir(ir)
87    }
88}
89
90impl FileWriter for RustUreqCodeGenerator {
91    fn source_dir(&self) -> Option<&str> {
92        Some("src")
93    }
94}
95
96fn cargo_toml_file(crate_name: &str, info: &IrInfo, config: &RustGeneratorConfig) -> FileInfo {
97    use crate::generators::rust::common::config::WorkspaceDepsMode;
98
99    let description = info
100        .description
101        .as_deref()
102        .unwrap_or("Generated Rust SDK.")
103        .lines()
104        .next()
105        .unwrap_or("Generated Rust SDK.")
106        .replace('"', r#"\""#);
107    let workspace = config.workspace_mode.unwrap_or(false);
108    let deps_mode = config.workspace_deps.as_ref().cloned().unwrap_or_default();
109
110    let pkg_section = if workspace {
111        format!(
112            r#"[package]
113name = "{crate_name}"
114version.workspace = true
115edition.workspace = true
116description = "{description}"
117"#,
118        )
119    } else {
120        format!(
121            r#"[package]
122name = "{crate_name}"
123version = "0.1.0"
124edition = "2024"
125description = "{description}"
126"#,
127        )
128    };
129
130    let deps_section = match deps_mode {
131        WorkspaceDepsMode::Full => r#"
132[dependencies]
133ureq.workspace = true
134serde.workspace = true
135serde_json.workspace = true
136serde_repr.workspace = true
137"#
138        .to_string(),
139        WorkspaceDepsMode::WorkspaceVersion => r#"
140[dependencies]
141ureq = { workspace = true, features = ["json"] }
142serde = { workspace = true, features = ["derive"] }
143serde_json.workspace = true
144serde_repr.workspace = true
145"#
146        .to_string(),
147        WorkspaceDepsMode::Explicit => r#"
148[dependencies]
149ureq = { version = "3", features = ["json"] }
150serde = { version = "1", features = ["derive"] }
151serde_json = "1"
152serde_repr = "0.1"
153"#
154        .to_string(),
155    };
156
157    let mut content = format!("{pkg_section}{deps_section}");
158
159    if let Some(extra) = &config.extra_derives {
160        extra.warn_missing_dependencies();
161        for (name, spec) in extra.all_dependencies() {
162            match deps_mode {
163                WorkspaceDepsMode::Full | WorkspaceDepsMode::WorkspaceVersion => {
164                    content.push_str(&format!("{name}.workspace = true\n"));
165                }
166                WorkspaceDepsMode::Explicit => {
167                    content.push_str(&format!("{name} = {spec}\n"));
168                }
169            }
170        }
171    }
172    if let Some(utoipa_cfg) = &config.utoipa
173        && utoipa_cfg.enabled
174    {
175        let spec = utoipa_cfg.dependency.as_deref().unwrap_or("\"*\"");
176        match deps_mode {
177            WorkspaceDepsMode::Full | WorkspaceDepsMode::WorkspaceVersion => {
178                content.push_str("utoipa.workspace = true\n");
179            }
180            WorkspaceDepsMode::Explicit => {
181                content.push_str(&format!("utoipa = {spec}\n"));
182            }
183        }
184    }
185
186    if workspace {
187        content.push_str("\n[lints]\nworkspace = true\n");
188    }
189
190    FileInfo::project("Cargo.toml".to_string(), content)
191}
openapi_nexus/generators/rust/ureq/codegen.rs

openapi_nexus/generators/rust/ureq/
codegen.rs
