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
|
use cmd;
const _DUMMY_DEPENDENCY: &'static str = include_str!("../Cargo.toml");
enum Command {
Sync {
username: String,
},
SQL {
query: String,
tsv: bool,
},
}
pub fn run() -> failure::Fallible<()> {
let command = get_options()?;
match command {
Command::Sync { username } => cmd::sync::run(&username),
Command::SQL { query, tsv } => cmd::sql::run(&query, tsv),
}
}
fn get_options() -> failure::Fallible<Command> {
let matches = app_from_crate!()
.subcommand(
clap::SubCommand::with_name("sync")
.about("Updates the local copy of track data from last.fm")
.arg(
clap::Arg::with_name("username")
.required(true)
.help("last.fm username to fetch tracks for")
)
)
.subcommand(
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")
)
)
.get_matches();
let command = match matches.subcommand() {
("sync", Some(matches)) => {
Command::Sync {
username: matches.value_of("username").unwrap().to_string(),
}
},
("sql", Some(matches)) => {
Command::SQL {
query: matches.value_of("query").unwrap().to_string(),
tsv: matches.is_present("tsv"),
}
},
(name, Some(_)) => bail!("unknown subcommand: {}", name),
(_, None) => bail!("no subcommand given"),
};
Ok(command)
}
|