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
107
108
109
110
|
use std::path::PathBuf;
use clap_v3 as clap;
use clap::ArgMatches;
use anyhow::anyhow;
use anyhow::Context;
use anyhow::Error;
use anyhow::Result;
use crate::config::Configuration;
use crate::db::DbConnectionConfig;
pub fn interface(db_connection_config: DbConnectionConfig, matches: &ArgMatches, config: &Configuration) -> Result<()> {
match matches.subcommand() {
("cli", Some(matches)) => cli(db_connection_config, matches, config),
(other, _) => return Err(anyhow!("Unknown subcommand: {}", other)),
}
}
fn cli(db_connection_config: DbConnectionConfig, matches: &ArgMatches, config: &Configuration) -> Result<()> {
use std::process::Command;
trait PgCliCommand {
fn run_for_uri(&self, dbcc: DbConnectionConfig) -> Result<()>;
}
struct Psql(PathBuf);
impl PgCliCommand for Psql {
fn run_for_uri(&self, dbcc: DbConnectionConfig) -> Result<()> {
Command::new(&self.0)
.arg(format!("--dbname={}", dbcc.database_name()))
.arg(format!("--host={}", dbcc.database_host()))
.arg(format!("--port={}", dbcc.database_port()))
.arg(format!("--username={}", dbcc.database_user()))
.stdin(std::process::Stdio::inherit())
.stdout(std::process::Stdio::inherit())
.stderr(std::process::Stdio::inherit())
.output()
.map_err(Error::from)
.and_then(|out| {
if out.status.success() {
info!("pgcli exited successfully");
Ok(())
} else {
Err(anyhow!("gpcli did not exit successfully"))
.with_context(|| {
match String::from_utf8(out.stderr) {
Ok(log) => anyhow!("{}", log),
Err(e) => anyhow!("Cannot parse log into valid UTF-8: {}", e),
}
})
.map_err(Error::from)
}
})
}
}
struct PgCli(PathBuf);
impl PgCliCommand for PgCli {
fn run_for_uri(&self, dbcc: DbConnectionConfig) -> Result<()> {
Command::new(&self.0)
.arg("--host")
.arg(dbcc.database_host())
.arg("--port")
.arg(dbcc.database_port())
.arg("--username")
.arg(dbcc.database_user())
.arg(dbcc.database_name())
.stdin(std::process::Stdio::inherit())
.stdout(std::process::Stdio::inherit())
.stderr(std::process::Stdio::inherit())
.output()
.map_err(Error::from)
.and_then(|out| {
if out.status.success() {
info!("pgcli exited successfully");
Ok(())
} else {
Err(anyhow!("gpcli did not exit successfully"))
.with_context(|| {
match String::from_utf8(out.stderr) {
Ok(log) => anyhow!("{}", log),
Err(e) => anyhow!("Cannot parse log into valid UTF-8: {}", e),
}
})
.map_err(Error::from)
}
})
}
}
matches.value_of("tool")
.map(|s| vec![s])
.unwrap_or_else(|| vec!["psql", "pgcli"])
.into_iter()
.filter_map(|s| which::which(&s).ok().map(|path| (path, s)))
.map(|(path, s)| {
match s {
"psql" => Ok(Box::new(Psql(path)) as Box<dyn PgCliCommand>),
"pgcli" => Ok(Box::new(PgCli(path)) as Box<dyn PgCliCommand>),
prog => Err(anyhow!("Unsupported pg CLI program: {}", prog)),
}
})
.next()
.transpose()?
.ok_or_else(|| anyhow!("No Program found"))?
.run_for_uri(db_connection_config)
}
|
