mirror of
https://github.com/nushell/nushell.git
synced 2025-05-30 06:39:33 +02:00
0f7e1d4d01
* WIP * changed to matches * fixed a bug with osc * changed back to if let because: clippy * fixed example test
245 lines
10 KiB
Rust
245 lines
10 KiB
Rust
use crate::commands::WholeStreamCommand;
|
|
use crate::prelude::*;
|
|
use ansi_term::Color;
|
|
use nu_errors::ShellError;
|
|
use nu_protocol::{ReturnSuccess, Signature, SyntaxShape, UntaggedValue, Value};
|
|
use nu_source::Tagged;
|
|
|
|
pub struct Ansi;
|
|
|
|
#[derive(Deserialize)]
|
|
struct AnsiArgs {
|
|
color: Value,
|
|
escape: Option<Tagged<String>>,
|
|
osc: Option<Tagged<String>>,
|
|
}
|
|
|
|
#[async_trait]
|
|
impl WholeStreamCommand for Ansi {
|
|
fn name(&self) -> &str {
|
|
"ansi"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("ansi")
|
|
.optional(
|
|
"color",
|
|
SyntaxShape::Any,
|
|
"the name of the color to use or 'reset' to reset the color",
|
|
)
|
|
.named(
|
|
"escape", // \x1b
|
|
SyntaxShape::Any,
|
|
"escape sequence without the escape character(s)",
|
|
Some('e'),
|
|
)
|
|
.named(
|
|
"osc",
|
|
SyntaxShape::Any,
|
|
"operating system command (ocs) escape sequence without the escape character(s)",
|
|
Some('o'),
|
|
)
|
|
}
|
|
|
|
fn usage(&self) -> &str {
|
|
r#"Output ANSI codes to change color
|
|
|
|
For escape sequences:
|
|
Escape: '\x1b[' is not required for --escape parameter
|
|
Format: #(;#)m
|
|
Example: 1;31m for bold red or 2;37;41m for dimmed white fg with red bg
|
|
There can be multiple text formatting sequence numbers
|
|
separated by a ; and ending with an m where the # is of the
|
|
following values:
|
|
attributes
|
|
0 reset / normal display
|
|
1 bold or increased intensity
|
|
2 faint or decreased intensity
|
|
3 italic on (non-mono font)
|
|
4 underline on
|
|
5 slow blink on
|
|
6 fast blink on
|
|
7 reverse video on
|
|
8 nondisplayed (invisible) on
|
|
9 strike-through on
|
|
|
|
foreground/bright colors background/bright colors
|
|
30/90 black 40/100 black
|
|
31/91 red 41/101 red
|
|
32/92 green 42/102 green
|
|
33/93 yellow 43/103 yellow
|
|
34/94 blue 44/104 blue
|
|
35/95 magenta 45/105 magenta
|
|
36/96 cyan 46/106 cyan
|
|
37/97 white 47/107 white
|
|
https://en.wikipedia.org/wiki/ANSI_escape_code
|
|
|
|
OSC: '\x1b]' is not required for --osc parameter
|
|
Example: echo [$(ansi -o '0') 'some title' $(char bel)] | str collect
|
|
Format: #
|
|
0 Set window title and icon name
|
|
1 Set icon name
|
|
2 Set window title
|
|
4 Set/read color palette
|
|
9 iTerm2 Grown notifications
|
|
10 Set foreground color (x11 color spec)
|
|
11 Set background color (x11 color spec)
|
|
... others"#
|
|
}
|
|
|
|
fn examples(&self) -> Vec<Example> {
|
|
vec![
|
|
Example {
|
|
description: "Change color to green",
|
|
example: r#"ansi green"#,
|
|
result: Some(vec![Value::from("\u{1b}[32m")]),
|
|
},
|
|
Example {
|
|
description: "Reset the color",
|
|
example: r#"ansi reset"#,
|
|
result: Some(vec![Value::from("\u{1b}[0m")]),
|
|
},
|
|
Example {
|
|
description:
|
|
"Use ansi to color text (rb = red bold, gb = green bold, pb = purple bold)",
|
|
example: r#"echo [$(ansi rb) Hello " " $(ansi gb) Nu " " $(ansi pb) World] | str collect"#,
|
|
result: Some(vec![Value::from(
|
|
"\u{1b}[1;31mHello \u{1b}[1;32mNu \u{1b}[1;35mWorld",
|
|
)]),
|
|
},
|
|
Example {
|
|
description:
|
|
"Use ansi to color text (rb = red bold, gb = green bold, pb = purple bold)",
|
|
example: r#"echo [$(ansi -e '3;93;41m') Hello $(ansi reset) " " $(ansi gb) Nu " " $(ansi pb) World] | str collect"#,
|
|
result: Some(vec![Value::from(
|
|
"\u{1b}[3;93;41mHello\u{1b}[0m \u{1b}[1;32mNu \u{1b}[1;35mWorld",
|
|
)]),
|
|
},
|
|
]
|
|
}
|
|
|
|
async fn run(
|
|
&self,
|
|
args: CommandArgs,
|
|
registry: &CommandRegistry,
|
|
) -> Result<OutputStream, ShellError> {
|
|
let (AnsiArgs { color, escape, osc }, _) = args.process(®istry).await?;
|
|
|
|
if let Some(e) = escape {
|
|
let esc_vec: Vec<char> = e.item.chars().collect();
|
|
if esc_vec[0] == '\\' {
|
|
return Err(ShellError::labeled_error(
|
|
"no need for escape characters",
|
|
"no need for escape characters",
|
|
e.tag(),
|
|
));
|
|
}
|
|
let output = format!("\x1b[{}", e.item);
|
|
return Ok(OutputStream::one(ReturnSuccess::value(
|
|
UntaggedValue::string(output).into_value(e.tag()),
|
|
)));
|
|
}
|
|
|
|
if let Some(o) = osc {
|
|
let osc_vec: Vec<char> = o.item.chars().collect();
|
|
if osc_vec[0] == '\\' {
|
|
return Err(ShellError::labeled_error(
|
|
"no need for escape characters",
|
|
"no need for escape characters",
|
|
o.tag(),
|
|
));
|
|
}
|
|
//Operating system command aka osc ESC ] <- note the right brace, not left brace for osc
|
|
// OCS's need to end with a bell '\x07' char
|
|
let output = format!("\x1b]{};", o.item);
|
|
return Ok(OutputStream::one(ReturnSuccess::value(
|
|
UntaggedValue::string(output).into_value(o.tag()),
|
|
)));
|
|
}
|
|
|
|
let color_string = color.as_string()?;
|
|
let ansi_code = str_to_ansi_color(color_string);
|
|
|
|
if let Some(output) = ansi_code {
|
|
Ok(OutputStream::one(ReturnSuccess::value(
|
|
UntaggedValue::string(output).into_value(color.tag()),
|
|
)))
|
|
} else {
|
|
Err(ShellError::labeled_error(
|
|
"Unknown color",
|
|
"unknown color",
|
|
color.tag(),
|
|
))
|
|
}
|
|
// }
|
|
}
|
|
}
|
|
|
|
pub fn str_to_ansi_color(s: String) -> Option<String> {
|
|
match s.as_str() {
|
|
"g" | "green" => Some(Color::Green.prefix().to_string()),
|
|
"gb" | "green_bold" => Some(Color::Green.bold().prefix().to_string()),
|
|
"gu" | "green_underline" => Some(Color::Green.underline().prefix().to_string()),
|
|
"gi" | "green_italic" => Some(Color::Green.italic().prefix().to_string()),
|
|
"gd" | "green_dimmed" => Some(Color::Green.dimmed().prefix().to_string()),
|
|
"gr" | "green_reverse" => Some(Color::Green.reverse().prefix().to_string()),
|
|
"r" | "red" => Some(Color::Red.prefix().to_string()),
|
|
"rb" | "red_bold" => Some(Color::Red.bold().prefix().to_string()),
|
|
"ru" | "red_underline" => Some(Color::Red.underline().prefix().to_string()),
|
|
"ri" | "red_italic" => Some(Color::Red.italic().prefix().to_string()),
|
|
"rd" | "red_dimmed" => Some(Color::Red.dimmed().prefix().to_string()),
|
|
"rr" | "red_reverse" => Some(Color::Red.reverse().prefix().to_string()),
|
|
"u" | "blue" => Some(Color::Blue.prefix().to_string()),
|
|
"ub" | "blue_bold" => Some(Color::Blue.bold().prefix().to_string()),
|
|
"uu" | "blue_underline" => Some(Color::Blue.underline().prefix().to_string()),
|
|
"ui" | "blue_italic" => Some(Color::Blue.italic().prefix().to_string()),
|
|
"ud" | "blue_dimmed" => Some(Color::Blue.dimmed().prefix().to_string()),
|
|
"ur" | "blue_reverse" => Some(Color::Blue.reverse().prefix().to_string()),
|
|
"b" | "black" => Some(Color::Black.prefix().to_string()),
|
|
"bb" | "black_bold" => Some(Color::Black.bold().prefix().to_string()),
|
|
"bu" | "black_underline" => Some(Color::Black.underline().prefix().to_string()),
|
|
"bi" | "black_italic" => Some(Color::Black.italic().prefix().to_string()),
|
|
"bd" | "black_dimmed" => Some(Color::Black.dimmed().prefix().to_string()),
|
|
"br" | "black_reverse" => Some(Color::Black.reverse().prefix().to_string()),
|
|
"y" | "yellow" => Some(Color::Yellow.prefix().to_string()),
|
|
"yb" | "yellow_bold" => Some(Color::Yellow.bold().prefix().to_string()),
|
|
"yu" | "yellow_underline" => Some(Color::Yellow.underline().prefix().to_string()),
|
|
"yi" | "yellow_italic" => Some(Color::Yellow.italic().prefix().to_string()),
|
|
"yd" | "yellow_dimmed" => Some(Color::Yellow.dimmed().prefix().to_string()),
|
|
"yr" | "yellow_reverse" => Some(Color::Yellow.reverse().prefix().to_string()),
|
|
"p" | "purple" => Some(Color::Purple.prefix().to_string()),
|
|
"pb" | "purple_bold" => Some(Color::Purple.bold().prefix().to_string()),
|
|
"pu" | "purple_underline" => Some(Color::Purple.underline().prefix().to_string()),
|
|
"pi" | "purple_italic" => Some(Color::Purple.italic().prefix().to_string()),
|
|
"pd" | "purple_dimmed" => Some(Color::Purple.dimmed().prefix().to_string()),
|
|
"pr" | "purple_reverse" => Some(Color::Purple.reverse().prefix().to_string()),
|
|
"c" | "cyan" => Some(Color::Cyan.prefix().to_string()),
|
|
"cb" | "cyan_bold" => Some(Color::Cyan.bold().prefix().to_string()),
|
|
"cu" | "cyan_underline" => Some(Color::Cyan.underline().prefix().to_string()),
|
|
"ci" | "cyan_italic" => Some(Color::Cyan.italic().prefix().to_string()),
|
|
"cd" | "cyan_dimmed" => Some(Color::Cyan.dimmed().prefix().to_string()),
|
|
"cr" | "cyan_reverse" => Some(Color::Cyan.reverse().prefix().to_string()),
|
|
"w" | "white" => Some(Color::White.prefix().to_string()),
|
|
"wb" | "white_bold" => Some(Color::White.bold().prefix().to_string()),
|
|
"wu" | "white_underline" => Some(Color::White.underline().prefix().to_string()),
|
|
"wi" | "white_italic" => Some(Color::White.italic().prefix().to_string()),
|
|
"wd" | "white_dimmed" => Some(Color::White.dimmed().prefix().to_string()),
|
|
"wr" | "white_reverse" => Some(Color::White.reverse().prefix().to_string()),
|
|
"reset" => Some("\x1b[0m".to_owned()),
|
|
_ => None,
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::Ansi;
|
|
use super::ShellError;
|
|
|
|
#[test]
|
|
fn examples_work_as_expected() -> Result<(), ShellError> {
|
|
use crate::examples::test as test_examples;
|
|
|
|
Ok(test_examples(Ansi {})?)
|
|
}
|
|
}
|