aboutsummaryrefslogtreecommitdiff
path: root/vendor/anstream/src/wincon.rs
diff options
context:
space:
mode:
authorValentin Popov <valentin@popov.link>2024-07-19 15:37:58 +0300
committerValentin Popov <valentin@popov.link>2024-07-19 15:37:58 +0300
commita990de90fe41456a23e58bd087d2f107d321f3a1 (patch)
tree15afc392522a9e85dc3332235e311b7d39352ea9 /vendor/anstream/src/wincon.rs
parent3d48cd3f81164bbfc1a755dc1d4a9a02f98c8ddd (diff)
downloadfparkan-a990de90fe41456a23e58bd087d2f107d321f3a1.tar.xz
fparkan-a990de90fe41456a23e58bd087d2f107d321f3a1.zip
Deleted vendor folder
Diffstat (limited to 'vendor/anstream/src/wincon.rs')
-rw-r--r--vendor/anstream/src/wincon.rs210
1 files changed, 0 insertions, 210 deletions
diff --git a/vendor/anstream/src/wincon.rs b/vendor/anstream/src/wincon.rs
deleted file mode 100644
index a5aa5e7..0000000
--- a/vendor/anstream/src/wincon.rs
+++ /dev/null
@@ -1,210 +0,0 @@
-use crate::adapter::WinconBytes;
-use crate::stream::AsLockedWrite;
-use crate::stream::RawStream;
-
-/// Only pass printable data to the inner `Write`
-#[cfg(feature = "wincon")] // here mostly for documentation purposes
-#[derive(Debug)]
-pub struct WinconStream<S>
-where
- S: RawStream,
-{
- raw: S,
- // `WinconBytes` is especially large compared to other variants of `AutoStream`, so boxing it
- // here so `AutoStream` doesn't have to discard one allocation and create another one when
- // calling `AutoStream::lock`
- state: Box<WinconBytes>,
-}
-
-impl<S> WinconStream<S>
-where
- S: RawStream,
-{
- /// Only pass printable data to the inner `Write`
- #[inline]
- pub fn new(raw: S) -> Self {
- Self {
- raw,
- state: Default::default(),
- }
- }
-
- /// Get the wrapped [`RawStream`]
- #[inline]
- pub fn into_inner(self) -> S {
- self.raw
- }
-
- #[inline]
- pub fn is_terminal(&self) -> bool {
- self.raw.is_terminal()
- }
-}
-
-impl WinconStream<std::io::Stdout> {
- /// Get exclusive access to the `WinconStream`
- ///
- /// Why?
- /// - Faster performance when writing in a loop
- /// - Avoid other threads interleaving output with the current thread
- #[inline]
- pub fn lock(self) -> WinconStream<std::io::StdoutLock<'static>> {
- WinconStream {
- raw: self.raw.lock(),
- state: self.state,
- }
- }
-}
-
-impl WinconStream<std::io::Stderr> {
- /// Get exclusive access to the `WinconStream`
- ///
- /// Why?
- /// - Faster performance when writing in a loop
- /// - Avoid other threads interleaving output with the current thread
- #[inline]
- pub fn lock(self) -> WinconStream<std::io::StderrLock<'static>> {
- WinconStream {
- raw: self.raw.lock(),
- state: self.state,
- }
- }
-}
-
-impl<S> std::io::Write for WinconStream<S>
-where
- S: RawStream + AsLockedWrite,
-{
- // Must forward all calls to ensure locking happens appropriately
- #[inline]
- fn write(&mut self, buf: &[u8]) -> std::io::Result<usize> {
- write(&mut self.raw.as_locked_write(), &mut self.state, buf)
- }
- #[inline]
- fn write_vectored(&mut self, bufs: &[std::io::IoSlice<'_>]) -> std::io::Result<usize> {
- let buf = bufs
- .iter()
- .find(|b| !b.is_empty())
- .map(|b| &**b)
- .unwrap_or(&[][..]);
- self.write(buf)
- }
- // is_write_vectored: nightly only
- #[inline]
- fn flush(&mut self) -> std::io::Result<()> {
- self.raw.as_locked_write().flush()
- }
- #[inline]
- fn write_all(&mut self, buf: &[u8]) -> std::io::Result<()> {
- write_all(&mut self.raw.as_locked_write(), &mut self.state, buf)
- }
- // write_all_vectored: nightly only
- #[inline]
- fn write_fmt(&mut self, args: std::fmt::Arguments<'_>) -> std::io::Result<()> {
- write_fmt(&mut self.raw.as_locked_write(), &mut self.state, args)
- }
-}
-
-fn write(raw: &mut dyn RawStream, state: &mut WinconBytes, buf: &[u8]) -> std::io::Result<usize> {
- for (style, printable) in state.extract_next(buf) {
- let fg = style.get_fg_color().and_then(cap_wincon_color);
- let bg = style.get_bg_color().and_then(cap_wincon_color);
- let written = raw.write_colored(fg, bg, printable.as_bytes())?;
- let possible = printable.len();
- if possible != written {
- // HACK: Unsupported atm
- break;
- }
- }
- Ok(buf.len())
-}
-
-fn write_all(raw: &mut dyn RawStream, state: &mut WinconBytes, buf: &[u8]) -> std::io::Result<()> {
- for (style, printable) in state.extract_next(buf) {
- let mut buf = printable.as_bytes();
- let fg = style.get_fg_color().and_then(cap_wincon_color);
- let bg = style.get_bg_color().and_then(cap_wincon_color);
- while !buf.is_empty() {
- match raw.write_colored(fg, bg, buf) {
- Ok(0) => {
- return Err(std::io::Error::new(
- std::io::ErrorKind::WriteZero,
- "failed to write whole buffer",
- ));
- }
- Ok(n) => buf = &buf[n..],
- Err(ref e) if e.kind() == std::io::ErrorKind::Interrupted => {}
- Err(e) => return Err(e),
- }
- }
- }
- Ok(())
-}
-
-fn write_fmt(
- raw: &mut dyn RawStream,
- state: &mut WinconBytes,
- args: std::fmt::Arguments<'_>,
-) -> std::io::Result<()> {
- let write_all = |buf: &[u8]| write_all(raw, state, buf);
- crate::fmt::Adapter::new(write_all).write_fmt(args)
-}
-
-fn cap_wincon_color(color: anstyle::Color) -> Option<anstyle::AnsiColor> {
- match color {
- anstyle::Color::Ansi(c) => Some(c),
- anstyle::Color::Ansi256(c) => c.into_ansi(),
- anstyle::Color::Rgb(_) => None,
- }
-}
-
-#[cfg(test)]
-mod test {
- use super::*;
- use proptest::prelude::*;
- use std::io::Write as _;
-
- proptest! {
- #[test]
- #[cfg_attr(miri, ignore)] // See https://github.com/AltSysrq/proptest/issues/253
- fn write_all_no_escapes(s in "\\PC*") {
- let buffer = Vec::new();
- let mut stream = WinconStream::new(buffer);
- stream.write_all(s.as_bytes()).unwrap();
- let buffer = stream.into_inner();
- let actual = std::str::from_utf8(buffer.as_ref()).unwrap();
- assert_eq!(s, actual);
- }
-
- #[test]
- #[cfg_attr(miri, ignore)] // See https://github.com/AltSysrq/proptest/issues/253
- fn write_byte_no_escapes(s in "\\PC*") {
- let buffer = Vec::new();
- let mut stream = WinconStream::new(buffer);
- for byte in s.as_bytes() {
- stream.write_all(&[*byte]).unwrap();
- }
- let buffer = stream.into_inner();
- let actual = std::str::from_utf8(buffer.as_ref()).unwrap();
- assert_eq!(s, actual);
- }
-
- #[test]
- #[cfg_attr(miri, ignore)] // See https://github.com/AltSysrq/proptest/issues/253
- fn write_all_random(s in any::<Vec<u8>>()) {
- let buffer = Vec::new();
- let mut stream = WinconStream::new(buffer);
- stream.write_all(s.as_slice()).unwrap();
- }
-
- #[test]
- #[cfg_attr(miri, ignore)] // See https://github.com/AltSysrq/proptest/issues/253
- fn write_byte_random(s in any::<Vec<u8>>()) {
- let buffer = Vec::new();
- let mut stream = WinconStream::new(buffer);
- for byte in s.as_slice() {
- stream.write_all(&[*byte]).unwrap();
- }
- }
- }
-}