openapi_nexus/generators/kotlin/okhttp/
codegen.rs

1use std::error::Error;
2
3use heck::ToKebabCase as _;
4
5use super::config::KotlinOkhttpConfig;
6use super::runtime::runtime_files;
7use super::{sigil_emit, sigil_emit_api};
8use crate::codegen::traits::code_generator::CodeGenerator;
9use crate::codegen::traits::file_writer::{FileInfo, FileWriter};
10use crate::codegen::{GeneratorType, Language};
11use crate::generators::request_inputs::plan_multipart_request_inputs;
12use crate::ir::types::{IrInfo, IrSpec};
13
14const DEFAULT_PACKAGE: &str = "com.example.sdk";
15
16#[derive(Debug, Clone)]
17pub struct KotlinOkhttpCodeGenerator {
18    config: KotlinOkhttpConfig,
19}
20
21impl KotlinOkhttpCodeGenerator {
22    pub fn new(config: toml::value::Table) -> Self {
23        Self {
24            config: KotlinOkhttpConfig::from(config),
25        }
26    }
27
28    fn package_name(&self) -> String {
29        self.config
30            .package_name
31            .clone()
32            .unwrap_or_else(|| DEFAULT_PACKAGE.to_string())
33    }
34
35    fn generate_ir(&self, ir: &IrSpec) -> Result<Vec<FileInfo>, Box<dyn Error + Send + Sync>> {
36        let package_name = self.package_name();
37        let header = render_file_header(&ir.info);
38        let request_inputs = plan_multipart_request_inputs(ir);
39
40        let mut files = Vec::new();
41
42        files.extend(
43            sigil_emit::generate_model_files(ir, &package_name, &header, &request_inputs).map_err(
44                |msg| Box::<dyn Error + Send + Sync>::from(format!("sigil_emit models: {msg}")),
45            )?,
46        );
47
48        files.extend(
49            sigil_emit_api::generate_api_files(ir, &package_name, &header, &request_inputs)
50                .map_err(|msg| {
51                    Box::<dyn Error + Send + Sync>::from(format!("sigil_emit_api: {msg}"))
52                })?,
53        );
54
55        files.extend(runtime_files(
56            &header,
57            &package_name,
58            request_inputs.has_uploads(),
59        ));
60
61        files.push(build_gradle_file(&package_name, &ir.info));
62        files.push(readme_file(&ir.info));
63
64        Ok(files)
65    }
66}
67
68impl CodeGenerator for KotlinOkhttpCodeGenerator {
69    fn language(&self) -> Language {
70        Language::Kotlin
71    }
72
73    fn generator_type(&self) -> GeneratorType {
74        GeneratorType::KotlinOkhttp
75    }
76
77    fn generate(&self, ir: &IrSpec) -> Result<Vec<FileInfo>, Box<dyn Error + Send + Sync>> {
78        self.generate_ir(ir)
79    }
80}
81
82impl FileWriter for KotlinOkhttpCodeGenerator {}
83
84fn render_file_header(info: &IrInfo) -> String {
85    let mut out = String::new();
86    out.push_str("// Code generated by openapi-nexus. DO NOT EDIT.\n");
87    out.push_str("//\n");
88    out.push_str(&format!("// {} — {}\n", info.title, info.version));
89    if let Some(desc) = &info.description {
90        for line in desc.lines() {
91            out.push_str(&format!("// {line}\n"));
92        }
93    }
94    out.push('\n');
95    out
96}
97
98fn build_gradle_file(package_name: &str, info: &IrInfo) -> FileInfo {
99    let content = format!(
100        r#"plugins {{
101    kotlin("jvm") version "1.9.22"
102}}
103
104group = "{package_name}"
105version = "{version}"
106
107repositories {{
108    mavenCentral()
109}}
110
111dependencies {{
112    implementation("com.squareup.okhttp3:okhttp:4.12.0")
113    implementation("com.google.code.gson:gson:2.11.0")
114}}
115
116kotlin {{
117    jvmToolchain(17)
118}}
119
120sourceSets {{
121    main {{
122        kotlin {{
123            srcDirs("apis", "models", "runtime")
124        }}
125    }}
126}}
127"#,
128        version = info.version,
129    );
130    FileInfo::project("build.gradle.kts".to_string(), content)
131}
132
133fn readme_file(info: &IrInfo) -> FileInfo {
134    let title = &info.title;
135    let version = &info.version;
136    let description = info
137        .description
138        .clone()
139        .unwrap_or_else(|| "Generated Kotlin SDK.".to_string());
140    let package = info.title.to_kebab_case();
141    let content = format!(
142        "# {title}\n\n{description}\n\nVersion: `{version}`\n\nGenerated by [openapi-nexus](https://github.com/rust-codegen-group/openapi-nexus) for `{package}`.\n"
143    );
144    FileInfo::readme("README.md".to_string(), content)
145}
openapi_nexus/generators/kotlin/okhttp/codegen.rs

openapi_nexus/generators/kotlin/okhttp/
codegen.rs
