sql-cli 1.73.1

SQL query tool for CSV/JSON with both interactive TUI and non-interactive CLI modes - perfect for exploration and automation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

use comfy_table::{Attribute, Cell, ContentArrangement, Table};
use crossterm::style::Stylize;
use serde_json::Value;

pub fn display_results(data: &[Value], fields: &[String]) {
    if data.is_empty() {
        println!("{}", "No results found.".yellow());
        return;
    }

    let mut table = Table::new();
    table.set_content_arrangement(ContentArrangement::Dynamic);

    // Set headers
    let headers: Vec<Cell> = if fields.contains(&"*".to_string()) {
        // Get all fields from first record
        if let Some(first) = data.first() {
            if let Some(obj) = first.as_object() {
                obj.keys()
                    .map(|k| Cell::new(k).add_attribute(Attribute::Bold))
                    .collect()
            } else {
                vec![]
            }
        } else {
            vec![]
        }
    } else {
        fields
            .iter()
            .map(|f| Cell::new(f).add_attribute(Attribute::Bold))
            .collect()
    };

    let field_names: Vec<String> = headers.iter().map(|h| h.content().to_string()).collect();
    table.set_header(headers);

    // Add rows
    for record in data {
        if let Some(obj) = record.as_object() {
            let row: Vec<String> = field_names
                .iter()
                .map(|field| match obj.get(field) {
                    Some(Value::String(s)) => s.clone(),
                    Some(Value::Number(n)) => n.to_string(),
                    Some(Value::Bool(b)) => b.to_string(),
                    Some(Value::Null) => "NULL".to_string(),
                    Some(v) => v.to_string(),
                    None => String::new(),
                })
                .collect();
            table.add_row(row);
        }
    }

    println!("{table}");
    println!("\n{}", format!("{} rows returned", data.len()).green());
}

pub fn export_to_csv(
    data: &[Value],
    fields: &[String],
    filename: &str,
) -> Result<(), Box<dyn std::error::Error>> {
    let mut wtr = csv::Writer::from_path(filename)?;

    // Write headers
    let headers: Vec<String> = if fields.contains(&"*".to_string()) {
        if let Some(first) = data.first() {
            if let Some(obj) = first.as_object() {
                obj.keys().cloned().collect()
            } else {
                return Err("Invalid data format".into());
            }
        } else {
            return Ok(());
        }
    } else {
        fields.to_vec()
    };

    wtr.write_record(&headers)?;

    // Write data
    for record in data {
        if let Some(obj) = record.as_object() {
            let row: Vec<String> = headers
                .iter()
                .map(|field| match obj.get(field) {
                    Some(Value::String(s)) => s.clone(),
                    Some(Value::Number(n)) => n.to_string(),
                    Some(Value::Bool(b)) => b.to_string(),
                    Some(Value::Null) => String::new(),
                    Some(v) => v.to_string(),
                    None => String::new(),
                })
                .collect();
            wtr.write_record(&row)?;
        }
    }

    wtr.flush()?;
    println!("{}", format!("Results exported to {filename}").green());
    Ok(())
}