use crate::controller::ControllerType;
mod tests;

pub fn determine_controller(
    text: &str,
) -> ControllerType {
    let text = text.trim();

    if let Some(prompt) = text.strip_prefix(&format!("create")) {
        return ControllerType::ImageGeneration(prompt.trim().to_owned());
    }

    if let Some(prompt) = text.strip_prefix(&format!("edit")) {
        return ControllerType::ImageEdit(prompt.trim().to_owned());
    }

    ControllerType::UsageHelp
}