clientext.rs
136 lines
| 4.0 KiB
| application/rls-services+xml
|
RustLexer
|
r40013 | // 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. | ||||
|
r45174 | use bytes::{BufMut, Bytes, BytesMut}; | ||
|
r40013 | use std::ffi::OsStr; | ||
|
r45171 | use std::io; | ||
|
r45174 | use std::mem; | ||
|
r40014 | use std::os::unix::ffi::OsStrExt; | ||
|
r40013 | use std::os::unix::io::AsRawFd; | ||
|
r40014 | use std::path::Path; | ||
|
r45171 | use tokio_hglib::protocol::{OneShotQuery, OneShotRequest}; | ||
|
r40013 | use tokio_hglib::{Client, Connection}; | ||
|
r45180 | use crate::attachio::AttachIo; | ||
use crate::message::{self, Instruction}; | ||||
use crate::runcommand::ChgRunCommand; | ||||
use crate::uihandler::SystemHandler; | ||||
|
r40013 | |||
pub trait ChgClientExt<C> | ||||
|
r44270 | where | ||
C: Connection + AsRawFd, | ||||
|
r40013 | { | ||
/// 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> | ||||
|
r44270 | where | ||
I: AsRawFd, | ||||
O: AsRawFd, | ||||
E: AsRawFd; | ||||
|
r40013 | |||
|
r40014 | /// Changes the working directory of the server. | ||
fn set_current_dir<P>(self, dir: P) -> OneShotRequest<C> | ||||
|
r44270 | where | ||
P: AsRef<Path>; | ||||
|
r40014 | |||
|
r45165 | /// Updates the environment variables of the server. | ||
fn set_env_vars_os<I, P>(self, vars: I) -> OneShotRequest<C> | ||||
where | ||||
I: IntoIterator<Item = (P, P)>, | ||||
P: AsRef<OsStr>; | ||||
|
r45175 | /// Changes the process title of the server. | ||
fn set_process_name<P>(self, name: P) -> OneShotRequest<C> | ||||
where | ||||
P: AsRef<OsStr>; | ||||
|
r45174 | /// Changes the umask of the server process. | ||
fn set_umask(self, mask: u32) -> OneShotRequest<C>; | ||||
|
r40013 | /// Runs the specified Mercurial command with cHg extension. | ||
fn run_command_chg<I, P, H>(self, handler: H, args: I) -> ChgRunCommand<C, H> | ||||
|
r44270 | where | ||
I: IntoIterator<Item = P>, | ||||
P: AsRef<OsStr>, | ||||
H: SystemHandler; | ||||
|
r45171 | |||
/// Validates if the server can run Mercurial commands with the expected | ||||
/// configuration. | ||||
/// | ||||
/// The `args` should contain early command arguments such as `--config` | ||||
/// and `-R`. | ||||
/// | ||||
/// Client-side environment must be sent prior to this request, by | ||||
/// `set_current_dir()` and `set_env_vars_os()`. | ||||
fn validate<I, P>(self, args: I) -> OneShotQuery<C, fn(Bytes) -> io::Result<Vec<Instruction>>> | ||||
where | ||||
I: IntoIterator<Item = P>, | ||||
P: AsRef<OsStr>; | ||||
|
r40013 | } | ||
impl<C> ChgClientExt<C> for Client<C> | ||||
|
r44270 | where | ||
C: Connection + AsRawFd, | ||||
|
r40013 | { | ||
fn attach_io<I, O, E>(self, stdin: I, stdout: O, stderr: E) -> AttachIo<C, I, O, E> | ||||
|
r44270 | where | ||
I: AsRawFd, | ||||
O: AsRawFd, | ||||
E: AsRawFd, | ||||
|
r40013 | { | ||
AttachIo::with_client(self, stdin, stdout, Some(stderr)) | ||||
} | ||||
|
r40014 | fn set_current_dir<P>(self, dir: P) -> OneShotRequest<C> | ||
|
r44270 | where | ||
P: AsRef<Path>, | ||||
|
r40014 | { | ||
OneShotRequest::start_with_args(self, b"chdir", dir.as_ref().as_os_str().as_bytes()) | ||||
} | ||||
|
r45165 | fn set_env_vars_os<I, P>(self, vars: I) -> OneShotRequest<C> | ||
where | ||||
I: IntoIterator<Item = (P, P)>, | ||||
P: AsRef<OsStr>, | ||||
{ | ||||
OneShotRequest::start_with_args(self, b"setenv", message::pack_env_vars_os(vars)) | ||||
} | ||||
|
r45175 | fn set_process_name<P>(self, name: P) -> OneShotRequest<C> | ||
where | ||||
P: AsRef<OsStr>, | ||||
{ | ||||
OneShotRequest::start_with_args(self, b"setprocname", name.as_ref().as_bytes()) | ||||
} | ||||
|
r45174 | fn set_umask(self, mask: u32) -> OneShotRequest<C> { | ||
let mut args = BytesMut::with_capacity(mem::size_of_val(&mask)); | ||||
args.put_u32_be(mask); | ||||
OneShotRequest::start_with_args(self, b"setumask2", args) | ||||
} | ||||
|
r40013 | fn run_command_chg<I, P, H>(self, handler: H, args: I) -> ChgRunCommand<C, H> | ||
|
r44270 | where | ||
I: IntoIterator<Item = P>, | ||||
P: AsRef<OsStr>, | ||||
H: SystemHandler, | ||||
|
r40013 | { | ||
ChgRunCommand::with_client(self, handler, message::pack_args_os(args)) | ||||
} | ||||
|
r45171 | |||
fn validate<I, P>(self, args: I) -> OneShotQuery<C, fn(Bytes) -> io::Result<Vec<Instruction>>> | ||||
where | ||||
I: IntoIterator<Item = P>, | ||||
P: AsRef<OsStr>, | ||||
{ | ||||
OneShotQuery::start_with_args( | ||||
self, | ||||
b"validate", | ||||
message::pack_args_os(args), | ||||
message::parse_instructions, | ||||
) | ||||
} | ||||
|
r40013 | } | ||