wizlib 3.3.9__py3-none-any.whl → 3.3.10__py3-none-any.whl

wizlib/ui/shell_ui.py

@@ -4,6 +4,7 @@ import sys
 from wizlib.ui import UI, Chooser, Emphasis
 from wizlib.ui.shell.line_editor import ShellLineEditor
 from wizlib.ui.shell import S
+from wizlib.ui.text_wrapper import StreamingTextWrapper
 import wizlib.io
 
 COLOR = {
@@ -23,12 +24,35 @@ class ShellUI(UI):
 
     name = "shell"
 
+    def __init__(self):
+        super().__init__()
+        self._wrapper = None
+
     def send(self, value: str = '', emphasis: Emphasis = Emphasis.GENERAL,
-             newline: bool = True):
-        """Output some text"""
-        end = '\n' if newline else ''
-        sys.stderr.write(COLOR[emphasis] + value + S.RESET + end)
-        sys.stderr.flush()
+             newline: bool = True, wrap: int = 0):
+        """Output some text
+
+        Args:
+            value: Text to output
+            emphasis: Color/emphasis style
+            newline: Whether to append a newline
+            wrap: Column width to wrap at (0 = no wrapping)
+        """
+        if wrap > 0:
+            # Initialize wrapper if needed or if width changed
+            if self._wrapper is None or self._wrapper.width != wrap:
+                self._wrapper = StreamingTextWrapper(
+                    width=wrap, output_stream=sys.stderr)
+
+            # Use streaming wrapper
+            self._wrapper.write_streaming(value, COLOR[emphasis])
+            if newline:
+                self._wrapper.write_newline()
+        else:
+            # Original behavior - no wrapping
+            end = '\n' if newline else ''
+            sys.stderr.write(COLOR[emphasis] + value + S.RESET + end)
+            sys.stderr.flush()
 
     def ask(self, value: str):
         """Prompt for input"""

wizlib/ui/text_wrapper.py

@@ -0,0 +1,103 @@
+"""
+Streaming text wrapper that provides word wrapping with backspace correction
+for a typewriter-like streaming experience.
+"""
+
+import sys
+from enum import Enum
+from wizlib.ui import Emphasis
+
+
+class StreamingTextWrapper:
+    """Handles streaming text output with word wrapping that uses backspace
+    to correct words that would overflow the line boundary.
+
+    Provides a typewriter-like streaming experience where characters are
+    written immediately, but words are moved to the next line if they
+    would overflow.
+    """
+
+    def __init__(self, width: int = 80, output_stream=None):
+        """Initialize the wrapper.
+
+        Args:
+            width: Column width to wrap at
+            output_stream: Stream to write to (defaults to sys.stderr)
+        """
+        self.width = width
+        self.output_stream = output_stream or sys.stderr
+        self._current_col = 0
+        self._word_buffer = []  # Buffer for current word being streamed
+
+    def _write_char(self, char: str, color_code: str = ''):
+        """Write a single character with optional color."""
+        if color_code:
+            self.output_stream.write(color_code + char + '\033[0m')
+        else:
+            self.output_stream.write(char)
+        self.output_stream.flush()
+
+    def _backspace_buffer(self):
+        """Backspace over the current word buffer."""
+        for _ in range(len(self._word_buffer)):
+            self.output_stream.write('\b \b')  # backspace, space, backspace
+        self.output_stream.flush()
+
+    def write_streaming(self, text: str, color_code: str = ''):
+        """Write text with streaming word wrapping.
+
+        Args:
+            text: Text to output
+            color_code: ANSI color code to apply
+        """
+        for char in text:
+            if char == '\n':
+                # Explicit newline - flush buffer and reset
+                self._word_buffer = []
+                self._write_char(char)
+                self._current_col = 0
+            elif char in ' \t':
+                # Word boundary - write the space and clear buffer
+                self._word_buffer = []
+                self._write_char(char, color_code)
+                self._current_col += 1
+            else:
+                # Regular character - add to buffer and write immediately
+                self._word_buffer.append(char)
+                self._write_char(char, color_code)
+                self._current_col += 1
+
+                # Check if this character pushed us over the width
+                if self._current_col > self.width:
+                    # Only backspace and wrap if the word can fit on a new line
+                    # If the word itself is longer than width, let it continue
+                    if len(self._word_buffer) <= self.width:
+                        # We've gone over - backspace the whole current word
+                        self._backspace_buffer()
+                        self._current_col -= len(self._word_buffer)
+
+                        # Write newline
+                        self._write_char('\n')
+                        self._current_col = 0
+
+                        # Rewrite word on new line
+                        for c in self._word_buffer:
+                            self._write_char(c, color_code)
+                            self._current_col += 1
+
+    def write_newline(self):
+        """Write a newline and reset position."""
+        self.output_stream.write('\n')
+        self.output_stream.flush()
+        self._current_col = 0
+        self._word_buffer = []
+
+    def reset_position(self):
+        """Reset the column position (useful for explicit positioning)."""
+        self._current_col = 0
+        self._word_buffer = []
+
+    @property
+    def current_column(self):
+        """Get the current column position."""
+        return self._current_col

{wizlib-3.3.9.dist-info → wizlib-3.3.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizlib
-Version: 3.3.9
+Version: 3.3.10
 Summary: Framework for flexible and powerful command-line applications
 License: MIT
 Author: Steampunk Wizard

{wizlib-3.3.9.dist-info → wizlib-3.3.10.dist-info}/RECORD

@@ -13,8 +13,9 @@ wizlib/test_case.py,sha256=T5cX7yWy94hf8JuherDeBCH9Gve5tx2x_PZOte4udC0,1017
 wizlib/ui/__init__.py,sha256=ve_p_g4aBujh4jIJMgKkJ6cE5PT0aeY5AgRlneDswGg,4241
 wizlib/ui/shell/__init__.py,sha256=VFtxcvTh5Mhyx6JHbR1C8yIDHlOsrNjdqSVfH5ZPAtQ,1154
 wizlib/ui/shell/line_editor.py,sha256=OXOgHBhp7EfheZK1mdOykYqB_aXDn2bHA12dvHN9kFg,8087
-wizlib/ui/shell_ui.py,sha256=2xrqR_hHuusOyUcXm7mF-Y-POkde-Pj-QIKOAWbC6A4,1983
+wizlib/ui/shell_ui.py,sha256=zsqmfL5LTnH_yqfqHUmBKhkZdb5cjRWL96vmfTrAn3U,2855
+wizlib/ui/text_wrapper.py,sha256=WYL59E9cMRv0gI2Yhrc5GJTvstRZO86uYUD7lmDiwjM,3812
 wizlib/ui_handler.py,sha256=JoZadtw9DKAtGvHKP3_BJF2NaYqmcQYNdsY4PeRnOjg,634
-wizlib-3.3.9.dist-info/METADATA,sha256=3VbJ_TIrxwiDPAKgGOKRG36MPVOodx5dU2rD_w4YyNc,995
-wizlib-3.3.9.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-wizlib-3.3.9.dist-info/RECORD,,
+wizlib-3.3.10.dist-info/METADATA,sha256=s7L0Q0x7p3Wi0-hTGlD0z2cz0XqcyAbe14YNzErmUJs,996
+wizlib-3.3.10.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+wizlib-3.3.10.dist-info/RECORD,,