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