tideway_cli/commands/

generate.rs

//! Generate command - creates frontend components from templates.

use anyhow::{Context, Result};
use colored::Colorize;
use std::fs;
use std::path::Path;

use crate::cli::{GenerateArgs, Module, Style};
use crate::templates::{TemplateContext, TemplateEngine};
use crate::{ensure_dir, is_json_output, print_info, print_success, print_warning, write_file};

/// Run the generate command
pub fn run(args: GenerateArgs) -> Result<()> {
    if !is_json_output() {
        println!(
            "\n{} Generating {} components with {} style\n",
            "tideway".cyan().bold(),
            args.module.to_string().green(),
            args.style.to_string().yellow()
        );
    }

    // Create output directory if it doesn't exist
    let output_path = Path::new(&args.output);
    if !output_path.exists() {
        ensure_dir(output_path)
            .with_context(|| format!("Failed to create output directory: {}", args.output))?;
        print_info(&format!("Created directory: {}", args.output));
    }

    // Create template context
    let context = TemplateContext {
        api_base_url: args.api_base.clone(),
        style: args.style.clone(),
    };

    // Initialize template engine
    let engine = TemplateEngine::new(context)?;

    // Track which shadcn components are needed
    let mut shadcn_components: Vec<&str> = Vec::new();

    // Generate modules based on selection
    match args.module {
        Module::Auth => {
            generate_auth(&engine, output_path, &args, &mut shadcn_components)?;
        }
        Module::Billing => {
            generate_billing(&engine, output_path, &args, &mut shadcn_components)?;
        }
        Module::Organizations => {
            generate_organizations(&engine, output_path, &args, &mut shadcn_components)?;
        }
        Module::Admin => {
            generate_admin(&engine, output_path, &args, &mut shadcn_components)?;
        }
        Module::All => {
            generate_auth(&engine, output_path, &args, &mut shadcn_components)?;
            generate_billing(&engine, output_path, &args, &mut shadcn_components)?;
            generate_organizations(&engine, output_path, &args, &mut shadcn_components)?;
            generate_admin(&engine, output_path, &args, &mut shadcn_components)?;
        }
    }

    // Generate shared files (types, composables) unless --no-shared is set
    if !args.no_shared {
        generate_shared(&engine, output_path, &args)?;
    } else {
        print_info("Skipping shared files (--no-shared)");
    }

    // Generate view files if --with-views is set
    if args.with_views {
        let views_path = Path::new(&args.views_output);
        if !views_path.exists() {
            ensure_dir(views_path).with_context(|| {
                format!("Failed to create views directory: {}", args.views_output)
            })?;
        }

        match args.module {
            Module::Auth => {
                generate_auth_views(&engine, views_path, &args)?;
            }
            Module::Billing => {
                generate_billing_views(&engine, views_path, &args)?;
            }
            Module::Organizations => {
                generate_org_views(&engine, views_path, &args)?;
            }
            Module::Admin => {
                generate_admin_views(&engine, views_path, &args)?;
            }
            Module::All => {
                generate_auth_views(&engine, views_path, &args)?;
                generate_billing_views(&engine, views_path, &args)?;
                generate_org_views(&engine, views_path, &args)?;
                generate_admin_views(&engine, views_path, &args)?;
            }
        }

        // Update router with all routes
        update_router_all()?;

        // Update App.vue with Toaster
        update_app_vue()?;
    }

    // Print shadcn component requirements if using shadcn style
    if args.style == Style::Shadcn && !shadcn_components.is_empty() {
        shadcn_components.sort();
        shadcn_components.dedup();

        // Detect which components are already installed
        let installed = detect_installed_shadcn_components();
        let missing: Vec<&str> = shadcn_components
            .iter()
            .filter(|c| !installed.contains(&c.to_string()))
            .copied()
            .collect();

        if !is_json_output() {
            if missing.is_empty() {
                println!(
                    "\n{} All required shadcn-vue components are installed",
                    "✓".green().bold()
                );
            } else {
                println!("\n{}", "Missing shadcn-vue components:".yellow().bold());
                println!("  npx shadcn-vue@latest add {}", missing.join(" "));
            }
        }
    }

    if !is_json_output() {
        println!(
            "\n{} Components generated in {}\n",
            "✓".green().bold(),
            args.output.cyan()
        );
    }

    Ok(())
}

