bastd.ui.purchase

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""UI related to purchasing items."""
  4
  5from __future__ import annotations
  6
  7from typing import TYPE_CHECKING
  8
  9import ba
 10import ba.internal
 11
 12if TYPE_CHECKING:
 13    from typing import Any
 14
 15
 16class PurchaseWindow(ba.Window):
 17    """Window for purchasing one or more items."""
 18
 19    def __init__(
 20        self,
 21        items: list[str],
 22        transition: str = 'in_right',
 23        header_text: ba.Lstr | None = None,
 24    ):
 25        from ba.internal import get_store_item_display_size
 26        from bastd.ui.store import item as storeitemui
 27
 28        if header_text is None:
 29            header_text = ba.Lstr(
 30                resource='unlockThisText',
 31                fallback_resource='unlockThisInTheStoreText',
 32            )
 33        if len(items) != 1:
 34            raise ValueError('expected exactly 1 item')
 35        self._items = list(items)
 36        self._width = 580
 37        self._height = 520
 38        uiscale = ba.app.ui.uiscale
 39        super().__init__(
 40            root_widget=ba.containerwidget(
 41                size=(self._width, self._height),
 42                transition=transition,
 43                toolbar_visibility='menu_currency',
 44                scale=(
 45                    1.2
 46                    if uiscale is ba.UIScale.SMALL
 47                    else 1.1
 48                    if uiscale is ba.UIScale.MEDIUM
 49                    else 1.0
 50                ),
 51                stack_offset=(0, -15)
 52                if uiscale is ba.UIScale.SMALL
 53                else (0, 0),
 54            )
 55        )
 56        self._is_double = False
 57        self._title_text = ba.textwidget(
 58            parent=self._root_widget,
 59            position=(self._width * 0.5, self._height - 30),
 60            size=(0, 0),
 61            text=header_text,
 62            h_align='center',
 63            v_align='center',
 64            maxwidth=self._width * 0.9 - 120,
 65            scale=1.2,
 66            color=(1, 0.8, 0.3, 1),
 67        )
 68        size = get_store_item_display_size(items[0])
 69        display: dict[str, Any] = {}
 70        storeitemui.instantiate_store_item_display(
 71            items[0],
 72            display,
 73            parent_widget=self._root_widget,
 74            b_pos=(
 75                self._width * 0.5
 76                - size[0] * 0.5
 77                + 10
 78                - ((size[0] * 0.5 + 30) if self._is_double else 0),
 79                self._height * 0.5
 80                - size[1] * 0.5
 81                + 30
 82                + (20 if self._is_double else 0),
 83            ),
 84            b_width=size[0],
 85            b_height=size[1],
 86            button=False,
 87        )
 88
 89        # Wire up the parts we need.
 90        if self._is_double:
 91            pass  # not working
 92        else:
 93            if self._items == ['pro']:
 94                price_str = ba.internal.get_price(self._items[0])
 95                pyoffs = -15
 96            else:
 97                pyoffs = 0
 98                price = self._price = ba.internal.get_v1_account_misc_read_val(
 99                    'price.' + str(items[0]), -1
100                )
101                price_str = ba.charstr(ba.SpecialChar.TICKET) + str(price)
102            self._price_text = ba.textwidget(
103                parent=self._root_widget,
104                position=(self._width * 0.5, 150 + pyoffs),
105                size=(0, 0),
106                text=price_str,
107                h_align='center',
108                v_align='center',
109                maxwidth=self._width * 0.9,
110                scale=1.4,
111                color=(0.2, 1, 0.2),
112            )
113
114        self._update_timer = ba.Timer(
115            1.0,
116            ba.WeakCall(self._update),
117            timetype=ba.TimeType.REAL,
118            repeat=True,
119        )
120
121        self._cancel_button = ba.buttonwidget(
122            parent=self._root_widget,
123            position=(50, 40),
124            size=(150, 60),
125            scale=1.0,
126            on_activate_call=self._cancel,
127            autoselect=True,
128            label=ba.Lstr(resource='cancelText'),
129        )
130        self._purchase_button = ba.buttonwidget(
131            parent=self._root_widget,
132            position=(self._width - 200, 40),
133            size=(150, 60),
134            scale=1.0,
135            on_activate_call=self._purchase,
136            autoselect=True,
137            label=ba.Lstr(resource='store.purchaseText'),
138        )
139
140        ba.containerwidget(
141            edit=self._root_widget,
142            cancel_button=self._cancel_button,
143            start_button=self._purchase_button,
144            selected_child=self._purchase_button,
145        )
146
147    def _update(self) -> None:
148        can_die = False
149
150        # We go away if we see that our target item is owned.
151        if self._items == ['pro']:
152            if ba.app.accounts_v1.have_pro():
153                can_die = True
154        else:
155            if ba.internal.get_purchased(self._items[0]):
156                can_die = True
157
158        if can_die:
159            ba.containerwidget(edit=self._root_widget, transition='out_left')
160
161    def _purchase(self) -> None:
162        from bastd.ui import getcurrency
163
164        if self._items == ['pro']:
165            ba.internal.purchase('pro')
166        else:
167            ticket_count: int | None
168            try:
169                ticket_count = ba.internal.get_v1_account_ticket_count()
170            except Exception:
171                ticket_count = None
172            if ticket_count is not None and ticket_count < self._price:
173                getcurrency.show_get_tickets_prompt()
174                ba.playsound(ba.getsound('error'))
175                return
176
177            def do_it() -> None:
178                ba.internal.in_game_purchase(self._items[0], self._price)
179
180            ba.playsound(ba.getsound('swish'))
181            do_it()
182
183    def _cancel(self) -> None:
184        ba.containerwidget(edit=self._root_widget, transition='out_right')