Show More
@@ -3,5 +3,49 b'' | |||||
3 | // This software may be used and distributed according to the terms of the |
|
3 | // This software may be used and distributed according to the terms of the | |
4 | // GNU General Public License version 2 or any later version. |
|
4 | // GNU General Public License version 2 or any later version. | |
5 |
|
5 | |||
|
6 | extern crate chg; | |||
|
7 | extern crate futures; | |||
|
8 | extern crate tokio; | |||
|
9 | extern crate tokio_hglib; | |||
|
10 | ||||
|
11 | use chg::{ChgClientExt, ChgUiHandler}; | |||
|
12 | use chg::locator; | |||
|
13 | use futures::sync::oneshot; | |||
|
14 | use std::env; | |||
|
15 | use std::io; | |||
|
16 | use std::process; | |||
|
17 | use tokio::prelude::*; | |||
|
18 | use tokio_hglib::UnixClient; | |||
|
19 | ||||
6 | fn main() { |
|
20 | fn main() { | |
|
21 | let code = run().unwrap_or_else(|err| { | |||
|
22 | eprintln!("chg: abort: {}", err); | |||
|
23 | 255 | |||
|
24 | }); | |||
|
25 | process::exit(code); | |||
7 | } |
|
26 | } | |
|
27 | ||||
|
28 | fn run() -> io::Result<i32> { | |||
|
29 | let current_dir = env::current_dir()?; | |||
|
30 | let sock_path = locator::prepare_server_socket_path()?; | |||
|
31 | let handler = ChgUiHandler::new(); | |||
|
32 | let (result_tx, result_rx) = oneshot::channel(); | |||
|
33 | let fut = UnixClient::connect(sock_path) | |||
|
34 | .and_then(|client| { | |||
|
35 | client.set_current_dir(current_dir) | |||
|
36 | }) | |||
|
37 | .and_then(|client| { | |||
|
38 | client.attach_io(io::stdin(), io::stdout(), io::stderr()) | |||
|
39 | }) | |||
|
40 | .and_then(|client| { | |||
|
41 | client.run_command_chg(handler, env::args_os().skip(1)) | |||
|
42 | }) | |||
|
43 | .map(|(_client, _handler, code)| { | |||
|
44 | Ok(code) | |||
|
45 | }) | |||
|
46 | .or_else(|err| Ok(Err(err))) // pass back error to caller | |||
|
47 | .map(|res| result_tx.send(res).unwrap()); | |||
|
48 | tokio::run(fut); | |||
|
49 | result_rx.wait().unwrap_or(Err(io::Error::new(io::ErrorKind::Other, | |||
|
50 | "no exit code set"))) | |||
|
51 | } |
General Comments 0
You need to be logged in to leave comments.
Login now