fn generate_auth(
    engine: &TemplateEngine,
    output_path: &Path,
    args: &GenerateArgs,
    shadcn_components: &mut Vec<&str>,
) -> Result<()> {
    let auth_path = output_path.join("auth");
    ensure_dir(&auth_path)?;

    let composables_path = auth_path.join("composables");
    ensure_dir(&composables_path)?;

    // Generate components
    let components = [
        ("LoginForm.vue", "auth/LoginForm"),
        ("RegisterForm.vue", "auth/RegisterForm"),
        ("ForgotPassword.vue", "auth/ForgotPassword"),
        ("ResetPassword.vue", "auth/ResetPassword"),
        ("MfaVerify.vue", "auth/MfaVerify"),
    ];

    for (filename, template_name) in components {
        let content = engine.render(template_name)?;
        let file_path = auth_path.join(filename);
        write_file_with_force(&file_path, &content, args.force)?;
        print_success(&format!("Generated auth/{}", filename));
    }

    // Generate composable
    let composable_content = engine.render("auth/composables/useAuth")?;
    let composable_path = composables_path.join("useAuth.ts");
    write_file_with_force(&composable_path, &composable_content, args.force)?;
    print_success("Generated auth/composables/useAuth.ts");

    // Track shadcn components needed for auth
    if args.style == Style::Shadcn {
        shadcn_components.extend(&[
            "button",
            "card",
            "input",
            "label",
            "form",
            "alert",
            "separator",
        ]);
    }

    Ok(())
}

fn generate_billing(
    engine: &TemplateEngine,
    output_path: &Path,
    args: &GenerateArgs,
    shadcn_components: &mut Vec<&str>,
) -> Result<()> {
    let billing_path = output_path.join("billing");
    ensure_dir(&billing_path)?;

    let composables_path = billing_path.join("composables");
    ensure_dir(&composables_path)?;

    // Generate components
    let components = [
        ("SubscriptionStatus.vue", "billing/SubscriptionStatus"),
        ("CheckoutButton.vue", "billing/CheckoutButton"),
        ("BillingPortalButton.vue", "billing/BillingPortalButton"),
        ("InvoiceHistory.vue", "billing/InvoiceHistory"),
        ("PlanSelector.vue", "billing/PlanSelector"),
        ("PlanList.vue", "billing/PlanList"),
        ("PlanForm.vue", "billing/PlanForm"),
        ("PricingTable.vue", "billing/PricingTable"),
    ];

    for (filename, template_name) in components {
        let content = engine.render(template_name)?;
        let file_path = billing_path.join(filename);
        write_file_with_force(&file_path, &content, args.force)?;
        print_success(&format!("Generated billing/{}", filename));
    }

    // Generate composables
    let composable_content = engine.render("billing/composables/useBilling")?;
    let composable_path = composables_path.join("useBilling.ts");
    write_file_with_force(&composable_path, &composable_content, args.force)?;
    print_success("Generated billing/composables/useBilling.ts");

    let plans_composable_content = engine.render("billing/composables/usePlans")?;
    let plans_composable_path = composables_path.join("usePlans.ts");
    write_file_with_force(
        &plans_composable_path,
        &plans_composable_content,
        args.force,
    )?;
    print_success("Generated billing/composables/usePlans.ts");

    // Track shadcn components needed for billing
    if args.style == Style::Shadcn {
        shadcn_components.extend(&[
            "button",
            "card",
            "badge",
            "table",
            "skeleton",
            "alert",
            "separator",
            "dropdown-menu",
            "select",
            "switch",
            "textarea",
            "tabs",
            "input",
            "label",
        ]);
    }

    Ok(())
}

