From 9d1ed9e9456a11f527a7a7dd15633db231640ea9 Mon Sep 17 00:00:00 2001
From: EqualMa <equalma@outlook.com>
Date: Thu, 9 Sep 2021 01:38:08 +0800
Subject: [PATCH] feat(lib): label

---
 runcc/src/label.rs | 48 +++++++++++++++++++++++++++++++++-------------
 1 file changed, 35 insertions(+), 13 deletions(-)

diff --git a/runcc/src/label.rs b/runcc/src/label.rs
index cc36c0d..b5781a2 100644
--- a/runcc/src/label.rs
+++ b/runcc/src/label.rs
@@ -1,15 +1,37 @@
-pub fn display_label(label: &str, max_label_length: usize) -> String {
-    let len = label.len();
-    if len > max_label_length {
-        let len_trim = std::cmp::min(len - max_label_length, 3);
-
-        let label = &label[0..(max_label_length - len_trim)];
-        let padding = ".".repeat(len_trim);
-        format!("{}{}", label, padding)
-    } else if len < max_label_length {
-        let padding = " ".repeat(max_label_length - len);
-        format!("{}{}", label, padding)
-    } else {
-        label.to_string()
+#[derive(Debug, Clone)]
+pub struct Label {
+    label: String,
+    display: Option<String>,
+}
+
+impl Label {
+    pub fn new(label: String, display: Option<String>) -> Self {
+        Self { label, display }
+    }
+
+    pub fn label(&self) -> &str {
+        &self.label
+    }
+
+    pub fn display(&self) -> &str {
+        self.display.as_ref().unwrap_or(&self.label)
+    }
+
+    pub fn from_label(label: String, max_label_length: usize) -> Self {
+        let len = label.len();
+        let display = if len > max_label_length {
+            let len_trim = std::cmp::min(len - max_label_length, 3);
+
+            let label = &label[0..(max_label_length - len_trim)];
+            let padding = ".".repeat(len_trim);
+            Some(format!("{}{}", label, padding))
+        } else if len < max_label_length {
+            let padding = " ".repeat(max_label_length - len);
+            Some(format!("{}{}", label, padding))
+        } else {
+            None
+        };
+
+        Self::new(label, display)
     }
 }