qail 0.28.0

//! Exec module - Execute QAIL AST for seeding/admin tasks
//!
//! Type-safe execution using native QAIL AST - no raw SQL.
//!
//! # Syntax
//!
//! ```text
//! add <table> fields <col1>, <col2> values <val1>, <val2>
//! ```
//!
//! # Multi-line Content
//!
//! Use triple quotes (`'''` or `"""`) for multi-line values:
//!
//! ```text
//! add articles fields title, content values 'My Title', '''
//! <article>
//!   <p>Multi-line HTML content here.</p>
//!   <p>Newlines are preserved inside triple quotes.</p>
//! </article>
//! '''
//! ```
//!
//! # File Format (.qail)
//!
//! - Each line is a separate statement (unless inside triple quotes)
//! - Comments start with `#` or `--`
//! - Blank lines are ignored
//!
//! # Examples
//!
//! ```bash
//! # Inline query
//! qail exec "add users fields name, email values 'Alice', 'a@test.com'" --url postgres://...
//!
//! # From file with SSH tunnel
//! qail exec -f seed.qail --ssh myserver --url postgres://...
//!
//! # Dry-run (preview SQL only)
//! qail exec -f data.qail --dry-run
//!
//! # Wrap in transaction
//! qail exec -f batch.qail --url postgres://... --tx
//! ```

use crate::colors::*;
use anyhow::Result;
use qail_core::prelude::*;
use qail_core::transpiler::ToSql;
use qail_pg::PgDriver;

/// Configuration for exec command
pub struct ExecConfig {
    pub query: Option<String>,
    pub file: Option<String>,
    pub url: Option<String>,
    pub ssh: Option<String>,
    pub tx: bool,
    pub dry_run: bool,
    pub json: bool,
}

/// SSH tunnel wrapper - kills tunnel on drop
struct SshTunnel {
    child: std::process::Child,
    local_port: u16,
}

impl SshTunnel {
    /// Create an SSH tunnel to a remote host
    /// Forwards local_port -> remote_host:remote_port via ssh_host
    async fn new(ssh_host: &str, remote_host: &str, remote_port: u16) -> Result<Self> {
        use std::process::{Command, Stdio};

        // Find available local port
        let local_port = Self::find_available_port()?;

        // Construct SSH tunnel command
        // ssh -N -L local_port:remote_host:remote_port ssh_host
        let child = Command::new("ssh")
            .args([
                "-N", // No remote command
                "-L",
                &format!("{}:{}:{}", local_port, remote_host, remote_port),
                ssh_host,
            ])
            .stdin(Stdio::null())
            .stdout(Stdio::null())
            .stderr(Stdio::piped())
            .spawn()
            .map_err(|e| anyhow::anyhow!("Failed to spawn SSH tunnel: {}", e))?;

        // Wait a moment for tunnel to establish
        tokio::time::sleep(tokio::time::Duration::from_millis(500)).await;

        Ok(Self { child, local_port })
    }

    fn find_available_port() -> Result<u16> {
        let listener = std::net::TcpListener::bind("127.0.0.1:0")?;
        let port = listener.local_addr()?.port();
        drop(listener);
        Ok(port)
    }

    fn local_port(&self) -> u16 {
        self.local_port
    }
}

impl Drop for SshTunnel {
    fn drop(&mut self) {
        // Kill the SSH tunnel process
        let _ = self.child.kill();
    }
}

fn split_qail_statements(content: &str) -> Vec<String> {
    let mut statements = Vec::new();
    let mut current = String::new();
    let mut in_triple_single = false;
    let mut in_triple_double = false;
    let mut chars = content.chars().peekable();

    while let Some(c) = chars.next() {
        // Check for triple quotes
        if c == '\'' && !in_triple_double {
            if chars.peek() == Some(&'\'') {
                chars.next();
                if chars.peek() == Some(&'\'') {
                    chars.next();
                    current.push_str("'''");
                    in_triple_single = !in_triple_single;
                    continue;
                } else {
                    current.push_str("''");
                    continue;
                }
            }
        } else if c == '"' && !in_triple_single && chars.peek() == Some(&'"') {
            chars.next();
            if chars.peek() == Some(&'"') {
                chars.next();
                current.push_str("\"\"\"");
                in_triple_double = !in_triple_double;
                continue;
            } else {
                current.push_str("\"\"");
                continue;
            }
        }

        // Handle newlines - statement boundary if not in multi-line string
        if c == '\n' && !in_triple_single && !in_triple_double {
            let trimmed = current.trim();
            if !trimmed.is_empty() && !trimmed.starts_with('#') && !trimmed.starts_with("--") {
                statements.push(current.trim().to_string());
            }
            current.clear();
            continue;
        }

        current.push(c);
    }

    // Don't forget the last statement
    let trimmed = current.trim();
    if !trimmed.is_empty() && !trimmed.starts_with('#') && !trimmed.starts_with("--") {
        statements.push(current.trim().to_string());
    }

    statements
}

