use crate::codegen::traits::file_writer::FileInfo;
use crate::generators::rust::aioduct::config::{AioductFeatureConfig, AioductTls};
use crate::generators::rust::common::project_files::with_header;
const CLIENT_RS_TEMPLATE: &str = include_str!("runtime/client.rs.txt");
const ERROR_RS: &str = include_str!("runtime/error.rs.txt");
const AUTH_RS: &str = include_str!("runtime/auth.rs.txt");
const MOD_RS: &str = include_str!("runtime/mod.rs.txt");
pub fn runtime_files(header: &str, aioduct_cfg: &AioductFeatureConfig) -> Vec<FileInfo> {
let client_rs = render_client_rs(aioduct_cfg);
vec![
FileInfo::runtime("client.rs".to_string(), with_header(header, &client_rs)),
FileInfo::runtime("error.rs".to_string(), with_header(header, ERROR_RS)),
FileInfo::runtime("auth.rs".to_string(), with_header(header, AUTH_RS)),
FileInfo::runtime("mod.rs".to_string(), with_header(header, MOD_RS)),
]
}
fn render_client_rs(cfg: &AioductFeatureConfig) -> String {
let has_http3 = cfg
.features
.as_ref()
.is_some_and(|f| f.contains(&"http3".to_string()));
let tls = cfg.tls.as_ref().unwrap_or(&AioductTls::RustlsRing);
let constructor = match (tls, has_http3) {
(AioductTls::RustlsRing | AioductTls::RustlsAwsLcRs, true) => {
"aioduct::Client::<R>::with_http3()"
}
(AioductTls::RustlsRing | AioductTls::RustlsAwsLcRs, false) => {
"aioduct::Client::<R>::with_rustls()"
}
(AioductTls::Disabled, _) => "aioduct::Client::<R>::new()",
};
CLIENT_RS_TEMPLATE.replace("{{CLIENT_CONSTRUCTOR}}", constructor)
}