From eaeeea2abbb2e56a014d7c0eca2d91fa367ac3ed Mon Sep 17 00:00:00 2001
From: Will McGugan <willmcgugan@gmail.com>
Date: Wed, 30 Oct 2024 16:12:51 +0000
Subject: [PATCH] fix justify center and right

---
 src/textual/content.py | 27 ++++++++++++++-------------
 1 file changed, 14 insertions(+), 13 deletions(-)

diff --git a/src/textual/content.py b/src/textual/content.py
index 71c26c8de4..3a7abbc413 100644
--- a/src/textual/content.py
+++ b/src/textual/content.py
@@ -55,20 +55,9 @@ def _justify_lines(
         lines = [line.truncate(width, overflow=overflow, pad=True) for line in lines]
 
     elif justify == "center":
-        lines = [
-            line.rstrip()
-            .truncate(width, overflow=overflow)
-            .pad_left((width - cell_len(line.plain)) // 2)
-            .pad_right(width - cell_len(line.plain))
-            for line in lines
-        ]
+        lines = [line.center(width) for line in lines]
     elif justify == "right":
-        lines = [
-            line.rstrip()
-            .truncate(width, overflow=overflow)
-            .pad_left(width - cell_len(line.plain))
-            for line in lines
-        ]
+        lines = [line.right(width) for line in lines]
 
     elif justify == "full":
         new_lines = lines.copy()
@@ -379,6 +368,17 @@ def pad_right(self, count: int, character: str = " ") -> Content:
             return Content(f"{self.plain}{character * count}", self._spans)
         return self
 
+    def center(self, width: int, overflow: OverflowMethod = "fold") -> Content:
+        content = self.rstrip().truncate(width, overflow=overflow)
+        content = content.pad_left((width - content.cell_length) // 2)
+        content = content.pad_right(width - content.cell_length)
+        return content
+
+    def right(self, width: int, overflow: OverflowMethod = "fold") -> Content:
+        content = self.rstrip().truncate(width, overflow=overflow)
+        content = content.pad_left(width - content.cell_length)
+        return content
+
     def right_crop(self, amount: int = 1) -> Content:
         """Remove a number of characters from the end of the text."""
         max_offset = len(self.plain) - amount
@@ -656,6 +656,7 @@ def highlight_regex(
     content = content.highlight_regex("F..r", Style(bold=True))
 
     lines = content.wrap(30, justify="left")
+    print("x" * 30)
     for line in lines:
         segments = Segments(line.render_segments(Style()))
         print(segments)