Initial commit

dist/ba_data/python/bastd/ui/account/v2proxy.py

@@ -0,0 +1,236 @@
+# Released under the MIT License. See LICENSE for details.
+#
+"""V2 account ui bits."""
+
+from __future__ import annotations
+
+import logging
+from typing import TYPE_CHECKING
+
+import ba
+import ba.internal
+
+from efro.error import CommunicationError
+import bacommon.cloud
+
+if TYPE_CHECKING:
+    pass
+
+STATUS_CHECK_INTERVAL_SECONDS = 2.0
+
+
+class V2ProxySignInWindow(ba.Window):
+    """A window allowing signing in to a v2 account."""
+
+    def __init__(self, origin_widget: ba.Widget):
+        self._width = 600
+        self._height = 550
+        self._proxyid: str | None = None
+        self._proxykey: str | None = None
+
+        uiscale = ba.app.ui.uiscale
+        super().__init__(
+            root_widget=ba.containerwidget(
+                size=(self._width, self._height),
+                transition='in_scale',
+                scale_origin_stack_offset=(
+                    origin_widget.get_screen_space_center()
+                ),
+                scale=(
+                    1.25
+                    if uiscale is ba.UIScale.SMALL
+                    else 1.05
+                    if uiscale is ba.UIScale.MEDIUM
+                    else 0.9
+                ),
+            )
+        )
+
+        self._loading_text = ba.textwidget(
+            parent=self._root_widget,
+            position=(self._width * 0.5, self._height * 0.5),
+            h_align='center',
+            v_align='center',
+            size=(0, 0),
+            maxwidth=0.9 * self._width,
+            text=ba.Lstr(
+                value='${A}...',
+                subs=[('${A}', ba.Lstr(resource='loadingText'))],
+            ),
+        )
+
+        self._cancel_button = ba.buttonwidget(
+            parent=self._root_widget,
+            position=(30, self._height - 65),
+            size=(130, 50),
+            scale=0.8,
+            label=ba.Lstr(resource='cancelText'),
+            on_activate_call=self._done,
+            autoselect=True,
+            color=(0.55, 0.5, 0.6),
+            textcolor=(0.75, 0.7, 0.8),
+        )
+
+        if bool(False):
+            ba.containerwidget(
+                edit=self._root_widget, cancel_button=self._cancel_button
+            )
+
+        self._update_timer: ba.Timer | None = None
+
+        # Ask the cloud for a proxy login id.
+        ba.app.cloud.send_message_cb(
+            bacommon.cloud.LoginProxyRequestMessage(),
+            on_response=ba.WeakCall(self._on_proxy_request_response),
+        )
+
+    def _on_proxy_request_response(
+        self, response: bacommon.cloud.LoginProxyRequestResponse | Exception
+    ) -> None:
+        from ba.internal import is_browser_likely_available
+
+        # Something went wrong. Show an error message and that's it.
+        if isinstance(response, Exception):
+            ba.textwidget(
+                edit=self._loading_text,
+                text=ba.Lstr(resource='internal.unavailableNoConnectionText'),
+                color=(1, 0, 0),
+            )
+            return
+
+        # Show link(s) the user can use to sign in.
+        address = (
+            ba.internal.get_master_server_address(version=2) + response.url
+        )
+        address_pretty = address.removeprefix('https://')
+
+        ba.textwidget(
+            parent=self._root_widget,
+            position=(self._width * 0.5, self._height - 95),
+            size=(0, 0),
+            text=ba.Lstr(
+                resource='accountSettingsWindow.v2LinkInstructionsText'
+            ),
+            color=ba.app.ui.title_color,
+            maxwidth=self._width * 0.9,
+            h_align='center',
+            v_align='center',
+        )
+        button_width = 450
+        if is_browser_likely_available():
+            ba.buttonwidget(
+                parent=self._root_widget,
+                position=(
+                    (self._width * 0.5 - button_width * 0.5),
+                    self._height - 185,
+                ),
+                autoselect=True,
+                size=(button_width, 60),
+                label=ba.Lstr(value=address_pretty),
+                color=(0.55, 0.5, 0.6),
+                textcolor=(0.75, 0.7, 0.8),
+                on_activate_call=lambda: ba.open_url(address),
+            )
+            qroffs = 0.0
+        else:
+            ba.textwidget(
+                parent=self._root_widget,
+                position=(self._width * 0.5, self._height - 145),
+                size=(0, 0),
+                text=ba.Lstr(value=address_pretty),
+                flatness=1.0,
+                maxwidth=self._width,
+                scale=0.75,
+                h_align='center',
+                v_align='center',
+            )
+            qroffs = 20.0
+
+        qr_size = 270
+        ba.imagewidget(
+            parent=self._root_widget,
+            position=(
+                self._width * 0.5 - qr_size * 0.5,
+                self._height * 0.36 + qroffs - qr_size * 0.5,
+            ),
+            size=(qr_size, qr_size),
+            texture=ba.internal.get_qrcode_texture(address),
+        )
+
+        # Start querying for results.
+        self._proxyid = response.proxyid
+        self._proxykey = response.proxykey
+        ba.timer(
+            STATUS_CHECK_INTERVAL_SECONDS, ba.WeakCall(self._ask_for_status)
+        )
+
+    def _ask_for_status(self) -> None:
+        assert self._proxyid is not None
+        assert self._proxykey is not None
+        ba.app.cloud.send_message_cb(
+            bacommon.cloud.LoginProxyStateQueryMessage(
+                proxyid=self._proxyid, proxykey=self._proxykey
+            ),
+            on_response=ba.WeakCall(self._got_status),
+        )
+
+    def _got_status(
+        self, response: bacommon.cloud.LoginProxyStateQueryResponse | Exception
+    ) -> None:
+
+        # For now, if anything goes wrong on the server-side, just abort
+        # with a vague error message. Can be more verbose later if need be.
+        if (
+            isinstance(response, bacommon.cloud.LoginProxyStateQueryResponse)
+            and response.state is response.State.FAIL
+        ):
+            ba.playsound(ba.getsound('error'))
+            ba.screenmessage(ba.Lstr(resource='errorText'), color=(1, 0, 0))
+            self._done()
+            return
+
+        # If we got a token, set ourself as signed in. Hooray!
+        if (
+            isinstance(response, bacommon.cloud.LoginProxyStateQueryResponse)
+            and response.state is response.State.SUCCESS
+        ):
+            assert response.credentials is not None
+            ba.app.accounts_v2.set_primary_credentials(response.credentials)
+
+            # As a courtesy, tell the server we're done with this proxy
+            # so it can clean up (not a huge deal if this fails)
+            assert self._proxyid is not None
+            try:
+                ba.app.cloud.send_message_cb(
+                    bacommon.cloud.LoginProxyCompleteMessage(
+                        proxyid=self._proxyid
+                    ),
+                    on_response=ba.WeakCall(self._proxy_complete_response),
+                )
+            except CommunicationError:
+                pass
+            except Exception:
+                logging.warning(
+                    'Unexpected error sending login-proxy-complete message',
+                    exc_info=True,
+                )
+
+            self._done()
+            return
+
+        # If we're still waiting, ask again soon.
+        if (
+            isinstance(response, Exception)
+            or response.state is response.State.WAITING
+        ):
+            ba.timer(
+                STATUS_CHECK_INTERVAL_SECONDS, ba.WeakCall(self._ask_for_status)
+            )
+
+    def _proxy_complete_response(self, response: None | Exception) -> None:
+        del response  # Not used.
+        # We could do something smart like retry on exceptions here, but
+        # this isn't critical so we'll just let anything slide.
+
+    def _done(self) -> None:
+        ba.containerwidget(edit=self._root_widget, transition='out_scale')