bauiv1lib.store.browser

   1# Released under the MIT License. See LICENSE for details.
   2#
   3"""UI for browsing the store."""
   4# pylint: disable=too-many-lines
   5from __future__ import annotations
   6
   7import os
   8import time
   9import copy
  10import math
  11import logging
  12import weakref
  13import datetime
  14from enum import Enum
  15from threading import Thread
  16from typing import TYPE_CHECKING
  17
  18from efro.error import CommunicationError
  19import bacommon.cloud
  20import bauiv1 as bui
  21
  22if TYPE_CHECKING:
  23    from typing import Any, Callable, Sequence
  24
  25MERCH_LINK_KEY = 'Merch Link'
  26
  27
  28class StoreBrowserWindow(bui.Window):
  29    """Window for browsing the store."""
  30
  31    class TabID(Enum):
  32        """Our available tab types."""
  33
  34        EXTRAS = 'extras'
  35        MAPS = 'maps'
  36        MINIGAMES = 'minigames'
  37        CHARACTERS = 'characters'
  38        ICONS = 'icons'
  39
  40    def __init__(
  41        self,
  42        transition: str = 'in_right',
  43        modal: bool = False,
  44        show_tab: StoreBrowserWindow.TabID | None = None,
  45        on_close_call: Callable[[], Any] | None = None,
  46        back_location: str | None = None,
  47        origin_widget: bui.Widget | None = None,
  48    ):
  49        # pylint: disable=too-many-statements
  50        # pylint: disable=too-many-locals
  51        from bauiv1lib.tabs import TabRow
  52        from bauiv1 import SpecialChar
  53
  54        app = bui.app
  55        assert app.classic is not None
  56        uiscale = app.ui_v1.uiscale
  57
  58        bui.set_analytics_screen('Store Window')
  59
  60        scale_origin: tuple[float, float] | None
  61
  62        # If they provided an origin-widget, scale up from that.
  63        if origin_widget is not None:
  64            self._transition_out = 'out_scale'
  65            scale_origin = origin_widget.get_screen_space_center()
  66            transition = 'in_scale'
  67        else:
  68            self._transition_out = 'out_right'
  69            scale_origin = None
  70
  71        self.button_infos: dict[str, dict[str, Any]] | None = None
  72        self.update_buttons_timer: bui.AppTimer | None = None
  73        self._status_textwidget_update_timer = None
  74
  75        self._back_location = back_location
  76        self._on_close_call = on_close_call
  77        self._show_tab = show_tab
  78        self._modal = modal
  79        self._width = 1240 if uiscale is bui.UIScale.SMALL else 1040
  80        self._x_inset = x_inset = 100 if uiscale is bui.UIScale.SMALL else 0
  81        self._height = (
  82            578
  83            if uiscale is bui.UIScale.SMALL
  84            else 645
  85            if uiscale is bui.UIScale.MEDIUM
  86            else 800
  87        )
  88        self._current_tab: StoreBrowserWindow.TabID | None = None
  89        extra_top = 30 if uiscale is bui.UIScale.SMALL else 0
  90
  91        self.request: Any = None
  92        self._r = 'store'
  93        self._last_buy_time: float | None = None
  94
  95        super().__init__(
  96            root_widget=bui.containerwidget(
  97                size=(self._width, self._height + extra_top),
  98                transition=transition,
  99                toolbar_visibility='menu_full',
 100                scale=(
 101                    1.3
 102                    if uiscale is bui.UIScale.SMALL
 103                    else 0.9
 104                    if uiscale is bui.UIScale.MEDIUM
 105                    else 0.8
 106                ),
 107                scale_origin_stack_offset=scale_origin,
 108                stack_offset=(
 109                    (0, -5)
 110                    if uiscale is bui.UIScale.SMALL
 111                    else (0, 0)
 112                    if uiscale is bui.UIScale.MEDIUM
 113                    else (0, 0)
 114                ),
 115            )
 116        )
 117
 118        self._back_button = btn = bui.buttonwidget(
 119            parent=self._root_widget,
 120            position=(70 + x_inset, self._height - 74),
 121            size=(140, 60),
 122            scale=1.1,
 123            autoselect=True,
 124            label=bui.Lstr(resource='doneText' if self._modal else 'backText'),
 125            button_type=None if self._modal else 'back',
 126            on_activate_call=self._back,
 127        )
 128        bui.containerwidget(edit=self._root_widget, cancel_button=btn)
 129
 130        self._ticket_count_text: bui.Widget | None = None
 131        self._get_tickets_button: bui.Widget | None = None
 132
 133        if app.classic.allow_ticket_purchases:
 134            self._get_tickets_button = bui.buttonwidget(
 135                parent=self._root_widget,
 136                size=(210, 65),
 137                on_activate_call=self._on_get_more_tickets_press,
 138                autoselect=True,
 139                scale=0.9,
 140                text_scale=1.4,
 141                left_widget=self._back_button,
 142                color=(0.7, 0.5, 0.85),
 143                textcolor=(0.2, 1.0, 0.2),
 144                label=bui.Lstr(resource='getTicketsWindow.titleText'),
 145            )
 146        else:
 147            self._ticket_count_text = bui.textwidget(
 148                parent=self._root_widget,
 149                size=(210, 64),
 150                color=(0.2, 1.0, 0.2),
 151                h_align='center',
 152                v_align='center',
 153            )
 154
 155        # Move this dynamically to keep it out of the way of the party icon.
 156        self._update_get_tickets_button_pos()
 157        self._get_ticket_pos_update_timer = bui.AppTimer(
 158            1.0,
 159            bui.WeakCall(self._update_get_tickets_button_pos),
 160            repeat=True,
 161        )
 162        if self._get_tickets_button:
 163            bui.widget(
 164                edit=self._back_button, right_widget=self._get_tickets_button
 165            )
 166        self._ticket_text_update_timer = bui.AppTimer(
 167            1.0, bui.WeakCall(self._update_tickets_text), repeat=True
 168        )
 169        self._update_tickets_text()
 170
 171        if (
 172            app.classic.platform in ['mac', 'ios']
 173            and app.classic.subplatform == 'appstore'
 174        ):
 175            bui.buttonwidget(
 176                parent=self._root_widget,
 177                position=(self._width * 0.5 - 70, 16),
 178                size=(230, 50),
 179                scale=0.65,
 180                on_activate_call=bui.WeakCall(self._restore_purchases),
 181                color=(0.35, 0.3, 0.4),
 182                selectable=False,
 183                textcolor=(0.55, 0.5, 0.6),
 184                label=bui.Lstr(
 185                    resource='getTicketsWindow.restorePurchasesText'
 186                ),
 187            )
 188
 189        bui.textwidget(
 190            parent=self._root_widget,
 191            position=(self._width * 0.5, self._height - 44),
 192            size=(0, 0),
 193            color=app.ui_v1.title_color,
 194            scale=1.5,
 195            h_align='center',
 196            v_align='center',
 197            text=bui.Lstr(resource='storeText'),
 198            maxwidth=420,
 199        )
 200
 201        if not self._modal:
 202            bui.buttonwidget(
 203                edit=self._back_button,
 204                button_type='backSmall',
 205                size=(60, 60),
 206                label=bui.charstr(SpecialChar.BACK),
 207            )
 208
 209        scroll_buffer_h = 130 + 2 * x_inset
 210        tab_buffer_h = 250 + 2 * x_inset
 211
 212        tabs_def = [
 213            (self.TabID.EXTRAS, bui.Lstr(resource=self._r + '.extrasText')),
 214            (self.TabID.MAPS, bui.Lstr(resource=self._r + '.mapsText')),
 215            (
 216                self.TabID.MINIGAMES,
 217                bui.Lstr(resource=self._r + '.miniGamesText'),
 218            ),
 219            (
 220                self.TabID.CHARACTERS,
 221                bui.Lstr(resource=self._r + '.charactersText'),
 222            ),
 223            (self.TabID.ICONS, bui.Lstr(resource=self._r + '.iconsText')),
 224        ]
 225
 226        self._tab_row = TabRow(
 227            self._root_widget,
 228            tabs_def,
 229            pos=(tab_buffer_h * 0.5, self._height - 130),
 230            size=(self._width - tab_buffer_h, 50),
 231            on_select_call=self._set_tab,
 232        )
 233
 234        self._purchasable_count_widgets: dict[
 235            StoreBrowserWindow.TabID, dict[str, Any]
 236        ] = {}
 237
 238        # Create our purchasable-items tags and have them update over time.
 239        for tab_id, tab in self._tab_row.tabs.items():
 240            pos = tab.position
 241            size = tab.size
 242            button = tab.button
 243            rad = 10
 244            center = (pos[0] + 0.1 * size[0], pos[1] + 0.9 * size[1])
 245            img = bui.imagewidget(
 246                parent=self._root_widget,
 247                position=(center[0] - rad * 1.04, center[1] - rad * 1.15),
 248                size=(rad * 2.2, rad * 2.2),
 249                texture=bui.gettexture('circleShadow'),
 250                color=(1, 0, 0),
 251            )
 252            txt = bui.textwidget(
 253                parent=self._root_widget,
 254                position=center,
 255                size=(0, 0),
 256                h_align='center',
 257                v_align='center',
 258                maxwidth=1.4 * rad,
 259                scale=0.6,
 260                shadow=1.0,
 261                flatness=1.0,
 262            )
 263            rad = 20
 264            sale_img = bui.imagewidget(
 265                parent=self._root_widget,
 266                position=(center[0] - rad, center[1] - rad),
 267                size=(rad * 2, rad * 2),
 268                draw_controller=button,
 269                texture=bui.gettexture('circleZigZag'),
 270                color=(0.5, 0, 1.0),
 271            )
 272            sale_title_text = bui.textwidget(
 273                parent=self._root_widget,
 274                position=(center[0], center[1] + 0.24 * rad),
 275                size=(0, 0),
 276                h_align='center',
 277                v_align='center',
 278                draw_controller=button,
 279                maxwidth=1.4 * rad,
 280                scale=0.6,
 281                shadow=0.0,
 282                flatness=1.0,
 283                color=(0, 1, 0),
 284            )
 285            sale_time_text = bui.textwidget(
 286                parent=self._root_widget,
 287                position=(center[0], center[1] - 0.29 * rad),
 288                size=(0, 0),
 289                h_align='center',
 290                v_align='center',
 291                draw_controller=button,
 292                maxwidth=1.4 * rad,
 293                scale=0.4,
 294                shadow=0.0,
 295                flatness=1.0,
 296                color=(0, 1, 0),
 297            )
 298            self._purchasable_count_widgets[tab_id] = {
 299                'img': img,
 300                'text': txt,
 301                'sale_img': sale_img,
 302                'sale_title_text': sale_title_text,
 303                'sale_time_text': sale_time_text,
 304            }
 305        self._tab_update_timer = bui.AppTimer(
 306            1.0, bui.WeakCall(self._update_tabs), repeat=True
 307        )
 308        self._update_tabs()
 309
 310        if self._get_tickets_button:
 311            last_tab_button = self._tab_row.tabs[tabs_def[-1][0]].button
 312            bui.widget(
 313                edit=self._get_tickets_button, down_widget=last_tab_button
 314            )
 315            bui.widget(
 316                edit=last_tab_button,
 317                up_widget=self._get_tickets_button,
 318                right_widget=self._get_tickets_button,
 319            )
 320
 321        self._scroll_width = self._width - scroll_buffer_h
 322        self._scroll_height = self._height - 180
 323
 324        self._scrollwidget: bui.Widget | None = None
 325        self._status_textwidget: bui.Widget | None = None
 326        self._restore_state()
 327
 328    def _update_get_tickets_button_pos(self) -> None:
 329        assert bui.app.classic is not None
 330        uiscale = bui.app.ui_v1.uiscale
 331        pos = (
 332            self._width
 333            - 252
 334            - (
 335                self._x_inset
 336                + (
 337                    47
 338                    if uiscale is bui.UIScale.SMALL
 339                    and bui.is_party_icon_visible()
 340                    else 0
 341                )
 342            ),
 343            self._height - 70,
 344        )
 345        if self._get_tickets_button:
 346            bui.buttonwidget(edit=self._get_tickets_button, position=pos)
 347        if self._ticket_count_text:
 348            bui.textwidget(edit=self._ticket_count_text, position=pos)
 349
 350    def _restore_purchases(self) -> None:
 351        from bauiv1lib import account
 352
 353        plus = bui.app.plus
 354        assert plus is not None
 355        if plus.get_v1_account_state() != 'signed_in':
 356            account.show_sign_in_prompt()
 357        else:
 358            plus.restore_purchases()
 359
 360    def _update_tabs(self) -> None:
 361        assert bui.app.classic is not None
 362        store = bui.app.classic.store
 363
 364        if not self._root_widget:
 365            return
 366        for tab_id, tab_data in list(self._purchasable_count_widgets.items()):
 367            sale_time = store.get_available_sale_time(tab_id.value)
 368
 369            if sale_time is not None:
 370                bui.textwidget(
 371                    edit=tab_data['sale_title_text'],
 372                    text=bui.Lstr(resource='store.saleText'),
 373                )
 374                bui.textwidget(
 375                    edit=tab_data['sale_time_text'],
 376                    text=bui.timestring(sale_time / 1000.0, centi=False),
 377                )
 378                bui.imagewidget(edit=tab_data['sale_img'], opacity=1.0)
 379                count = 0
 380            else:
 381                bui.textwidget(edit=tab_data['sale_title_text'], text='')
 382                bui.textwidget(edit=tab_data['sale_time_text'], text='')
 383                bui.imagewidget(edit=tab_data['sale_img'], opacity=0.0)
 384                count = store.get_available_purchase_count(tab_id.value)
 385
 386            if count > 0:
 387                bui.textwidget(edit=tab_data['text'], text=str(count))
 388                bui.imagewidget(edit=tab_data['img'], opacity=1.0)
 389            else:
 390                bui.textwidget(edit=tab_data['text'], text='')
 391                bui.imagewidget(edit=tab_data['img'], opacity=0.0)
 392
 393    def _update_tickets_text(self) -> None:
 394        from bauiv1 import SpecialChar
 395
 396        if not self._root_widget:
 397            return
 398        plus = bui.app.plus
 399        assert plus is not None
 400        sval: str | bui.Lstr
 401        if plus.get_v1_account_state() == 'signed_in':
 402            sval = bui.charstr(SpecialChar.TICKET) + str(
 403                plus.get_v1_account_ticket_count()
 404            )
 405        else:
 406            sval = bui.Lstr(resource='getTicketsWindow.titleText')
 407        if self._get_tickets_button:
 408            bui.buttonwidget(edit=self._get_tickets_button, label=sval)
 409        if self._ticket_count_text:
 410            bui.textwidget(edit=self._ticket_count_text, text=sval)
 411
 412    def _set_tab(self, tab_id: TabID) -> None:
 413        if self._current_tab is tab_id:
 414            return
 415        self._current_tab = tab_id
 416
 417        # We wanna preserve our current tab between runs.
 418        cfg = bui.app.config
 419        cfg['Store Tab'] = tab_id.value
 420        cfg.commit()
 421
 422        # Update tab colors based on which is selected.
 423        self._tab_row.update_appearance(tab_id)
 424
 425        # (Re)create scroll widget.
 426        if self._scrollwidget:
 427            self._scrollwidget.delete()
 428
 429        self._scrollwidget = bui.scrollwidget(
 430            parent=self._root_widget,
 431            highlight=False,
 432            position=(
 433                (self._width - self._scroll_width) * 0.5,
 434                self._height - self._scroll_height - 79 - 48,
 435            ),
 436            size=(self._scroll_width, self._scroll_height),
 437            claims_left_right=True,
 438            claims_tab=True,
 439            selection_loops_to_parent=True,
 440        )
 441
 442        # NOTE: this stuff is modified by the _Store class.
 443        # Should maybe clean that up.
 444        self.button_infos = {}
 445        self.update_buttons_timer = None
 446
 447        # Show status over top.
 448        if self._status_textwidget:
 449            self._status_textwidget.delete()
 450        self._status_textwidget = bui.textwidget(
 451            parent=self._root_widget,
 452            position=(self._width * 0.5, self._height * 0.5),
 453            size=(0, 0),
 454            color=(1, 0.7, 1, 0.5),
 455            h_align='center',
 456            v_align='center',
 457            text=bui.Lstr(resource=self._r + '.loadingText'),
 458            maxwidth=self._scroll_width * 0.9,
 459        )
 460
 461        class _Request:
 462            def __init__(self, window: StoreBrowserWindow):
 463                self._window = weakref.ref(window)
 464                data = {'tab': tab_id.value}
 465                bui.apptimer(0.1, bui.WeakCall(self._on_response, data))
 466
 467            def _on_response(self, data: dict[str, Any] | None) -> None:
 468                # FIXME: clean this up.
 469                # pylint: disable=protected-access
 470                window = self._window()
 471                if window is not None and (window.request is self):
 472                    window.request = None
 473                    # noinspection PyProtectedMember
 474                    window._on_response(data)
 475
 476        # Kick off a server request.
 477        self.request = _Request(self)
 478
 479    # Actually start the purchase locally.
 480    def _purchase_check_result(
 481        self, item: str, is_ticket_purchase: bool, result: dict[str, Any] | None
 482    ) -> None:
 483        plus = bui.app.plus
 484        assert plus is not None
 485        if result is None:
 486            bui.getsound('error').play()
 487            bui.screenmessage(
 488                bui.Lstr(resource='internal.unavailableNoConnectionText'),
 489                color=(1, 0, 0),
 490            )
 491        else:
 492            if is_ticket_purchase:
 493                if result['allow']:
 494                    price = plus.get_v1_account_misc_read_val(
 495                        'price.' + item, None
 496                    )
 497                    if (
 498                        price is None
 499                        or not isinstance(price, int)
 500                        or price <= 0
 501                    ):
 502                        print(
 503                            'Error; got invalid local price of',
 504                            price,
 505                            'for item',
 506                            item,
 507                        )
 508                        bui.getsound('error').play()
 509                    else:
 510                        bui.getsound('click01').play()
 511                        plus.in_game_purchase(item, price)
 512                else:
 513                    if result['reason'] == 'versionTooOld':
 514                        bui.getsound('error').play()
 515                        bui.screenmessage(
 516                            bui.Lstr(
 517                                resource='getTicketsWindow.versionTooOldText'
 518                            ),
 519                            color=(1, 0, 0),
 520                        )
 521                    else:
 522                        bui.getsound('error').play()
 523                        bui.screenmessage(
 524                            bui.Lstr(
 525                                resource='getTicketsWindow.unavailableText'
 526                            ),
 527                            color=(1, 0, 0),
 528                        )
 529            # Real in-app purchase.
 530            else:
 531                if result['allow']:
 532                    plus.purchase(item)
 533                else:
 534                    if result['reason'] == 'versionTooOld':
 535                        bui.getsound('error').play()
 536                        bui.screenmessage(
 537                            bui.Lstr(
 538                                resource='getTicketsWindow.versionTooOldText'
 539                            ),
 540                            color=(1, 0, 0),
 541                        )
 542                    else:
 543                        bui.getsound('error').play()
 544                        bui.screenmessage(
 545                            bui.Lstr(
 546                                resource='getTicketsWindow.unavailableText'
 547                            ),
 548                            color=(1, 0, 0),
 549                        )
 550
 551    def _do_purchase_check(
 552        self, item: str, is_ticket_purchase: bool = False
 553    ) -> None:
 554        app = bui.app
 555        if app.classic is None:
 556            logging.warning('_do_purchase_check() requires classic.')
 557            return
 558
 559        # Here we ping the server to ask if it's valid for us to
 560        # purchase this. Better to fail now than after we've
 561        # paid locally.
 562
 563        app.classic.master_server_v1_get(
 564            'bsAccountPurchaseCheck',
 565            {
 566                'item': item,
 567                'platform': app.classic.platform,
 568                'subplatform': app.classic.subplatform,
 569                'version': app.env.version,
 570                'buildNumber': app.env.build_number,
 571                'purchaseType': 'ticket' if is_ticket_purchase else 'real',
 572            },
 573            callback=bui.WeakCall(
 574                self._purchase_check_result, item, is_ticket_purchase
 575            ),
 576        )
 577
 578    def buy(self, item: str) -> None:
 579        """Attempt to purchase the provided item."""
 580        from bauiv1lib import account
 581        from bauiv1lib.confirm import ConfirmWindow
 582        from bauiv1lib import getcurrency
 583
 584        assert bui.app.classic is not None
 585        store = bui.app.classic.store
 586
 587        plus = bui.app.plus
 588        assert plus is not None
 589
 590        # Prevent pressing buy within a few seconds of the last press
 591        # (gives the buttons time to disable themselves and whatnot).
 592        curtime = bui.apptime()
 593        if (
 594            self._last_buy_time is not None
 595            and (curtime - self._last_buy_time) < 2.0
 596        ):
 597            bui.getsound('error').play()
 598        else:
 599            if plus.get_v1_account_state() != 'signed_in':
 600                account.show_sign_in_prompt()
 601            else:
 602                self._last_buy_time = curtime
 603
 604                # Merch is a special case - just a link.
 605                if item == 'merch':
 606                    url = bui.app.config.get('Merch Link')
 607                    if isinstance(url, str):
 608                        bui.open_url(url)
 609
 610                # Pro is an actual IAP, and the rest are ticket purchases.
 611                elif item == 'pro':
 612                    bui.getsound('click01').play()
 613
 614                    # Purchase either pro or pro_sale depending on whether
 615                    # there is a sale going on.
 616                    self._do_purchase_check(
 617                        'pro'
 618                        if store.get_available_sale_time('extras') is None
 619                        else 'pro_sale'
 620                    )
 621                else:
 622                    price = plus.get_v1_account_misc_read_val(
 623                        'price.' + item, None
 624                    )
 625                    our_tickets = plus.get_v1_account_ticket_count()
 626                    if price is not None and our_tickets < price:
 627                        bui.getsound('error').play()
 628                        getcurrency.show_get_tickets_prompt()
 629                    else:
 630
 631                        def do_it() -> None:
 632                            self._do_purchase_check(
 633                                item, is_ticket_purchase=True
 634                            )
 635
 636                        bui.getsound('swish').play()
 637                        ConfirmWindow(
 638                            bui.Lstr(
 639                                resource='store.purchaseConfirmText',
 640                                subs=[
 641                                    (
 642                                        '${ITEM}',
 643                                        store.get_store_item_name_translated(
 644                                            item
 645                                        ),
 646                                    )
 647                                ],
 648                            ),
 649                            width=400,
 650                            height=120,
 651                            action=do_it,
 652                            ok_text=bui.Lstr(
 653                                resource='store.purchaseText',
 654                                fallback_resource='okText',
 655                            ),
 656                        )
 657
 658    def _print_already_own(self, charname: str) -> None:
 659        bui.screenmessage(
 660            bui.Lstr(
 661                resource=self._r + '.alreadyOwnText',
 662                subs=[('${NAME}', charname)],
 663            ),
 664            color=(1, 0, 0),
 665        )
 666        bui.getsound('error').play()
 667
 668    def update_buttons(self) -> None:
 669        """Update our buttons."""
 670        # pylint: disable=too-many-statements
 671        # pylint: disable=too-many-branches
 672        # pylint: disable=too-many-locals
 673        from bauiv1 import SpecialChar
 674
 675        assert bui.app.classic is not None
 676        store = bui.app.classic.store
 677
 678        plus = bui.app.plus
 679        assert plus is not None
 680
 681        if not self._root_widget:
 682            return
 683
 684        sales_raw = plus.get_v1_account_misc_read_val('sales', {})
 685        sales = {}
 686        try:
 687            # Look at the current set of sales; filter any with time remaining.
 688            for sale_item, sale_info in list(sales_raw.items()):
 689                to_end = (
 690                    datetime.datetime.utcfromtimestamp(sale_info['e'])
 691                    - datetime.datetime.utcnow()
 692                ).total_seconds()
 693                if to_end > 0:
 694                    sales[sale_item] = {
 695                        'to_end': to_end,
 696                        'original_price': sale_info['op'],
 697                    }
 698        except Exception:
 699            logging.exception('Error parsing sales.')
 700
 701        assert self.button_infos is not None
 702        for b_type, b_info in self.button_infos.items():
 703            if b_type == 'merch':
 704                purchased = False
 705            elif b_type in ['upgrades.pro', 'pro']:
 706                assert bui.app.classic is not None
 707                purchased = bui.app.classic.accounts.have_pro()
 708            else:
 709                purchased = plus.get_purchased(b_type)
 710
 711            sale_opacity = 0.0
 712            sale_title_text: str | bui.Lstr = ''
 713            sale_time_text: str | bui.Lstr = ''
 714
 715            if purchased:
 716                title_color = (0.8, 0.7, 0.9, 1.0)
 717                color = (0.63, 0.55, 0.78)
 718                extra_image_opacity = 0.5
 719                call = bui.WeakCall(self._print_already_own, b_info['name'])
 720                price_text = ''
 721                price_text_left = ''
 722                price_text_right = ''
 723                show_purchase_check = True
 724                description_color: Sequence[float] = (0.4, 1.0, 0.4, 0.4)
 725                description_color2: Sequence[float] = (0.0, 0.0, 0.0, 0.0)
 726                price_color = (0.5, 1, 0.5, 0.3)
 727            else:
 728                title_color = (0.7, 0.9, 0.7, 1.0)
 729                color = (0.4, 0.8, 0.1)
 730                extra_image_opacity = 1.0
 731                call = b_info['call'] if 'call' in b_info else None
 732                if b_type == 'merch':
 733                    price_text = ''
 734                    price_text_left = ''
 735                    price_text_right = ''
 736                elif b_type in ['upgrades.pro', 'pro']:
 737                    sale_time = store.get_available_sale_time('extras')
 738                    if sale_time is not None:
 739                        priceraw = plus.get_price('pro')
 740                        price_text_left = (
 741                            priceraw if priceraw is not None else '?'
 742                        )
 743                        priceraw = plus.get_price('pro_sale')
 744                        price_text_right = (
 745                            priceraw if priceraw is not None else '?'
 746                        )
 747                        sale_opacity = 1.0
 748                        price_text = ''
 749                        sale_title_text = bui.Lstr(resource='store.saleText')
 750                        sale_time_text = bui.timestring(
 751                            sale_time / 1000.0, centi=False
 752                        )
 753                    else:
 754                        priceraw = plus.get_price('pro')
 755                        price_text = priceraw if priceraw is not None else '?'
 756                        price_text_left = ''
 757                        price_text_right = ''
 758                else:
 759                    price = plus.get_v1_account_misc_read_val(
 760                        'price.' + b_type, 0
 761                    )
 762
 763                    # Color the button differently if we cant afford this.
 764                    if plus.get_v1_account_state() == 'signed_in':
 765                        if plus.get_v1_account_ticket_count() < price:
 766                            color = (0.6, 0.61, 0.6)
 767                    price_text = bui.charstr(bui.SpecialChar.TICKET) + str(
 768                        plus.get_v1_account_misc_read_val(
 769                            'price.' + b_type, '?'
 770                        )
 771                    )
 772                    price_text_left = ''
 773                    price_text_right = ''
 774
 775                    # TESTING:
 776                    if b_type in sales:
 777                        sale_opacity = 1.0
 778                        price_text_left = bui.charstr(SpecialChar.TICKET) + str(
 779                            sales[b_type]['original_price']
 780                        )
 781                        price_text_right = price_text
 782                        price_text = ''
 783                        sale_title_text = bui.Lstr(resource='store.saleText')
 784                        sale_time_text = bui.timestring(
 785                            sales[b_type]['to_end'], centi=False
 786                        )
 787
 788                description_color = (0.5, 1.0, 0.5)
 789                description_color2 = (0.3, 1.0, 1.0)
 790                price_color = (0.2, 1, 0.2, 1.0)
 791                show_purchase_check = False
 792
 793            if 'title_text' in b_info:
 794                bui.textwidget(edit=b_info['title_text'], color=title_color)
 795            if 'purchase_check' in b_info:
 796                bui.imagewidget(
 797                    edit=b_info['purchase_check'],
 798                    opacity=1.0 if show_purchase_check else 0.0,
 799                )
 800            if 'price_widget' in b_info:
 801                bui.textwidget(
 802                    edit=b_info['price_widget'],
 803                    text=price_text,
 804                    color=price_color,
 805                )
 806            if 'price_widget_left' in b_info:
 807                bui.textwidget(
 808                    edit=b_info['price_widget_left'], text=price_text_left
 809                )
 810            if 'price_widget_right' in b_info:
 811                bui.textwidget(
 812                    edit=b_info['price_widget_right'], text=price_text_right
 813                )
 814            if 'price_slash_widget' in b_info:
 815                bui.imagewidget(
 816                    edit=b_info['price_slash_widget'], opacity=sale_opacity
 817                )
 818            if 'sale_bg_widget' in b_info:
 819                bui.imagewidget(
 820                    edit=b_info['sale_bg_widget'], opacity=sale_opacity
 821                )
 822            if 'sale_title_widget' in b_info:
 823                bui.textwidget(
 824                    edit=b_info['sale_title_widget'], text=sale_title_text
 825                )
 826            if 'sale_time_widget' in b_info:
 827                bui.textwidget(
 828                    edit=b_info['sale_time_widget'], text=sale_time_text
 829                )
 830            if 'button' in b_info:
 831                bui.buttonwidget(
 832                    edit=b_info['button'], color=color, on_activate_call=call
 833                )
 834            if 'extra_backings' in b_info:
 835                for bck in b_info['extra_backings']:
 836                    bui.imagewidget(
 837                        edit=bck, color=color, opacity=extra_image_opacity
 838                    )
 839            if 'extra_images' in b_info:
 840                for img in b_info['extra_images']:
 841                    bui.imagewidget(edit=img, opacity=extra_image_opacity)
 842            if 'extra_texts' in b_info:
 843                for etxt in b_info['extra_texts']:
 844                    bui.textwidget(edit=etxt, color=description_color)
 845            if 'extra_texts_2' in b_info:
 846                for etxt in b_info['extra_texts_2']:
 847                    bui.textwidget(edit=etxt, color=description_color2)
 848            if 'descriptionText' in b_info:
 849                bui.textwidget(
 850                    edit=b_info['descriptionText'], color=description_color
 851                )
 852
 853    def _on_response(self, data: dict[str, Any] | None) -> None:
 854        # pylint: disable=too-many-statements
 855
 856        assert bui.app.classic is not None
 857        cstore = bui.app.classic.store
 858
 859        # clear status text..
 860        if self._status_textwidget:
 861            self._status_textwidget.delete()
 862            self._status_textwidget_update_timer = None
 863
 864        if data is None:
 865            self._status_textwidget = bui.textwidget(
 866                parent=self._root_widget,
 867                position=(self._width * 0.5, self._height * 0.5),
 868                size=(0, 0),
 869                scale=1.3,
 870                transition_delay=0.1,
 871                color=(1, 0.3, 0.3, 1.0),
 872                h_align='center',
 873                v_align='center',
 874                text=bui.Lstr(resource=self._r + '.loadErrorText'),
 875                maxwidth=self._scroll_width * 0.9,
 876            )
 877        else:
 878
 879            class _Store:
 880                def __init__(
 881                    self,
 882                    store_window: StoreBrowserWindow,
 883                    sdata: dict[str, Any],
 884                    width: float,
 885                ):
 886                    self._store_window = store_window
 887                    self._width = width
 888                    store_data = cstore.get_store_layout()
 889                    self._tab = sdata['tab']
 890                    self._sections = copy.deepcopy(store_data[sdata['tab']])
 891                    self._height: float | None = None
 892
 893                    assert bui.app.classic is not None
 894                    uiscale = bui.app.ui_v1.uiscale
 895
 896                    # Pre-calc a few things and add them to store-data.
 897                    for section in self._sections:
 898                        if self._tab == 'characters':
 899                            dummy_name = 'characters.foo'
 900                        elif self._tab == 'extras':
 901                            dummy_name = 'pro'
 902                        elif self._tab == 'maps':
 903                            dummy_name = 'maps.foo'
 904                        elif self._tab == 'icons':
 905                            dummy_name = 'icons.foo'
 906                        else:
 907                            dummy_name = ''
 908                        section[
 909                            'button_size'
 910                        ] = cstore.get_store_item_display_size(dummy_name)
 911                        section['v_spacing'] = (
 912                            -25
 913                            if (
 914                                self._tab == 'extras'
 915                                and uiscale is bui.UIScale.SMALL
 916                            )
 917                            else -17
 918                            if self._tab == 'characters'
 919                            else 0
 920                        )
 921                        if 'title' not in section:
 922                            section['title'] = ''
 923                        section['x_offs'] = (
 924                            130
 925                            if self._tab == 'extras'
 926                            else 270
 927                            if self._tab == 'maps'
 928                            else 0
 929                        )
 930                        section['y_offs'] = (
 931                            20
 932                            if (
 933                                self._tab == 'extras'
 934                                and uiscale is bui.UIScale.SMALL
 935                                and bui.app.config.get('Merch Link')
 936                            )
 937                            else 55
 938                            if (
 939                                self._tab == 'extras'
 940                                and uiscale is bui.UIScale.SMALL
 941                            )
 942                            else -20
 943                            if self._tab == 'icons'
 944                            else 0
 945                        )
 946
 947                def instantiate(
 948                    self, scrollwidget: bui.Widget, tab_button: bui.Widget
 949                ) -> None:
 950                    """Create the store."""
 951                    # pylint: disable=too-many-locals
 952                    # pylint: disable=too-many-branches
 953                    # pylint: disable=too-many-nested-blocks
 954                    from bauiv1lib.store.item import (
 955                        instantiate_store_item_display,
 956                    )
 957
 958                    title_spacing = 40
 959                    button_border = 20
 960                    button_spacing = 4
 961                    boffs_h = 40
 962                    self._height = 80.0
 963
 964                    # Calc total height.
 965                    for i, section in enumerate(self._sections):
 966                        if section['title'] != '':
 967                            assert self._height is not None
 968                            self._height += title_spacing
 969                        b_width, b_height = section['button_size']
 970                        b_column_count = int(
 971                            math.floor(
 972                                (self._width - boffs_h - 20)
 973                                / (b_width + button_spacing)
 974                            )
 975                        )
 976                        b_row_count = int(
 977                            math.ceil(
 978                                float(len(section['items'])) / b_column_count
 979                            )
 980                        )
 981                        b_height_total = (
 982                            2 * button_border
 983                            + b_row_count * b_height
 984                            + (b_row_count - 1) * section['v_spacing']
 985                        )
 986                        self._height += b_height_total
 987
 988                    assert self._height is not None
 989                    cnt2 = bui.containerwidget(
 990                        parent=scrollwidget,
 991                        scale=1.0,
 992                        size=(self._width, self._height),
 993                        background=False,
 994                        claims_left_right=True,
 995                        claims_tab=True,
 996                        selection_loops_to_parent=True,
 997                    )
 998                    v = self._height - 20
 999
