136 lines
2.5 KiB
Rust
136 lines
2.5 KiB
Rust
use crate::common::*;
|
|
|
|
use ansi_term::{ANSIGenericString, Color::*, Prefix, Style, Suffix};
|
|
use atty::Stream;
|
|
|
|
#[derive(Copy, Clone, Debug, PartialEq)]
|
|
pub(crate) struct Color {
|
|
use_color: UseColor,
|
|
atty: bool,
|
|
style: Style,
|
|
}
|
|
|
|
impl Color {
|
|
fn restyle(self, style: Style) -> Self {
|
|
Self { style, ..self }
|
|
}
|
|
|
|
fn redirect(self, stream: Stream) -> Self {
|
|
Self {
|
|
atty: atty::is(stream),
|
|
..self
|
|
}
|
|
}
|
|
|
|
fn effective_style(&self) -> Style {
|
|
if self.active() {
|
|
self.style
|
|
} else {
|
|
Style::new()
|
|
}
|
|
}
|
|
|
|
pub(crate) fn fmt(fmt: &Formatter) -> Self {
|
|
if fmt.alternate() {
|
|
Self::always()
|
|
} else {
|
|
Self::never()
|
|
}
|
|
}
|
|
|
|
pub(crate) fn auto() -> Self {
|
|
Self {
|
|
use_color: UseColor::Auto,
|
|
..default()
|
|
}
|
|
}
|
|
|
|
pub(crate) fn always() -> Self {
|
|
Self {
|
|
use_color: UseColor::Always,
|
|
..default()
|
|
}
|
|
}
|
|
|
|
pub(crate) fn never() -> Self {
|
|
Self {
|
|
use_color: UseColor::Never,
|
|
..default()
|
|
}
|
|
}
|
|
|
|
pub(crate) fn stderr(self) -> Self {
|
|
self.redirect(Stream::Stderr)
|
|
}
|
|
|
|
pub(crate) fn stdout(self) -> Self {
|
|
self.redirect(Stream::Stdout)
|
|
}
|
|
|
|
pub(crate) fn doc(self) -> Self {
|
|
self.restyle(Style::new().fg(Blue))
|
|
}
|
|
|
|
pub(crate) fn error(self) -> Self {
|
|
self.restyle(Style::new().fg(Red).bold())
|
|
}
|
|
|
|
pub(crate) fn warning(self) -> Self {
|
|
self.restyle(Style::new().fg(Yellow).bold())
|
|
}
|
|
|
|
pub(crate) fn banner(self) -> Self {
|
|
self.restyle(Style::new().fg(Cyan).bold())
|
|
}
|
|
|
|
pub(crate) fn command(self) -> Self {
|
|
self.restyle(Style::new().bold())
|
|
}
|
|
|
|
pub(crate) fn parameter(self) -> Self {
|
|
self.restyle(Style::new().fg(Cyan))
|
|
}
|
|
|
|
pub(crate) fn message(self) -> Self {
|
|
self.restyle(Style::new().bold())
|
|
}
|
|
|
|
pub(crate) fn annotation(self) -> Self {
|
|
self.restyle(Style::new().fg(Purple))
|
|
}
|
|
|
|
pub(crate) fn string(self) -> Self {
|
|
self.restyle(Style::new().fg(Green))
|
|
}
|
|
|
|
pub(crate) fn active(&self) -> bool {
|
|
match self.use_color {
|
|
UseColor::Always => true,
|
|
UseColor::Never => false,
|
|
UseColor::Auto => self.atty,
|
|
}
|
|
}
|
|
|
|
pub(crate) fn paint<'a>(&self, text: &'a str) -> ANSIGenericString<'a, str> {
|
|
self.effective_style().paint(text)
|
|
}
|
|
|
|
pub(crate) fn prefix(&self) -> Prefix {
|
|
self.effective_style().prefix()
|
|
}
|
|
|
|
pub(crate) fn suffix(&self) -> Suffix {
|
|
self.effective_style().suffix()
|
|
}
|
|
}
|
|
|
|
impl Default for Color {
|
|
fn default() -> Self {
|
|
Self {
|
|
use_color: UseColor::Auto,
|
|
atty: false,
|
|
style: Style::new(),
|
|
}
|
|
}
|
|
}
|