bauiv1lib.chest

   1# Released under the MIT License. See LICENSE for details.
   2#
   3# pylint: disable=too-many-lines
   4"""Provides chest related ui."""
   5
   6from __future__ import annotations
   7
   8import math
   9import random
  10from typing import override, TYPE_CHECKING
  11
  12from efro.util import strict_partial
  13import bacommon.bs
  14import bauiv1 as bui
  15
  16if TYPE_CHECKING:
  17    import datetime
  18
  19    import baclassic
  20
  21_g_open_voices: list[tuple[float, str, float]] = []
  22
  23
  24class ChestWindow(bui.MainWindow):
  25    """Allows viewing and performing operations on a chest."""
  26
  27    def __init__(
  28        self,
  29        index: int,
  30        transition: str | None = 'in_right',
  31        origin_widget: bui.Widget | None = None,
  32    ):
  33        self._index = index
  34
  35        assert bui.app.classic is not None
  36        uiscale = bui.app.ui_v1.uiscale
  37        self._width = 1200 if uiscale is bui.UIScale.SMALL else 650
  38        self._height = 800 if uiscale is bui.UIScale.SMALL else 450
  39        self._action_in_flight = False
  40        self._open_now_button: bui.Widget | None = None
  41        self._open_now_spinner: bui.Widget | None = None
  42        self._open_now_texts: list[bui.Widget] = []
  43        self._open_now_images: list[bui.Widget] = []
  44        self._watch_ad_button: bui.Widget | None = None
  45        self._time_string_timer: bui.AppTimer | None = None
  46        self._time_string_text: bui.Widget | None = None
  47        self._prizesets: list[bacommon.bs.ChestInfoResponse.Chest.PrizeSet] = []
  48        self._prizeindex = -1
  49        self._prizesettxts: dict[int, list[bui.Widget]] = {}
  50        self._prizesetimgs: dict[int, list[bui.Widget]] = {}
  51        self._chestdisplayinfo: baclassic.ChestAppearanceDisplayInfo | None = (
  52            None
  53        )
  54
  55        # The set of widgets we keep when doing a clear.
  56        self._core_widgets: list[bui.Widget] = []
  57
  58        # Do some fancy math to fill all available screen area up to the
  59        # size of our backing container. This lets us fit to the exact
  60        # screen shape at small ui scale.
  61        screensize = bui.get_virtual_screen_size()
  62        scale = (
  63            1.8
  64            if uiscale is bui.UIScale.SMALL
  65            else 1.1 if uiscale is bui.UIScale.MEDIUM else 0.9
  66        )
  67        # Calc screen size in our local container space and clamp to a
  68        # bit smaller than our container size.
  69        target_height = min(self._height - 120, screensize[1] / scale)
  70
  71        # To get top/left coords, go to the center of our window and
  72        # offset by half the width/height of our target area.
  73        self._yoffstop = 0.5 * self._height + 0.5 * target_height + 18
  74
  75        # Offset for stuff we want centered.
  76        self._yoffs = 0.5 * self._height + (
  77            220 if uiscale is bui.UIScale.SMALL else 190
  78        )
  79
  80        self._chest_yoffs = self._yoffs - 223
  81
  82        super().__init__(
  83            root_widget=bui.containerwidget(
  84                size=(self._width, self._height),
  85                toolbar_visibility='menu_full',
  86                scale=scale,
  87            ),
  88            transition=transition,
  89            origin_widget=origin_widget,
  90            # We're affected by screen size only at small ui-scale.
  91            refresh_on_screen_size_changes=uiscale is bui.UIScale.SMALL,
  92        )
  93
  94        # Tell the root-ui to stop updating toolbar values immediately;
  95        # this allows it to run animations based on the results of our
  96        # chest opening.
  97        bui.root_ui_pause_updates()
  98        self._root_ui_updates_paused = True
  99
 100        self._title_text = bui.textwidget(
 101            parent=self._root_widget,
 102            position=(
 103                self._width * 0.5,
 104                self._yoffstop - (36 if uiscale is bui.UIScale.SMALL else 10),
 105            ),
 106            size=(0, 0),
 107            text=bui.Lstr(
 108                resource='chests.slotText',
 109                subs=[('${NUM}', str(index + 1))],
 110            ),
 111            color=bui.app.ui_v1.title_color,
 112            maxwidth=110.0 if uiscale is bui.UIScale.SMALL else 200,
 113            scale=0.9 if uiscale is bui.UIScale.SMALL else 1.1,
 114            h_align='center',
 115            v_align='center',
 116        )
 117        self._core_widgets.append(self._title_text)
 118
 119        if uiscale is bui.UIScale.SMALL:
 120            bui.containerwidget(
 121                edit=self._root_widget, on_cancel_call=self.main_window_back
 122            )
 123        else:
 124            btn = bui.buttonwidget(
 125                parent=self._root_widget,
 126                position=(50, self._yoffs - 44),
 127                size=(60, 55),
 128                scale=0.8,
 129                label=bui.charstr(bui.SpecialChar.BACK),
 130                button_type='backSmall',
 131                extra_touch_border_scale=2.0,
 132                autoselect=True,
 133                on_activate_call=self.main_window_back,
 134            )
 135            bui.containerwidget(edit=self._root_widget, cancel_button=btn)
 136            self._core_widgets.append(btn)
 137
 138        # Note: Don't need to explicitly clean this up. Just not adding
 139        # it to core_widgets so it will go away on next reset.
 140        self._loadingspinner = bui.spinnerwidget(
 141            parent=self._root_widget,
 142            position=(self._width * 0.5, self._height * 0.5),
 143            size=48,
 144            style='bomb',
 145        )
 146
 147        self._infotext = bui.textwidget(
 148            parent=self._root_widget,
 149            position=(self._width * 0.5, self._yoffs - 200),
 150            size=(0, 0),
 151            text='',
 152            maxwidth=700,
 153            scale=0.8,
 154            color=(0.6, 0.5, 0.6),
 155            h_align='center',
 156            v_align='center',
 157        )
 158        self._core_widgets.append(self._infotext)
 159
 160        plus = bui.app.plus
 161        if plus is None:
 162            self._error('Plus feature-set is not present.')
 163            return
 164
 165        if plus.accounts.primary is None:
 166            self._error(bui.Lstr(resource='notSignedInText'))
 167            return
 168
 169        # Start by showing info/options for our target chest. Note that
 170        # we always ask the server for these values even though we may
 171        # have them through our appmode subscription which updates the
 172        # chest UI. This is because the wait_for_connectivity()
 173        # mechanism will often bring our window up a split second before
 174        # the chest subscription receives its first values which would
 175        # lead us to incorrectly think there is no chest there. If we
 176        # want to optimize this in the future we could perhaps use local
 177        # values only if there is a chest present in them.
 178        assert not self._action_in_flight
 179        self._action_in_flight = True
 180        with plus.accounts.primary:
 181            plus.cloud.send_message_cb(
 182                bacommon.bs.ChestInfoMessage(chest_id=str(self._index)),
 183                on_response=bui.WeakCall(self._on_chest_info_response),
 184            )
 185
 186    def __del__(self) -> None:
 187
 188        # Make sure UI updates are resumed if we haven't done so.
 189        if self._root_ui_updates_paused:
 190            bui.root_ui_resume_updates()
 191
 192    @override
 193    def get_main_window_state(self) -> bui.MainWindowState:
 194        # Support recreating our window for back/refresh purposes.
 195        cls = type(self)
 196
 197        # Pull anything we need from self out here; if we do it in the
 198        # lambda we keep self alive which is bad.
 199        index = self._index
 200
 201        return bui.BasicMainWindowState(
 202            create_call=lambda transition, origin_widget: cls(
 203                index=index, transition=transition, origin_widget=origin_widget
 204            )
 205        )
 206
 207    def _update_time_display(self, unlock_time: datetime.datetime) -> None:
 208        # Once text disappears, kill our timer.
 209        if not self._time_string_text:
 210            self._time_string_timer = None
 211            return
 212        now = bui.utc_now_cloud()
 213        secs_till_open = max(0.0, (unlock_time - now).total_seconds())
 214        tstr = (
 215            bui.timestring(secs_till_open, centi=False)
 216            if secs_till_open > 0
 217            else ''
 218        )
 219        bui.textwidget(edit=self._time_string_text, text=tstr)
 220
 221    def _on_chest_info_response(
 222        self, response: bacommon.bs.ChestInfoResponse | Exception
 223    ) -> None:
 224        assert self._action_in_flight  # Should be us.
 225        self._action_in_flight = False
 226
 227        if isinstance(response, Exception):
 228            self._error(
 229                bui.Lstr(resource='internal.unableToCompleteTryAgainText'),
 230                minor=True,
 231            )
 232            return
 233
 234        if response.chest is None:
 235            self._show_about_chest_slots()
 236            return
 237
 238        assert response.user_tokens is not None
 239        self._show_chest_actions(response.user_tokens, response.chest)
 240
 241    def _on_chest_action_response(
 242        self, response: bacommon.bs.ChestActionResponse | Exception
 243    ) -> None:
 244        assert self._action_in_flight  # Should be us.
 245        self._action_in_flight = False
 246
 247        # Communication/local error:
 248        if isinstance(response, Exception):
 249            self._error(
 250                bui.Lstr(resource='internal.unableToCompleteTryAgainText'),
 251                minor=True,
 252            )
 253            return
 254
 255        # Server-side error:
 256        if response.error is not None:
 257            self._error(bui.Lstr(translate=('serverResponses', response.error)))
 258            return
 259
 260        # Show any bundled success message.
 261        if response.success_msg is not None:
 262            bui.screenmessage(
 263                bui.Lstr(translate=('serverResponses', response.success_msg)),
 264                color=(0, 1.0, 0),
 265            )
 266            bui.getsound('cashRegister').play()
 267
 268        # Show any bundled warning.
 269        if response.warning is not None:
 270            bui.screenmessage(
 271                bui.Lstr(translate=('serverResponses', response.warning)),
 272                color=(1, 0.5, 0),
 273            )
 274            bui.getsound('error').play()
 275
 276        # If we just paid for something, make a sound accordingly.
 277        if bool(False):  # Hmm maybe this feels odd.
 278            if response.tokens_charged > 0:
 279                bui.getsound('cashRegister').play()
 280
 281        # If there's contents listed in the response, show them.
 282        if response.contents is not None:
 283            self._show_chest_contents(response)
 284        else:
 285            # Otherwise we're done here; just close out our UI.
 286            self.main_window_back()
 287
 288    def _show_chest_actions(
 289        self, user_tokens: int, chest: bacommon.bs.ChestInfoResponse.Chest
 290    ) -> None:
 291        """Show state for our chest."""
 292        # pylint: disable=too-many-locals
 293        # pylint: disable=cyclic-import
 294        from baclassic import (
 295            ClassicAppMode,
 296            CHEST_APPEARANCE_DISPLAY_INFOS,
 297            CHEST_APPEARANCE_DISPLAY_INFO_DEFAULT,
 298        )
 299
 300        plus = bui.app.plus
 301        assert plus is not None
 302
 303        # We expect to be run under classic app mode.
 304        mode = bui.app.mode
 305        if not isinstance(mode, ClassicAppMode):
 306            self._error('Classic app mode not active.')
 307            return
 308
 309        self._reset()
 310
 311        self._chestdisplayinfo = CHEST_APPEARANCE_DISPLAY_INFOS.get(
 312            chest.appearance, CHEST_APPEARANCE_DISPLAY_INFO_DEFAULT
 313        )
 314
 315        bui.textwidget(
 316            edit=self._title_text,
 317            text=bui.Lstr(
 318                translate=('displayItemNames', chest.appearance.pretty_name)
 319            ),
 320        )
 321
 322        imgsize = 145
 323        bui.imagewidget(
 324            parent=self._root_widget,
 325            position=(
 326                self._width * 0.5 - imgsize * 0.5,
 327                # self._height - 223 + self._yoffs,
 328                self._chest_yoffs,
 329            ),
 330            color=self._chestdisplayinfo.color,
 331            size=(imgsize, imgsize),
 332            texture=bui.gettexture(self._chestdisplayinfo.texclosed),
 333            tint_texture=bui.gettexture(self._chestdisplayinfo.texclosedtint),
 334            tint_color=self._chestdisplayinfo.tint,
 335            tint2_color=self._chestdisplayinfo.tint2,
 336        )
 337
 338        # Store the prize-sets so we can display odds/etc. Sort them
 339        # with largest weights first.
 340        self._prizesets = sorted(
 341            chest.prizesets, key=lambda s: s.weight, reverse=True
 342        )
 343
 344        if chest.unlock_tokens > 0:
 345            lsize = 30
 346            bui.imagewidget(
 347                parent=self._root_widget,
 348                position=(
 349                    self._width * 0.5 - imgsize * 0.4 - lsize * 0.5,
 350                    # self._height - 223 + 27.0 + self._yoffs,
 351                    self._chest_yoffs + 27.0,
 352                ),
 353                size=(lsize, lsize),
 354                texture=bui.gettexture('lock'),
 355            )
 356
 357        # Time string.
 358        if chest.unlock_tokens != 0:
 359            self._time_string_text = bui.textwidget(
 360                parent=self._root_widget,
 361                position=(
 362                    self._width * 0.5,
 363                    # self._height - 85 + self._yoffs
 364                    self._yoffs - 85,
 365                ),
 366                size=(0, 0),
 367                text='',
 368                maxwidth=700,
 369                scale=0.6,
 370                color=(0.6, 1.0, 0.6),
 371                h_align='center',
 372                v_align='center',
 373            )
 374            self._update_time_display(chest.unlock_time)
 375            self._time_string_timer = bui.AppTimer(
 376                1.0,
 377                repeat=True,
 378                call=bui.WeakCall(self._update_time_display, chest.unlock_time),
 379            )
 380
 381        # Allow watching an ad IF the server tells us we can AND we have
 382        # an ad ready to show.
 383        show_ad_button = (
 384            chest.unlock_tokens > 0
 385            and chest.ad_allow
 386            and plus.have_incentivized_ad()
 387        )
 388
 389        bwidth = 130
 390        bheight = 90
 391        bposy = -330 if chest.unlock_tokens == 0 else -340
 392        hspace = 20
 393        boffsx = (hspace * -0.5 - bwidth * 0.5) if show_ad_button else 0.0
 394
 395        self._open_now_button = bui.buttonwidget(
 396            parent=self._root_widget,
 397            position=(
 398                self._width * 0.5 - bwidth * 0.5 + boffsx,
 399                self._yoffs + bposy,
 400            ),
 401            size=(bwidth, bheight),
 402            label='',
 403            button_type='square',
 404            autoselect=True,
 405            on_activate_call=bui.WeakCall(
 406                self._open_press, user_tokens, chest.unlock_tokens
 407            ),
 408            enable_sound=False,
 409        )
 410        self._open_now_images = []
 411        self._open_now_texts = []
 412
 413        iconsize = 50
 414        if chest.unlock_tokens == 0:
 415            self._open_now_texts.append(
 416                bui.textwidget(
 417                    parent=self._root_widget,
 418                    text=bui.Lstr(resource='openText'),
 419                    position=(
 420                        self._width * 0.5 + boffsx,
 421                        self._yoffs + bposy + bheight * 0.5,
 422                    ),
 423                    color=(0, 1, 0),
 424                    draw_controller=self._open_now_button,
 425                    scale=0.7,
 426                    maxwidth=bwidth * 0.8,
 427                    size=(0, 0),
 428                    h_align='center',
 429                    v_align='center',
 430                )
 431            )
 432        else:
 433            self._open_now_texts.append(
 434                bui.textwidget(
 435                    parent=self._root_widget,
 436                    text=bui.Lstr(resource='openNowText'),
 437                    position=(
 438                        self._width * 0.5 + boffsx,
 439                        self._yoffs + bposy + bheight * 1.15,
 440                    ),
 441                    maxwidth=bwidth * 0.8,
 442                    scale=0.7,
 443                    color=(0.7, 1, 0.7),
 444                    size=(0, 0),
 445                    h_align='center',
 446                    v_align='center',
 447                )
 448            )
 449            self._open_now_images.append(
 450                bui.imagewidget(
 451                    parent=self._root_widget,
 452                    size=(iconsize, iconsize),
 453                    position=(
 454                        self._width * 0.5 - iconsize * 0.5 + boffsx,
 455                        self._yoffs + bposy + bheight * 0.35,
 456                    ),
 457                    draw_controller=self._open_now_button,
 458                    texture=bui.gettexture('coin'),
 459                )
 460            )
 461            self._open_now_texts.append(
 462                bui.textwidget(
 463                    parent=self._root_widget,
 464                    text=bui.Lstr(
 465                        resource='tokens.numTokensText',
 466                        subs=[('${COUNT}', str(chest.unlock_tokens))],
 467                    ),
 468                    position=(
 469                        self._width * 0.5 + boffsx,
 470                        self._yoffs + bposy + bheight * 0.25,
 471                    ),
 472                    scale=0.65,
 473                    color=(0, 1, 0),
 474                    draw_controller=self._open_now_button,
 475                    maxwidth=bwidth * 0.8,
 476                    size=(0, 0),
 477                    h_align='center',
 478                    v_align='center',
 479                )
 480            )
 481        self._open_now_spinner = bui.spinnerwidget(
 482            parent=self._root_widget,
 483            position=(
 484                self._width * 0.5 + boffsx,
 485                self._yoffs + bposy + 0.5 * bheight,
 486            ),
 487            visible=False,
 488        )
 489
 490        if show_ad_button:
 491            bui.textwidget(
 492                parent=self._root_widget,
 493                text=bui.Lstr(resource='chests.reduceWaitText'),
 494                position=(
 495                    self._width * 0.5 + hspace * 0.5 + bwidth * 0.5,
 496                    self._yoffs + bposy + bheight * 1.15,
 497                ),
 498                maxwidth=bwidth * 0.8,
 499                scale=0.7,
 500                color=(0.7, 1, 0.7),
 501                size=(0, 0),
 502                h_align='center',
 503                v_align='center',
 504            )
 505            self._watch_ad_button = bui.buttonwidget(
 506                parent=self._root_widget,
 507                position=(
 508                    self._width * 0.5 + hspace * 0.5,
 509                    self._yoffs + bposy,
 510                ),
 511                size=(bwidth, bheight),
 512                label='',
 513                button_type='square',
 514                autoselect=True,
 515                on_activate_call=bui.WeakCall(self._watch_ad_press),
 516                enable_sound=False,
 517            )
 518            bui.imagewidget(
 519                parent=self._root_widget,
 520                size=(iconsize, iconsize),
 521                position=(
 522                    self._width * 0.5
 523                    + hspace * 0.5
 524                    + bwidth * 0.5
 525                    - iconsize * 0.5,
 526                    self._yoffs + bposy + bheight * 0.35,
 527                ),
 528                draw_controller=self._watch_ad_button,
 529                color=(1.5, 1.0, 2.0),
 530                texture=bui.gettexture('tv'),
 531            )
 532            # Note to self: AdMob requires rewarded ad usage
 533            # specifically says 'Ad' in it.
 534            bui.textwidget(
 535                parent=self._root_widget,
 536                text=bui.Lstr(resource='watchAnAdText'),
 537                position=(
 538                    self._width * 0.5 + hspace * 0.5 + bwidth * 0.5,
 539                    self._yoffs + bposy + bheight * 0.25,
 540                ),
 541                scale=0.65,
 542                color=(0, 1, 0),
 543                draw_controller=self._watch_ad_button,
 544                maxwidth=bwidth * 0.8,
 545                size=(0, 0),
 546                h_align='center',
 547                v_align='center',
 548            )
 549
 550        self._show_odds(initial_highlighted_row=-1)
 551
 552    def _highlight_odds_row(self, row: int, extra: bool = False) -> None:
 553
 554        for rindex, imgs in self._prizesetimgs.items():
 555            opacity = (
 556                (0.9 if extra else 0.75)
 557                if rindex == row
 558                else (0.4 if extra else 0.5)
 559            )
 560            for img in imgs:
 561                if img:
 562                    bui.imagewidget(edit=img, opacity=opacity)
 563
 564        for rindex, txts in self._prizesettxts.items():
 565            opacity = (
 566                (0.9 if extra else 0.75)
 567                if rindex == row
 568                else (0.4 if extra else 0.5)
 569            )
 570            for txt in txts:
 571                if txt:
 572                    bui.textwidget(edit=txt, color=(0.7, 0.65, 1, opacity))
 573
 574    def _show_odds(
 575        self,
 576        *,
 577        initial_highlighted_row: int,
 578        initial_highlighted_extra: bool = False,
 579    ) -> None:
 580        # pylint: disable=too-many-locals
 581        xoffs = 110
 582
 583        totalweight = max(0.001, sum(t.weight for t in self._prizesets))
 584
 585        rowheight = 25
 586        totalheight = (len(self._prizesets) + 1) * rowheight
 587        x = self._width * 0.5 + xoffs
 588        y = self._yoffs - 150.0 + totalheight * 0.5
 589
 590        # Title.
 591        bui.textwidget(
 592            parent=self._root_widget,
 593            text=bui.Lstr(resource='chests.prizeOddsText'),
 594            color=(0.7, 0.65, 1, 0.5),
 595            flatness=1.0,
 596            shadow=1.0,
 597            position=(x, y),
 598            scale=0.55,
 599            size=(0, 0),
 600            h_align='left',
 601            v_align='center',
 602        )
 603        y -= 5.0
 604
 605        prizesettxts: list[bui.Widget]
 606        prizesetimgs: list[bui.Widget]
 607
 608        def _mkicon(img: str) -> None:
 609            iconsize = 20.0
 610            nonlocal x
 611            nonlocal prizesetimgs
 612            prizesetimgs.append(
 613                bui.imagewidget(
 614                    parent=self._root_widget,
 615                    size=(iconsize, iconsize),
 616                    position=(x, y - iconsize * 0.5),
 617                    texture=bui.gettexture(img),
 618                    opacity=0.4,
 619                )
 620            )
 621            x += iconsize
 622
 623        def _mktxt(txt: str, advance: bool = True) -> None:
 624            tscale = 0.45
 625            nonlocal x
 626            nonlocal prizesettxts
 627            prizesettxts.append(
 628                bui.textwidget(
 629                    parent=self._root_widget,
 630                    text=txt,
 631                    flatness=1.0,
 632                    shadow=1.0,
 633                    position=(x, y),
 634                    scale=tscale,
 635                    size=(0, 0),
 636                    h_align='left',
 637                    v_align='center',
 638                )
 639            )
 640            if advance:
 641                x += (bui.get_string_width(txt, suppress_warning=True)) * tscale
 642
 643        self._prizesettxts = {}
 644        self._prizesetimgs = {}
 645
 646        for i, p in enumerate(self._prizesets):
 647            prizesettxts = self._prizesettxts.setdefault(i, [])
 648            prizesetimgs = self._prizesetimgs.setdefault(i, [])
 649            x = self._width * 0.5 + xoffs
 650            y -= rowheight
 651            percent = 100.0 * p.weight / totalweight
 652
 653            # Show decimals only if we get very small percentages (looks
 654            # better than rounding as '0%').
 655            percenttxt = (
 656                f'{percent:.2f}%:'
 657                if percent < 0.095
 658                else (
 659                    f'{percent:.1f}%:'
 660                    if percent < 0.95
 661                    else f'{round(percent)}%:'
 662                )
 663            )
 664
 665            # We advance manually here to keep values lined up
 666            # (otherwise single digit percent rows don't line up with
 667            # double digit ones).
 668            _mktxt(percenttxt, advance=False)
 669            x += 35.0
 670
 671            for item in p.contents:
 672                x += 5.0
 673                if isinstance(item.item, bacommon.bs.TicketsDisplayItem):
 674                    _mktxt(str(item.item.count))
 675                    _mkicon('tickets')
 676                elif isinstance(item.item, bacommon.bs.TokensDisplayItem):
 677                    _mktxt(str(item.item.count))
 678                    _mkicon('coin')
 679                else:
 680                    # For other cases just fall back on text desc.
 681                    #
 682                    # Translate the wrapper description and apply any subs.
 683                    descfin = bui.Lstr(
 684                        translate=('serverResponses', item.description)
 685                    ).evaluate()
 686                    subs = (
 687                        []
 688                        if item.description_subs is None
 689                        else item.description_subs
 690                    )
 691                    assert len(subs) % 2 == 0  # Should always be even.
 692                    for j in range(0, len(subs) - 1, 2):
 693                        descfin = descfin.replace(subs[j], subs[j + 1])
 694                    _mktxt(descfin)
 695        self._highlight_odds_row(
 696            initial_highlighted_row, extra=initial_highlighted_extra
 697        )
 698
 699    def _open_press(self, user_tokens: int, token_payment: int) -> None:
 700        from bauiv1lib.gettokens import show_get_tokens_prompt
 701
 702        bui.getsound('click01').play()
 703
 704        # Allow only one in-flight action at once.
 705        if self._action_in_flight:
 706            bui.screenmessage(
 707                bui.Lstr(resource='pleaseWaitText'), color=(1, 0, 0)
 708            )
 709            bui.getsound('error').play()
 710            return
 711
 712        plus = bui.app.plus
 713        assert plus is not None
 714
 715        if plus.accounts.primary is None:
 716            self._error(bui.Lstr(resource='notSignedInText'))
 717            return
 718
 719        # Offer to purchase tokens if they don't have enough.
 720        if user_tokens < token_payment:
 721            # Hack: We disable normal swish for the open button and it
 722            # seems weird without a swish here, so explicitly do one.
 723            bui.getsound('swish').play()
 724            show_get_tokens_prompt()
 725            return
 726
 727        self._action_in_flight = True
 728        with plus.accounts.primary:
 729            plus.cloud.send_message_cb(
 730                bacommon.bs.ChestActionMessage(
 731                    chest_id=str(self._index),
 732                    action=bacommon.bs.ChestActionMessage.Action.UNLOCK,
 733                    token_payment=token_payment,
 734                ),
 735                on_response=bui.WeakCall(self._on_chest_action_response),
 736            )
 737
 738        # Convey that something is in progress.
 739        if self._open_now_button:
 740            bui.spinnerwidget(edit=self._open_now_spinner, visible=True)
 741            for twidget in self._open_now_texts:
 742                bui.textwidget(edit=twidget, color=(1, 1, 1, 0.2))
 743            for iwidget in self._open_now_images:
 744                bui.imagewidget(edit=iwidget, opacity=0.2)
 745
 746    def _watch_ad_press(self) -> None:
 747
 748        bui.getsound('click01').play()
 749
 750        # Allow only one in-flight action at once.
 751        if self._action_in_flight:
 752            bui.screenmessage(
 753                bui.Lstr(resource='pleaseWaitText'), color=(1, 0, 0)
 754            )
 755            bui.getsound('error').play()
 756            return
 757
 758        assert bui.app.classic is not None
 759
 760        self._action_in_flight = True
 761        bui.app.classic.ads.show_ad_2(
 762            'reduce_chest_wait',
 763            on_completion_call=bui.WeakCall(self._watch_ad_complete),
 764        )
 765
 766        # Convey that something is in progress.
 767        if self._watch_ad_button:
 768            bui.buttonwidget(edit=self._watch_ad_button, color=(0.4, 0.4, 0.4))
 769
 770    def _watch_ad_complete(self, actually_showed: bool) -> None:
 771
 772        assert self._action_in_flight  # Should be ad view.
 773        self._action_in_flight = False
 774
 775        if not actually_showed:
 776            return
 777
 778        # Allow only one in-flight action at once.
 779        if self._action_in_flight:
 780            bui.screenmessage(
 781                bui.Lstr(resource='pleaseWaitText'), color=(1, 0, 0)
 782            )
 783            bui.getsound('error').play()
 784            return
 785
 786        plus = bui.app.plus
 787        assert plus is not None
 788
 789        if plus.accounts.primary is None:
 790            self._error(bui.Lstr(resource='notSignedInText'))
 791            return
 792
 793        self._action_in_flight = True
 794        with plus.accounts.primary:
 795            plus.cloud.send_message_cb(
 796                bacommon.bs.ChestActionMessage(
 797                    chest_id=str(self._index),
 798                    action=bacommon.bs.ChestActionMessage.Action.AD,
 799                    token_payment=0,
 800                ),
 801                on_response=bui.WeakCall(self._on_chest_action_response),
 802            )
 803
 804    def _reset(self) -> None:
 805        """Clear all non-permanent widgets and clear infotext."""
 806        for widget in self._root_widget.get_children():
 807            if widget not in self._core_widgets:
 808                widget.delete()
 809        bui.textwidget(edit=self._infotext, text='', color=(1, 1, 1))
 810
 811    def _error(self, msg: str | bui.Lstr, minor: bool = False) -> None:
 812        """Put ourself in an error state with a visible error message."""
 813        self._reset()
 814        bui.textwidget(
 815            edit=self._infotext,
 816            text=msg,
 817            color=(1, 0.5, 0.5) if minor else (1, 0, 0),
 818        )
 819
 820    def _show_about_chest_slots(self) -> None:
 821        # No-op if our ui is dead.
 822        if not self._root_widget:
 823            return
 824
 825        self._reset()
 826        bui.textwidget(
 827            edit=self._infotext,
 828            text=bui.Lstr(resource='chests.slotDescriptionText'),
 829            color=(1, 1, 1),
 830        )
 831
 832    def _show_chest_contents(
 833        self, response: bacommon.bs.ChestActionResponse
 834    ) -> None:
 835        # pylint: disable=too-many-locals
 836
 837        from baclassic import show_display_item
 838
 839        # No-op if our ui is dead.
 840        if not self._root_widget:
 841            return
 842
 843        assert response.contents is not None
 844
 845        # Insert test items for testing.
 846        if bool(False):
 847            response.contents += [
 848                bacommon.bs.DisplayItemWrapper.for_display_item(
 849                    bacommon.bs.TestDisplayItem()
 850                )
 851            ]
 852
 853        tincr = 0.4
 854        tendoffs = tincr * 4.0
 855        toffs = 0.0
 856
 857        bui.getsound('revUp').play(volume=2.0)
 858
 859        # Show nothing but the chest icon and animate it shaking.
 860        self._reset()
 861        imgsize = 145
 862        assert self._chestdisplayinfo is not None
 863        img = bui.imagewidget(
 864            parent=self._root_widget,
 865            color=self._chestdisplayinfo.color,
 866            texture=bui.gettexture(self._chestdisplayinfo.texclosed),
 867            tint_texture=bui.gettexture(self._chestdisplayinfo.texclosedtint),
 868            tint_color=self._chestdisplayinfo.tint,
 869            tint2_color=self._chestdisplayinfo.tint2,
 870        )
 871
 872        def _set_img(x: float, scale: float) -> None:
 873            if not img:
 874                return
 875            bui.imagewidget(
 876                edit=img,
 877                position=(
 878                    self._width * 0.5 - imgsize * scale * 0.5 + x,
 879                    self._yoffs - 223 + imgsize * 0.5 - imgsize * scale * 0.5,
 880                ),
 881                size=(imgsize * scale, imgsize * scale),
 882            )
 883
 884        # Set initial place.
 885        _set_img(0.0, 1.0)
 886
 887        sign = 1.0
 888        while toffs < tendoffs:
 889            toffs += 0.03 * random.uniform(0.5, 1.5)
 890            sign = -sign
 891            bui.apptimer(
 892                toffs,
 893                bui.Call(
 894                    _set_img,
 895                    x=(
 896                        20.0
 897                        * random.uniform(0.3, 1.0)
 898                        * math.pow(toffs / tendoffs, 2.0)
 899                        * sign
 900                    ),
 901                    scale=1.0 - 0.2 * math.pow(toffs / tendoffs, 2.0),
 902                ),
 903            )
 904
 905        xspacing = 100
 906        xoffs = -0.5 * (len(response.contents) - 1) * xspacing
 907        bui.apptimer(
 908            toffs - 0.2, lambda: bui.getsound('corkPop2').play(volume=4.0)
 909        )
 910        # Play a variety of voice sounds.
 911
 912        # We keep a global list of voice options which we randomly pull
 913        # from and refill when empty. This ensures everything gets
 914        # played somewhat frequently and minimizes annoying repeats.
 915        global _g_open_voices  # pylint: disable=global-statement
 916        if not _g_open_voices:
 917            _g_open_voices = [
 918                (0.3, 'woo3', 2.5),
 919                (0.1, 'gasp', 1.3),
 920                (0.2, 'woo2', 2.0),
 921                (0.2, 'wow', 2.0),
 922                (0.2, 'kronk2', 2.0),
 923                (0.2, 'mel03', 2.0),
 924                (0.2, 'aww', 2.0),
 925                (0.4, 'nice', 2.0),
 926                (0.3, 'yeah', 1.5),
 927                (0.2, 'woo', 1.0),
 928                (0.5, 'ooh', 0.8),
 929            ]
 930
 931        voicetimeoffs, voicename, volume = _g_open_voices.pop(
 932            random.randrange(len(_g_open_voices))
 933        )
 934        bui.apptimer(
 935            toffs + voicetimeoffs,
 936            lambda: bui.getsound(voicename).play(volume=volume),
 937        )
 938
 939        toffsopen = toffs
 940        bui.apptimer(toffs, bui.WeakCall(self._show_chest_opening))
 941        toffs += tincr * 1.0
 942        width = xspacing * 0.95
 943
 944        for item in response.contents:
 945            toffs += tincr
 946            bui.apptimer(
 947                toffs - 0.1, lambda: bui.getsound('cashRegister').play()
 948            )
 949            bui.apptimer(
 950                toffs,
 951                strict_partial(
 952                    show_display_item,
 953                    item,
 954                    self._root_widget,
 955                    pos=(
 956                        self._width * 0.5 + xoffs,
 957                        self._yoffs - 250.0,
 958                    ),
 959                    width=width,
 960                ),
 961            )
 962            xoffs += xspacing
 963        toffs += tincr
 964        bui.apptimer(toffs, bui.WeakCall(self._show_done_button))
 965
 966        self._show_odds(initial_highlighted_row=-1)
 967
 968        # Store this for later
 969        self._prizeindex = response.prizeindex
 970
 971        # The final result was already randomly selected on the server,
 972        # but we want to give the illusion of randomness here, so cycle
 973        # through highlighting our options and stop on the winner when
 974        # the chest opens. To do this, we start at the end at the prize
 975        # and work backwards setting timers.
 976        if self._prizesets:
 977            toffs2 = toffsopen - 0.01
 978            amt = 0.02
 979            i = self._prizeindex
 980            while toffs2 > 0.0:
 981                bui.apptimer(
 982                    toffs2,
 983                    bui.WeakCall(self._highlight_odds_row, i),
 984                )
 985                toffs2 -= amt
 986                amt *= 1.05 * random.uniform(0.9, 1.1)
 987                i = (i - 1) % len(self._prizesets)
 988
 989    def _show_chest_opening(self) -> None:
 990
 991        # No-op if our ui is dead.
 992        if not self._root_widget:
 993            return
 994
 995        self._reset()
 996        imgsize = 145
 997        bui.getsound('hiss').play()
 998        assert self._chestdisplayinfo is not None
 999        img = bui.imagewidget(
1000            parent=self._root_widget,
1001            color=self._chestdisplayinfo.color,
1002            texture=bui.gettexture(self._chestdisplayinfo.texopen),
1003            tint_texture=bui.gettexture(self._chestdisplayinfo.texopentint),
1004            tint_color=self._chestdisplayinfo.tint,
1005            tint2_color=self._chestdisplayinfo.tint2,
1006        )
1007        tincr = 0.8
1008        tendoffs = tincr * 2.0
1009        toffs = 0.0
1010
1011        def _set_img(x: float, scale: float) -> None:
1012            if not img:
1013                return
1014            bui.imagewidget(
1015                edit=img,
1016                position=(
1017                    self._width * 0.5 - imgsize * scale * 0.5 + x,
1018                    self._yoffs - 223 + imgsize * 0.5 - imgsize * scale * 0.5,
1019                ),
1020                size=(imgsize * scale, imgsize * scale),
1021            )
1022
1023        # Set initial place.
1024        _set_img(0.0, 1.0)
1025
1026        sign = 1.0
1027        while toffs < tendoffs:
1028            toffs += 0.03 * random.uniform(0.5, 1.5)
1029            sign = -sign
1030            # Note: we speed x along here (multing toffs) so position
1031            # comes to rest before scale.
1032            bui.apptimer(
1033                toffs,
1034                bui.Call(
1035                    _set_img,
1036                    x=(
1037                        1.0
1038                        * random.uniform(0.3, 1.0)
1039                        * (
1040                            1.0
1041                            - math.pow(min(1.0, 3.0 * toffs / tendoffs), 2.0)
1042                        )
1043                        * sign
1044                    ),
1045                    scale=1.0 - 0.1 * math.pow(toffs / tendoffs, 0.5),
1046                ),
1047            )
1048
1049        self._show_odds(
1050            initial_highlighted_row=self._prizeindex,
1051            initial_highlighted_extra=True,
1052        )
1053
1054    def _show_done_button(self) -> None:
1055        # No-op if our ui is dead.
1056        if not self._root_widget:
1057            return
1058
1059        bwidth = 200
1060        bheight = 60
1061
1062        btn = bui.buttonwidget(
1063            parent=self._root_widget,
1064            position=(
1065                self._width * 0.5 - bwidth * 0.5,
1066                self._yoffs - 350,
1067            ),
1068            size=(bwidth, bheight),
1069            label=bui.Lstr(resource='doneText'),
1070            autoselect=True,
1071            on_activate_call=self.main_window_back,
1072        )
1073        bui.containerwidget(edit=self._root_widget, start_button=btn)
1074
1075
1076# Slight hack: we define window different classes for our different
1077# chest slots so that the default UI behavior is to replace each other
1078# when different ones are pressed. If they are all the same window class
1079# then the default behavior for such presses is to toggle the existing
1080# one back off.
1081
1082
1083class ChestWindow0(ChestWindow):
1084    """Child class of ChestWindow for slighty hackish reasons."""
1085
1086
1087class ChestWindow1(ChestWindow):
1088    """Child class of ChestWindow for slighty hackish reasons."""
1089
1090
1091class ChestWindow2(ChestWindow):
1092    """Child class of ChestWindow for slighty hackish reasons."""
1093
1094
1095class ChestWindow3(ChestWindow):
1096    """Child class of ChestWindow for slighty hackish reasons."""