From 05210b9510b797f7738d5b2d51e8a6877f2d5283 Mon Sep 17 00:00:00 2001
From: Matan Kushner <hello@matchai.me>
Date: Fri, 4 Oct 2019 22:30:46 +0900
Subject: refactor: Go from Rust workspaces to a package with nested packages
 (#480)

---
 src/segment.rs | 66 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 src/segment.rs

(limited to 'src/segment.rs')

diff --git a/src/segment.rs b/src/segment.rs
new file mode 100644
index 000000000..8c0140ac1
--- /dev/null
+++ b/src/segment.rs
@@ -0,0 +1,66 @@
+use ansi_term::{ANSIString, Style};
+use std::fmt;
+
+/// A segment is a single configurable element in a module. This will usually
+/// contain a data point to provide context for the prompt's user
+/// (e.g. The version that software is running).
+pub struct Segment {
+    /// The segment's name, to be used in configuration and logging.
+    _name: String,
+
+    /// The segment's style. If None, will inherit the style of the module containing it.
+    style: Option<Style>,
+
+    /// The string value of the current segment.
+    value: String,
+}
+
+impl Segment {
+    /// Creates a new segment with default fields.
+    pub fn new(name: &str) -> Self {
+        Self {
+            _name: name.to_string(),
+            style: None,
+            value: "".to_string(),
+        }
+    }
+
+    /// Sets the style of the segment.
+    ///
+    /// Accepts either `Color` or `Style`.
+    pub fn set_style<T>(&mut self, style: T) -> &mut Self
+    where
+        T: Into<Style>,
+    {
+        self.style = Some(style.into());
+        self
+    }
+
+    /// Sets the value of the segment.
+    pub fn set_value<T>(&mut self, value: T) -> &mut Self
+    where
+        T: Into<String>,
+    {
+        self.value = value.into();
+        self
+    }
+
+    // Returns the ANSIString of the segment value, not including its prefix and suffix
+    pub fn ansi_string(&self) -> ANSIString {
+        match self.style {
+            Some(style) => style.paint(&self.value),
+            None => ANSIString::from(&self.value),
+        }
+    }
+
+    /// Determines if the segment contains a value.
+    pub fn is_empty(&self) -> bool {
+        self.value.trim().is_empty()
+    }
+}
+
+impl fmt::Display for Segment {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        write!(f, "{}", self.ansi_string())
+    }
+}
-- 
cgit v1.2.3