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