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
106
|
use db;
use util;
use clap;
pub struct Command {
query: String,
tsv: bool,
}
pub fn subcommand<'a, 'b>() -> clap::App<'a, 'b> {
clap::SubCommand::with_name("sql")
.about("Run a query against the local database")
.arg(
clap::Arg::with_name("query")
.required(true)
.help("query to run")
)
.arg(
clap::Arg::with_name("tsv")
.long("tsv")
.help("format output as tsv")
)
}
impl Command {
pub fn new<'a>(matches: &clap::ArgMatches<'a>) -> Command {
Command {
query: matches.value_of("query").unwrap().to_string(),
tsv: matches.is_present("tsv"),
}
}
}
impl super::Command for Command {
fn run(&self) -> failure::Fallible<()> {
let db = db::DB::new(&util::db_path()?)?;
let mut rows = vec![];
let cols = db.query(&self.query, |row| {
let display_row: Vec<String> = (0..row.column_count())
.map(|i| row.get_raw(i))
.map(|v| format_value(&v))
.collect();
rows.push(display_row);
})?;
if self.tsv {
print_tsv(&rows);
}
else {
print_table(&cols, &rows);
}
Ok(())
}
}
fn print_table(cols: &[String], rows: &[Vec<String>]) {
let widths = column_widths(&cols, &rows);
print_row(&widths, &cols);
let border: Vec<String> = widths.iter().map(|l| "-".repeat(*l)).collect();
println!("{}", &border.join("-+-"));
for row in rows {
print_row(&widths, &row);
}
}
fn print_tsv(rows: &[Vec<String>]) {
for row in rows {
println!("{}", row.join("\t"));
}
}
fn format_value(v: &rusqlite::types::ValueRef) -> String {
match v {
rusqlite::types::ValueRef::Null => "null".to_string(),
rusqlite::types::ValueRef::Integer(i) => format!("{}", i),
rusqlite::types::ValueRef::Real(f) => format!("{}", f),
rusqlite::types::ValueRef::Text(s) => format!("{}", s),
rusqlite::types::ValueRef::Blob(b) => format!("{:?}", b),
}
}
fn column_widths(cols: &[String], rows: &[Vec<String>]) -> Vec<usize> {
let mut max_widths: Vec<usize> = cols.iter().map(|s| s.len()).collect();
for row in rows {
for (i, col) in row.iter().enumerate() {
if col.len() > max_widths[i] {
max_widths[i] = col.len();
}
}
}
max_widths
}
fn print_row(widths: &[usize], row: &[String]) {
let fixed_width_row: Vec<String> = row
.iter()
.zip(widths.iter())
.map(|(s, width)| format!("{:width$}", s, width=width))
.collect();
println!("{}", &fixed_width_row.join(" | "));
}
|