vendor/anstream/src/adapter/wincon.rs - toolchain/rustc - Git at Google

 /// Incrementally convert to wincon calls for non-contiguous data
 #[derive(Default, Clone, Debug, PartialEq, Eq)]
 pub struct WinconBytes {
     parser: anstyle_parse::Parser,
     capture: WinconCapture,
 }

 impl WinconBytes {
     /// Initial state
     pub fn new() -> Self {
         Default::default()
     }

     /// Strip the next segment of data
     pub fn extract_next<'s>(&'s mut self, bytes: &'s [u8]) -> WinconBytesIter<'s> {
         self.capture.reset();
         self.capture.printable.reserve(bytes.len());
         WinconBytesIter {
             bytes,
             parser: &mut self.parser,
             capture: &mut self.capture,
         }
     }
 }

 /// See [`WinconBytes`]
 #[derive(Debug, PartialEq, Eq)]
 pub struct WinconBytesIter<'s> {
     bytes: &'s [u8],
     parser: &'s mut anstyle_parse::Parser,
     capture: &'s mut WinconCapture,
 }

 impl<'s> Iterator for WinconBytesIter<'s> {
     type Item = (anstyle::Style, String);

     #[inline]
     fn next(&mut self) -> Option<Self::Item> {
         next_bytes(&mut self.bytes, self.parser, self.capture)
     }
 }

 #[inline]
 fn next_bytes(
     bytes: &mut &[u8],
     parser: &mut anstyle_parse::Parser,
     capture: &mut WinconCapture,
 ) -> Option<(anstyle::Style, String)> {
     capture.reset();
     while capture.ready.is_none() {
         let byte = if let Some((byte, remainder)) = (*bytes).split_first() {
             *bytes = remainder;
             *byte
         } else {
             break;
         };
         parser.advance(capture, byte);
     }
     if capture.printable.is_empty() {
         return None;
     }

     let style = capture.ready.unwrap_or(capture.style);
     Some((style, std::mem::take(&mut capture.printable)))
 }

 #[derive(Default, Clone, Debug, PartialEq, Eq)]
 struct WinconCapture {
     style: anstyle::Style,
     printable: String,
     ready: Option<anstyle::Style>,
 }

 impl WinconCapture {
     fn reset(&mut self) {
         self.ready = None;
     }
 }

 impl anstyle_parse::Perform for WinconCapture {
     /// Draw a character to the screen and update states.
     fn print(&mut self, c: char) {
         self.printable.push(c);
     }

     /// Execute a C0 or C1 control function.
     fn execute(&mut self, byte: u8) {
         if byte.is_ascii_whitespace() {
             self.printable.push(byte as char);
         }
     }

     fn csi_dispatch(
         &mut self,
         params: &anstyle_parse::Params,
         _intermediates: &[u8],
         ignore: bool,
         action: u8,
     ) {
         if ignore {
             return;
         }
         if action != b'm' {
             return;
         }

         let mut style = self.style;
         // param/value differences are dependent on the escape code
         let mut state = State::Normal;
         let mut r = None;
         let mut g = None;
         let mut color_target = ColorTarget::Fg;
         for param in params {
             for value in param {
                 match (state, *value) {
                     (State::Normal, 0) => {
                         style = anstyle::Style::default();
                         break;
                     }
                     (State::Normal, 1) => {
                         style = style.bold();
                         break;
                     }
                     (State::Normal, 2) => {
                         style = style.dimmed();
                         break;
                     }
                     (State::Normal, 3) => {
                         style = style.italic();
                         break;
                     }
                     (State::Normal, 4) => {
                         style = style.underline();
                         state = State::Underline;
                     }
                     (State::Normal, 21) => {
                         style |= anstyle::Effects::DOUBLE_UNDERLINE;
                         break;
                     }
                     (State::Normal, 7) => {
                         style = style.invert();
                         break;
                     }
                     (State::Normal, 8) => {
                         style = style.hidden();
                         break;
                     }
                     (State::Normal, 9) => {
                         style = style.strikethrough();
                         break;
                     }
                     (State::Normal, 30..=37) => {
                         let color = to_ansi_color(value - 30).unwrap();
                         style = style.fg_color(Some(color.into()));
                         break;
                     }
                     (State::Normal, 38) => {
                         color_target = ColorTarget::Fg;
                         state = State::PrepareCustomColor;
                     }
                     (State::Normal, 39) => {
                         style = style.fg_color(None);
                         break;
                     }
                     (State::Normal, 40..=47) => {
                         let color = to_ansi_color(value - 40).unwrap();
                         style = style.bg_color(Some(color.into()));
                         break;
                     }
                     (State::Normal, 48) => {
                         color_target = ColorTarget::Bg;
                         state = State::PrepareCustomColor;
                     }
                     (State::Normal, 49) => {
                         style = style.bg_color(None);
                         break;
                     }
                     (State::Normal, 58) => {
                         color_target = ColorTarget::Underline;
                         state = State::PrepareCustomColor;
                     }
                     (State::Normal, 90..=97) => {
                         let color = to_ansi_color(value - 90).unwrap().bright(true);
                         style = style.fg_color(Some(color.into()));
                         break;
                     }
                     (State::Normal, 100..=107) => {
                         let color = to_ansi_color(value - 100).unwrap().bright(true);
                         style = style.bg_color(Some(color.into()));
                         break;
                     }
                     (State::PrepareCustomColor, 5) => {
                         state = State::Ansi256;
                     }
                     (State::PrepareCustomColor, 2) => {
                         state = State::Rgb;
                         r = None;
                         g = None;
                     }
                     (State::Ansi256, n) => {
                         let color = anstyle::Ansi256Color(n as u8);
                         style = match color_target {
                             ColorTarget::Fg => style.fg_color(Some(color.into())),
                             ColorTarget::Bg => style.bg_color(Some(color.into())),
                             ColorTarget::Underline => style.underline_color(Some(color.into())),
                         };
                         break;
                     }
                     (State::Rgb, b) => match (r, g) {
                         (None, _) => {
                             r = Some(b);
                         }
                         (Some(_), None) => {
                             g = Some(b);
                         }
                         (Some(r), Some(g)) => {
                             let color = anstyle::RgbColor(r as u8, g as u8, b as u8);
                             style = match color_target {
                                 ColorTarget::Fg => style.fg_color(Some(color.into())),
                                 ColorTarget::Bg => style.bg_color(Some(color.into())),
                                 ColorTarget::Underline => style.underline_color(Some(color.into())),
                             };
                             break;
                         }
                     },
                     (State::Underline, 0) => {
                         style =
                             style.effects(style.get_effects().remove(anstyle::Effects::UNDERLINE));
                     }
                     (State::Underline, 1) => {
                         // underline already set
                     }
                     (State::Underline, 2) => {
                         style = style
                             .effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
                             | anstyle::Effects::DOUBLE_UNDERLINE;
                     }
                     (State::Underline, 3) => {
                         style = style
                             .effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
                             | anstyle::Effects::CURLY_UNDERLINE;
                     }
                     (State::Underline, 4) => {
                         style = style
                             .effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
                             | anstyle::Effects::DOTTED_UNDERLINE;
                     }
                     (State::Underline, 5) => {
                         style = style
                             .effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
                             | anstyle::Effects::DASHED_UNDERLINE;
                     }
                     _ => {
                         break;
                     }
                 }
             }
         }

         if style != self.style && !self.printable.is_empty() {
             self.ready = Some(self.style);
         }
         self.style = style;
     }
 }

 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
 enum State {
     Normal,
     PrepareCustomColor,
     Ansi256,
     Rgb,
     Underline,
 }

 #[derive(Copy, Clone, PartialEq, Eq, Debug)]
 enum ColorTarget {
     Fg,
     Bg,
     Underline,
 }

 fn to_ansi_color(digit: u16) -> Option<anstyle::AnsiColor> {
     match digit {
         0 => Some(anstyle::AnsiColor::Black),
         1 => Some(anstyle::AnsiColor::Red),
         2 => Some(anstyle::AnsiColor::Green),
         3 => Some(anstyle::AnsiColor::Yellow),
         4 => Some(anstyle::AnsiColor::Blue),
         5 => Some(anstyle::AnsiColor::Magenta),
         6 => Some(anstyle::AnsiColor::Cyan),
         7 => Some(anstyle::AnsiColor::White),
         _ => None,
     }
 }

 #[cfg(test)]
 mod test {
     use super::*;
     use owo_colors::OwoColorize as _;
     use proptest::prelude::*;

     #[track_caller]
     fn verify(input: &str, expected: Vec<(anstyle::Style, &str)>) {
         let expected = expected
             .into_iter()
             .map(|(style, value)| (style, value.to_owned()))
             .collect::<Vec<_>>();
         let mut state = WinconBytes::new();
         let actual = state.extract_next(input.as_bytes()).collect::<Vec<_>>();
         assert_eq!(expected, actual, "{input:?}");
     }

     #[test]
     fn start() {
         let input = format!("{} world!", "Hello".green().on_red());
         let expected = vec![
             (
                 anstyle::AnsiColor::Green.on(anstyle::AnsiColor::Red),
                 "Hello",
             ),
             (anstyle::Style::default(), " world!"),
         ];
         verify(&input, expected);
     }

     #[test]
     fn middle() {
         let input = format!("Hello {}!", "world".green().on_red());
         let expected = vec![
             (anstyle::Style::default(), "Hello "),
             (
                 anstyle::AnsiColor::Green.on(anstyle::AnsiColor::Red),
                 "world",
             ),
             (anstyle::Style::default(), "!"),
         ];
         verify(&input, expected);
     }

     #[test]
     fn end() {
         let input = format!("Hello {}", "world!".green().on_red());
         let expected = vec![
             (anstyle::Style::default(), "Hello "),
             (
                 anstyle::AnsiColor::Green.on(anstyle::AnsiColor::Red),
                 "world!",
             ),
         ];
         verify(&input, expected);
     }

     #[test]
     fn ansi256_colors() {
         // termcolor only supports "brights" via these
         let input = format!(
             "Hello {}!",
             "world".color(owo_colors::XtermColors::UserBrightYellow)
         );
         let expected = vec![
             (anstyle::Style::default(), "Hello "),
             (anstyle::Ansi256Color(11).on_default(), "world"),
             (anstyle::Style::default(), "!"),
         ];
         verify(&input, expected);
     }

     proptest! {
         #[test]
         #[cfg_attr(miri, ignore)]  // See https://github.com/AltSysrq/proptest/issues/253
         fn wincon_no_escapes(s in "\\PC*") {
             let expected = if s.is_empty() {
                 vec![]
             } else {
                 vec![(anstyle::Style::default(), s.clone())]
             };
             let mut state = WinconBytes::new();
             let actual = state.extract_next(s.as_bytes()).collect::<Vec<_>>();
             assert_eq!(expected, actual);
         }
     }
 }
	/// Incrementally convert to wincon calls for non-contiguous data
	#[derive(Default, Clone, Debug, PartialEq, Eq)]
	pub struct WinconBytes {
	parser: anstyle_parse::Parser,
	capture: WinconCapture,
	}

	impl WinconBytes {
	/// Initial state
	pub fn new() -> Self {
	Default::default()
	}

	/// Strip the next segment of data
	pub fn extract_next<'s>(&'s mut self, bytes: &'s [u8]) -> WinconBytesIter<'s> {
	self.capture.reset();
	self.capture.printable.reserve(bytes.len());
	WinconBytesIter {
	bytes,
	parser: &mut self.parser,
	capture: &mut self.capture,
	}
	}
	}

	/// See [`WinconBytes`]
	#[derive(Debug, PartialEq, Eq)]
	pub struct WinconBytesIter<'s> {
	bytes: &'s [u8],
	parser: &'s mut anstyle_parse::Parser,
	capture: &'s mut WinconCapture,
	}

	impl<'s> Iterator for WinconBytesIter<'s> {
	type Item = (anstyle::Style, String);

	#[inline]
	fn next(&mut self) -> Option<Self::Item> {
	next_bytes(&mut self.bytes, self.parser, self.capture)
	}
	}

	#[inline]
	fn next_bytes(
	bytes: &mut &[u8],
	parser: &mut anstyle_parse::Parser,
	capture: &mut WinconCapture,
	) -> Option<(anstyle::Style, String)> {
	capture.reset();
	while capture.ready.is_none() {
	let byte = if let Some((byte, remainder)) = (*bytes).split_first() {
	*bytes = remainder;
	*byte
	} else {
	break;
	};
	parser.advance(capture, byte);
	}
	if capture.printable.is_empty() {
	return None;
	}

	let style = capture.ready.unwrap_or(capture.style);
	Some((style, std::mem::take(&mut capture.printable)))
	}

	#[derive(Default, Clone, Debug, PartialEq, Eq)]
	struct WinconCapture {
	style: anstyle::Style,
	printable: String,
	ready: Option<anstyle::Style>,
	}

	impl WinconCapture {
	fn reset(&mut self) {
	self.ready = None;
	}
	}

	impl anstyle_parse::Perform for WinconCapture {
	/// Draw a character to the screen and update states.
	fn print(&mut self, c: char) {
	self.printable.push(c);
	}

	/// Execute a C0 or C1 control function.
	fn execute(&mut self, byte: u8) {
	if byte.is_ascii_whitespace() {
	self.printable.push(byte as char);
	}
	}

	fn csi_dispatch(
	&mut self,
	params: &anstyle_parse::Params,
	_intermediates: &[u8],
	ignore: bool,
	action: u8,
	) {
	if ignore {
	return;
	}
	if action != b'm' {
	return;
	}

	let mut style = self.style;
	// param/value differences are dependent on the escape code
	let mut state = State::Normal;
	let mut r = None;
	let mut g = None;
	let mut color_target = ColorTarget::Fg;
	for param in params {
	for value in param {
	match (state, *value) {
	(State::Normal, 0) => {
	style = anstyle::Style::default();
	break;
	}
	(State::Normal, 1) => {
	style = style.bold();
	break;
	}
	(State::Normal, 2) => {
	style = style.dimmed();
	break;
	}
	(State::Normal, 3) => {
	style = style.italic();
	break;
	}
	(State::Normal, 4) => {
	style = style.underline();
	state = State::Underline;
	}
	(State::Normal, 21) => {
	style \|= anstyle::Effects::DOUBLE_UNDERLINE;
	break;
	}
	(State::Normal, 7) => {
	style = style.invert();
	break;
	}
	(State::Normal, 8) => {
	style = style.hidden();
	break;
	}
	(State::Normal, 9) => {
	style = style.strikethrough();
	break;
	}
	(State::Normal, 30..=37) => {
	let color = to_ansi_color(value - 30).unwrap();
	style = style.fg_color(Some(color.into()));
	break;
	}
	(State::Normal, 38) => {
	color_target = ColorTarget::Fg;
	state = State::PrepareCustomColor;
	}
	(State::Normal, 39) => {
	style = style.fg_color(None);
	break;
	}
	(State::Normal, 40..=47) => {
	let color = to_ansi_color(value - 40).unwrap();
	style = style.bg_color(Some(color.into()));
	break;
	}
	(State::Normal, 48) => {
	color_target = ColorTarget::Bg;
	state = State::PrepareCustomColor;
	}
	(State::Normal, 49) => {
	style = style.bg_color(None);
	break;
	}
	(State::Normal, 58) => {
	color_target = ColorTarget::Underline;
	state = State::PrepareCustomColor;
	}
	(State::Normal, 90..=97) => {
	let color = to_ansi_color(value - 90).unwrap().bright(true);
	style = style.fg_color(Some(color.into()));
	break;
	}
	(State::Normal, 100..=107) => {
	let color = to_ansi_color(value - 100).unwrap().bright(true);
	style = style.bg_color(Some(color.into()));
	break;
	}
	(State::PrepareCustomColor, 5) => {
	state = State::Ansi256;
	}
	(State::PrepareCustomColor, 2) => {
	state = State::Rgb;
	r = None;
	g = None;
	}
	(State::Ansi256, n) => {
	let color = anstyle::Ansi256Color(n as u8);
	style = match color_target {
	ColorTarget::Fg => style.fg_color(Some(color.into())),
	ColorTarget::Bg => style.bg_color(Some(color.into())),
	ColorTarget::Underline => style.underline_color(Some(color.into())),
	};
	break;
	}
	(State::Rgb, b) => match (r, g) {
	(None, _) => {
	r = Some(b);
	}
	(Some(_), None) => {
	g = Some(b);
	}
	(Some(r), Some(g)) => {
	let color = anstyle::RgbColor(r as u8, g as u8, b as u8);
	style = match color_target {
	ColorTarget::Fg => style.fg_color(Some(color.into())),
	ColorTarget::Bg => style.bg_color(Some(color.into())),
	ColorTarget::Underline => style.underline_color(Some(color.into())),
	};
	break;
	}
	},
	(State::Underline, 0) => {
	style =
	style.effects(style.get_effects().remove(anstyle::Effects::UNDERLINE));
	}
	(State::Underline, 1) => {
	// underline already set
	}
	(State::Underline, 2) => {
	style = style
	.effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
	\| anstyle::Effects::DOUBLE_UNDERLINE;
	}
	(State::Underline, 3) => {
	style = style
	.effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
	\| anstyle::Effects::CURLY_UNDERLINE;
	}
	(State::Underline, 4) => {
	style = style
	.effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
	\| anstyle::Effects::DOTTED_UNDERLINE;
	}
	(State::Underline, 5) => {
	style = style
	.effects(style.get_effects().remove(anstyle::Effects::UNDERLINE))
	\| anstyle::Effects::DASHED_UNDERLINE;
	}
	_ => {
	break;
	}
	}
	}
	}

	if style != self.style && !self.printable.is_empty() {
	self.ready = Some(self.style);
	}
	self.style = style;
	}
	}

	#[derive(Copy, Clone, PartialEq, Eq, Debug)]
	enum State {
	Normal,
	PrepareCustomColor,
	Ansi256,
	Rgb,
	Underline,
	}

	#[derive(Copy, Clone, PartialEq, Eq, Debug)]
	enum ColorTarget {
	Fg,
	Bg,
	Underline,
	}

	fn to_ansi_color(digit: u16) -> Option<anstyle::AnsiColor> {
	match digit {
	0 => Some(anstyle::AnsiColor::Black),
	1 => Some(anstyle::AnsiColor::Red),
	2 => Some(anstyle::AnsiColor::Green),
	3 => Some(anstyle::AnsiColor::Yellow),
	4 => Some(anstyle::AnsiColor::Blue),
	5 => Some(anstyle::AnsiColor::Magenta),
	6 => Some(anstyle::AnsiColor::Cyan),
	7 => Some(anstyle::AnsiColor::White),
	_ => None,
	}
	}

	#[cfg(test)]
	mod test {
	use super::*;
	use owo_colors::OwoColorize as _;
	use proptest::prelude::*;

	#[track_caller]
	fn verify(input: &str, expected: Vec<(anstyle::Style, &str)>) {
	let expected = expected
	.into_iter()
	.map(\|(style, value)\| (style, value.to_owned()))
	.collect::<Vec<_>>();
	let mut state = WinconBytes::new();
	let actual = state.extract_next(input.as_bytes()).collect::<Vec<_>>();
	assert_eq!(expected, actual, "{input:?}");
	}

	#[test]
	fn start() {
	let input = format!("{} world!", "Hello".green().on_red());
	let expected = vec![
	(
	anstyle::AnsiColor::Green.on(anstyle::AnsiColor::Red),
	"Hello",
	),
	(anstyle::Style::default(), " world!"),
	];
	verify(&input, expected);
	}

	#[test]
	fn middle() {
	let input = format!("Hello {}!", "world".green().on_red());
	let expected = vec![
	(anstyle::Style::default(), "Hello "),
	(
	anstyle::AnsiColor::Green.on(anstyle::AnsiColor::Red),
	"world",
	),
	(anstyle::Style::default(), "!"),
	];
	verify(&input, expected);
	}

	#[test]
	fn end() {
	let input = format!("Hello {}", "world!".green().on_red());
	let expected = vec![
	(anstyle::Style::default(), "Hello "),
	(
	anstyle::AnsiColor::Green.on(anstyle::AnsiColor::Red),
	"world!",
	),
	];
	verify(&input, expected);
	}

	#[test]
	fn ansi256_colors() {
	// termcolor only supports "brights" via these
	let input = format!(
	"Hello {}!",
	"world".color(owo_colors::XtermColors::UserBrightYellow)
	);
	let expected = vec![
	(anstyle::Style::default(), "Hello "),
	(anstyle::Ansi256Color(11).on_default(), "world"),
	(anstyle::Style::default(), "!"),
	];
	verify(&input, expected);
	}

	proptest! {
	#[test]
	#[cfg_attr(miri, ignore)] // See https://github.com/AltSysrq/proptest/issues/253
	fn wincon_no_escapes(s in "\\PC*") {
	let expected = if s.is_empty() {
	vec![]
	} else {
	vec![(anstyle::Style::default(), s.clone())]
	};
	let mut state = WinconBytes::new();
	let actual = state.extract_next(s.as_bytes()).collect::<Vec<_>>();
	assert_eq!(expected, actual);
	}
	}
	}