fn generate_organizations(
    engine: &TemplateEngine,
    output_path: &Path,
    args: &GenerateArgs,
    shadcn_components: &mut Vec<&str>,
) -> Result<()> {
    let orgs_path = output_path.join("organizations");
    ensure_dir(&orgs_path)?;

    let composables_path = orgs_path.join("composables");
    ensure_dir(&composables_path)?;

    // Generate components
    let components = [
        ("OrgSwitcher.vue", "organizations/OrgSwitcher"),
        ("OrgSettings.vue", "organizations/OrgSettings"),
        ("MemberList.vue", "organizations/MemberList"),
        ("InviteMember.vue", "organizations/InviteMember"),
    ];

    for (filename, template_name) in components {
        let content = engine.render(template_name)?;
        let file_path = orgs_path.join(filename);
        write_file_with_force(&file_path, &content, args.force)?;
        print_success(&format!("Generated organizations/{}", filename));
    }

    // Generate composable
    let composable_content = engine.render("organizations/composables/useOrganization")?;
    let composable_path = composables_path.join("useOrganization.ts");
    write_file_with_force(&composable_path, &composable_content, args.force)?;
    print_success("Generated organizations/composables/useOrganization.ts");

    // Track shadcn components needed for organizations
    if args.style == Style::Shadcn {
        shadcn_components.extend(&[
            "button",
            "card",
            "input",
            "label",
            "form",
            "avatar",
            "dropdown-menu",
            "table",
            "badge",
            "dialog",
            "alert",
        ]);
    }

    Ok(())
}

fn generate_admin(
    engine: &TemplateEngine,
    output_path: &Path,
    args: &GenerateArgs,
    shadcn_components: &mut Vec<&str>,
) -> Result<()> {
    let admin_path = output_path.join("admin");
    ensure_dir(&admin_path)?;

    let composables_path = admin_path.join("composables");
    ensure_dir(&composables_path)?;

    // Generate components
    let components = [
        ("AdminDashboard.vue", "admin/AdminDashboard"),
        ("UserList.vue", "admin/UserList"),
        ("UserDetail.vue", "admin/UserDetail"),
        ("OrganizationList.vue", "admin/OrganizationList"),
        ("OrganizationDetail.vue", "admin/OrganizationDetail"),
        ("ImpersonationBanner.vue", "admin/ImpersonationBanner"),
    ];

    for (filename, template_name) in components {
        let content = engine.render(template_name)?;
        let file_path = admin_path.join(filename);
        write_file_with_force(&file_path, &content, args.force)?;
        print_success(&format!("Generated admin/{}", filename));
    }

    // Generate composable
    let composable_content = engine.render("admin/composables/useAdmin")?;
    let composable_path = composables_path.join("useAdmin.ts");
    write_file_with_force(&composable_path, &composable_content, args.force)?;
    print_success("Generated admin/composables/useAdmin.ts");

    // Track shadcn components needed for admin
    if args.style == Style::Shadcn {
        shadcn_components.extend(&[
            "button",
            "card",
            "input",
            "label",
            "table",
            "badge",
            "skeleton",
            "alert",
            "separator",
            "switch",
        ]);
    }

    Ok(())
}

fn generate_admin_views(
    engine: &TemplateEngine,
    views_path: &Path,
    args: &GenerateArgs,
) -> Result<()> {
    let admin_views_path = views_path.join("admin");
    ensure_dir(&admin_views_path)?;

    // Generate view files
    let views = [
        ("AdminLayout.vue", "views/admin/AdminLayout"),
        ("AdminDashboardView.vue", "views/admin/AdminDashboardView"),
        ("AdminUsersView.vue", "views/admin/AdminUsersView"),
        (
            "AdminOrganizationsView.vue",
            "views/admin/AdminOrganizationsView",
        ),
    ];

    for (filename, template_name) in views {
        let content = engine.render(template_name)?;
        let file_path = admin_views_path.join(filename);
        write_file_with_force(&file_path, &content, args.force)?;
        print_success(&format!("Generated views/admin/{}", filename));
    }

    Ok(())
}

