Fix center align in header.

1 files changed, 8 insertions, 3 deletions

diff --git a/text_utils.py b/text_utils.py
index afe0f63..a6337d6 100644
--- a/text_utils.py
+++ b/text_utils.py
@@ -336,16 +336,21 @@ def header(
         width = get_console_rows_columns().columns
     if not align:
         align = 'left'
+    if not style:
+        style = 'ascii'
 
+    text_len = len(string_utils.strip_ansi_sequences(title))
     if align == 'left':
         left = 4
-        right = width - (left + len(string_utils.strip_ansi_sequences(title)) + 4)
+        right = width - (left + text_len + 4)
     elif align == 'right':
         right = 4
-        left = width - (right + len(string_utils.strip_ansi_sequences(title)) + 4)
+        left = width - (right + text_len + 4)
     else:
-        left = int((width - (len(string_utils.strip_ansi_sequences(title)) + 4)) / 2)
+        left = int((width - (text_len + 4)) / 2)
         right = left
+        while left + text_len + 4 + right < width:
+            right += 1
 
     if style == 'solid':
         line_char = '━'