/// Run the exec command (type-safe QAIL AST only)
pub async fn run_exec(config: ExecConfig) -> Result<()> {
    // Get content from file or inline
    let content = if let Some(file) = &config.file {
        std::fs::read_to_string(file)
            .map_err(|e| anyhow::anyhow!("Failed to read file '{}': {}", file, e))?
    } else if let Some(query) = &config.query {
        query.clone()
    } else {
        anyhow::bail!("Either QAIL query or --file must be provided");
    };

    // Split into statements, handling multi-line strings
    let statements_str = split_qail_statements(&content);

    if statements_str.is_empty() {
        println!("{}", "No QAIL statements to execute.".yellow());
        return Ok(());
    }

    // Parse all QAIL statements into ASTs
    let mut statements: Vec<Qail> = Vec::new();
    for (i, stmt) in statements_str.iter().enumerate() {
        let ast = qail_core::parse(stmt)
            .map_err(|e| anyhow::anyhow!("Parse error at statement {}: {}", i + 1, e))?;
        statements.push(ast);
    }

    if !config.json {
        println!(
            "{} Parsed {} QAIL statement(s)",
            "📋".cyan(),
            statements.len().to_string().green()
        );
    }

    // Dry-run mode: show generated SQL
    if config.dry_run {
        println!("\n{}", "🔍 DRY-RUN MODE - Generated SQL:".yellow().bold());
        for (i, ast) in statements.iter().enumerate() {
            let sql = ast.to_sql();
            println!("\n{}{}:", "Statement ".dimmed(), (i + 1).to_string().cyan());
            println!("  {}", sql.white());
        }
        println!("\n{}", "No changes made.".yellow());
        return Ok(());
    }

    // Get database URL (priority: --url > DATABASE_URL > qail.toml)
    let db_url = crate::resolve::resolve_db_url(config.url.as_deref())?;

    // Set up SSH tunnel if requested
    let _tunnel: Option<SshTunnel>;
    let connect_url = if let Some(ssh_host) = &config.ssh {
        println!(
            "{} Opening SSH tunnel to {}...",
            "🔐".cyan(),
            ssh_host.green()
        );

        // Parse the URL to extract host and port
        let (_scheme, remote_host, remote_port, _path) = crate::util::parse_url_parts(&db_url)?;

        // Create tunnel
        let tunnel = SshTunnel::new(ssh_host, &remote_host, remote_port).await?;
        let local_port = tunnel.local_port();

        // Rewrite URL to use tunnel
        let tunneled_url = crate::util::rewrite_url_host(&db_url, "127.0.0.1", local_port)?;

        println!(
            "{} Tunnel established: localhost:{} -> {}:{}",
            "✓".green(),
            local_port,
            remote_host,
            remote_port
        );

        _tunnel = Some(tunnel);
        tunneled_url
    } else {
        _tunnel = None;
        db_url
    };

    // Connect to database
    if !config.json {
        println!("{} Connecting to database...", "🔌".cyan());
    }
    let mut driver = PgDriver::connect_url(&connect_url)
        .await
        .map_err(|e| anyhow::anyhow!("Connection failed: {}", e))?;

    // Execute statements using type-safe AST
    let mut success_count = 0;
    let mut error_count = 0;

    if config.tx {
        println!("{} Starting transaction...", "🔒".cyan());
        driver
            .begin()
            .await
            .map_err(|e| anyhow::anyhow!("BEGIN failed: {}", e))?;
    }

    for (i, ast) in statements.iter().enumerate() {
        let stmt_num = i + 1;
        if !config.json {
            print!("  {} Executing statement {}... ", "→".dimmed(), stmt_num);
        }

        if matches!(ast.action, Action::Get) {
            // SELECT query — use query_ast to get rows back
            match driver.query_ast(ast).await {
                Ok(result) => {
                    if config.json {
                        // JSON output mode — clean, pipe-friendly
                        let mut json_rows: Vec<String> = Vec::new();
                        for row in &result.rows {
                            let fields: Vec<String> = result
                                .columns
                                .iter()
                                .enumerate()
                                .map(|(j, col)| {
                                    let val = row
                                        .get(j)
                                        .and_then(|v| v.as_ref())
                                        .map(|s| {
                                            format!(
                                                "\"{}\"",
                                                s.replace('\\', "\\\\").replace('"', "\\\"")
                                            )
                                        })
                                        .unwrap_or_else(|| "null".to_string());
                                    format!("\"{}\":{}", col, val)
                                })
                                .collect();
                            json_rows.push(format!("{{{}}}", fields.join(",")));
                        }
                        println!("[{}]", json_rows.join(","));
                        success_count += 1;
                    } else {
                        println!("{}", "✓".green());
                        success_count += 1;

                        if result.columns.is_empty() {
                            println!("  {}", "(no columns)".dimmed());
                        } else {
                            // Calculate column widths
                            let mut widths: Vec<usize> =
                                result.columns.iter().map(|c| c.len()).collect();
                            for row in &result.rows {
                                for (j, col) in row.iter().enumerate() {
                                    if j < widths.len() {
                                        let len = col.as_ref().map(|s| s.len()).unwrap_or(1); // "∅"
                                        if len > widths[j] {
                                            widths[j] = len;
                                        }
                                    }
                                }
                            }
                            // Cap column widths at 40 chars for readability
                            for w in widths.iter_mut() {
                                if *w > 40 {
                                    *w = 40;
                                }
                            }

                            // Print header
                            println!();
                            let header: Vec<String> = result
                                .columns
                                .iter()
                                .enumerate()
                                .map(|(j, c)| format!("{:<width$}", c, width = widths[j]))
                                .collect();
                            println!("  {}", header.join(" │ ").cyan().bold());

                            // Print separator
                            let sep: Vec<String> = widths.iter().map(|w| "─".repeat(*w)).collect();
                            println!("  {}", sep.join("─┼─").dimmed());

                            // Print rows
                            for row in &result.rows {
                                let cells: Vec<String> = row
                                    .iter()
                                    .enumerate()
                                    .map(|(j, col)| {
                                        let val = col
                                            .as_ref()
                                            .map(|s| {
                                                if s.len() > 40 {
                                                    format!("{}…", &s[..39])
                                                } else {
                                                    s.clone()
                                                }
                                            })
                                            .unwrap_or_else(|| "∅".to_string());
                                        let w = if j < widths.len() {
                                            widths[j]
                                        } else {
                                            val.len()
                                        };
                                        format!("{:<width$}", val, width = w)
                                    })
                                    .collect();
                                println!("  {}", cells.join(" │ "));
                            }

                            // Row count
                            println!(
                                "\n  {} {} row(s)",
                                "→".dimmed(),
                                result.rows.len().to_string().green()
                            );
                        }
                    }
                }
                Err(e) => {
                    println!("{} {}", "✗".red(), e.to_string().red());
                    error_count += 1;

                    if config.tx {
                        println!("{} Rolling back transaction...", "⚠️".yellow());
                        let _ = driver.rollback().await;
                        anyhow::bail!("Execution failed at statement {}: {}", stmt_num, e);
                    }
                }
            }
        } else {
            // Mutation query — use execute
            match driver.execute(ast).await {
                Ok(_) => {
                    println!("{}", "✓".green());
                    success_count += 1;
                }
                Err(e) => {
                    println!("{} {}", "✗".red(), e.to_string().red());
                    error_count += 1;

                    if config.tx {
                        println!("{} Rolling back transaction...", "⚠️".yellow());
                        let _ = driver.rollback().await;
                        anyhow::bail!("Execution failed at statement {}: {}", stmt_num, e);
                    }
                }
            }
        }
    }

    if config.tx {
        println!("{} Committing transaction...", "🔓".cyan());
        driver
            .commit()
            .await
            .map_err(|e| anyhow::anyhow!("COMMIT failed: {}", e))?;
    }

    if !config.json {
        println!();
        if error_count == 0 {
            println!(
                "{} All {} statement(s) executed successfully!",
                "✅".green(),
                success_count.to_string().green()
            );
        } else {
            println!(
                "{} {} succeeded, {} failed",
                "⚠️".yellow(),
                success_count.to_string().green(),
                error_count.to_string().red()
            );
        }
    }

    Ok(())
}