fn generate_auth_views(
    _engine: &TemplateEngine,
    views_path: &Path,
    args: &GenerateArgs,
) -> Result<()> {
    let auth_views_path = views_path.join("auth");
    ensure_dir(&auth_views_path)?;

    // Generate auth view files inline (simple wrappers around components)
    let views = [
        (
            "LoginView.vue",
            include_str!("../templates_inline/views/auth/LoginView.vue"),
        ),
        (
            "RegisterView.vue",
            include_str!("../templates_inline/views/auth/RegisterView.vue"),
        ),
        (
            "ForgotPasswordView.vue",
            include_str!("../templates_inline/views/auth/ForgotPasswordView.vue"),
        ),
        (
            "ResetPasswordView.vue",
            include_str!("../templates_inline/views/auth/ResetPasswordView.vue"),
        ),
        (
            "MfaView.vue",
            include_str!("../templates_inline/views/auth/MfaView.vue"),
        ),
    ];

    for (filename, content) in views {
        let file_path = auth_views_path.join(filename);
        write_file_with_force(&file_path, content, args.force)?;
        print_success(&format!("Generated views/auth/{}", filename));
    }

    Ok(())
}

fn generate_billing_views(
    _engine: &TemplateEngine,
    views_path: &Path,
    args: &GenerateArgs,
) -> Result<()> {
    let billing_views_path = views_path.join("billing");
    ensure_dir(&billing_views_path)?;

    let views = [(
        "BillingView.vue",
        include_str!("../templates_inline/views/billing/BillingView.vue"),
    )];

    for (filename, content) in views {
        let file_path = billing_views_path.join(filename);
        write_file_with_force(&file_path, content, args.force)?;
        print_success(&format!("Generated views/billing/{}", filename));
    }

    Ok(())
}

fn generate_org_views(
    _engine: &TemplateEngine,
    views_path: &Path,
    args: &GenerateArgs,
) -> Result<()> {
    let org_views_path = views_path.join("settings");
    ensure_dir(&org_views_path)?;

    let views = [
        (
            "OrganizationSettingsView.vue",
            include_str!("../templates_inline/views/settings/OrganizationSettingsView.vue"),
        ),
        (
            "MembersView.vue",
            include_str!("../templates_inline/views/settings/MembersView.vue"),
        ),
    ];

    for (filename, content) in views {
        let file_path = org_views_path.join(filename);
        write_file_with_force(&file_path, content, args.force)?;
        print_success(&format!("Generated views/settings/{}", filename));
    }

    Ok(())
}

fn update_router_all() -> Result<()> {
    let router_path = Path::new("./src/router/index.ts");

    if !router_path.exists() {
        print_warning("Router file not found at src/router/index.ts - skipping router update");
        return Ok(());
    }

    let content = fs::read_to_string(router_path)?;

    // Check if routes already added
    if content.contains("LoginView") {
        print_info("Routes already configured in router");
        return Ok(());
    }

    // All routes to add
    let all_routes = r#"
  // Auth routes (public)
  { path: '/login', name: 'login', component: () => import('@/views/auth/LoginView.vue') },
  { path: '/register', name: 'register', component: () => import('@/views/auth/RegisterView.vue') },
  { path: '/forgot-password', name: 'forgot-password', component: () => import('@/views/auth/ForgotPasswordView.vue') },
  { path: '/reset-password', name: 'reset-password', component: () => import('@/views/auth/ResetPasswordView.vue') },
  { path: '/mfa', name: 'mfa', component: () => import('@/views/auth/MfaView.vue') },

  // Billing (protected)
  { path: '/billing', name: 'billing', component: () => import('@/views/billing/BillingView.vue'), meta: { requiresAuth: true } },

  // Organization settings (protected)
  { path: '/settings/organization', name: 'organization-settings', component: () => import('@/views/settings/OrganizationSettingsView.vue'), meta: { requiresAuth: true } },
  { path: '/settings/members', name: 'members', component: () => import('@/views/settings/MembersView.vue'), meta: { requiresAuth: true } },

  // Admin routes (protected, admin only)
  {
    path: '/admin',
    component: () => import('@/views/admin/AdminLayout.vue'),
    meta: { requiresAuth: true, requiresAdmin: true },
    children: [
      { path: '', name: 'admin-dashboard', component: () => import('@/views/admin/AdminDashboardView.vue') },
      { path: 'users', name: 'admin-users', component: () => import('@/views/admin/AdminUsersView.vue') },
      { path: 'organizations', name: 'admin-organizations', component: () => import('@/views/admin/AdminOrganizationsView.vue') },
    ]
  },"#;

    // Try to insert after the first route in the routes array
    let updated = if content.contains("routes: [") {
        content.replace("routes: [", &format!("routes: [{}", all_routes))
    } else if content.contains("const routes") {
        // Handle const routes = [ ... ] pattern
        content.replace(
            "const routes = [",
            &format!("const routes = [{}", all_routes),
        )
    } else {
        print_warning("Could not find routes array in router file");
        if !is_json_output() {
            println!("\n{}", "Add to your router manually:".yellow().bold());
            println!("{}", all_routes);
        }
        return Ok(());
    };

    write_file(router_path, &updated)?;
    print_success("Updated router with all routes (auth, billing, settings, admin)");

    Ok(())
}