1000                    if self._tab == 'characters':
1001                        txt = bui.Lstr(
1002                            resource='store.howToSwitchCharactersText',
1003                            subs=[
1004                                (
1005                                    '${SETTINGS}',
1006                                    bui.Lstr(
1007                                        resource=(
1008                                            'accountSettingsWindow.titleText'
1009                                        )
1010                                    ),
1011                                ),
1012                                (
1013                                    '${PLAYER_PROFILES}',
1014                                    bui.Lstr(
1015                                        resource=(
1016                                            'playerProfilesWindow.titleText'
1017                                        )
1018                                    ),
1019                                ),
1020                            ],
1021                        )
1022                        bui.textwidget(
1023                            parent=cnt2,
1024                            text=txt,
1025                            size=(0, 0),
1026                            position=(self._width * 0.5, self._height - 28),
1027                            h_align='center',
1028                            v_align='center',
1029                            color=(0.7, 1, 0.7, 0.4),
1030                            scale=0.7,
1031                            shadow=0,
1032                            flatness=1.0,
1033                            maxwidth=700,
1034                            transition_delay=0.4,
1035                        )
1036                    elif self._tab == 'icons':
1037                        txt = bui.Lstr(
1038                            resource='store.howToUseIconsText',
1039                            subs=[
1040                                (
1041                                    '${SETTINGS}',
1042                                    bui.Lstr(resource='mainMenu.settingsText'),
1043                                ),
1044                                (
1045                                    '${PLAYER_PROFILES}',
1046                                    bui.Lstr(
1047                                        resource=(
1048                                            'playerProfilesWindow.titleText'
1049                                        )
1050                                    ),
1051                                ),
1052                            ],
1053                        )
1054                        bui.textwidget(
1055                            parent=cnt2,
1056                            text=txt,
1057                            size=(0, 0),
1058                            position=(self._width * 0.5, self._height - 28),
1059                            h_align='center',
1060                            v_align='center',
1061                            color=(0.7, 1, 0.7, 0.4),
1062                            scale=0.7,
1063                            shadow=0,
1064                            flatness=1.0,
1065                            maxwidth=700,
1066                            transition_delay=0.4,
1067                        )
1068                    elif self._tab == 'maps':
1069                        assert self._width is not None
1070                        assert self._height is not None
1071                        txt = bui.Lstr(resource='store.howToUseMapsText')
1072                        bui.textwidget(
1073                            parent=cnt2,
1074                            text=txt,
1075                            size=(0, 0),
1076                            position=(self._width * 0.5, self._height - 28),
1077                            h_align='center',
1078                            v_align='center',
1079                            color=(0.7, 1, 0.7, 0.4),
1080                            scale=0.7,
1081                            shadow=0,
1082                            flatness=1.0,
1083                            maxwidth=700,
1084                            transition_delay=0.4,
1085                        )
1086
1087                    prev_row_buttons: list | None = None
1088                    this_row_buttons = []
1089
1090                    delay = 0.3
1091                    for section in self._sections:
1092                        if section['title'] != '':
1093                            bui.textwidget(
1094                                parent=cnt2,
1095                                position=(60, v - title_spacing * 0.8),
1096                                size=(0, 0),
1097                                scale=1.0,
1098                                transition_delay=delay,
1099                                color=(0.7, 0.9, 0.7, 1),
1100                                h_align='left',
1101                                v_align='center',
1102                                text=bui.Lstr(resource=section['title']),
1103                                maxwidth=self._width * 0.7,
1104                            )
1105                            v -= title_spacing
1106                        delay = max(0.100, delay - 0.100)
1107                        v -= button_border
1108                        b_width, b_height = section['button_size']
1109                        b_count = len(section['items'])
1110                        b_column_count = int(
1111                            math.floor(
1112                                (self._width - boffs_h - 20)
1113                                / (b_width + button_spacing)
1114                            )
1115                        )
1116                        col = 0
1117                        item: dict[str, Any]
1118                        assert self._store_window.button_infos is not None
1119                        for i, item_name in enumerate(section['items']):
1120                            item = self._store_window.button_infos[
1121                                item_name
1122                            ] = {}
1123                            item['call'] = bui.WeakCall(
1124                                self._store_window.buy, item_name
1125                            )
1126                            if 'x_offs' in section:
1127                                boffs_h2 = section['x_offs']
1128                            else:
1129                                boffs_h2 = 0
1130
1131                            if 'y_offs' in section:
1132                                boffs_v2 = section['y_offs']
1133                            else:
1134                                boffs_v2 = 0
1135                            b_pos = (
1136                                boffs_h
1137                                + boffs_h2
1138                                + (b_width + button_spacing) * col,
1139                                v - b_height + boffs_v2,
1140                            )
1141                            instantiate_store_item_display(
1142                                item_name,
1143                                item,
1144                                parent_widget=cnt2,
1145                                b_pos=b_pos,
1146                                boffs_h=boffs_h,
1147                                b_width=b_width,
1148                                b_height=b_height,
1149                                boffs_h2=boffs_h2,
1150                                boffs_v2=boffs_v2,
1151                                delay=delay,
1152                            )
1153                            btn = item['button']
1154                            delay = max(0.1, delay - 0.1)
1155                            this_row_buttons.append(btn)
1156
1157                            # Wire this button to the equivalent in the
1158                            # previous row.
1159                            if prev_row_buttons is not None:
1160                                if len(prev_row_buttons) > col:
1161                                    bui.widget(
1162                                        edit=btn,
1163                                        up_widget=prev_row_buttons[col],
1164                                    )
1165                                    bui.widget(
1166                                        edit=prev_row_buttons[col],
1167                                        down_widget=btn,
1168                                    )
1169
1170                                    # If we're the last button in our row,
1171                                    # wire any in the previous row past
1172                                    # our position to go to us if down is
1173                                    # pressed.
1174                                    if (
1175                                        col + 1 == b_column_count
1176                                        or i == b_count - 1
1177                                    ):
1178                                        for b_prev in prev_row_buttons[
1179                                            col + 1 :
1180                                        ]:
1181                                            bui.widget(
1182                                                edit=b_prev, down_widget=btn
1183                                            )
1184                                else:
1185                                    bui.widget(
1186                                        edit=btn, up_widget=prev_row_buttons[-1]
1187                                    )
1188                            else:
1189                                bui.widget(edit=btn, up_widget=tab_button)
1190
1191                            col += 1
1192                            if col == b_column_count or i == b_count - 1:
1193                                prev_row_buttons = this_row_buttons
1194                                this_row_buttons = []
1195                                col = 0
1196                                v -= b_height
1197                                if i < b_count - 1:
1198                                    v -= section['v_spacing']
1199
1200                        v -= button_border
1201
1202                    # Set a timer to update these buttons periodically as long
1203                    # as we're alive (so if we buy one it will grey out, etc).
1204                    self._store_window.update_buttons_timer = bui.AppTimer(
1205                        0.5,
1206                        bui.WeakCall(self._store_window.update_buttons),
1207                        repeat=True,
1208                    )
1209
1210                    # Also update them immediately.
1211                    self._store_window.update_buttons()
1212
1213            if self._current_tab in (
1214                self.TabID.EXTRAS,
1215                self.TabID.MINIGAMES,
1216                self.TabID.CHARACTERS,
1217                self.TabID.MAPS,
1218                self.TabID.ICONS,
1219            ):
1220                store = _Store(self, data, self._scroll_width)
1221                assert self._scrollwidget is not None
1222                store.instantiate(
1223                    scrollwidget=self._scrollwidget,
1224                    tab_button=self._tab_row.tabs[self._current_tab].button,
1225                )
1226            else:
1227                cnt = bui.containerwidget(
1228                    parent=self._scrollwidget,
1229                    scale=1.0,
1230                    size=(self._scroll_width, self._scroll_height * 0.95),
1231                    background=False,
1232                    claims_left_right=True,
1233                    claims_tab=True,
1234                    selection_loops_to_parent=True,
1235                )
1236                self._status_textwidget = bui.textwidget(
1237                    parent=cnt,
1238                    position=(
1239                        self._scroll_width * 0.5,
1240                        self._scroll_height * 0.5,
1241                    ),
1242                    size=(0, 0),
1243                    scale=1.3,
1244                    transition_delay=0.1,
1245                    color=(1, 1, 0.3, 1.0),
1246                    h_align='center',
1247                    v_align='center',
1248                    text=bui.Lstr(resource=self._r + '.comingSoonText'),
1249                    maxwidth=self._scroll_width * 0.9,
1250                )
1251
1252    def _save_state(self) -> None:
1253        try:
1254            sel = self._root_widget.get_selected_child()
1255            selected_tab_ids = [
1256                tab_id
1257                for tab_id, tab in self._tab_row.tabs.items()
1258                if sel == tab.button
1259            ]
1260            if sel == self._get_tickets_button:
1261                sel_name = 'GetTickets'
1262            elif sel == self._scrollwidget:
1263                sel_name = 'Scroll'
1264            elif sel == self._back_button:
1265                sel_name = 'Back'
1266            elif selected_tab_ids:
1267                assert len(selected_tab_ids) == 1
1268                sel_name = f'Tab:{selected_tab_ids[0].value}'
1269            else:
1270                raise ValueError(f'unrecognized selection \'{sel}\'')
1271            assert bui.app.classic is not None
1272            bui.app.ui_v1.window_states[type(self)] = {
1273                'sel_name': sel_name,
1274            }
1275        except Exception:
1276            logging.exception('Error saving state for %s.', self)
1277
1278    def _restore_state(self) -> None:
1279        from efro.util import enum_by_value
1280
1281        try:
1282            sel: bui.Widget | None
1283            assert bui.app.classic is not None
1284            sel_name = bui.app.ui_v1.window_states.get(type(self), {}).get(
1285                'sel_name'
1286            )
1287            assert isinstance(sel_name, (str, type(None)))
1288
1289            try:
1290                current_tab = enum_by_value(
1291                    self.TabID, bui.app.config.get('Store Tab')
1292                )
1293            except ValueError:
1294                current_tab = self.TabID.CHARACTERS
1295
1296            if self._show_tab is not None:
1297                current_tab = self._show_tab
1298            if sel_name == 'GetTickets' and self._get_tickets_button:
1299                sel = self._get_tickets_button
1300            elif sel_name == 'Back':
1301                sel = self._back_button
1302            elif sel_name == 'Scroll':
1303                sel = self._scrollwidget
1304            elif isinstance(sel_name, str) and sel_name.startswith('Tab:'):
1305                try:
1306                    sel_tab_id = enum_by_value(
1307                        self.TabID, sel_name.split(':')[-1]
1308                    )
1309                except ValueError:
1310                    sel_tab_id = self.TabID.CHARACTERS
1311                sel = self._tab_row.tabs[sel_tab_id].button
1312            else:
1313                sel = self._tab_row.tabs[current_tab].button
1314
1315            # If we were requested to show a tab, select it too..
1316            if (
1317                self._show_tab is not None
1318                and self._show_tab in self._tab_row.tabs
1319            ):
1320                sel = self._tab_row.tabs[self._show_tab].button
1321            self._set_tab(current_tab)
1322            if sel is not None:
1323                bui.containerwidget(edit=self._root_widget, selected_child=sel)
1324        except Exception:
1325            logging.exception('Error restoring state for %s.', self)
1326
1327    def _on_get_more_tickets_press(self) -> None:
1328        # pylint: disable=cyclic-import
1329        from bauiv1lib.account import show_sign_in_prompt
1330        from bauiv1lib.getcurrency import GetCurrencyWindow
1331
1332        plus = bui.app.plus
1333        assert plus is not None
1334
1335        if plus.get_v1_account_state() != 'signed_in':
1336            show_sign_in_prompt()
1337            return
1338        self._save_state()
1339        bui.containerwidget(edit=self._root_widget, transition='out_left')
1340        window = GetCurrencyWindow(
1341            from_modal_store=self._modal,
1342            store_back_location=self._back_location,
1343        ).get_root_widget()
1344        if not self._modal:
1345            assert bui.app.classic is not None
1346            bui.app.ui_v1.set_main_menu_window(window)
1347
1348    def _back(self) -> None:
1349        # pylint: disable=cyclic-import
1350        from bauiv1lib.coop.browser import CoopBrowserWindow
1351        from bauiv1lib.mainmenu import MainMenuWindow
1352
1353        self._save_state()
1354        bui.containerwidget(
1355            edit=self._root_widget, transition=self._transition_out
1356        )
1357        if not self._modal:
1358            assert bui.app.classic is not None
1359            if self._back_location == 'CoopBrowserWindow':
1360                bui.app.ui_v1.set_main_menu_window(
1361                    CoopBrowserWindow(transition='in_left').get_root_widget()
1362                )
1363            else:
1364                bui.app.ui_v1.set_main_menu_window(
1365                    MainMenuWindow(transition='in_left').get_root_widget()
1366                )
1367        if self._on_close_call is not None:
1368            self._on_close_call()
1369
1370
1371def _check_merch_availability_in_bg_thread() -> None:
1372    # pylint: disable=cell-var-from-loop
1373
1374    # Merch is available from some countries only.
1375    # Make a reasonable check to ask the master-server about this at
1376    # launch and store the results.
1377    plus = bui.app.plus
1378    assert plus is not None
1379
1380    for _i in range(15):
1381        try:
1382            if plus.cloud.is_connected():
1383                response = plus.cloud.send_message(
1384                    bacommon.cloud.MerchAvailabilityMessage()
1385                )
1386
1387                def _store_in_logic_thread() -> None:
1388                    cfg = bui.app.config
1389                    current: str | None = cfg.get(MERCH_LINK_KEY)
1390                    if not isinstance(current, str | None):
1391                        current = None
1392                    if current != response.url:
1393                        cfg[MERCH_LINK_KEY] = response.url
1394                        cfg.commit()
1395
1396                # If we successfully get a response, kick it over to the
1397                # logic thread to store and we're done.
1398                bui.pushcall(_store_in_logic_thread, from_other_thread=True)
1399                return
1400        except CommunicationError:
1401            pass
1402        except Exception:
1403            logging.warning(
1404                'Unexpected error in merch-availability-check.', exc_info=True
1405            )
1406        time.sleep(1.1934)  # A bit randomized to avoid aliasing.
1407
1408
1409# Slight hack; start checking merch availability in the bg (but only if
1410# it looks like we've been imported for use in a running app; don't want
1411# to do this during docs generation/etc.)
1412if (
1413    os.environ.get('BA_RUNNING_WITH_DUMMY_MODULES') != '1'
1414    and bui.app.state is not bui.app.State.NOT_RUNNING
1415):
1416    Thread(target=_check_merch_availability_in_bg_thread, daemon=True).start()