2018-07-23 22:49:25 +02:00
|
|
|
use std::env;
|
2018-09-06 19:11:38 +05:30
|
|
|
use std::ffi::OsString;
|
2018-05-21 14:59:42 +02:00
|
|
|
use std::io::{self, Write};
|
2018-09-06 19:11:38 +05:30
|
|
|
use std::path::PathBuf;
|
2018-05-21 14:59:42 +02:00
|
|
|
use std::process::{Child, Command, Stdio};
|
|
|
|
|
2018-10-17 20:10:29 +02:00
|
|
|
use shell_words;
|
|
|
|
|
2019-10-06 08:44:14 +07:00
|
|
|
use crate::errors::*;
|
2019-12-23 09:54:18 +01:00
|
|
|
use crate::less::retrieve_less_version;
|
2019-10-09 11:38:09 +07:00
|
|
|
use crate::PagingMode;
|
2018-08-22 22:29:12 +02:00
|
|
|
|
2019-10-15 08:25:53 +07:00
|
|
|
#[derive(Debug)]
|
2018-05-21 14:59:42 +02:00
|
|
|
pub enum OutputType {
|
|
|
|
Pager(Child),
|
|
|
|
Stdout(io::Stdout),
|
|
|
|
}
|
|
|
|
|
|
|
|
impl OutputType {
|
2018-10-21 19:52:29 +09:00
|
|
|
pub fn from_mode(mode: PagingMode, pager: Option<&str>) -> Result<Self> {
|
2018-05-21 14:59:42 +02:00
|
|
|
use self::PagingMode::*;
|
2018-10-17 20:21:58 +02:00
|
|
|
Ok(match mode {
|
2018-10-21 19:52:29 +09:00
|
|
|
Always => OutputType::try_pager(false, pager)?,
|
|
|
|
QuitIfOneScreen => OutputType::try_pager(true, pager)?,
|
2018-05-21 14:59:42 +02:00
|
|
|
_ => OutputType::stdout(),
|
2018-10-17 20:21:58 +02:00
|
|
|
})
|
2018-05-21 14:59:42 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Try to launch the pager. Fall back to stdout in case of errors.
|
2018-10-21 19:52:29 +09:00
|
|
|
fn try_pager(quit_if_one_screen: bool, pager_from_config: Option<&str>) -> Result<Self> {
|
2019-02-07 22:12:41 +01:00
|
|
|
let mut replace_arguments_to_less = false;
|
|
|
|
|
|
|
|
let pager_from_env = match (env::var("BAT_PAGER"), env::var("PAGER")) {
|
|
|
|
(Ok(bat_pager), _) => Some(bat_pager),
|
|
|
|
(_, Ok(pager)) => {
|
|
|
|
// less needs to be called with the '-R' option in order to properly interpret the
|
|
|
|
// ANSI color sequences printed by bat. If someone has set PAGER="less -F", we
|
|
|
|
// therefore need to overwrite the arguments and add '-R'.
|
|
|
|
//
|
|
|
|
// We only do this for PAGER (as it is not specific to 'bat'), not for BAT_PAGER
|
|
|
|
// or bats '--pager' command line option.
|
|
|
|
replace_arguments_to_less = true;
|
|
|
|
Some(pager)
|
|
|
|
}
|
|
|
|
_ => None,
|
|
|
|
};
|
2018-10-31 20:22:56 -03:00
|
|
|
|
2019-02-18 19:53:14 +01:00
|
|
|
let pager_from_config = pager_from_config.map(|p| p.to_string());
|
|
|
|
|
|
|
|
if pager_from_config.is_some() {
|
|
|
|
replace_arguments_to_less = false;
|
|
|
|
}
|
|
|
|
|
2018-11-05 18:41:51 -02:00
|
|
|
let pager = pager_from_config
|
2019-02-07 22:12:41 +01:00
|
|
|
.or(pager_from_env)
|
2019-03-08 10:46:49 +00:00
|
|
|
.unwrap_or_else(|| String::from("less"));
|
2018-07-23 22:49:25 +02:00
|
|
|
|
2019-02-07 22:12:41 +01:00
|
|
|
let pagerflags =
|
|
|
|
shell_words::split(&pager).chain_err(|| "Could not parse pager command.")?;
|
2018-10-12 17:37:57 +02:00
|
|
|
|
2018-10-17 20:31:22 +02:00
|
|
|
match pagerflags.split_first() {
|
2019-02-07 22:12:41 +01:00
|
|
|
Some((pager_name, args)) => {
|
2018-11-06 19:53:32 +01:00
|
|
|
let mut pager_path = PathBuf::from(pager_name);
|
|
|
|
|
|
|
|
if pager_path.file_stem() == Some(&OsString::from("bat")) {
|
|
|
|
pager_path = PathBuf::from("less");
|
2018-11-05 18:54:49 -02:00
|
|
|
}
|
2018-11-05 18:41:51 -02:00
|
|
|
|
2018-10-17 20:31:22 +02:00
|
|
|
let is_less = pager_path.file_stem() == Some(&OsString::from("less"));
|
2018-09-06 19:11:38 +05:30
|
|
|
|
2018-10-17 20:31:22 +02:00
|
|
|
let mut process = if is_less {
|
|
|
|
let mut p = Command::new(&pager_path);
|
2019-02-07 22:12:41 +01:00
|
|
|
if args.is_empty() || replace_arguments_to_less {
|
2019-12-23 09:54:18 +01:00
|
|
|
p.arg("--RAW-CONTROL-CHARS");
|
2018-10-17 20:31:22 +02:00
|
|
|
if quit_if_one_screen {
|
|
|
|
p.arg("--quit-if-one-screen");
|
|
|
|
}
|
2019-12-23 09:54:18 +01:00
|
|
|
|
|
|
|
// Passing '--no-init' fixes a bug with '--quit-if-one-screen' in older
|
|
|
|
// versions of 'less'. Unfortunately, it also breaks mouse-wheel support.
|
|
|
|
//
|
|
|
|
// See: http://www.greenwoodsoftware.com/less/news.530.html
|
|
|
|
match retrieve_less_version() {
|
|
|
|
Some(version) if version < 530 => {
|
|
|
|
p.arg("--no-init");
|
|
|
|
}
|
|
|
|
_ => {}
|
|
|
|
}
|
2019-02-07 22:12:41 +01:00
|
|
|
} else {
|
|
|
|
p.args(args);
|
2018-10-17 20:31:22 +02:00
|
|
|
}
|
|
|
|
p.env("LESSCHARSET", "UTF-8");
|
|
|
|
p
|
|
|
|
} else {
|
2019-02-07 22:12:41 +01:00
|
|
|
let mut p = Command::new(&pager_path);
|
|
|
|
p.args(args);
|
|
|
|
p
|
2018-10-17 20:31:22 +02:00
|
|
|
};
|
2018-07-23 22:49:25 +02:00
|
|
|
|
2018-10-17 20:31:22 +02:00
|
|
|
Ok(process
|
|
|
|
.stdin(Stdio::piped())
|
|
|
|
.spawn()
|
|
|
|
.map(OutputType::Pager)
|
|
|
|
.unwrap_or_else(|_| OutputType::stdout()))
|
|
|
|
}
|
|
|
|
None => Ok(OutputType::stdout()),
|
|
|
|
}
|
2018-05-21 14:59:42 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
fn stdout() -> Self {
|
|
|
|
OutputType::Stdout(io::stdout())
|
|
|
|
}
|
|
|
|
|
2019-08-02 09:14:57 +02:00
|
|
|
pub fn handle(&mut self) -> Result<&mut dyn Write> {
|
2018-05-21 14:59:42 +02:00
|
|
|
Ok(match *self {
|
|
|
|
OutputType::Pager(ref mut command) => command
|
|
|
|
.stdin
|
|
|
|
.as_mut()
|
|
|
|
.chain_err(|| "Could not open stdin for pager")?,
|
|
|
|
OutputType::Stdout(ref mut handle) => handle,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Drop for OutputType {
|
|
|
|
fn drop(&mut self) {
|
|
|
|
if let OutputType::Pager(ref mut command) = *self {
|
|
|
|
let _ = command.wait();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|