bauiv1lib.partyqueue

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""UI related to waiting in line for a party."""
  4
  5from __future__ import annotations
  6
  7import time
  8import random
  9import logging
 10from typing import TYPE_CHECKING
 11
 12import bauiv1 as bui
 13import bascenev1 as bs
 14
 15if TYPE_CHECKING:
 16    from typing import Any, Sequence
 17
 18
 19class PartyQueueWindow(bui.Window):
 20    """Window showing players waiting to join a server."""
 21
 22    # Ewww this needs quite a bit of de-linting if/when i revisit it..
 23    # pylint: disable=invalid-name
 24    # pylint: disable=consider-using-dict-comprehension
 25    class Dude:
 26        """Represents a single dude waiting in a server line."""
 27
 28        def __init__(
 29            self,
 30            parent: PartyQueueWindow,
 31            distance: float,
 32            initial_offset: float,
 33            is_player: bool,
 34            account_id: str,
 35            name: str,
 36        ):
 37            self.claimed = False
 38            self._line_left = parent.get_line_left()
 39            self._line_width = parent.get_line_width()
 40            self._line_bottom = parent.get_line_bottom()
 41            self._target_distance = distance
 42            self._distance = distance + initial_offset
 43            self._boost_brightness = 0.0
 44            self._debug = False
 45            self._sc = sc = 1.1 if is_player else 0.6 + random.random() * 0.2
 46            self._y_offs = -30.0 if is_player else -47.0 * sc
 47            self._last_boost_time = 0.0
 48            self._color = (
 49                (0.2, 1.0, 0.2)
 50                if is_player
 51                else (
 52                    0.5 + 0.3 * random.random(),
 53                    0.4 + 0.2 * random.random(),
 54                    0.5 + 0.3 * random.random(),
 55                )
 56            )
 57            self._eye_color = (
 58                0.7 * 1.0 + 0.3 * self._color[0],
 59                0.7 * 1.0 + 0.3 * self._color[1],
 60                0.7 * 1.0 + 0.3 * self._color[2],
 61            )
 62            self._body_image = bui.buttonwidget(
 63                parent=parent.get_root_widget(),
 64                selectable=True,
 65                label='',
 66                size=(sc * 60, sc * 80),
 67                color=self._color,
 68                texture=parent.lineup_tex,
 69                mesh_transparent=parent.lineup_1_transparent_mesh,
 70            )
 71            bui.buttonwidget(
 72                edit=self._body_image,
 73                on_activate_call=bui.WeakCall(
 74                    parent.on_account_press, account_id, self._body_image
 75                ),
 76            )
 77            bui.widget(edit=self._body_image, autoselect=True)
 78            self._eyes_image = bui.imagewidget(
 79                parent=parent.get_root_widget(),
 80                size=(sc * 36, sc * 18),
 81                texture=parent.lineup_tex,
 82                color=self._eye_color,
 83                mesh_transparent=parent.eyes_mesh,
 84            )
 85            self._name_text = bui.textwidget(
 86                parent=parent.get_root_widget(),
 87                size=(0, 0),
 88                shadow=0,
 89                flatness=1.0,
 90                text=name,
 91                maxwidth=100,
 92                h_align='center',
 93                v_align='center',
 94                scale=0.75,
 95                color=(1, 1, 1, 0.6),
 96            )
 97            self._update_image()
 98
 99            # DEBUG: vis target pos..
100            self._body_image_target: bui.Widget | None
101            self._eyes_image_target: bui.Widget | None
102            if self._debug:
103                self._body_image_target = bui.imagewidget(
104                    parent=parent.get_root_widget(),
105                    size=(sc * 60, sc * 80),
106                    color=self._color,
107                    texture=parent.lineup_tex,
108                    mesh_transparent=parent.lineup_1_transparent_mesh,
109                )
110                self._eyes_image_target = bui.imagewidget(
111                    parent=parent.get_root_widget(),
112                    size=(sc * 36, sc * 18),
113                    texture=parent.lineup_tex,
114                    color=self._eye_color,
115                    mesh_transparent=parent.eyes_mesh,
116                )
117                # (updates our image positions)
118                self.set_target_distance(self._target_distance)
119            else:
120                self._body_image_target = self._eyes_image_target = None
121
122        def __del__(self) -> None:
123            # ew.  our destructor here may get called as part of an internal
124            # widget tear-down.
125            # running further widget calls here can quietly break stuff, so we
126            # need to push a deferred call to kill these as necessary instead.
127            # (should bulletproof internal widget code to give a clean error
128            # in this case)
129            def kill_widgets(widgets: Sequence[bui.Widget | None]) -> None:
130                for widget in widgets:
131                    if widget:
132                        widget.delete()
133
134            bui.pushcall(
135                bui.Call(
136                    kill_widgets,
137                    [
138                        self._body_image,
139                        self._eyes_image,
140                        self._body_image_target,
141                        self._eyes_image_target,
142                        self._name_text,
143                    ],
144                )
145            )
146
147        def set_target_distance(self, dist: float) -> None:
148            """Set distance for a dude."""
149            self._target_distance = dist
150            if self._debug:
151                sc = self._sc
152                position = (
153                    self._line_left
154                    + self._line_width * (1.0 - self._target_distance),
155                    self._line_bottom - 30,
156                )
157                bui.imagewidget(
158                    edit=self._body_image_target,
159                    position=(
160                        position[0] - sc * 30,
161                        position[1] - sc * 25 - 70,
162                    ),
163                )
164                bui.imagewidget(
165                    edit=self._eyes_image_target,
166                    position=(
167                        position[0] - sc * 18,
168                        position[1] + sc * 31 - 70,
169                    ),
170                )
171
172        def step(self, smoothing: float) -> None:
173            """Step this dude."""
174            self._distance = (
175                smoothing * self._distance
176                + (1.0 - smoothing) * self._target_distance
177            )
178            self._update_image()
179            self._boost_brightness *= 0.9
180
181        def _update_image(self) -> None:
182            sc = self._sc
183            position = (
184                self._line_left + self._line_width * (1.0 - self._distance),
185                self._line_bottom + 40,
186            )
187            brightness = 1.0 + self._boost_brightness
188            bui.buttonwidget(
189                edit=self._body_image,
190                position=(
191                    position[0] - sc * 30,
192                    position[1] - sc * 25 + self._y_offs,
193                ),
194                color=(
195                    self._color[0] * brightness,
196                    self._color[1] * brightness,
197                    self._color[2] * brightness,
198                ),
199            )
200            bui.imagewidget(
201                edit=self._eyes_image,
202                position=(
203                    position[0] - sc * 18,
204                    position[1] + sc * 31 + self._y_offs,
205                ),
206                color=(
207                    self._eye_color[0] * brightness,
208                    self._eye_color[1] * brightness,
209                    self._eye_color[2] * brightness,
210                ),
211            )
212            bui.textwidget(
213                edit=self._name_text,
214                position=(position[0] - sc * 0, position[1] + sc * 40.0),
215            )
216
217        def boost(self, amount: float, smoothing: float) -> None:
218            """Boost this dude."""
219            del smoothing  # unused arg
220            self._distance = max(0.0, self._distance - amount)
221            self._update_image()
222            self._last_boost_time = time.time()
223            self._boost_brightness += 0.6
224
225    def __init__(self, queue_id: str, address: str, port: int):
226        assert bui.app.classic is not None
227        bui.app.ui_v1.have_party_queue_window = True
228        self._address = address
229        self._port = port
230        self._queue_id = queue_id
231        self._width = 800
232        self._height = 400
233        self._last_connect_attempt_time: float | None = None
234        self._last_transaction_time: float | None = None
235        self._boost_button: bui.Widget | None = None
236        self._boost_price: bui.Widget | None = None
237        self._boost_label: bui.Widget | None = None
238        self._field_shown = False
239        self._dudes: list[PartyQueueWindow.Dude] = []
240        self._dudes_by_id: dict[int, PartyQueueWindow.Dude] = {}
241        self._line_left = 40.0
242        self._line_width = self._width - 190
243        self._line_bottom = self._height * 0.4
244        self.lineup_tex: bui.Texture = bui.gettexture('playerLineup')
245        self._smoothing = 0.0
246        self._initial_offset = 0.0
247        self._boost_tickets = 0
248        self._boost_strength = 0.0
249        self._angry_computer_transparent_mesh = bui.getmesh(
250            'angryComputerTransparent'
251        )
252        self._angry_computer_image: bui.Widget | None = None
253        self.lineup_1_transparent_mesh: bui.Mesh = bui.getmesh(
254            'playerLineup1Transparent'
255        )
256        self._lineup_2_transparent_mesh: bui.Mesh = bui.getmesh(
257            'playerLineup2Transparent'
258        )
259
260        self._lineup_3_transparent_mesh = bui.getmesh(
261            'playerLineup3Transparent'
262        )
263        self._lineup_4_transparent_mesh = bui.getmesh(
264            'playerLineup4Transparent'
265        )
266        self._line_image: bui.Widget | None = None
267        self.eyes_mesh: bui.Mesh = bui.getmesh('plasticEyesTransparent')
268        self._white_tex = bui.gettexture('white')
269        uiscale = bui.app.ui_v1.uiscale
270        super().__init__(
271            root_widget=bui.containerwidget(
272                size=(self._width, self._height),
273                color=(0.45, 0.63, 0.15),
274                transition='in_scale',
275                scale=(
276                    1.4
277                    if uiscale is bui.UIScale.SMALL
278                    else 1.2
279                    if uiscale is bui.UIScale.MEDIUM
280                    else 1.0
281                ),
282            )
283        )
284
285        self._cancel_button = bui.buttonwidget(
286            parent=self._root_widget,
287            scale=1.0,
288            position=(60, self._height - 80),
289            size=(50, 50),
290            label='',
291            on_activate_call=self.close,
292            autoselect=True,
293            color=(0.45, 0.63, 0.15),
294            icon=bui.gettexture('crossOut'),
295            iconscale=1.2,
296        )
297        bui.containerwidget(
298            edit=self._root_widget, cancel_button=self._cancel_button
299        )
300
301        self._title_text = bui.textwidget(
302            parent=self._root_widget,
303            position=(self._width * 0.5, self._height * 0.55),
304            size=(0, 0),
305            color=(1.0, 3.0, 1.0),
306            scale=1.3,
307            h_align='center',
308            v_align='center',
309            text=bui.Lstr(resource='internal.connectingToPartyText'),
310            maxwidth=self._width * 0.65,
311        )
312
313        self._tickets_text = bui.textwidget(
314            parent=self._root_widget,
315            position=(self._width - 180, self._height - 20),
316            size=(0, 0),
317            color=(0.2, 1.0, 0.2),
318            scale=0.7,
319            h_align='center',
320            v_align='center',
321            text='',
322        )
323
324        # update at roughly 30fps
325        self._update_timer = bui.AppTimer(
326            0.033, bui.WeakCall(self.update), repeat=True
327        )
328        self.update()
329
330    def __del__(self) -> None:
331        try:
332            plus = bui.app.plus
333            assert plus is not None
334
335            assert bui.app.classic is not None
336            bui.app.ui_v1.have_party_queue_window = False
337            plus.add_v1_account_transaction(
338                {'type': 'PARTY_QUEUE_REMOVE', 'q': self._queue_id}
339            )
340            plus.run_v1_account_transactions()
341        except Exception:
342            logging.exception('Error removing self from party queue.')
343
344    def get_line_left(self) -> float:
345        """(internal)"""
346        return self._line_left
347
348    def get_line_width(self) -> float:
349        """(internal)"""
350        return self._line_width
351
352    def get_line_bottom(self) -> float:
353        """(internal)"""
354        return self._line_bottom
355
356    def on_account_press(
357        self, account_id: str | None, origin_widget: bui.Widget
358    ) -> None:
359        """A dude was clicked so we should show his account info."""
360        from bauiv1lib.account import viewer
361
362        if account_id is None:
363            bui.getsound('error').play()
364            return
365        viewer.AccountViewerWindow(
366            account_id=account_id,
367            position=origin_widget.get_screen_space_center(),
368        )
369
370    def close(self) -> None:
371        """Close the ui."""
372        bui.containerwidget(edit=self._root_widget, transition='out_scale')
373
374    def _update_field(self, response: dict[str, Any]) -> None:
375        plus = bui.app.plus
376        assert plus is not None
377
378        if self._angry_computer_image is None:
379            self._angry_computer_image = bui.imagewidget(
380                parent=self._root_widget,
381                position=(self._width - 180, self._height * 0.5 - 65),
382                size=(150, 150),
383                texture=self.lineup_tex,
384                mesh_transparent=self._angry_computer_transparent_mesh,
385            )
386        if self._line_image is None:
387            self._line_image = bui.imagewidget(
388                parent=self._root_widget,
389                color=(0.0, 0.0, 0.0),
390                opacity=0.2,
391                position=(self._line_left, self._line_bottom - 2.0),
392                size=(self._line_width, 4.0),
393                texture=self._white_tex,
394            )
395
396        # now go through the data they sent, creating dudes for us and our
397        # enemies as needed and updating target positions on all of them..
398
399        # mark all as unclaimed so we know which ones to kill off..
400        for dude in self._dudes:
401            dude.claimed = False
402
403        # always have a dude for ourself..
404        if -1 not in self._dudes_by_id:
405            dude = self.Dude(
406                self,
407                response['d'],
408                self._initial_offset,
409                True,
410                plus.get_v1_account_misc_read_val_2('resolvedAccountID', None),
411                plus.get_v1_account_display_string(),
412            )
413            self._dudes_by_id[-1] = dude
414            self._dudes.append(dude)
415        else:
416            self._dudes_by_id[-1].set_target_distance(response['d'])
417        self._dudes_by_id[-1].claimed = True
418
419        # now create/destroy enemies
420        for (
421            enemy_id,
422            enemy_distance,
423            enemy_account_id,
424            enemy_name,
425        ) in response['e']:
426            if enemy_id not in self._dudes_by_id:
427                dude = self.Dude(
428                    self,
429                    enemy_distance,
430                    self._initial_offset,
431                    False,
432                    enemy_account_id,
433                    enemy_name,
434                )
435                self._dudes_by_id[enemy_id] = dude
436                self._dudes.append(dude)
437            else:
438                self._dudes_by_id[enemy_id].set_target_distance(enemy_distance)
439            self._dudes_by_id[enemy_id].claimed = True
440
441        # remove unclaimed dudes from both of our lists
442        # noinspection PyUnresolvedReferences
443        self._dudes_by_id = dict(
444            [
445                item
446                for item in list(self._dudes_by_id.items())
447                if item[1].claimed
448            ]
449        )
450        self._dudes = [dude for dude in self._dudes if dude.claimed]
451
452    def _hide_field(self) -> None:
453        if self._angry_computer_image:
454            self._angry_computer_image.delete()
455        self._angry_computer_image = None
456        if self._line_image:
457            self._line_image.delete()
458        self._line_image = None
459        self._dudes = []
460        self._dudes_by_id = {}
461
462    def on_update_response(self, response: dict[str, Any] | None) -> None:
463        """We've received a response from an update to the server."""
464        # pylint: disable=too-many-branches
465        if not self._root_widget:
466            return
467
468        # Seeing this in logs; debugging...
469        if not self._title_text:
470            print('PartyQueueWindows update: Have root but no title_text.')
471            return
472
473        if response is not None:
474            should_show_field = response.get('d') is not None
475            self._smoothing = response['s']
476            self._initial_offset = response['o']
477
478            # If they gave us a position, show the field.
479            if should_show_field:
480                bui.textwidget(
481                    edit=self._title_text,
482                    text=bui.Lstr(resource='waitingInLineText'),
483                    position=(self._width * 0.5, self._height * 0.85),
484                )
485                self._update_field(response)
486                self._field_shown = True
487            if not should_show_field and self._field_shown:
488                bui.textwidget(
489                    edit=self._title_text,
490                    text=bui.Lstr(resource='internal.connectingToPartyText'),
491                    position=(self._width * 0.5, self._height * 0.55),
492                )
493                self._hide_field()
494                self._field_shown = False
495
496            # if they told us there's a boost button, update..
497            if response.get('bt') is not None:
498                self._boost_tickets = response['bt']
499                self._boost_strength = response['ba']
500                if self._boost_button is None:
501                    self._boost_button = bui.buttonwidget(
502                        parent=self._root_widget,
503                        scale=1.0,
504                        position=(self._width * 0.5 - 75, 20),
505                        size=(150, 100),
506                        button_type='square',
507                        label='',
508                        on_activate_call=self.on_boost_press,
509                        enable_sound=False,
510                        color=(0, 1, 0),
511                        autoselect=True,
512                    )
513                    self._boost_label = bui.textwidget(
514                        parent=self._root_widget,
515                        draw_controller=self._boost_button,
516                        position=(self._width * 0.5, 88),
517                        size=(0, 0),
518                        color=(0.8, 1.0, 0.8),
519                        scale=1.5,
520                        h_align='center',
521                        v_align='center',
522                        text=bui.Lstr(resource='boostText'),
523                        maxwidth=150,
524                    )
525                    self._boost_price = bui.textwidget(
526                        parent=self._root_widget,
527                        draw_controller=self._boost_button,
528                        position=(self._width * 0.5, 50),
529                        size=(0, 0),
530                        color=(0, 1, 0),
531                        scale=0.9,
532                        h_align='center',
533                        v_align='center',
534                        text=bui.charstr(bui.SpecialChar.TICKET)
535                        + str(self._boost_tickets),
536                        maxwidth=150,
537                    )
538            else:
539                if self._boost_button is not None:
540                    self._boost_button.delete()
541                    self._boost_button = None
542                if self._boost_price is not None:
543                    self._boost_price.delete()
544                    self._boost_price = None
545                if self._boost_label is not None:
546                    self._boost_label.delete()
547                    self._boost_label = None
548
549            # if they told us to go ahead and try and connect, do so..
550            # (note: servers will disconnect us if we try to connect before
551            # getting this go-ahead, so don't get any bright ideas...)
552            if response.get('c', False):
553                # enforce a delay between connection attempts
554                # (in case they're jamming on the boost button)
555                now = time.time()
556                if (
557                    self._last_connect_attempt_time is None
558                    or now - self._last_connect_attempt_time > 10.0
559                ):
560                    bs.connect_to_party(
561                        address=self._address,
562                        port=self._port,
563                        print_progress=False,
564                    )
565                    self._last_connect_attempt_time = now
566
567    def on_boost_press(self) -> None:
568        """Boost was pressed."""
569        from bauiv1lib import account
570        from bauiv1lib import getcurrency
571
572        plus = bui.app.plus
573        assert plus is not None
574
575        if plus.get_v1_account_state() != 'signed_in':
576            account.show_sign_in_prompt()
577            return
578
579        if plus.get_v1_account_ticket_count() < self._boost_tickets:
580            bui.getsound('error').play()
581            getcurrency.show_get_tickets_prompt()
582            return
583
584        bui.getsound('laserReverse').play()
585        plus.add_v1_account_transaction(
586            {
587                'type': 'PARTY_QUEUE_BOOST',
588                't': self._boost_tickets,
589                'q': self._queue_id,
590            },
591            callback=bui.WeakCall(self.on_update_response),
592        )
593        # lets not run these immediately (since they may be rapid-fire,
594        # just bucket them until the next tick)
595
596        # the transaction handles the local ticket change, but we apply our
597        # local boost vis manually here..
598        # (our visualization isn't really wired up to be transaction-based)
599        our_dude = self._dudes_by_id.get(-1)
600        if our_dude is not None:
601            our_dude.boost(self._boost_strength, self._smoothing)
602
603    def update(self) -> None:
604        """Update!"""
605        plus = bui.app.plus
606        assert plus is not None
607
608        if not self._root_widget:
609            return
610
611        # Update boost-price.
612        if self._boost_price is not None:
613            bui.textwidget(
614                edit=self._boost_price,
615                text=bui.charstr(bui.SpecialChar.TICKET)
616                + str(self._boost_tickets),
617            )
618
619        # Update boost button color based on if we have enough moola.
620        if self._boost_button is not None:
621            can_boost = (
622                plus.get_v1_account_state() == 'signed_in'
623                and plus.get_v1_account_ticket_count() >= self._boost_tickets
624            )
625            bui.buttonwidget(
626                edit=self._boost_button,
627                color=(0, 1, 0) if can_boost else (0.7, 0.7, 0.7),
628            )
629
630        # Update ticket-count.
631        if self._tickets_text is not None:
632            if self._boost_button is not None:
633                if plus.get_v1_account_state() == 'signed_in':
634                    val = bui.charstr(bui.SpecialChar.TICKET) + str(
635                        plus.get_v1_account_ticket_count()
636                    )
637                else:
638                    val = bui.charstr(bui.SpecialChar.TICKET) + '???'
639                bui.textwidget(edit=self._tickets_text, text=val)
640            else:
641                bui.textwidget(edit=self._tickets_text, text='')
642
643        current_time = bui.apptime()
644        if (
645            self._last_transaction_time is None
646            or current_time - self._last_transaction_time
647            > 0.001 * plus.get_v1_account_misc_read_val('pqInt', 5000)
648        ):
649            self._last_transaction_time = current_time
650            plus.add_v1_account_transaction(
651                {'type': 'PARTY_QUEUE_QUERY', 'q': self._queue_id},
652                callback=bui.WeakCall(self.on_update_response),
653            )
654            plus.run_v1_account_transactions()
655
656        # step our dudes
657        for dude in self._dudes:
658            dude.step(self._smoothing)