1 files changed, 102 insertions, 1 deletions
diff --git a/mumctl/src/main.rs b/mumctl/src/main.rs
index e7a11a9..124cc8c 100644
--- a/mumctl/src/main.rs
+++ b/mumctl/src/main.rs
@@ -1,3 +1,104 @@
+use clap::{App, AppSettings, Arg, Shell, SubCommand};
+use ipc_channel::ipc::{self, IpcReceiver, IpcSender};
+use log::*;
+use mumlib::command::{Command, CommandResponse};
+use mumlib::setup_logger;
+use std::{fs, io};
+
 fn main() {
-    println!("Hello, world!");
+    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();
+
+    debug!("Matching clap");
+    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();
+            send_command(Command::ServerConnect {
+                host: host.to_string(),
+                port: 64738u16, //TODO
+                username: username.to_string(),
+                accept_invalid_cert: true, //TODO
+            }).unwrap();
+        } else if let Some(_) = matches.subcommand_matches("disconnect") {
+            send_command(Command::ServerDisconnect).unwrap();
+        }
+    } else if let Some(matches) = matches.subcommand_matches("channel") {
+        if let Some(_matches) = matches.subcommand_matches("list") {
+            let res = send_command(Command::ChannelList).unwrap().unwrap();
+            println!("{:#?}", res);
+            /*if matches.is_present("short") {
+                None //TODO
+            } else {
+                None //TODO
+            };*/
+        } else if let Some(matches) = matches.subcommand_matches("connect") {
+            send_command(Command::ChannelJoin {
+                channel_id: matches.value_of("channel").unwrap().parse::<u32>().unwrap()
+            }).unwrap();
+        }
+    } else if let Some(_matches) = matches.subcommand_matches("status") {
+        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) -> Result<Option<CommandResponse>, ()> {
+    let (tx_client, rx_client): (IpcSender<Result<Option<CommandResponse>, ()>>,
+                                 IpcReceiver<Result<Option<CommandResponse>, ()>>) = ipc::channel().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()
 }