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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
use crate::prelude::*;
mod play;
mod record;
mod server;
mod stream;
mod watch;
mod web;
struct Command {
name: &'static str,
cmd: &'static dyn for<'a, 'b> Fn(clap::App<'a, 'b>) -> clap::App<'a, 'b>,
config: &'static dyn Fn(
Option<config::Config>,
) -> Result<Box<dyn crate::config::Config>>,
log_level: &'static str,
}
const COMMANDS: &[Command] = &[
Command {
name: "stream",
cmd: &stream::cmd,
config: &stream::config,
log_level: "error",
},
Command {
name: "server",
cmd: &server::cmd,
config: &server::config,
log_level: "info",
},
Command {
name: "web",
cmd: &web::cmd,
config: &web::config,
log_level: "info",
},
Command {
name: "watch",
cmd: &watch::cmd,
config: &watch::config,
log_level: "error",
},
Command {
name: "record",
cmd: &record::cmd,
config: &record::config,
log_level: "error",
},
Command {
name: "play",
cmd: &play::cmd,
config: &play::config,
log_level: "error",
},
];
pub fn parse<'a>() -> Result<clap::ArgMatches<'a>> {
let mut app = clap::App::new(program_name()?)
.about("Stream your terminal for other people to watch")
.author(clap::crate_authors!())
.version(clap::crate_version!())
.arg(
clap::Arg::with_name("config-file")
.long("config-file")
.takes_value(true)
.value_name("FILE")
.help("Read configuration from FILE"),
)
.global_setting(clap::AppSettings::DontCollapseArgsInUsage)
.global_setting(clap::AppSettings::GlobalVersion)
.global_setting(clap::AppSettings::UnifiedHelpMessage)
.global_setting(clap::AppSettings::VersionlessSubcommands);
for cmd in COMMANDS {
let subcommand = clap::SubCommand::with_name(cmd.name);
app = app.subcommand(
(cmd.cmd)(subcommand).setting(clap::AppSettings::NextLineHelp),
);
}
app.get_matches_safe().context(crate::error::ParseArgs)
}
pub fn run(matches: &clap::ArgMatches<'_>) -> Result<()> {
let mut chosen_cmd = &COMMANDS[0];
let mut chosen_submatches = &clap::ArgMatches::<'_>::default();
for cmd in COMMANDS {
if let Some(submatches) = matches.subcommand_matches(cmd.name) {
chosen_cmd = cmd;
chosen_submatches = submatches;
}
}
env_logger::from_env(
env_logger::Env::default().default_filter_or(chosen_cmd.log_level),
)
.init();
let config = crate::config::config(
matches.value_of("config-file").map(std::path::Path::new),
)?;
let mut cmd_config = (chosen_cmd.config)(config)?;
cmd_config.merge_args(chosen_submatches)?;
log::debug!("{:?}", cmd_config);
// XXX ideally we'd be able to run everything on the current_thread
// runtime, but this is blocked on
// https://github.com/tokio-rs/tokio/issues/1356 (fixed in the 0.2 branch
// which is not yet stable)
// let mut runtime = tokio::runtime::current_thread::Runtime::new()
// .unwrap();
// runtime
// .block_on(cmd_config.run().map_err(|e| {
// log::error!("{}", e);
// }))
// .unwrap();
tokio::run(cmd_config.run().map_err(|e| {
log::error!("{}", e);
}));
Ok(())
}
fn program_name() -> Result<String> {
let program =
std::env::args().next().context(crate::error::MissingArgv)?;
let path = std::path::Path::new(&program);
let filename = path.file_name();
Ok(filename
.ok_or_else(|| Error::NotAFileName {
path: path.to_string_lossy().to_string(),
})?
.to_string_lossy()
.to_string())
}
|
