| 1 | //! Terminal formatting module. |
| 2 | //! |
| 3 | //! This module provides the `Terminal` trait, which abstracts over an [ANSI |
| 4 | //! Terminal][ansi] to provide color printing, among other things. There are two |
| 5 | //! implementations, the `TerminfoTerminal`, which uses control characters from |
| 6 | //! a [terminfo][ti] database, and `WinConsole`, which uses the [Win32 Console |
| 7 | //! API][win]. |
| 8 | //! |
| 9 | //! [ansi]: https://en.wikipedia.org/wiki/ANSI_escape_code |
| 10 | //! [win]: https://docs.microsoft.com/en-us/windows/console/character-mode-applications |
| 11 | //! [ti]: https://en.wikipedia.org/wiki/Terminfo |
| 12 | |
| 13 | #![deny (missing_docs)] |
| 14 | |
| 15 | use std::io::prelude::*; |
| 16 | use std::io::{self}; |
| 17 | |
| 18 | pub(crate) use terminfo::TerminfoTerminal; |
| 19 | #[cfg (windows)] |
| 20 | pub(crate) use win::WinConsole; |
| 21 | |
| 22 | pub(crate) mod terminfo; |
| 23 | |
| 24 | #[cfg (windows)] |
| 25 | mod win; |
| 26 | |
| 27 | /// Alias for stdout terminals. |
| 28 | pub(crate) type StdoutTerminal = dyn Terminal + Send; |
| 29 | |
| 30 | #[cfg (not(windows))] |
| 31 | /// Returns a Terminal wrapping stdout, or None if a terminal couldn't be |
| 32 | /// opened. |
| 33 | pub(crate) fn stdout() -> Option<Box<StdoutTerminal>> { |
| 34 | TerminfoTerminal::new(out:io::stdout()).map(|t: TerminfoTerminal| Box::new(t) as Box<StdoutTerminal>) |
| 35 | } |
| 36 | |
| 37 | #[cfg (windows)] |
| 38 | /// Returns a Terminal wrapping stdout, or None if a terminal couldn't be |
| 39 | /// opened. |
| 40 | pub(crate) fn stdout() -> Option<Box<StdoutTerminal>> { |
| 41 | TerminfoTerminal::new(io::stdout()) |
| 42 | .map(|t| Box::new(t) as Box<StdoutTerminal>) |
| 43 | .or_else(|| Some(Box::new(WinConsole::new(io::stdout())) as Box<StdoutTerminal>)) |
| 44 | } |
| 45 | |
| 46 | /// Terminal color definitions |
| 47 | #[allow (missing_docs)] |
| 48 | #[cfg_attr (not(windows), allow(dead_code))] |
| 49 | pub(crate) mod color { |
| 50 | /// Number for a terminal color |
| 51 | pub(crate) type Color = u32; |
| 52 | |
| 53 | pub(crate) const BLACK: Color = 0; |
| 54 | pub(crate) const RED: Color = 1; |
| 55 | pub(crate) const GREEN: Color = 2; |
| 56 | pub(crate) const YELLOW: Color = 3; |
| 57 | pub(crate) const BLUE: Color = 4; |
| 58 | pub(crate) const MAGENTA: Color = 5; |
| 59 | pub(crate) const CYAN: Color = 6; |
| 60 | pub(crate) const WHITE: Color = 7; |
| 61 | } |
| 62 | |
| 63 | /// A terminal with similar capabilities to an ANSI Terminal |
| 64 | /// (foreground/background colors etc). |
| 65 | pub(crate) trait Terminal: Write { |
| 66 | /// Sets the foreground color to the given color. |
| 67 | /// |
| 68 | /// If the color is a bright color, but the terminal only supports 8 colors, |
| 69 | /// the corresponding normal color will be used instead. |
| 70 | /// |
| 71 | /// Returns `Ok(true)` if the color was set, `Ok(false)` otherwise, and `Err(e)` |
| 72 | /// if there was an I/O error. |
| 73 | fn fg(&mut self, color: color::Color) -> io::Result<bool>; |
| 74 | |
| 75 | /// Resets all terminal attributes and colors to their defaults. |
| 76 | /// |
| 77 | /// Returns `Ok(true)` if the terminal was reset, `Ok(false)` otherwise, and `Err(e)` if there |
| 78 | /// was an I/O error. |
| 79 | /// |
| 80 | /// *Note: This does not flush.* |
| 81 | /// |
| 82 | /// That means the reset command may get buffered so, if you aren't planning on doing anything |
| 83 | /// else that might flush stdout's buffer (e.g., writing a line of text), you should flush after |
| 84 | /// calling reset. |
| 85 | fn reset(&mut self) -> io::Result<bool>; |
| 86 | } |
| 87 | |