git.iko.soy/bat
on 448bfaf help sign in
bat/src/style.rs
122 lines 3.7 kB history raw
1use std::collections::HashSet;
2use std::str::FromStr;
4use crate::error::*;
6#[non_exhaustive]
7#[derive(Debug, Eq, PartialEq, Copy, Clone, Hash)]
8pub enum StyleComponent {
9 Auto,
10 #[cfg(feature = "git")]
11 Changes,
12 Grid,
13 Rule,
14 Header,
15 HeaderFilename,
16 HeaderFilesize,
17 LineNumbers,
18 Snip,
19 Full,
20 Plain,
23impl StyleComponent {
24 pub fn components(self, interactive_terminal: bool) -> &'static [StyleComponent] {
25 match self {
26 StyleComponent::Auto => {
27 if interactive_terminal {
28 StyleComponent::Full.components(interactive_terminal)
29 } else {
30 StyleComponent::Plain.components(interactive_terminal)
31 }
32 }
33 #[cfg(feature = "git")]
34 StyleComponent::Changes => &[StyleComponent::Changes],
35 StyleComponent::Grid => &[StyleComponent::Grid],
36 StyleComponent::Rule => &[StyleComponent::Rule],
37 StyleComponent::Header => &[StyleComponent::HeaderFilename],
38 StyleComponent::HeaderFilename => &[StyleComponent::HeaderFilename],
39 StyleComponent::HeaderFilesize => &[StyleComponent::HeaderFilesize],
40 StyleComponent::LineNumbers => &[StyleComponent::LineNumbers],
41 StyleComponent::Snip => &[StyleComponent::Snip],
42 StyleComponent::Full => &[
43 #[cfg(feature = "git")]
44 StyleComponent::Changes,
45 StyleComponent::Grid,
46 StyleComponent::HeaderFilename,
47 StyleComponent::HeaderFilesize,
48 StyleComponent::LineNumbers,
49 StyleComponent::Snip,
50 ],
51 StyleComponent::Plain => &[],
52 }
53 }
56impl FromStr for StyleComponent {
57 type Err = Error;
59 fn from_str(s: &str) -> Result<Self> {
60 match s {
61 "auto" => Ok(StyleComponent::Auto),
62 #[cfg(feature = "git")]
63 "changes" => Ok(StyleComponent::Changes),
64 "grid" => Ok(StyleComponent::Grid),
65 "rule" => Ok(StyleComponent::Rule),
66 "header" => Ok(StyleComponent::Header),
67 "header-filename" => Ok(StyleComponent::HeaderFilename),
68 "header-filesize" => Ok(StyleComponent::HeaderFilesize),
69 "numbers" => Ok(StyleComponent::LineNumbers),
70 "snip" => Ok(StyleComponent::Snip),
71 "full" => Ok(StyleComponent::Full),
72 "plain" => Ok(StyleComponent::Plain),
73 _ => Err(format!("Unknown style '{}'", s).into()),
74 }
75 }
78#[derive(Debug, Clone, Default)]
79pub struct StyleComponents(pub HashSet<StyleComponent>);
81impl StyleComponents {
82 pub fn new(components: &[StyleComponent]) -> StyleComponents {
83 StyleComponents(components.iter().cloned().collect())
84 }
86 #[cfg(feature = "git")]
87 pub fn changes(&self) -> bool {
88 self.0.contains(&StyleComponent::Changes)
89 }
91 pub fn grid(&self) -> bool {
92 self.0.contains(&StyleComponent::Grid)
93 }
95 pub fn rule(&self) -> bool {
96 self.0.contains(&StyleComponent::Rule)
97 }
99 pub fn header(&self) -> bool {
100 self.header_filename() || self.header_filesize()
101 }
103 pub fn header_filename(&self) -> bool {
104 self.0.contains(&StyleComponent::HeaderFilename)
105 }
107 pub fn header_filesize(&self) -> bool {
108 self.0.contains(&StyleComponent::HeaderFilesize)
109 }
111 pub fn numbers(&self) -> bool {
112 self.0.contains(&StyleComponent::LineNumbers)
113 }
115 pub fn snip(&self) -> bool {
116 self.0.contains(&StyleComponent::Snip)
117 }
119 pub fn plain(&self) -> bool {
120 self.0.iter().all(|c| c == &StyleComponent::Plain)
121 }