add paused state to progress items in TUI

Author: Nejc Stebe

tui/components.py

@@ -135,10 +135,12 @@ class ProgressItem(Vertical):
     PROCESSING = "PROCESSING"
     COMPLETED = "COMPLETED"
     STOPPED = "STOPPED"
+    PAUSED = "PAUSED"
 
     def __init__(self, initial_text: str, **kwargs):
         super().__init__(**kwargs)
         self.initial_text = initial_text
+        self.current_status = self.PENDING
 
     def compose(self):
         # Main row with status and description
@@ -156,11 +158,14 @@ def _get_status_text(self, status: str) -> str:
             return "◉ processing"
         elif status == self.STOPPED:
             return "◼ stopped"
+        elif status == self.PAUSED:
+            return "⏸ paused"
         else:
             return "○ pending"
 
     async def update_status(self, status: str):
         # TODO: Move to plain2code_tui.py
+        self.current_status = status
         try:
             # Get the main row container
             main_row = self.query_one(f"#{self.id}-main-row", Horizontal)

tui/plain2code_tui.py

@@ -1,3 +1,4 @@
+import threading
 from typing import Callable, Optional
 
 from textual.app import App, ComposeResult
@@ -18,7 +19,7 @@
     RenderStateUpdated,
 )
 from render_machine.states import States
-from tui.widget_helpers import log_to_widget
+from tui.widget_helpers import log_to_widget, set_frid_progress_to_paused
 
 from .components import (
     CustomFooter,
@@ -120,7 +121,15 @@ def get_active_script_types(self) -> list[ScriptOutputType]:
 
     def on_mount(self) -> None:
         """Called when the app is mounted."""
-        self.event_bus.register_dispatch_wrapper(self.call_from_thread)
+        main_thread_id = threading.get_ident()
+
+        def dispatch_wrapper(fn):
+            if threading.get_ident() == main_thread_id:
+                fn()
+            else:
+                self.call_from_thread(fn)
+
+        self.event_bus.register_dispatch_wrapper(dispatch_wrapper)
 
         self.event_bus.subscribe(RenderStateUpdated, self.on_render_state_updated)
         self.event_bus.subscribe(RenderCompleted, self.on_render_completed)
@@ -275,6 +284,7 @@ async def action_copy_selection(self) -> None:
     def action_pause(self) -> None:
         """Handle ctrl+shift+p: request the render machine to pause."""
         if not self._render_finished:
+            set_frid_progress_to_paused(self)
             self.event_bus.publish(PauseRequested())
 
     def action_enter_exit(self) -> None:

tui/styles.css

@@ -194,6 +194,11 @@ TestScriptsContainer {
   background: #c77777;
 }
 
+.status.PAUSED {
+  color: #fff;
+  background: #5b8bf5;
+}
+
 .description {
   margin-left: 1;
 }

tui/widget_helpers.py

@@ -124,6 +124,23 @@ def set_frid_progress_to_stopped(tui):
         clear_progress_item_substates(tui, widget_id)
 
 
+def set_frid_progress_to_paused(tui):
+    progress_ids = [
+        TUIComponents.FRID_PROGRESS_RENDER_FR.value,
+        TUIComponents.FRID_PROGRESS_UNIT_TEST.value,
+        TUIComponents.FRID_PROGRESS_REFACTORING.value,
+        TUIComponents.FRID_PROGRESS_CONFORMANCE_TEST.value,
+    ]
+
+    for widget_id in progress_ids:
+        try:
+            widget = tui.query_one(f"#{widget_id}", ProgressItem)
+            if widget.current_status == ProgressItem.PROCESSING:
+                update_progress_item_status(tui, widget_id, ProgressItem.PAUSED)
+        except Exception:
+            pass
+
+
 def display_error_message(tui, error_message: str):
     widget: Static = tui.query_one(f"#{TUIComponents.RENDER_STATUS_WIDGET.value}", Static)
     widget.add_class("error")