bauiv1lib.profile.upgrade

  1# Released under the MIT License. See LICENSE for details.
  2#
  3"""UI for player profile upgrades."""
  4
  5from __future__ import annotations
  6
  7import time
  8import weakref
  9from typing import TYPE_CHECKING
 10
 11import bauiv1 as bui
 12
 13if TYPE_CHECKING:
 14    from typing import Any
 15
 16    from bauiv1lib.profile.edit import EditProfileWindow
 17
 18
 19class ProfileUpgradeWindow(bui.Window):
 20    """Window for player profile upgrades to global."""
 21
 22    def __init__(
 23        self,
 24        edit_profile_window: EditProfileWindow,
 25        transition: str = 'in_right',
 26    ):
 27        if bui.app.classic is None:
 28            raise RuntimeError('This requires classic.')
 29
 30        plus = bui.app.plus
 31        assert plus is not None
 32
 33        self._r = 'editProfileWindow'
 34
 35        self._width = 680
 36        self._height = 350
 37        assert bui.app.classic is not None
 38        uiscale = bui.app.ui_v1.uiscale
 39        self._base_scale = (
 40            2.05
 41            if uiscale is bui.UIScale.SMALL
 42            else 1.5 if uiscale is bui.UIScale.MEDIUM else 1.2
 43        )
 44        self._upgrade_start_time: float | None = None
 45        self._name = edit_profile_window.getname()
 46        self._edit_profile_window = weakref.ref(edit_profile_window)
 47
 48        top_extra = 15 if uiscale is bui.UIScale.SMALL else 15
 49        super().__init__(
 50            root_widget=bui.containerwidget(
 51                size=(self._width, self._height + top_extra),
 52                toolbar_visibility='menu_currency',
 53                transition=transition,
 54                scale=self._base_scale,
 55                stack_offset=(
 56                    (0, 15) if uiscale is bui.UIScale.SMALL else (0, 0)
 57                ),
 58            )
 59        )
 60        cancel_button = bui.buttonwidget(
 61            parent=self._root_widget,
 62            position=(52, 30),
 63            size=(155, 60),
 64            scale=0.8,
 65            autoselect=True,
 66            label=bui.Lstr(resource='cancelText'),
 67            on_activate_call=self._cancel,
 68        )
 69        self._upgrade_button = bui.buttonwidget(
 70            parent=self._root_widget,
 71            position=(self._width - 190, 30),
 72            size=(155, 60),
 73            scale=0.8,
 74            autoselect=True,
 75            label=bui.Lstr(resource='upgradeText'),
 76            on_activate_call=self._on_upgrade_press,
 77        )
 78        bui.containerwidget(
 79            edit=self._root_widget,
 80            cancel_button=cancel_button,
 81            start_button=self._upgrade_button,
 82            selected_child=self._upgrade_button,
 83        )
 84
 85        assert bui.app.classic is not None
 86        bui.textwidget(
 87            parent=self._root_widget,
 88            position=(self._width * 0.5, self._height - 38),
 89            size=(0, 0),
 90            text=bui.Lstr(resource=self._r + '.upgradeToGlobalProfileText'),
 91            color=bui.app.ui_v1.title_color,
 92            maxwidth=self._width * 0.45,
 93            scale=1.0,
 94            h_align='center',
 95            v_align='center',
 96        )
 97
 98        assert bui.app.classic is not None
 99        bui.textwidget(
100            parent=self._root_widget,
101            position=(self._width * 0.5, self._height - 100),
102            size=(0, 0),
103            text=bui.Lstr(resource=self._r + '.upgradeProfileInfoText'),
104            color=bui.app.ui_v1.infotextcolor,
105            maxwidth=self._width * 0.8,
106            scale=0.7,
107            h_align='center',
108            v_align='center',
109        )
110
111        self._status_text = bui.textwidget(
112            parent=self._root_widget,
113            position=(self._width * 0.5, self._height - 160),
114            size=(0, 0),
115            text=bui.Lstr(
116                resource=self._r + '.checkingAvailabilityText',
117                subs=[('${NAME}', self._name)],
118            ),
119            color=(0.8, 0.4, 0.0),
120            maxwidth=self._width * 0.8,
121            scale=0.65,
122            h_align='center',
123            v_align='center',
124        )
125
126        self._price_text = bui.textwidget(
127            parent=self._root_widget,
128            position=(self._width * 0.5, self._height - 230),
129            size=(0, 0),
130            text='',
131            color=(0.2, 1, 0.2),
132            maxwidth=self._width * 0.8,
133            scale=1.5,
134            h_align='center',
135            v_align='center',
136        )
137
138        self._tickets_text: bui.Widget | None
139        if not bui.app.ui_v1.use_toolbars:
140            self._tickets_text = bui.textwidget(
141                parent=self._root_widget,
142                position=(self._width * 0.9 - 5, self._height - 30),
143                size=(0, 0),
144                text=bui.charstr(bui.SpecialChar.TICKET) + '123',
145                color=(0.2, 1, 0.2),
146                maxwidth=100,
147                scale=0.5,
148                h_align='right',
149                v_align='center',
150            )
151        else:
152            self._tickets_text = None
153
154        bui.app.classic.master_server_v1_get(
155            'bsGlobalProfileCheck',
156            {'name': self._name, 'b': bui.app.env.build_number},
157            callback=bui.WeakCall(self._profile_check_result),
158        )
159        self._cost = plus.get_v1_account_misc_read_val(
160            'price.global_profile', 500
161        )
162        self._status: str | None = 'waiting'
163        self._update_timer = bui.AppTimer(
164            1.0, bui.WeakCall(self._update), repeat=True
165        )
166        self._update()
167
168    def _profile_check_result(self, result: dict[str, Any] | None) -> None:
169        if result is None:
170            bui.textwidget(
171                edit=self._status_text,
172                text=bui.Lstr(resource='internal.unavailableNoConnectionText'),
173                color=(1, 0, 0),
174            )
175            self._status = 'error'
176            bui.buttonwidget(
177                edit=self._upgrade_button,
178                color=(0.4, 0.4, 0.4),
179                textcolor=(0.5, 0.5, 0.5),
180            )
181        else:
182            if result['available']:
183                bui.textwidget(
184                    edit=self._status_text,
185                    text=bui.Lstr(
186                        resource=self._r + '.availableText',
187                        subs=[('${NAME}', self._name)],
188                    ),
189                    color=(0, 1, 0),
190                )
191                bui.textwidget(
192                    edit=self._price_text,
193                    text=bui.charstr(bui.SpecialChar.TICKET) + str(self._cost),
194                )
195                self._status = None
196            else:
197                bui.textwidget(
198                    edit=self._status_text,
199                    text=bui.Lstr(
200                        resource=self._r + '.unavailableText',
201                        subs=[('${NAME}', self._name)],
202                    ),
203                    color=(1, 0, 0),
204                )
205                self._status = 'unavailable'
206                bui.buttonwidget(
207                    edit=self._upgrade_button,
208                    color=(0.4, 0.4, 0.4),
209                    textcolor=(0.5, 0.5, 0.5),
210                )
211
212    def _on_upgrade_press(self) -> None:
213        from bauiv1lib import getcurrency
214
215        if self._status is None:
216            plus = bui.app.plus
217            assert plus is not None
218
219            # If it appears we don't have enough tickets, offer to buy more.
220            tickets = plus.get_v1_account_ticket_count()
221            if tickets < self._cost:
222                bui.getsound('error').play()
223                getcurrency.show_get_tickets_prompt()
224                return
225            bui.screenmessage(
226                bui.Lstr(resource='purchasingText'), color=(0, 1, 0)
227            )
228            self._status = 'pre_upgrading'
229
230            # Now we tell the original editor to save the profile, add an
231            # upgrade transaction, and then sit and wait for everything to
232            # go through.
233            edit_profile_window = self._edit_profile_window()
234            if edit_profile_window is None:
235                print('profile upgrade: original edit window gone')
236                return
237            success = edit_profile_window.save(transition_out=False)
238            if not success:
239                print('profile upgrade: error occurred saving profile')
240                bui.screenmessage(
241                    bui.Lstr(resource='errorText'), color=(1, 0, 0)
242                )
243                bui.getsound('error').play()
244                return
245            plus.add_v1_account_transaction(
246                {'type': 'UPGRADE_PROFILE', 'name': self._name}
247            )
248            plus.run_v1_account_transactions()
249            self._status = 'upgrading'
250            self._upgrade_start_time = time.time()
251        else:
252            bui.getsound('error').play()
253
254    def _update(self) -> None:
255        plus = bui.app.plus
256        assert plus is not None
257
258        try:
259            t_str = str(plus.get_v1_account_ticket_count())
260        except Exception:
261            t_str = '?'
262        if self._tickets_text is not None:
263            bui.textwidget(
264                edit=self._tickets_text,
265                text=bui.Lstr(
266                    resource='getTicketsWindow.youHaveShortText',
267                    subs=[
268                        (
269                            '${COUNT}',
270                            bui.charstr(bui.SpecialChar.TICKET) + t_str,
271                        )
272                    ],
273                ),
274            )
275
276        # Once we've kicked off an upgrade attempt and all transactions go
277        # through, we're done.
278        if (
279            self._status == 'upgrading'
280            and not plus.have_outstanding_v1_account_transactions()
281        ):
282            self._status = 'exiting'
283            bui.containerwidget(edit=self._root_widget, transition='out_right')
284            edit_profile_window = self._edit_profile_window()
285            if edit_profile_window is None:
286                print(
287                    'profile upgrade transition out:'
288                    ' original edit window gone'
289                )
290                return
291            bui.getsound('gunCocking').play()
292            edit_profile_window.reload_window()
293
294    def _cancel(self) -> None:
295        # If we recently sent out an upgrade request, disallow canceling
296        # for a bit.
297        if (
298            self._upgrade_start_time is not None
299            and time.time() - self._upgrade_start_time < 10.0
300        ):
301            bui.getsound('error').play()
302            return
303        bui.containerwidget(edit=self._root_widget, transition='out_right')