romance_core/addon/

oauth.rs

use crate::addon::Addon;
use anyhow::Result;
use std::path::Path;

pub struct OauthAddon {
    pub provider: String,
}

impl Addon for OauthAddon {
    fn name(&self) -> &str {
        "oauth"
    }

    fn check_prerequisites(&self, project_root: &Path) -> Result<()> {
        super::check_romance_project(project_root)?;
        super::check_auth_exists(project_root)
    }

    fn is_already_installed(&self, project_root: &Path) -> bool {
        project_root.join("backend/src/oauth.rs").exists()
    }

    fn install(&self, project_root: &Path) -> Result<()> {
        install_oauth(project_root, &self.provider)
    }

    fn uninstall(&self, project_root: &Path) -> Result<()> {
        use colored::Colorize;

        println!("{}", "Uninstalling OAuth...".bold());

        // Delete files
        if super::remove_file_if_exists(&project_root.join("backend/src/oauth.rs"))? {
            println!("  {} backend/src/oauth.rs", "delete".red());
        }
        if super::remove_file_if_exists(&project_root.join("backend/src/handlers/oauth.rs"))? {
            println!("  {} backend/src/handlers/oauth.rs", "delete".red());
        }
        if super::remove_file_if_exists(&project_root.join("backend/src/routes/oauth.rs"))? {
            println!("  {} backend/src/routes/oauth.rs", "delete".red());
        }
        if super::remove_file_if_exists(
            &project_root.join("frontend/src/features/auth/OAuthButton.tsx"),
        )? {
            println!(
                "  {} frontend/src/features/auth/OAuthButton.tsx",
                "delete".red()
            );
        }

        // Remove mod declaration from main.rs
        super::remove_mod_from_main(project_root, "oauth")?;

        // Remove from handlers/mod.rs
        super::remove_line_from_file(
            &project_root.join("backend/src/handlers/mod.rs"),
            "pub mod oauth;",
        )?;

        // Remove from routes/mod.rs
        super::remove_line_from_file(
            &project_root.join("backend/src/routes/mod.rs"),
            "pub mod oauth;",
        )?;
        super::remove_line_from_file(
            &project_root.join("backend/src/routes/mod.rs"),
            ".merge(oauth::router())",
        )?;

        // Regenerate AI context
        crate::ai_context::regenerate(project_root).ok();

        println!();
        println!("{}", "OAuth uninstalled successfully.".green().bold());

        Ok(())
    }

    fn dependencies(&self) -> Vec<&str> {
        vec!["auth"]
    }
}

