added get_style_at and get_widget_at

Author: willmcgugan

src/textual/app.py

@@ -18,7 +18,7 @@
 from . import events
 from . import actions
 from ._animator import Animator
-from .geometry import Point
+from .geometry import Point, Region
 from ._context import active_app
 from .keys import Binding
 from .driver import Driver
@@ -258,6 +258,9 @@ def display(self, renderable: RenderableType) -> None:
             except Exception:
                 log.exception("display failed")
 
+    def get_widget_at(self, x: int, y: int) -> tuple[Widget, Region]:
+        return self.view.get_widget_at(x, y)
+
     async def on_event(self, event: events.Event) -> None:
         if isinstance(event, events.Key):
             binding = self._bindings.get(event.key, None)

src/textual/layout.py

@@ -18,7 +18,7 @@
 from ._profile import timer
 from ._types import Lines
 
-from .geometry import clamp, Region
+from .geometry import clamp, Region, Point
 
 log = logging.getLogger("rich")
 
@@ -85,12 +85,34 @@ def __iter__(self) -> Iterable[tuple[Widget, Region]]:
         for widget, (region, _) in layers:
             yield widget, region
 
+    def get_offset(self, widget: Widget) -> Point:
+        try:
+            return self._layout_map[widget].region.origin
+        except KeyError:
+            raise NoWidget("Widget is not in layout")
+
     def get_widget_at(self, x: int, y: int) -> tuple[Widget, Region]:
         """Get the widget under the given point or None."""
         for widget, region in self:
             if region.contains(x, y):
                 return widget, region
-        raise NoWidget
+        raise NoWidget(f"No widget under screen coordinate ({x}, {y})")
+
+    def get_style_at(self, x: int, y: int) -> Style:
+        try:
+            widget, region = self.get_widget_at(x, y)
+        except NoWidget:
+            return Style.null()
+        _region, lines = self.renders[widget]
+        x -= region.x
+        y -= region.y
+        line = lines[y]
+        end = 0
+        for segment in line:
+            end += segment.cell_length
+            if x < end:
+                return segment.style or Style.null()
+        return Style.null()
 
     @property
     def cuts(self) -> list[list[int]]:

src/textual/view.py

@@ -10,12 +10,13 @@
 from rich.region import Region as LayoutRegion
 import rich.repr
 from rich.segment import Segments
+from rich.style import Style
 
 from . import events
 from ._context import active_app
 from .layout import Layout, NoWidget
 from .layouts.dock import DockEdge, DockLayout, Dock
-from .geometry import Dimensions, Region
+from .geometry import Dimensions, Point, Region
 from .messages import UpdateMessage, LayoutMessage
 
 from .widget import StaticWidget, Widget, WidgetID, Widget
@@ -57,6 +58,9 @@ def is_mounted(self, widget: Widget) -> bool:
     def render(self) -> RenderableType:
         return self.layout
 
+    def get_offset(self, widget: Widget) -> Point:
+        return self.layout.get_offset(widget)
+
     async def message_update(self, message: UpdateMessage) -> None:
         widget = message.widget
         assert isinstance(widget, Widget)
@@ -109,9 +113,15 @@ async def on_resize(self, event: events.Resize) -> None:
         self.size = Dimensions(event.width, event.height)
         await self.refresh_layout()
 
+    def get_widget_at(self, x: int, y: int) -> tuple[Widget, Region]:
+        return self.layout.get_widget_at(x, y)
+
+    def get_style_at(self, x: int, y: int) -> Style:
+        return self.layout.get_style_at(x, y)
+
     async def _on_mouse_move(self, event: events.MouseMove) -> None:
         try:
-            widget, region = self.layout.get_widget_at(event.x, event.y)
+            widget, region = self.get_widget_at(event.x, event.y)
         except NoWidget:
             await self.app.set_mouse_over(None)
         else:
@@ -140,26 +150,24 @@ async def forward_event(self, event: events.Event) -> None:
             await self._on_mouse_move(event)
 
         elif isinstance(event, events.MouseEvent):
-            pass
-            # try:
-            #     widget, region = self.get_widget_at(event.x, event.y, deep=True)
-            # except NoWidget:
-            #     if isinstance(event, events.MouseDown):
-            #         await self.app.set_focus(None)
-            # else:
-            #     if isinstance(event, events.MouseDown):
-            #         await self.app.set_focus(widget)
-            #     await widget.forward_event(event.offset(-region.x, -region.y))
+
+            try:
+                widget, region = self.get_widget_at(event.x, event.y)
+            except NoWidget:
+                pass
+            else:
+                if isinstance(event, events.MouseDown) and widget.can_focus:
+                    await self.app.set_focus(widget)
+                await widget.forward_event(event.offset(-region.x, -region.y))
 
         elif isinstance(event, (events.MouseScrollDown, events.MouseScrollUp)):
-            pass
-            # try:
-            #     widget, _region = self.get_widget_at(event.x, event.y, deep=True)
-            # except NoWidget:
-            #     return
-            # scroll_widget = widget or self.focused
-            # if scroll_widget is not None:
-            #     await scroll_widget.forward_event(event)
+            try:
+                widget, _region = self.get_widget_at(event.x, event.y)
+            except NoWidget:
+                return
+            scroll_widget = widget or self.focused
+            if scroll_widget is not None:
+                await scroll_widget.forward_event(event)
         else:
             if self.focused is not None:
                 await self.focused.forward_event(event)

src/textual/widget.py

@@ -177,6 +177,10 @@ def reset_check_repaint(self) -> None:
     def reset_check_layout(self) -> None:
         self._layout_required = False
 
+    def get_style_at(self, x: int, y: int) -> Style:
+        offset_x, offset_y = self.root_view.get_offset(self)
+        return self.root_view.get_style_at(x + offset_x, y + offset_y)
+
     async def forward_event(self, event: events.Event) -> None:
         await self.post_message(event)
 
@@ -235,14 +239,9 @@ async def focus(self) -> None:
     async def capture_mouse(self, capture: bool = True) -> None:
         await self.app.capture_mouse(self if capture else None)
 
-    def get_style_at(self, x: int, y: int) -> Style:
-        return
-        return self.line_cache.get_style_at(x, y)
-
-    # async def on_mouse_move(self, event: events.MouseMove) -> None:
-    #     style_under_cursor = self.get_style_at(event.x, event.y)
-    #     if style_under_cursor:
-    #         log.debug("%r", style_under_cursor)
+    async def on_mouse_move(self, event: events.MouseMove) -> None:
+        style_under_cursor = self.get_style_at(event.x, event.y)
+        log.debug("%r", style_under_cursor)
 
     # async def on_mouse_up(self, event: events.MouseUp) -> None:
     #     style = self.get_style_at(event.x, event.y)