rust-chg: add Client extensions to run cHg-specific requests
authorYuya Nishihara <yuya@tcha.org>
Mon, 24 Sep 2018 18:57:54 +0900
changeset 39977 74da9d999cd7
parent 39976 44840bcc411a
child 39978 045ea159418d
rust-chg: add Client extensions to run cHg-specific requests This just provides a nicer way to issue command-server requests.
rust/chg/src/clientext.rs
rust/chg/src/lib.rs
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/rust/chg/src/clientext.rs	Mon Sep 24 18:57:54 2018 +0900
@@ -0,0 +1,51 @@
+// Copyright 2018 Yuya Nishihara <yuya@tcha.org>
+//
+// This software may be used and distributed according to the terms of the
+// GNU General Public License version 2 or any later version.
+
+//! cHg extensions to command server client.
+
+use std::ffi::OsStr;
+use std::os::unix::io::AsRawFd;
+use tokio_hglib::{Client, Connection};
+
+use super::attachio::AttachIo;
+use super::message;
+use super::runcommand::ChgRunCommand;
+use super::uihandler::SystemHandler;
+
+pub trait ChgClientExt<C>
+    where C: Connection + AsRawFd,
+{
+    /// Attaches the client file descriptors to the server.
+    fn attach_io<I, O, E>(self, stdin: I, stdout: O, stderr: E) -> AttachIo<C, I, O, E>
+        where I: AsRawFd,
+              O: AsRawFd,
+              E: AsRawFd;
+
+    /// Runs the specified Mercurial command with cHg extension.
+    fn run_command_chg<I, P, H>(self, handler: H, args: I) -> ChgRunCommand<C, H>
+        where I: IntoIterator<Item = P>,
+              P: AsRef<OsStr>,
+              H: SystemHandler;
+}
+
+impl<C> ChgClientExt<C> for Client<C>
+    where C: Connection + AsRawFd,
+{
+    fn attach_io<I, O, E>(self, stdin: I, stdout: O, stderr: E) -> AttachIo<C, I, O, E>
+        where I: AsRawFd,
+              O: AsRawFd,
+              E: AsRawFd,
+    {
+        AttachIo::with_client(self, stdin, stdout, Some(stderr))
+    }
+
+    fn run_command_chg<I, P, H>(self, handler: H, args: I) -> ChgRunCommand<C, H>
+        where I: IntoIterator<Item = P>,
+              P: AsRef<OsStr>,
+              H: SystemHandler,
+    {
+        ChgRunCommand::with_client(self, handler, message::pack_args_os(args))
+    }
+}
--- a/rust/chg/src/lib.rs	Mon Sep 24 18:33:46 2018 +0900
+++ b/rust/chg/src/lib.rs	Mon Sep 24 18:57:54 2018 +0900
@@ -11,11 +11,13 @@
 extern crate tokio_hglib;
 extern crate tokio_process;
 
-pub mod attachio;
+mod attachio;
+mod clientext;
 pub mod locator;
 pub mod message;
 pub mod procutil;
-pub mod runcommand;
+mod runcommand;
 mod uihandler;
 
+pub use clientext::ChgClientExt;
 pub use uihandler::{ChgUiHandler, SystemHandler};