Skip to main content

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