Method format_table

gql-cli/src/cli/output.rs:24–84

Format results as a table using comfy-table

(result: &QueryResult)

    /// Format results as a table using comfy-table
    fn format_table(result: &QueryResult) -> String {
        // Check if this is a session command
        if result.is_session_command() {
            if let Some(msg) = result.get_session_message() {
                return format!("{}\n", msg.to_string().green());
            }
        }

        if result.rows.is_empty() {
            return format!("{}\n", "No results found".yellow());
        }

        let mut output = String::new();

        // Header
        output.push_str(&format!("{}\n", "Query Results".bold().green()));
        output.push_str(&format!(
            "Execution time: {} ms\n",
            result.execution_time_ms
        ));
        output.push_str(&format!("Rows returned: {}\n\n", result.rows.len()));

        // Create table
        let mut table = Table::new();
        table.load_preset(UTF8_FULL);

        // Table header
        let header_cells: Vec<Cell> = result
            .variables
            .iter()
            .map(|col| Cell::new(col).fg(Color::Green))
            .collect();
        table.set_header(header_cells);

        // Table rows
        for row in &result.rows {
            let row_values: Vec<String> = result
                .variables
                .iter()
                .map(|col| {
                    row.get_value(col)
                        .map(Self::value_to_string)
                        .unwrap_or_else(|| "NULL".to_string())
                })
                .collect();
            table.add_row(row_values);
        }

        output.push_str(&table.to_string());
        output.push('\n');

        // Display warnings if any
        if !result.warnings.is_empty() {
            output.push_str(&format!("\n{}\n", "Warnings:".bold().yellow()));
            for (i, warning) in result.warnings.iter().enumerate() {
                output.push_str(&format!("  {}. {}\n", i + 1, warning.yellow()));
            }
        }