openapi_nexus/generators/go/http/
codegen.rs1use std::error::Error;
10
11use heck::ToKebabCase as _;
12
13use super::config::GoHttpConfig;
14use super::runtime::runtime_files;
15use super::{sigil_emit, sigil_emit_api};
16use crate::codegen::traits::code_generator::CodeGenerator;
17use crate::codegen::traits::file_writer::{FileInfo, FileWriter};
18use crate::codegen::{GeneratorType, Language};
19use crate::ir::types::{IrInfo, IrSpec};
20
21const DEFAULT_MODULE_PATH: &str = "example.com/sdk";
22
23#[derive(Debug, Clone)]
25pub struct GoHttpCodeGenerator {
26 config: GoHttpConfig,
27}
28
29impl GoHttpCodeGenerator {
30 pub fn new(config: toml::value::Table) -> Self {
32 Self {
33 config: GoHttpConfig::from(config),
34 }
35 }
36
37 fn module_path(&self) -> String {
38 self.config
39 .module_path
40 .clone()
41 .unwrap_or_else(|| DEFAULT_MODULE_PATH.to_string())
42 }
43
44 fn generate_ir(&self, ir: &IrSpec) -> Result<Vec<FileInfo>, Box<dyn Error + Send + Sync>> {
46 let module_path = self.module_path();
47 let header = render_file_header(&ir.info);
48
49 let mut files = Vec::new();
50
51 files.extend(
53 sigil_emit::generate_model_files(ir, &header).map_err(|msg| {
54 Box::<dyn Error + Send + Sync>::from(format!("sigil_emit models: {msg}"))
55 })?,
56 );
57
58 files.extend(
60 sigil_emit_api::generate_api_files(ir, &module_path, &header).map_err(|msg| {
61 Box::<dyn Error + Send + Sync>::from(format!("sigil_emit_api: {msg}"))
62 })?,
63 );
64
65 files.extend(runtime_files(&header));
67
68 files.push(go_mod_file(&module_path));
70 files.push(readme_file(&ir.info));
71
72 Ok(files)
73 }
74}
75
76impl CodeGenerator for GoHttpCodeGenerator {
77 fn language(&self) -> Language {
78 Language::Go
79 }
80
81 fn generator_type(&self) -> GeneratorType {
82 GeneratorType::GoHttp
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 GoHttpCodeGenerator {}
91
92fn render_file_header(info: &IrInfo) -> String {
93 let mut out = String::new();
94 out.push_str("// Code generated by openapi-nexus. DO NOT EDIT.\n");
95 out.push_str("//\n");
96 out.push_str(&format!("// {} — {}\n", info.title, info.version));
97 if let Some(desc) = &info.description {
98 for line in desc.lines() {
99 out.push_str(&format!("// {line}\n"));
100 }
101 }
102 out.push('\n');
103 out
104}
105
106fn go_mod_file(module_path: &str) -> FileInfo {
107 let content = format!("module {module_path}\n\ngo 1.21\n");
108 FileInfo::project("go.mod".to_string(), content)
109}
110
111fn readme_file(info: &IrInfo) -> FileInfo {
112 let title = &info.title;
113 let version = &info.version;
114 let description = info
115 .description
116 .clone()
117 .unwrap_or_else(|| "Generated Go SDK.".to_string());
118 let package = info.title.to_kebab_case();
119 let content = format!(
120 "# {title}\n\n{description}\n\nVersion: `{version}`\n\nGenerated by [openapi-nexus](https://github.com/adamcavendish/openapi-nexus) for `{package}`.\n"
121 );
122 FileInfo::readme("README.md".to_string(), content)
123}