rust/rhg/src/ui.rs

48728:3199b575375d

use format_bytes::format_bytes;
use std::borrow::Cow;
use std::env;
use std::io;
use std::io::{ErrorKind, Write};

    ///
    /// The return value can either be
    /// - False if HGPLAIN is not set, or feature is in HGPLAINEXCEPT
    /// - False if feature is disabled by default and not included in HGPLAIN
    /// - True otherwise
    pub fn plain(&self) -> bool {
        // TODO: add support for HGPLAINEXCEPT
        env::var_os("HGPLAIN").is_some()
    }
}

/// A buffered stdout writer for faster batch printing operations.

48729:99b1dfc06571

use format_bytes::format_bytes;
use hg::utils::files::get_bytes_from_os_string;
use std::borrow::Cow;
use std::env;
use std::io;
use std::io::{ErrorKind, Write};

    ///
    /// The return value can either be
    /// - False if HGPLAIN is not set, or feature is in HGPLAINEXCEPT
    /// - False if feature is disabled by default and not included in HGPLAIN
    /// - True otherwise
    pub fn plain(&self, feature: Option<&str>) -> bool {
        plain(feature)
    }
}

fn plain(opt_feature: Option<&str>) -> bool {
    if let Some(except) = env::var_os("HGPLAINEXCEPT") {
        opt_feature.map_or(true, |feature| {
            get_bytes_from_os_string(except)
                .split(|&byte| byte == b',')
                .all(|exception| exception != feature.as_bytes())
        })
    } else {
        env::var_os("HGPLAIN").is_some()
    }
}

/// A buffered stdout writer for faster batch printing operations.