fn update_app_vue() -> Result<()> {
    let app_path = Path::new("./src/App.vue");

    if !app_path.exists() {
        print_warning("App.vue not found - skipping App.vue update");
        return Ok(());
    }

    let content = fs::read_to_string(app_path)?;

    // Check if Toaster already added
    if content.contains("Toaster") {
        print_info("Toaster already in App.vue");
        return Ok(());
    }

    let mut updated = content;

    // Add Toaster import (shadcn-vue exports Sonner component as Toaster)
    if updated.contains("<script setup") {
        updated = updated.replace(
            "<script setup lang=\"ts\">",
            "<script setup lang=\"ts\">\nimport 'vue-sonner/style.css'\nimport { Toaster } from '@/components/ui/sonner'"
        );
    } else if updated.contains("<script setup>") {
        updated = updated.replace(
            "<script setup>",
            "<script setup>\nimport 'vue-sonner/style.css'\nimport { Toaster } from '@/components/ui/sonner'"
        );
    }

    // Add Toaster component before </template>
    if updated.contains("</template>") {
        updated = updated.replace("</template>", "  <Toaster />\n</template>");
    }

    write_file(app_path, &updated)?;
    print_success("Updated App.vue with Sonner");

    Ok(())
}

fn generate_shared(engine: &TemplateEngine, output_path: &Path, args: &GenerateArgs) -> Result<()> {
    // Generate shared types
    let types_path = output_path.join("types");
    ensure_dir(&types_path)?;

    let types_content = engine.render("shared/types/index")?;
    let types_file = types_path.join("index.ts");
    write_file_with_force(&types_file, &types_content, args.force)?;
    print_success("Generated types/index.ts");

    // Generate shared API composable
    let composables_path = output_path.join("composables");
    ensure_dir(&composables_path)?;

    let api_content = engine.render("shared/composables/useApi")?;
    let api_file = composables_path.join("useApi.ts");
    write_file_with_force(&api_file, &api_content, args.force)?;
    print_success("Generated composables/useApi.ts");

    Ok(())
}

fn write_file_with_force(path: &Path, content: &str, force: bool) -> Result<()> {
    if path.exists() && !force {
        print_warning(&format!(
            "Skipping {} (use --force to overwrite)",
            path.display()
        ));
        return Ok(());
    }
    write_file(path, content).with_context(|| format!("Failed to write {}", path.display()))?;
    Ok(())
}

/// Detect installed shadcn-vue components by checking ./src/components/ui/ subdirectories
fn detect_installed_shadcn_components() -> Vec<String> {
    let ui_path = Path::new("./src/components/ui");

    if !ui_path.exists() {
        return Vec::new();
    }

    let Ok(entries) = fs::read_dir(ui_path) else {
        return Vec::new();
    };

    entries
        .filter_map(|entry| {
            let entry = entry.ok()?;
            if entry.file_type().ok()?.is_dir() {
                entry.file_name().to_str().map(String::from)
            } else {
                None
            }
        })
        .collect()
}