use crate::database::introspection::Schema;
use crate::generators::rust::casing;
use crate::generators::rust::model::Model;
pub fn with_key(model: &Model, schema: &Schema) -> String {
let mut file = format!("use super::{};\n", model.name);
let declared = model
.enums(schema)
.iter()
.map(|item| casing::to_pascal(&item.name))
.find(|name| model.key().is_some_and(|key| key.mapping.names(name)));
if let Some(name) = declared {
file.push_str(&format!("use super::super::enums::{name};\n"));
}
file
}
pub fn struct_only(model: &Model) -> String {
format!("use super::{};\n", model.name)
}