bastd.ui.gather.nearbytab

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""Defines the nearby tab in the gather UI."""
  4
  5from __future__ import annotations
  6
  7import weakref
  8from typing import TYPE_CHECKING
  9
 10import ba
 11import ba.internal
 12from bastd.ui.gather import GatherTab
 13
 14if TYPE_CHECKING:
 15    from typing import Any
 16    from bastd.ui.gather import GatherWindow
 17
 18
 19class NetScanner:
 20    """Class for scanning for games on the lan."""
 21
 22    def __init__(
 23        self,
 24        tab: GatherTab,
 25        scrollwidget: ba.Widget,
 26        tab_button: ba.Widget,
 27        width: float,
 28    ):
 29        self._tab = weakref.ref(tab)
 30        self._scrollwidget = scrollwidget
 31        self._tab_button = tab_button
 32        self._columnwidget = ba.columnwidget(
 33            parent=self._scrollwidget, border=2, margin=0, left_border=10
 34        )
 35        ba.widget(edit=self._columnwidget, up_widget=tab_button)
 36        self._width = width
 37        self._last_selected_host: dict[str, Any] | None = None
 38
 39        self._update_timer = ba.Timer(
 40            1.0,
 41            ba.WeakCall(self.update),
 42            timetype=ba.TimeType.REAL,
 43            repeat=True,
 44        )
 45        # Go ahead and run a few *almost* immediately so we don't
 46        # have to wait a second.
 47        self.update()
 48        ba.timer(0.25, ba.WeakCall(self.update), timetype=ba.TimeType.REAL)
 49
 50    def __del__(self) -> None:
 51        ba.internal.end_host_scanning()
 52
 53    def _on_select(self, host: dict[str, Any]) -> None:
 54        self._last_selected_host = host
 55
 56    def _on_activate(self, host: dict[str, Any]) -> None:
 57        ba.internal.connect_to_party(host['address'])
 58
 59    def update(self) -> None:
 60        """(internal)"""
 61
 62        # In case our UI was killed from under us.
 63        if not self._columnwidget:
 64            print(
 65                f'ERROR: NetScanner running without UI at time'
 66                f' {ba.time(timetype=ba.TimeType.REAL)}.'
 67            )
 68            return
 69
 70        t_scale = 1.6
 71        for child in self._columnwidget.get_children():
 72            child.delete()
 73
 74        # Grab this now this since adding widgets will change it.
 75        last_selected_host = self._last_selected_host
 76        hosts = ba.internal.host_scan_cycle()
 77        for i, host in enumerate(hosts):
 78            txt3 = ba.textwidget(
 79                parent=self._columnwidget,
 80                size=(self._width / t_scale, 30),
 81                selectable=True,
 82                color=(1, 1, 1),
 83                on_select_call=ba.Call(self._on_select, host),
 84                on_activate_call=ba.Call(self._on_activate, host),
 85                click_activate=True,
 86                text=host['display_string'],
 87                h_align='left',
 88                v_align='center',
 89                corner_scale=t_scale,
 90                maxwidth=(self._width / t_scale) * 0.93,
 91            )
 92            if host == last_selected_host:
 93                ba.containerwidget(
 94                    edit=self._columnwidget,
 95                    selected_child=txt3,
 96                    visible_child=txt3,
 97                )
 98            if i == 0:
 99                ba.widget(edit=txt3, up_widget=self._tab_button)
100
101
102class NearbyGatherTab(GatherTab):
103    """The nearby tab in the gather UI"""
104
105    def __init__(self, window: GatherWindow) -> None:
106        super().__init__(window)
107        self._net_scanner: NetScanner | None = None
108        self._container: ba.Widget | None = None
109
110    def on_activate(
111        self,
112        parent_widget: ba.Widget,
113        tab_button: ba.Widget,
114        region_width: float,
115        region_height: float,
116        region_left: float,
117        region_bottom: float,
118    ) -> ba.Widget:
119        c_width = region_width
120        c_height = region_height - 20
121        sub_scroll_height = c_height - 85
122        sub_scroll_width = 650
123        self._container = ba.containerwidget(
124            parent=parent_widget,
125            position=(
126                region_left,
127                region_bottom + (region_height - c_height) * 0.5,
128            ),
129            size=(c_width, c_height),
130            background=False,
131            selection_loops_to_parent=True,
132        )
133        v = c_height - 30
134        ba.textwidget(
135            parent=self._container,
136            position=(c_width * 0.5, v - 3),
137            color=(0.6, 1.0, 0.6),
138            scale=1.3,
139            size=(0, 0),
140            maxwidth=c_width * 0.9,
141            h_align='center',
142            v_align='center',
143            text=ba.Lstr(
144                resource='gatherWindow.' 'localNetworkDescriptionText'
145            ),
146        )
147        v -= 15
148        v -= sub_scroll_height + 23
149        scrollw = ba.scrollwidget(
150            parent=self._container,
151            position=((region_width - sub_scroll_width) * 0.5, v),
152            size=(sub_scroll_width, sub_scroll_height),
153        )
154
155        self._net_scanner = NetScanner(
156            self, scrollw, tab_button, width=sub_scroll_width
157        )
158
159        ba.widget(edit=scrollw, autoselect=True, up_widget=tab_button)
160        return self._container
161
162    def on_deactivate(self) -> None:
163        self._net_scanner = None