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
|
use clap::{App, AppSettings, Arg, Shell, SubCommand};
use ipc_channel::ipc::{self, IpcSender};
use log::*;
use mumlib::command::{Command, CommandResponse};
use mumlib::setup_logger;
use std::{fs, io};
use colored::Colorize;
macro_rules! err_print {
($func:expr) => {
if let Err(e) = $func {
println!("{} {}", "error:".red(), e);
}
};
}
fn main() {
setup_logger();
debug!("Logger up!");
let mut app = App::new("mumctl")
.setting(AppSettings::ArgRequiredElseHelp)
.subcommand(SubCommand::with_name("server")
.setting(AppSettings::ArgRequiredElseHelp)
.subcommand(SubCommand::with_name("connect")
.setting(AppSettings::ArgRequiredElseHelp)
.arg(Arg::with_name("host")
.required(true)
.index(1))
.arg(Arg::with_name("username")
.required(true)
.index(2)))
.subcommand(SubCommand::with_name("disconnect")))
.subcommand(SubCommand::with_name("channel")
.setting(AppSettings::ArgRequiredElseHelp)
.subcommand(SubCommand::with_name("list")
.arg(Arg::with_name("short")
.short("s")
.long("short")))
.subcommand(SubCommand::with_name("connect")
.arg(Arg::with_name("channel")
.required(true))))
.subcommand(SubCommand::with_name("status"))
.subcommand(SubCommand::with_name("completions")
.arg(Arg::with_name("zsh")
.long("zsh"))
.arg(Arg::with_name("bash")
.long("bash"))
.arg(Arg::with_name("fish")
.long("fish")));
let matches = app.clone().get_matches();
if let Some(matches) = matches.subcommand_matches("server") {
if let Some(matches) = matches.subcommand_matches("connect") {
let host = matches.value_of("host").unwrap();
let username = matches.value_of("username").unwrap();
err_print!(send_command(Command::ServerConnect {
host: host.to_string(),
port: 64738u16, //TODO
username: username.to_string(),
accept_invalid_cert: true, //TODO
}));
} else if let Some(_) = matches.subcommand_matches("disconnect") {
err_print!(send_command(Command::ServerDisconnect));
}
} else if let Some(matches) = matches.subcommand_matches("channel") {
if let Some(_matches) = matches.subcommand_matches("list") {
match send_command(Command::ChannelList) {
Ok(res) => {
println!("{:#?}", res.unwrap());
}
Err(e) => println!("{} {}", "error:".red(), e),
}
} else if let Some(matches) = matches.subcommand_matches("connect") {
err_print!(send_command(Command::ChannelJoin {
channel_id: matches.value_of("channel").unwrap().parse::<u32>().unwrap()
}));
}
} else if let Some(_matches) = matches.subcommand_matches("status") {
match send_command(Command::Status) {
Ok(res) => {
println!("{:#?}", res.unwrap());
}
Err(e) => println!("{} {}", "error:".red(), e),
}
let res = send_command(Command::Status).unwrap().unwrap();
println!("{:#?}", res);
} else if let Some(matches) = matches.subcommand_matches("completions") {
app.gen_completions_to("mumctl",
match matches.value_of("shell").unwrap_or("zsh") {
"bash" => {
Shell::Bash
},
"fish" => {
Shell::Fish
},
_ => {
Shell::Zsh
},
},
&mut io::stdout());
return;
};
}
fn send_command(command: Command) -> mumlib::error::Result<Option<CommandResponse>> {
let (tx_client, rx_client) = ipc::channel::<mumlib::error::Result<Option<CommandResponse>>>().unwrap();
let server_name = fs::read_to_string("/var/tmp/mumd-oneshot").unwrap(); //TODO don't panic
let tx0 = IpcSender::connect(server_name).unwrap();
tx0.send((command, tx_client)).unwrap();
rx_client.recv().unwrap()
}
|
