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 | |