fn install_oauth(project_root: &Path, provider: &str) -> Result<()> {
    use crate::template::TemplateEngine;
    use crate::utils;
    use colored::Colorize;
    use heck::ToPascalCase;
    use tera::Context;

    let valid_providers = ["google", "github", "discord"];
    if !valid_providers.contains(&provider) {
        anyhow::bail!(
            "Unsupported OAuth provider '{}'. Supported: {}",
            provider,
            valid_providers.join(", ")
        );
    }

    println!(
        "{}",
        format!("Installing OAuth ({})...", provider).bold()
    );

    let engine = TemplateEngine::new()?;
    let timestamp = crate::generator::migration::next_timestamp();

    let mut ctx = Context::new();
    ctx.insert("provider", provider);
    ctx.insert("provider_pascal", &provider.to_pascal_case());
    ctx.insert("timestamp", &timestamp);

    // Generate OAuth module
    let content = engine.render("addon/oauth/oauth.rs.tera", &ctx)?;
    utils::write_file(&project_root.join("backend/src/oauth.rs"), &content)?;
    println!("  {} backend/src/oauth.rs", "create".green());

    // Generate OAuth handlers
    let content = engine.render("addon/oauth/oauth_handlers.rs.tera", &ctx)?;
    utils::write_file(
        &project_root.join("backend/src/handlers/oauth.rs"),
        &content,
    )?;
    println!("  {} backend/src/handlers/oauth.rs", "create".green());

    // Generate OAuth routes
    let content = engine.render("addon/oauth/oauth_routes.rs.tera", &ctx)?;
    utils::write_file(
        &project_root.join("backend/src/routes/oauth.rs"),
        &content,
    )?;
    println!("  {} backend/src/routes/oauth.rs", "create".green());

    // Generate migration to add oauth columns to users table
    let content = engine.render("addon/oauth/oauth_migration.rs.tera", &ctx)?;
    let migration_module = format!("m{}_add_oauth_to_users", timestamp);
    utils::write_file(
        &project_root.join(format!("backend/migration/src/{}.rs", migration_module)),
        &content,
    )?;
    println!(
        "  {} backend/migration/src/{}.rs",
        "create".green(),
        migration_module
    );

    // Generate frontend OAuth button
    let content = engine.render("addon/oauth/OAuthButton.tsx.tera", &ctx)?;
    utils::write_file(
        &project_root.join("frontend/src/features/auth/OAuthButton.tsx"),
        &content,
    )?;
    println!(
        "  {} frontend/src/features/auth/OAuthButton.tsx",
        "create".green()
    );

    // Register modules
    let mods_marker = "// === ROMANCE:MODS ===";
    utils::insert_at_marker(
        &project_root.join("backend/src/handlers/mod.rs"),
        mods_marker,
        "pub mod oauth;",
    )?;
    utils::insert_at_marker(
        &project_root.join("backend/src/routes/mod.rs"),
        mods_marker,
        "pub mod oauth;",
    )?;
    utils::insert_at_marker(
        &project_root.join("backend/src/routes/mod.rs"),
        "// === ROMANCE:ROUTES ===",
        "        .merge(oauth::router())",
    )?;

    // Register migration
    let lib_path = project_root.join("backend/migration/src/lib.rs");
    utils::insert_at_marker(
        &lib_path,
        "// === ROMANCE:MIGRATION_MODS ===",
        &format!("mod {};", migration_module),
    )?;
    utils::insert_at_marker(
        &lib_path,
        "// === ROMANCE:MIGRATIONS ===",
        &format!("            Box::new({}::Migration),", migration_module),
    )?;

    // Inject oauth fields into user entity model (both Model and UserPublic)
    let user_model_path = project_root.join("backend/src/entities/user.rs");
    if user_model_path.exists() {
        let mut user_content = std::fs::read_to_string(&user_model_path)?;
        if !user_content.contains("oauth_provider") {
            // Add oauth fields to Model struct: insert before "pub created_at" in "pub struct Model"
            if let Some(model_pos) = user_content.find("pub struct Model") {
                if let Some(rel_pos) = user_content[model_pos..].find("    pub created_at:") {
                    let insert_pos = model_pos + rel_pos;
                    user_content.insert_str(insert_pos, "    pub oauth_provider: Option<String>,\n    pub oauth_id: Option<String>,\n");
                }
            }
            // Add oauth fields to UserPublic struct: insert before "pub created_at" in "pub struct UserPublic"
            if let Some(up_pos) = user_content.find("pub struct UserPublic") {
                if let Some(rel_pos) = user_content[up_pos..].find("    pub created_at:") {
                    let insert_pos = up_pos + rel_pos;
                    user_content.insert_str(insert_pos, "    pub oauth_provider: Option<String>,\n    pub oauth_id: Option<String>,\n");
                }
            }
            std::fs::write(&user_model_path, user_content)?;
            println!("  {} backend/src/entities/user.rs (added oauth fields)", "update".green());
        }
    }

    // Patch auth handlers to include oauth fields in UserPublic and ActiveModel
    let auth_handlers_path = project_root.join("backend/src/handlers/auth.rs");
    if auth_handlers_path.exists() {
        let mut auth_content = std::fs::read_to_string(&auth_handlers_path)?;
        // Add ..Default::default() to ActiveModel in register handler so new optional fields are handled
        if !auth_content.contains("..Default::default()") {
            // Find the ActiveModel block and add ..Default::default() before its closing brace
            // Pattern: "created_at: Set(now),\n    };" in the register function's ActiveModel
            if let Some(pos) = auth_content.find("created_at: Set(now),\n        updated_at: Set(now),\n    };") {
                let insert_pos = pos + "created_at: Set(now),\n        updated_at: Set(now),\n".len();
                auth_content.insert_str(insert_pos, "        ..Default::default()\n");
            } else if let Some(pos) = auth_content.find("updated_at: Set(now),\n    };") {
                let insert_pos = pos + "updated_at: Set(now),\n".len();
                auth_content.insert_str(insert_pos, "        ..Default::default()\n");
            }
        }
        // Add oauth fields to all UserPublic constructions
        if !auth_content.contains("oauth_provider:") {
            // Replace "created_at: *.created_at," patterns in UserPublic structs
            // to include oauth fields before created_at
            auth_content = auth_content.replace(
                "        created_at: created.created_at,",
                "        oauth_provider: created.oauth_provider,\n        oauth_id: created.oauth_id,\n        created_at: created.created_at,",
            );
            auth_content = auth_content.replace(
                "        created_at: user.created_at,",
                "        oauth_provider: user.oauth_provider.clone(),\n        oauth_id: user.oauth_id.clone(),\n        created_at: user.created_at,",
            );
            auth_content = auth_content.replace(
                "        created_at: updated.created_at,",
                "        oauth_provider: updated.oauth_provider,\n        oauth_id: updated.oauth_id,\n        created_at: updated.created_at,",
            );
            auth_content = auth_content.replace(
                "            created_at: u.created_at,",
                "            oauth_provider: u.oauth_provider,\n            oauth_id: u.oauth_id,\n            created_at: u.created_at,",
            );
        }
        std::fs::write(&auth_handlers_path, auth_content)?;
    }

    // Add mod oauth to main.rs
    super::add_mod_to_main(project_root, "oauth")?;

    // Add dependencies
    crate::generator::auth::insert_cargo_dependency(
        &project_root.join("backend/Cargo.toml"),
        &[
            ("oauth2", r#"{ version = "4", features = ["reqwest"] }"#),
            ("reqwest", r#"{ version = "0.12", features = ["json"] }"#),
        ],
    )?;

    // Add env vars
    let provider_upper = provider.to_uppercase();
    super::append_env_var(
        &project_root.join("backend/.env"),
        &format!("{}_CLIENT_ID=your-client-id", provider_upper),
    )?;
    super::append_env_var(
        &project_root.join("backend/.env"),
        &format!("{}_CLIENT_SECRET=your-client-secret", provider_upper),
    )?;
    super::append_env_var(
        &project_root.join("backend/.env.example"),
        &format!("{}_CLIENT_ID=your-client-id", provider_upper),
    )?;
    super::append_env_var(
        &project_root.join("backend/.env.example"),
        &format!("{}_CLIENT_SECRET=your-client-secret", provider_upper),
    )?;

    println!();
    println!(
        "{}",
        format!("OAuth ({}) installed successfully!", provider)
            .green()
            .bold()
    );
    println!(
        "  Set {}_CLIENT_ID and {}_CLIENT_SECRET in backend/.env",
        provider_upper, provider_upper
    );

    Ok(())
}