Redraw progress bar on resize

Author: kovidgoyal

kittens/transfer/receive.py

@@ -18,7 +18,7 @@
     IdentityDecompressor, NameReprEnum, TransmissionType, ZlibDecompressor,
     encode_bypass, split_for_transfer
 )
-from kitty.typing import KeyEventType
+from kitty.typing import KeyEventType, ScreenSize
 from kitty.utils import sanitize_control_codes
 
 from ..tui.handler import Handler
@@ -619,6 +619,11 @@ def on_writing_finished(self) -> None:
         elif self.transmit_iterator is not None:
             self.transmit_one()
 
+    def on_resize(self, screen_size: ScreenSize) -> None:
+        super().on_resize(screen_size)
+        if self.progress_drawn:
+            self.refresh_progress()
+
 
 def receive_main(cli_opts: TransferCLIOptions, args: List[str]) -> None:
     dest = ''

kittens/transfer/send.py

@@ -20,7 +20,7 @@
     Action, Compression, FileTransmissionCommand, FileType, NameReprEnum,
     TransmissionType, encode_bypass, split_for_transfer
 )
-from kitty.typing import KeyEventType
+from kitty.typing import KeyEventType, ScreenSize
 from kitty.utils import sanitize_control_codes
 
 from ..tui.handler import Handler
@@ -703,6 +703,11 @@ def draw_progress(self) -> None:
         self.schedule_progress_update(self.spinner.interval)
         self.progress_drawn = True
 
+    def on_resize(self, screen_size: ScreenSize) -> None:
+        super().on_resize(screen_size)
+        if self.progress_drawn:
+            self.refresh_progress()
+
     def refresh_progress(self) -> None:
         if not self.transmit_started:
             return