113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
"""
|
|
This file is part of the SplendidBear Websites' projects.
|
|
|
|
Copyright (c) 2026 @ www.splendidbear.org
|
|
|
|
For the full copyright and license information, please view the LICENSE
|
|
file that was distributed with this source code.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import threading
|
|
from collections.abc import Callable
|
|
|
|
import gi
|
|
gi.require_version("Gtk", "4.0")
|
|
gi.require_version("Adw", "1")
|
|
from gi.repository import Gtk, Adw, GLib
|
|
|
|
from mineseeker.api.auth import submit_totp, AuthError
|
|
|
|
|
|
class TotpPage(Gtk.Box):
|
|
"""6-digit TOTP code entry shown after a successful password login."""
|
|
|
|
def __init__(
|
|
self,
|
|
on_success: Callable[[], None],
|
|
on_back: Callable[[], None],
|
|
) -> None:
|
|
super().__init__(orientation=Gtk.Orientation.VERTICAL, spacing=0)
|
|
self._on_success = on_success
|
|
self._on_back = on_back
|
|
|
|
self.set_valign(Gtk.Align.CENTER)
|
|
self.set_halign(Gtk.Align.CENTER)
|
|
|
|
clamp = Adw.Clamp()
|
|
clamp.set_maximum_size(360)
|
|
|
|
inner = Gtk.Box(orientation=Gtk.Orientation.VERTICAL, spacing=16)
|
|
inner.set_margin_top(32)
|
|
inner.set_margin_bottom(32)
|
|
inner.set_margin_start(16)
|
|
inner.set_margin_end(16)
|
|
|
|
title = Gtk.Label(label="Two-Factor Authentication")
|
|
title.add_css_class("title-2")
|
|
inner.append(title)
|
|
|
|
subtitle = Gtk.Label(label="Enter the 6-digit code from your authenticator app.")
|
|
subtitle.set_wrap(True)
|
|
subtitle.add_css_class("dim-label")
|
|
inner.append(subtitle)
|
|
|
|
group = Adw.PreferencesGroup()
|
|
self._code_row = Adw.EntryRow(title="Authentication Code")
|
|
self._code_row.set_input_purpose(Gtk.InputPurpose.DIGITS)
|
|
self._code_row.connect("entry-activated", self._on_verify_clicked)
|
|
group.add(self._code_row)
|
|
inner.append(group)
|
|
|
|
self._error_label = Gtk.Label(label="")
|
|
self._error_label.add_css_class("error")
|
|
self._error_label.set_visible(False)
|
|
inner.append(self._error_label)
|
|
|
|
self._verify_btn = Gtk.Button(label="Verify")
|
|
self._verify_btn.add_css_class("suggested-action")
|
|
self._verify_btn.add_css_class("pill")
|
|
self._verify_btn.connect("clicked", self._on_verify_clicked)
|
|
inner.append(self._verify_btn)
|
|
|
|
back_btn = Gtk.Button(label="Back to Login")
|
|
back_btn.add_css_class("pill")
|
|
back_btn.connect("clicked", lambda *_: self._on_back())
|
|
inner.append(back_btn)
|
|
|
|
clamp.set_child(inner)
|
|
self.append(clamp)
|
|
|
|
def _set_busy(self, busy: bool) -> None:
|
|
self._verify_btn.set_sensitive(not busy)
|
|
self._code_row.set_sensitive(not busy)
|
|
if busy:
|
|
self._error_label.set_visible(False)
|
|
|
|
def _show_error(self, message: str) -> None:
|
|
self._error_label.set_label(message)
|
|
self._error_label.set_visible(True)
|
|
|
|
def _on_verify_clicked(self, *_) -> None:
|
|
code = self._code_row.get_text().strip()
|
|
if len(code) != 6 or not code.isdigit():
|
|
self._show_error("Code must be exactly 6 digits.")
|
|
return
|
|
self._set_busy(True)
|
|
threading.Thread(target=self._do_verify, args=(code,), daemon=True).start()
|
|
|
|
def _do_verify(self, code: str) -> None:
|
|
try:
|
|
submit_totp(code)
|
|
GLib.idle_add(self._on_success)
|
|
except AuthError as e:
|
|
GLib.idle_add(self._handle_error, str(e))
|
|
except Exception as e:
|
|
GLib.idle_add(self._handle_error, f"Connection error: {e}")
|
|
|
|
def _handle_error(self, message: str) -> bool:
|
|
self._set_busy(False)
|
|
self._show_error(message)
|
|
return GLib.SOURCE_REMOVE
|