bastd.ui.specialoffer

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""UI for presenting sales/etc."""
  4
  5from __future__ import annotations
  6
  7import copy
  8from typing import TYPE_CHECKING
  9
 10import _ba
 11import ba
 12
 13if TYPE_CHECKING:
 14    from typing import Any
 15
 16
 17class SpecialOfferWindow(ba.Window):
 18    """Window for presenting sales/etc."""
 19
 20    def __init__(self, offer: dict[str, Any], transition: str = 'in_right'):
 21        # pylint: disable=too-many-statements
 22        # pylint: disable=too-many-branches
 23        # pylint: disable=too-many-locals
 24        from ba.internal import (get_store_item_display_size, get_clean_price)
 25        from ba import SpecialChar
 26        from bastd.ui.store import item as storeitemui
 27        self._cancel_delay = offer.get('cancelDelay', 0)
 28
 29        # First thing: if we're offering pro or an IAP, see if we have a
 30        # price for it.
 31        # If not, abort and go into zombie mode (the user should never see
 32        # us that way).
 33
 34        real_price: str | None
 35
 36        # Misnomer: 'pro' actually means offer 'pro_sale'.
 37        if offer['item'] in ['pro', 'pro_fullprice']:
 38            real_price = _ba.get_price('pro' if offer['item'] ==
 39                                       'pro_fullprice' else 'pro_sale')
 40            if real_price is None and ba.app.debug_build:
 41                print('NOTE: Faking prices for debug build.')
 42                real_price = '$1.23'
 43            zombie = real_price is None
 44        elif isinstance(offer['price'], str):
 45            # (a string price implies IAP id)
 46            real_price = _ba.get_price(offer['price'])
 47            if real_price is None and ba.app.debug_build:
 48                print('NOTE: Faking price for debug build.')
 49                real_price = '$1.23'
 50            zombie = real_price is None
 51        else:
 52            real_price = None
 53            zombie = False
 54        if real_price is None:
 55            real_price = '?'
 56
 57        if offer['item'] in ['pro', 'pro_fullprice']:
 58            self._offer_item = 'pro'
 59        else:
 60            self._offer_item = offer['item']
 61
 62        # If we wanted a real price but didn't find one, go zombie.
 63        if zombie:
 64            return
 65
 66        # This can pop up suddenly, so lets block input for 1 second.
 67        _ba.lock_all_input()
 68        ba.timer(1.0, _ba.unlock_all_input, timetype=ba.TimeType.REAL)
 69        ba.playsound(ba.getsound('ding'))
 70        ba.timer(0.3,
 71                 lambda: ba.playsound(ba.getsound('ooh')),
 72                 timetype=ba.TimeType.REAL)
 73        self._offer = copy.deepcopy(offer)
 74        self._width = 580
 75        self._height = 590
 76        uiscale = ba.app.ui.uiscale
 77        super().__init__(root_widget=ba.containerwidget(
 78            size=(self._width, self._height),
 79            transition=transition,
 80            scale=(1.2 if uiscale is ba.UIScale.SMALL else
 81                   1.15 if uiscale is ba.UIScale.MEDIUM else 1.0),
 82            stack_offset=(0, -15) if uiscale is ba.UIScale.SMALL else (0, 0)))
 83        self._is_bundle_sale = False
 84        try:
 85            if offer['item'] in ['pro', 'pro_fullprice']:
 86                original_price_str = _ba.get_price('pro')
 87                if original_price_str is None:
 88                    original_price_str = '?'
 89                new_price_str = _ba.get_price('pro_sale')
 90                if new_price_str is None:
 91                    new_price_str = '?'
 92                percent_off_text = ''
 93            else:
 94                # If the offer includes bonus tickets, it's a bundle-sale.
 95                if ('bonusTickets' in offer
 96                        and offer['bonusTickets'] is not None):
 97                    self._is_bundle_sale = True
 98                original_price = _ba.get_v1_account_misc_read_val(
 99                    'price.' + self._offer_item, 9999)
100
101                # For pure ticket prices we can show a percent-off.
102                if isinstance(offer['price'], int):
103                    new_price = offer['price']
104                    tchar = ba.charstr(SpecialChar.TICKET)
105                    original_price_str = tchar + str(original_price)
106                    new_price_str = tchar + str(new_price)
107                    percent_off = int(
108                        round(100.0 -
109                              (float(new_price) / original_price) * 100.0))
110                    percent_off_text = ' ' + ba.Lstr(
111                        resource='store.salePercentText').evaluate().replace(
112                            '${PERCENT}', str(percent_off))
113                else:
114                    original_price_str = new_price_str = '?'
115                    percent_off_text = ''
116
117        except Exception:
118            print(f'Offer: {offer}')
119            ba.print_exception('Error setting up special-offer')
120            original_price_str = new_price_str = '?'
121            percent_off_text = ''
122
123        # If its a bundle sale, change the title.
124        if self._is_bundle_sale:
125            sale_text = ba.Lstr(resource='store.saleBundleText',
126                                fallback_resource='store.saleText').evaluate()
127        else:
128            # For full pro we say 'Upgrade?' since its not really a sale.
129            if offer['item'] == 'pro_fullprice':
130                sale_text = ba.Lstr(
131                    resource='store.upgradeQuestionText',
132                    fallback_resource='store.saleExclaimText').evaluate()
133            else:
134                sale_text = ba.Lstr(
135                    resource='store.saleExclaimText',
136                    fallback_resource='store.saleText').evaluate()
137
138        self._title_text = ba.textwidget(
139            parent=self._root_widget,
140            position=(self._width * 0.5, self._height - 40),
141            size=(0, 0),
142            text=sale_text +
143            ((' ' + ba.Lstr(resource='store.oneTimeOnlyText').evaluate())
144             if self._offer['oneTimeOnly'] else '') + percent_off_text,
145            h_align='center',
146            v_align='center',
147            maxwidth=self._width * 0.9 - 220,
148            scale=1.4,
149            color=(0.3, 1, 0.3))
150
151        self._flash_on = False
152        self._flashing_timer: ba.Timer | None = ba.Timer(
153            0.05,
154            ba.WeakCall(self._flash_cycle),
155            repeat=True,
156            timetype=ba.TimeType.REAL)
157        ba.timer(0.6,
158                 ba.WeakCall(self._stop_flashing),
159                 timetype=ba.TimeType.REAL)
160
161        size = get_store_item_display_size(self._offer_item)
162        display: dict[str, Any] = {}
163        storeitemui.instantiate_store_item_display(
164            self._offer_item,
165            display,
166            parent_widget=self._root_widget,
167            b_pos=(self._width * 0.5 - size[0] * 0.5 + 10 -
168                   ((size[0] * 0.5 + 30) if self._is_bundle_sale else 0),
169                   self._height * 0.5 - size[1] * 0.5 + 20 +
170                   (20 if self._is_bundle_sale else 0)),
171            b_width=size[0],
172            b_height=size[1],
173            button=not self._is_bundle_sale)
174
175        # Wire up the parts we need.
176        if self._is_bundle_sale:
177            self._plus_text = ba.textwidget(parent=self._root_widget,
178                                            position=(self._width * 0.5,
179                                                      self._height * 0.5 + 50),
180                                            size=(0, 0),
181                                            text='+',
182                                            h_align='center',
183                                            v_align='center',
184                                            maxwidth=self._width * 0.9,
185                                            scale=1.4,
186                                            color=(0.5, 0.5, 0.5))
187            self._plus_tickets = ba.textwidget(
188                parent=self._root_widget,
189                position=(self._width * 0.5 + 120, self._height * 0.5 + 50),
190                size=(0, 0),
191                text=ba.charstr(SpecialChar.TICKET_BACKING) +
192                str(offer['bonusTickets']),
193                h_align='center',
194                v_align='center',
195                maxwidth=self._width * 0.9,
196                scale=2.5,
197                color=(0.2, 1, 0.2))
198            self._price_text = ba.textwidget(parent=self._root_widget,
199                                             position=(self._width * 0.5, 150),
200                                             size=(0, 0),
201                                             text=real_price,
202                                             h_align='center',
203                                             v_align='center',
204                                             maxwidth=self._width * 0.9,
205                                             scale=1.4,
206                                             color=(0.2, 1, 0.2))
207            # Total-value if they supplied it.
208            total_worth_item = offer.get('valueItem', None)
209            if total_worth_item is not None:
210                price = _ba.get_price(total_worth_item)
211                total_worth_price = (get_clean_price(price)
212                                     if price is not None else None)
213                if total_worth_price is not None:
214                    total_worth_text = ba.Lstr(resource='store.totalWorthText',
215                                               subs=[('${TOTAL_WORTH}',
216                                                      total_worth_price)])
217                    self._total_worth_text = ba.textwidget(
218                        parent=self._root_widget,
219                        text=total_worth_text,
220                        position=(self._width * 0.5, 210),
221                        scale=0.9,
222                        maxwidth=self._width * 0.7,
223                        size=(0, 0),
224                        h_align='center',
225                        v_align='center',
226                        shadow=1.0,
227                        flatness=1.0,
228                        color=(0.3, 1, 1))
229
230        elif offer['item'] == 'pro_fullprice':
231            # for full-price pro we simply show full price
232            ba.textwidget(edit=display['price_widget'], text=real_price)
233            ba.buttonwidget(edit=display['button'],
234                            on_activate_call=self._purchase)
235        else:
236            # Show old/new prices otherwise (for pro sale).
237            ba.buttonwidget(edit=display['button'],
238                            on_activate_call=self._purchase)
239            ba.imagewidget(edit=display['price_slash_widget'], opacity=1.0)
240            ba.textwidget(edit=display['price_widget_left'],
241                          text=original_price_str)
242            ba.textwidget(edit=display['price_widget_right'],
243                          text=new_price_str)
244
245        # Add ticket button only if this is ticket-purchasable.
246        if isinstance(offer.get('price'), int):
247            self._get_tickets_button = ba.buttonwidget(
248                parent=self._root_widget,
249                position=(self._width - 125, self._height - 68),
250                size=(90, 55),
251                scale=1.0,
252                button_type='square',
253                color=(0.7, 0.5, 0.85),
254                textcolor=(0.2, 1, 0.2),
255                autoselect=True,
256                label=ba.Lstr(resource='getTicketsWindow.titleText'),
257                on_activate_call=self._on_get_more_tickets_press)
258
259            self._ticket_text_update_timer = ba.Timer(
260                1.0,
261                ba.WeakCall(self._update_tickets_text),
262                timetype=ba.TimeType.REAL,
263                repeat=True)
264            self._update_tickets_text()
265
266        self._update_timer = ba.Timer(1.0,
267                                      ba.WeakCall(self._update),
268                                      timetype=ba.TimeType.REAL,
269                                      repeat=True)
270
271        self._cancel_button = ba.buttonwidget(
272            parent=self._root_widget,
273            position=(50, 40) if self._is_bundle_sale else
274            (self._width * 0.5 - 75, 40),
275            size=(150, 60),
276            scale=1.0,
277            on_activate_call=self._cancel,
278            autoselect=True,
279            label=ba.Lstr(resource='noThanksText'))
280        self._cancel_countdown_text = ba.textwidget(
281            parent=self._root_widget,
282            text='',
283            position=(50 + 150 + 20, 40 + 27) if self._is_bundle_sale else
284            (self._width * 0.5 - 75 + 150 + 20, 40 + 27),
285            scale=1.1,
286            size=(0, 0),
287            h_align='left',
288            v_align='center',
289            shadow=1.0,
290            flatness=1.0,
291            color=(0.6, 0.5, 0.5))
292        self._update_cancel_button_graphics()
293
294        if self._is_bundle_sale:
295            self._purchase_button = ba.buttonwidget(
296                parent=self._root_widget,
297                position=(self._width - 200, 40),
298                size=(150, 60),
299                scale=1.0,
300                on_activate_call=self._purchase,
301                autoselect=True,
302                label=ba.Lstr(resource='store.purchaseText'))
303
304        ba.containerwidget(edit=self._root_widget,
305                           cancel_button=self._cancel_button,
306                           start_button=self._purchase_button
307                           if self._is_bundle_sale else None,
308                           selected_child=self._purchase_button
309                           if self._is_bundle_sale else display['button'])
310
311    def _stop_flashing(self) -> None:
312        self._flashing_timer = None
313        ba.textwidget(edit=self._title_text, color=(0.3, 1, 0.3))
314
315    def _flash_cycle(self) -> None:
316        if not self._root_widget:
317            return
318        self._flash_on = not self._flash_on
319        ba.textwidget(edit=self._title_text,
320                      color=(0.3, 1, 0.3) if self._flash_on else (1, 0.5, 0))
321
322    def _update_cancel_button_graphics(self) -> None:
323        ba.buttonwidget(edit=self._cancel_button,
324                        color=(0.5, 0.5, 0.5) if self._cancel_delay > 0 else
325                        (0.7, 0.4, 0.34),
326                        textcolor=(0.5, 0.5,
327                                   0.5) if self._cancel_delay > 0 else
328                        (0.9, 0.9, 1.0))
329        ba.textwidget(
330            edit=self._cancel_countdown_text,
331            text=str(self._cancel_delay) if self._cancel_delay > 0 else '')
332
333    def _update(self) -> None:
334
335        # If we've got seconds left on our countdown, update it.
336        if self._cancel_delay > 0:
337            self._cancel_delay = max(0, self._cancel_delay - 1)
338            self._update_cancel_button_graphics()
339
340        can_die = False
341
342        # We go away if we see that our target item is owned.
343        if self._offer_item == 'pro':
344            if _ba.app.accounts_v1.have_pro():
345                can_die = True
346        else:
347            if _ba.get_purchased(self._offer_item):
348                can_die = True
349
350        if can_die:
351            self._transition_out('out_left')
352
353    def _transition_out(self, transition: str = 'out_left') -> None:
354        # Also clear any pending-special-offer we've stored at this point.
355        cfg = ba.app.config
356        if 'pendingSpecialOffer' in cfg:
357            del cfg['pendingSpecialOffer']
358            cfg.commit()
359
360        ba.containerwidget(edit=self._root_widget, transition=transition)
361
362    def _update_tickets_text(self) -> None:
363        from ba import SpecialChar
364        if not self._root_widget:
365            return
366        sval: str | ba.Lstr
367        if _ba.get_v1_account_state() == 'signed_in':
368            sval = (ba.charstr(SpecialChar.TICKET) +
369                    str(_ba.get_v1_account_ticket_count()))
370        else:
371            sval = ba.Lstr(resource='getTicketsWindow.titleText')
372        ba.buttonwidget(edit=self._get_tickets_button, label=sval)
373
374    def _on_get_more_tickets_press(self) -> None:
375        from bastd.ui import account
376        from bastd.ui import getcurrency
377        if _ba.get_v1_account_state() != 'signed_in':
378            account.show_sign_in_prompt()
379            return
380        getcurrency.GetCurrencyWindow(modal=True).get_root_widget()
381
382    def _purchase(self) -> None:
383        from ba.internal import get_store_item_name_translated
384        from bastd.ui import getcurrency
385        from bastd.ui import confirm
386        if self._offer['item'] == 'pro':
387            _ba.purchase('pro_sale')
388        elif self._offer['item'] == 'pro_fullprice':
389            _ba.purchase('pro')
390        elif self._is_bundle_sale:
391            # With bundle sales, the price is the name of the IAP.
392            _ba.purchase(self._offer['price'])
393        else:
394            ticket_count: int | None
395            try:
396                ticket_count = _ba.get_v1_account_ticket_count()
397            except Exception:
398                ticket_count = None
399            if (ticket_count is not None
400                    and ticket_count < self._offer['price']):
401                getcurrency.show_get_tickets_prompt()
402                ba.playsound(ba.getsound('error'))
403                return
404
405            def do_it() -> None:
406                _ba.in_game_purchase('offer:' + str(self._offer['id']),
407                                     self._offer['price'])
408
409            ba.playsound(ba.getsound('swish'))
410            confirm.ConfirmWindow(ba.Lstr(
411                resource='store.purchaseConfirmText',
412                subs=[('${ITEM}',
413                       get_store_item_name_translated(self._offer['item']))]),
414                                  width=400,
415                                  height=120,
416                                  action=do_it,
417                                  ok_text=ba.Lstr(
418                                      resource='store.purchaseText',
419                                      fallback_resource='okText'))
420
421    def _cancel(self) -> None:
422        if self._cancel_delay > 0:
423            ba.playsound(ba.getsound('error'))
424            return
425        self._transition_out('out_right')
426
427
428def show_offer() -> bool:
429    """(internal)"""
430    try:
431        from bastd.ui import feedback
432        app = ba.app
433
434        # Space things out a bit so we don't hit the poor user with an ad and
435        # then an in-game offer.
436        has_been_long_enough_since_ad = True
437        if (app.ads.last_ad_completion_time is not None and
438            (ba.time(ba.TimeType.REAL) - app.ads.last_ad_completion_time <
439             30.0)):
440            has_been_long_enough_since_ad = False
441
442        if app.special_offer is not None and has_been_long_enough_since_ad:
443
444            # Special case: for pro offers, store this in our prefs so we
445            # can re-show it if the user kills us (set phasers to 'NAG'!!!).
446            if app.special_offer.get('item') == 'pro_fullprice':
447                cfg = app.config
448                cfg['pendingSpecialOffer'] = {
449                    'a': _ba.get_public_login_id(),
450                    'o': app.special_offer
451                }
452                cfg.commit()
453
454            with ba.Context('ui'):
455                if app.special_offer['item'] == 'rating':
456                    feedback.ask_for_rating()
457                else:
458                    SpecialOfferWindow(app.special_offer)
459
460            app.special_offer = None
461            return True
462    except Exception:
463        ba.print_exception('Error showing offer.')
464
465    return False