openapi_nexus/generators/rust/reqwest/
codegen.rs

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

openapi_nexus/generators/rust/reqwest/
codegen.rs
