#[macro_use]
extern crate clap;
use clap::{App, Arg};
extern crate tokio;
use tokio::prelude::*;
extern crate tokio_uds;
use tokio_uds::UnixStream;
extern crate serde;
#[macro_use]
extern crate serde_derive;
extern crate daemon_engine;
use daemon_engine::{Connection, DaemonError, JsonCodec};
mod common;
use common::{Request, Response};
fn main() {
let matches = App::new("rustd-client")
.author("Ryan Kurte <diot@kurte.nz>")
.about("rust-daemon example client")
.version(crate_version!())
.arg(
Arg::with_name("Socket Address")
.short("s")
.long("socket-address")
.help("Sets unix socket address")
.takes_value(true)
.default_value("/tmp/rustd.sock"),
).arg(
Arg::with_name("Key")
.short("k")
.long("key")
.help("key to set / get")
.takes_value(true),
).arg(
Arg::with_name("Value")
.short("v")
.long("value")
.help("value to set")
.takes_value(true),
).get_matches();
let addr = matches.value_of("Socket Address").unwrap().to_owned();
let key = match matches.value_of("Key") {
Some(k) => k.to_string(),
None => panic!("--key,-k argument required"),
};
let client = UnixConnection::<JsonCodec<Request, Response>>::new(&addr);
let (tx, rx) = client.split();
match matches.value_of("Value") {
Some(value) => {
println!("Set key: '{}'", key);
tx.send(Request::Set(key, value.to_string()))
}
None => {
println!("Get key: '{}'", key);
tx.send(Request::Get(key))
}
}.wait()
.unwrap();
rx.map(|resp| -> Result<(), DaemonError> {
println!("Response: {:?}", resp);
Ok(())
}).wait()
.next();
}