comparison rust/chg/src/clientext.rs @ 44683:065048e66f32

rust-chg: send client side umask to server

This is equivalent to forwardumask() of hgclient.c.

Differential Revision: https://phab.mercurial-scm.org/D8382

44682:9ce613d648de

// 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 bytes::Bytes;
use std::ffi::OsStr;
use std::io;
use std::os::unix::ffi::OsStrExt;
use std::os::unix::io::AsRawFd;
use std::path::Path;
use tokio_hglib::protocol::{OneShotQuery, OneShotRequest};
use tokio_hglib::{Client, Connection};

    /// 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>;

    /// 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>,
    {
        OneShotRequest::start_with_args(self, b"setenv", message::pack_env_vars_os(vars))
    }

    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,

44683:065048e66f32

// 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 bytes::{BufMut, Bytes, BytesMut};
use std::ffi::OsStr;
use std::io;
use std::mem;
use std::os::unix::ffi::OsStrExt;
use std::os::unix::io::AsRawFd;
use std::path::Path;
use tokio_hglib::protocol::{OneShotQuery, OneShotRequest};
use tokio_hglib::{Client, Connection};

    /// 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>;

    /// Changes the umask of the server process.
    fn set_umask(self, mask: u32) -> OneShotRequest<C>;

    /// 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>,
    {
        OneShotRequest::start_with_args(self, b"setenv", message::pack_env_vars_os(vars))
    }

    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)
    }

    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,