From bf3404043cea7b4891a8da2768de1f56e8742b37 Mon Sep 17 00:00:00 2001
From: Andreas Steding <steding.andreas@gmail.com>
Date: Tue, 23 Feb 2021 20:41:15 +0100
Subject: [PATCH] implemented title alignment

---
 src/widgets/block.rs | 30 +++++++++++++++++++++++++++++-
 1 file changed, 29 insertions(+), 1 deletion(-)

diff --git a/src/widgets/block.rs b/src/widgets/block.rs
index 70610134..05ad60d4 100644
--- a/src/widgets/block.rs
+++ b/src/widgets/block.rs
@@ -26,6 +26,13 @@ impl BorderType {
     }
 }
 
+#[derive(Debug, Clone, Copy, PartialEq)]
+pub enum TitleAlignment {
+    Left,
+    Center,
+    Right,
+}
+
 /// Base widget to be used with all upper level ones. It may be used to display a box border around
 /// the widget and/or add a title.
 ///
@@ -54,6 +61,8 @@ pub struct Block<'a> {
     border_type: BorderType,
     /// Widget style
     style: Style,
+    /// Title alignment
+    title_alignment: TitleAlignment
 }
 
 impl<'a> Default for Block<'a> {
@@ -64,6 +73,7 @@ impl<'a> Default for Block<'a> {
             border_style: Default::default(),
             border_type: BorderType::Plain,
             style: Default::default(),
+            title_alignment: TitleAlignment::Left,
         }
     }
 }
@@ -108,6 +118,11 @@ impl<'a> Block<'a> {
         self.border_type = border_type;
         self
     }
+    
+    pub fn title_alignment(mut self, title_alignment: TitleAlignment) -> Block<'a> {
+        self.title_alignment = title_alignment;
+        self
+    }
 
     /// Compute the inner area of a block based on its border visibility rules.
     pub fn inner(&self, area: Rect) -> Rect {
@@ -192,6 +207,7 @@ impl<'a> Widget for Block<'a> {
                 .set_style(self.border_style);
         }
 
+        // Title
         if let Some(title) = self.title {
             let lx = if self.borders.intersects(Borders::LEFT) {
                 1
@@ -204,7 +220,19 @@ impl<'a> Widget for Block<'a> {
                 0
             };
             let width = area.width.saturating_sub(lx).saturating_sub(rx);
-            buf.set_spans(area.left() + lx, area.top(), &title, width);
+            
+            // Title alignment
+            match self.title_alignment {
+                TitleAlignment::Left => {
+                    buf.set_spans(area.left() + lx, area.top(), &title, width);
+                }
+                TitleAlignment::Center => {
+                    buf.set_spans(area.left() + width/2 - (title.width() as u16)/2 + lx, area.top(), &title, width);
+                }
+                TitleAlignment::Right => {
+                    buf.set_spans(area.right() - title.width() as u16 - rx, area.top(), &title, width);
+                }
+            }
         }
     }
 }