Input should release mouse on screen change

Author: willmcgugan

src/textual/app.py

@@ -2736,6 +2736,7 @@ def push_screen(
         else:
             future = loop.create_future()
 
+        self.app.capture_mouse(None)
         if self._screen_stack:
             self.screen.post_message(events.ScreenSuspend())
             self.screen.refresh()
@@ -2804,6 +2805,7 @@ def switch_screen(self, screen: Screen | str) -> AwaitComplete:
             self.log.system(f"Screen {screen} is already current.")
             return AwaitComplete.nothing()
 
+        self.app.capture_mouse(None)
         top_screen = self._screen_stack.pop()
 
         top_screen._pop_result_callback()

src/textual/screen.py

@@ -1607,6 +1607,7 @@ def _forward_event(self, event: events.Event) -> None:
             except errors.NoWidget:
                 self.set_focus(None)
             else:
+
                 if isinstance(event, events.MouseDown):
                     focusable_widget = self.get_focusable_widget_at(event.x, event.y)
                     if focusable_widget:

src/textual/widget.py

@@ -4341,7 +4341,8 @@ def release_mouse(self) -> None:
 
         Mouse events will only be sent when the mouse is over the widget.
         """
-        self.app.capture_mouse(None)
+        if self.app.mouse_captured is self:
+            self.app.capture_mouse(None)
 
     def text_select_all(self) -> None:
         """Select the entire widget."""

src/textual/widgets/_input.py

@@ -769,12 +769,18 @@ async def _on_mouse_down(self, event: events.MouseDown) -> None:
         self._selecting = True
         self.capture_mouse()
 
-    async def _on_mouse_up(self, event: events.MouseUp) -> None:
+    def _end_selecting(self):
         if self._selecting:
             self._selecting = False
             self.release_mouse()
             self._restart_blink()
 
+    async def _on_mouse_release(self, _event: events.MouseRelease) -> None:
+        self._end_selecting()
+
+    async def _on_mouse_up(self, _event: events.MouseUp) -> None:
+        self._end_selecting()
+
     async def _on_mouse_move(self, event: events.MouseMove) -> None:
         if self._selecting:
             # As we drag the mouse, we update the end position of the selection,