use crate::backends::kotlin_android::naming;
use crate::core::backend::GeneratedFile;
use crate::core::config::ResolvedCrateConfig;
use crate::e2e::config::E2eConfig;
use crate::e2e::escape::sanitize_filename;
use crate::e2e::fixture::{Fixture, FixtureGroup};
use anyhow::Result;
use heck::ToUpperCamelCase;
use std::collections::HashSet;
use std::path::PathBuf;
use super::enum_fixtures::is_enum_typed;
use super::gradle::{
render_build_gradle_kotlin_android, render_gradle_properties, render_settings_gradle_kotlin_android,
};
use super::gradle_wrapper::{
GRADLE_WRAPPER_UNIX, GRADLE_WRAPPER_WINDOWS, get_gradle_wrapper_jar_base64, render_gradle_wrapper_properties,
};
use crate::e2e::codegen::kotlin;
pub(super) fn generate(
groups: &[FixtureGroup],
e2e_config: &E2eConfig,
config: &ResolvedCrateConfig,
type_defs: &[crate::core::ir::TypeDef],
) -> Result<Vec<GeneratedFile>> {
let lang = "kotlin_android";
let output_base = PathBuf::from(e2e_config.effective_output()).join(lang);
let mut files = Vec::new();
let call = &e2e_config.call;
let overrides = call.overrides.get(lang);
let _module_path = overrides
.and_then(|o| o.module.as_ref())
.cloned()
.unwrap_or_else(|| call.module.clone());
let function_name = overrides
.and_then(|o| o.function.as_ref())
.cloned()
.unwrap_or_else(|| call.function.clone());
let class_name = overrides
.and_then(|o| o.class.as_ref())
.cloned()
.unwrap_or_else(|| config.name.to_upper_camel_case());
let result_is_simple = overrides.is_some_and(|o| o.result_is_simple);
let result_var = &call.result_var;
let kotlin_android_pkg = e2e_config.resolve_package("kotlin_android");
let pkg_name = kotlin_android_pkg
.as_ref()
.and_then(|p| p.name.as_ref())
.cloned()
.unwrap_or_else(|| config.name.clone());
let _kotlin_android_pkg_path = kotlin_android_pkg
.as_ref()
.and_then(|p| p.path.as_deref())
.unwrap_or("../../packages/kotlin-android");
let kotlin_android_version = kotlin_android_pkg
.as_ref()
.and_then(|p| p.version.as_ref())
.cloned()
.or_else(|| config.resolved_version())
.unwrap_or_else(|| "0.1.0".to_string());
let maven_group_id = naming::aar_group_id(config);
let maven_artifact_id = naming::aar_artifact_id(config);
let maven_coordinate = format!("{}:{}:{}", maven_group_id, maven_artifact_id, kotlin_android_version);
let kotlin_pkg_id = kotlin_android_pkg
.as_ref()
.and_then(|p| p.module.clone())
.or_else(|| config.kotlin_android.as_ref().and_then(|c| c.package.clone()))
.unwrap_or_else(|| config.kotlin_package());
let needs_mock_server = groups
.iter()
.flat_map(|g| g.fixtures.iter())
.any(|f| f.needs_mock_server());
let jni_lib_name = config.jni_lib_name();
let jni_crate_path = config.jni_crate_path();
files.push(GeneratedFile {
path: output_base.join("build.gradle.kts"),
content: render_build_gradle_kotlin_android(
&kotlin_pkg_id,
&maven_coordinate,
e2e_config.dep_mode,
needs_mock_server,
&jni_lib_name,
&jni_crate_path,
),
generated_header: false,
});
files.push(GeneratedFile {
path: output_base.join("gradle.properties"),
content: render_gradle_properties(),
generated_header: false,
});
files.push(GeneratedFile {
path: output_base.join("settings.gradle.kts"),
content: render_settings_gradle_kotlin_android(&pkg_name),
generated_header: false,
});
if e2e_config.dep_mode == crate::e2e::config::DependencyMode::Registry {
files.push(GeneratedFile {
path: output_base.join("gradle/wrapper/gradle-wrapper.properties"),
content: render_gradle_wrapper_properties(),
generated_header: false,
});
files.push(GeneratedFile {
path: output_base.join("gradlew"),
content: GRADLE_WRAPPER_UNIX.to_string(),
generated_header: false,
});
files.push(GeneratedFile {
path: output_base.join("gradlew.bat"),
content: GRADLE_WRAPPER_WINDOWS.to_string(),
generated_header: false,
});
files.push(GeneratedFile {
path: output_base.join("gradle/wrapper/gradle-wrapper.jar"),
content: get_gradle_wrapper_jar_base64(),
generated_header: false,
});
}
let mut test_base = output_base.join("src").join("test").join("kotlin");
for segment in kotlin_pkg_id.split('.') {
test_base = test_base.join(segment);
}
let test_base = test_base.join("e2e");
if needs_mock_server {
files.push(GeneratedFile {
path: test_base.join("MockServerListener.kt"),
content: kotlin::render_mock_server_listener_kt(&kotlin_pkg_id),
generated_header: true,
});
files.push(GeneratedFile {
path: output_base
.join("src")
.join("test")
.join("resources")
.join("META-INF")
.join("services")
.join("org.junit.platform.launcher.LauncherSessionListener"),
content: format!("{kotlin_pkg_id}.e2e.MockServerListener\n"),
generated_header: false,
});
}
let options_type = overrides.and_then(|o| o.options_type.clone());
let struct_names: HashSet<&str> = type_defs.iter().map(|td| td.name.as_str()).collect();
let type_enum_fields: std::collections::HashMap<String, HashSet<String>> = type_defs
.iter()
.filter_map(|td| {
let enum_field_names: HashSet<String> = td
.fields
.iter()
.filter(|field| is_enum_typed(&field.ty, &struct_names))
.map(|field| field.name.clone())
.collect();
if enum_field_names.is_empty() {
None
} else {
Some((td.name.clone(), enum_field_names))
}
})
.collect();
for group in groups {
let active: Vec<&Fixture> = group
.fixtures
.iter()
.filter(|f| crate::e2e::codegen::should_include_fixture(f, lang, e2e_config))
.filter(|f| f.visitor.is_none())
.collect();
if active.is_empty() {
continue;
}
let class_file_name = format!("{}Test.kt", sanitize_filename(&group.category).to_upper_camel_case());
let content = kotlin::render_test_file_android(
&group.category,
&active,
&class_name,
&function_name,
&kotlin_pkg_id,
result_var,
&e2e_config.call.args,
options_type.as_deref(),
result_is_simple,
e2e_config,
&type_enum_fields,
config,
type_defs,
);
files.push(GeneratedFile {
path: test_base.join(&class_file_name),
content,
generated_header: true,
});
}
Ok(files)
}