diff --git a/.forgejo/workflows/release.yml b/.forgejo/workflows/release.yml index bf1f8c6..6d12cf4 100644 --- a/.forgejo/workflows/release.yml +++ b/.forgejo/workflows/release.yml @@ -46,6 +46,8 @@ jobs: needs: test if: github.ref == 'refs/heads/main' && github.event_name == 'push' runs-on: docker + container: + image: node:20-slim outputs: new_tag: ${{ steps.version.outputs.new_tag }} new_version: ${{ steps.version.outputs.new_version }} diff --git a/.gitignore b/.gitignore index 3c981dd..f59019a 100644 --- a/.gitignore +++ b/.gitignore @@ -4,3 +4,5 @@ __pycache__/ .env .nicegui/ logs/ +.idea/ +.coverage diff --git a/TODO.md b/TODO.md index 17e555b..22d6fa7 100644 --- a/TODO.md +++ b/TODO.md @@ -188,9 +188,68 @@ Source: `/home/stefanob/PycharmProjects/personal/wirezone` - [x] Loguru configured (wiregui/logging.py), no print statements - [x] File logging to `logs/` when `WG_LOG_TO_FILE=true` -### Deployment -- [ ] Dockerfile (multi-stage) -- [ ] compose.prod.yml (app + postgres + valkey + caddy) -- [ ] Health endpoint `GET /api/health` +### Deployment ✅ +- [x] Dockerfile (multi-stage python:3.13-slim) +- [x] compose.prod.yml (bridge networking, NET_ADMIN, nftables) +- [x] Health endpoint `GET /api/health` +- [x] Forgejo CI: test → semver → Docker registry push - [ ] First-run CLI setup command - [ ] README.md + +--- + +## UI Polish — Account Page (`/account`) + +Redesign from tabbed layout to single scrollable page (matching original wirezone pattern). +Leverage Quasar components + Tailwind utility classes for modern look. + +### Layout change +- [ ] Remove tabs — render all sections stacked vertically on one page +- [ ] Page header: "Account Settings" with subtitle description + +### Section 1: Account Details +- [ ] Quasar `q-card` with clean table layout (not grid) for user info +- [ ] Rows: Email, Role, Last Sign-in, Method, Created +- [ ] Tailwind: rounded borders, hover states on rows, subtle dividers +- [ ] "Edit" button to open email change dialog (future) + +### Section 2: Change Password +- [ ] Separate `q-card` below details +- [ ] Outlined inputs with proper validation feedback +- [ ] Min 8 chars, confirmation match check shown inline +- [ ] Success/error toast notifications + +### Section 3: Connected SSO Providers +- [ ] `q-card` showing OIDC connections as a proper table +- [ ] Columns: Provider, Last Refreshed, Status +- [ ] "Disconnect" action per provider (future) +- [ ] Empty state: "No SSO providers connected" + +### Section 4: Multi-Factor Authentication +- [ ] `q-card` with MFA methods table +- [ ] Columns: Name, Type, Last Used, Actions (delete) +- [ ] Styled delete button (red outline, confirmation dialog) +- [ ] "Add TOTP Method" and "Add Security Key" buttons below table +- [ ] TOTP registration renders inline (QR + verify code) inside an expansion panel +- [ ] Empty state with icon + message + +### Section 5: API Tokens +- [ ] `q-card` with tokens table +- [ ] Columns: Created, Expires, Status (chip: green "Active" / red "Expired"), Actions +- [ ] Quasar `q-chip` for status badges +- [ ] Create token: inline row with expiry input + button (not a dialog) +- [ ] Token display after creation: `q-banner` with copy-to-clipboard button +- [ ] Empty state message + +### Section 6: Danger Zone +- [ ] `q-card` with red left border accent (`border-l-4 border-red-500`) +- [ ] "Delete Your Account" button with `q-btn color=negative outline` +- [ ] Confirmation dialog with typed email verification +- [ ] Disabled if user is the only admin + +### General styling improvements +- [ ] Consistent card spacing (`q-mt-lg` between sections) +- [ ] Section titles: `text-h6 text-weight-medium` +- [ ] Descriptive subtitles below each section title in `text-caption text-grey-7` +- [ ] Responsive: max-width container centered (`max-w-3xl mx-auto`) +- [ ] Smooth scroll between sections diff --git a/compose.prod.yml b/compose.prod.yml index f5394d3..ddcef0b 100644 --- a/compose.prod.yml +++ b/compose.prod.yml @@ -39,6 +39,8 @@ services: postgres: image: postgres:17 restart: unless-stopped + ports: + - "5432:5432" environment: POSTGRES_USER: wiregui POSTGRES_PASSWORD: wiregui diff --git a/tests/test_services.py b/tests/test_services.py index f0cf173..1c32f0e 100644 --- a/tests/test_services.py +++ b/tests/test_services.py @@ -29,6 +29,7 @@ def _make_device(**kwargs) -> Device: async def test_on_device_created_calls_add_peer(mock_wg, mock_fw, mock_settings): mock_settings.return_value.wg_enabled = True mock_wg.add_peer = AsyncMock() + mock_fw.add_user_chain = AsyncMock() mock_fw.add_device_jump_rule = AsyncMock() device = _make_device() diff --git a/wiregui/pages/account.py b/wiregui/pages/account.py index 8448b0a..941b849 100644 --- a/wiregui/pages/account.py +++ b/wiregui/pages/account.py @@ -1,12 +1,12 @@ -"""User account page — password change, MFA management, API tokens.""" +"""User account page — compact, single-page layout matching Firezone's density.""" +import json +from datetime import timedelta from uuid import UUID from loguru import logger from nicegui import app, ui -from sqlmodel import select - -import json +from sqlmodel import func, select from wiregui.auth.api_token import generate_api_token from wiregui.auth.mfa import generate_totp_qr_svg, generate_totp_secret, get_totp_uri, verify_totp_code @@ -14,13 +14,37 @@ from wiregui.auth.passwords import hash_password, verify_password from wiregui.auth.webauthn import create_registration_options, verify_registration from wiregui.db import async_session from wiregui.models.api_token import ApiToken +from wiregui.models.device import Device from wiregui.models.mfa_method import MFAMethod from wiregui.models.oidc_connection import OIDCConnection +from wiregui.models.rule import Rule from wiregui.models.user import User from wiregui.pages.layout import layout from wiregui.utils.time import utcnow +def _section_header(title: str, description: str = ""): + """Compact section header — bold title + optional subtitle.""" + ui.label(title).classes("text-lg text-weight-bold q-mt-lg q-mb-none") + if description: + ui.label(description).classes("text-caption text-grey-7 q-mb-sm") + + +def _kv_row(label: str, value): + """Compact key-value row.""" + with ui.row().classes("w-full items-baseline gap-4"): + ui.label(label).classes("text-sm text-grey-8 w-36 shrink-0") + if isinstance(value, str): + ui.label(value).classes("text-sm") + + +# Consistent button style: proper padding, readable size, no cramped text +BTN = "unelevated padding=8px 20px" +BTN_PRIMARY = f"color=primary {BTN}" +BTN_OUTLINE = f"outline color=primary {BTN}" +BTN_DANGER = f"color=negative {BTN}" + + @ui.page("/account") async def account_page(): if not app.storage.user.get("authenticated"): @@ -31,358 +55,291 @@ async def account_page(): async with async_session() as session: user = await session.get(User, user_id) + device_count = (await session.execute( + select(func.count()).select_from(Device).where(Device.user_id == user_id) + )).scalar() + rule_count = (await session.execute( + select(func.count()).select_from(Rule).where(Rule.user_id == user_id) + )).scalar() + oidc_conns = (await session.execute( + select(OIDCConnection).where(OIDCConnection.user_id == user_id) + )).scalars().all() - with ui.column().classes("w-full p-4"): - ui.label("Account Settings").classes("text-h5 q-mb-md") + with ui.column().classes("w-full p-6"): + ui.label("Account Settings").classes("text-sm text-weight-bold") + ui.label("Configure settings related to your WireGUI account.").classes("text-caption text-grey-7") - with ui.tabs().classes("w-full") as tabs: - profile_tab = ui.tab("Profile") - mfa_tab = ui.tab("Two-Factor Auth") - tokens_tab = ui.tab("API Tokens") + # ===== Details ===== + _section_header("Details") + with ui.column().classes("gap-1"): + _kv_row("Email", user.email) + _kv_row("Role", user.role) + _kv_row("Last Signed In", str(user.last_signed_in_at)[:19] if user.last_signed_in_at else "-") + _kv_row("Created", str(user.inserted_at)[:19]) + _kv_row("Number of Devices", str(device_count)) + _kv_row("Number of Rules", str(rule_count)) - with ui.tab_panels(tabs, value=profile_tab).classes("w-full"): + ui.button("Change Email or Password", icon="edit", on_click=lambda: pw_dialog.open()).props( + BTN_PRIMARY + ).classes("q-mt-sm") - # === Profile === - with ui.tab_panel(profile_tab): - with ui.card().classes("w-full"): - ui.label("Account Details").classes("text-subtitle1 text-bold") - ui.separator() - with ui.grid(columns=2).classes("w-full gap-2 q-pa-sm"): - ui.label("Email:").classes("text-bold") - ui.label(user.email) - ui.label("Role:").classes("text-bold") - ui.label(user.role) - ui.label("Last Sign-in:").classes("text-bold") - ui.label(str(user.last_signed_in_at)[:19] if user.last_signed_in_at else "-") - ui.label("Method:").classes("text-bold") - ui.label(user.last_signed_in_method or "-") + # ===== SSO ===== + if oidc_conns: + _section_header("Connected SSO Providers") + cols = [ + {"name": "provider", "label": "Provider", "field": "provider", "align": "left"}, + {"name": "refreshed", "label": "Last Refreshed", "field": "refreshed", "align": "left"}, + ] + rows = [{"provider": c.provider, "refreshed": str(c.refreshed_at)[:19] if c.refreshed_at else "Never"} for c in oidc_conns] + ui.table(columns=cols, rows=rows, row_key="provider").props("dense flat bordered").classes("w-full text-xs") - with ui.card().classes("w-full q-mt-md"): - ui.label("Change Password").classes("text-subtitle1 text-bold") - ui.separator() + # ===== API Tokens ===== + _section_header("API Tokens", "Manage API tokens.") - current_pw = ui.input("Current Password", password=True, password_toggle_button=True).props("outlined dense").classes("w-full") - new_pw = ui.input("New Password", password=True, password_toggle_button=True).props("outlined dense").classes("w-full") - confirm_pw = ui.input("Confirm New Password", password=True, password_toggle_button=True).props("outlined dense").classes("w-full") + tokens_container = ui.column().classes("w-full") + token_banner = ui.column().classes("w-full") - async def change_password(): - if not current_pw.value or not new_pw.value: - ui.notify("Fill in all password fields", type="negative") - return - if new_pw.value != confirm_pw.value: - ui.notify("New passwords do not match", type="negative") - return - if len(new_pw.value) < 8: - ui.notify("Password must be at least 8 characters", type="negative") - return + async def refresh_tokens(): + async with async_session() as session: + tokens = (await session.execute( + select(ApiToken).where(ApiToken.user_id == user_id).order_by(ApiToken.inserted_at.desc()) + )).scalars().all() + tokens_container.clear() + with tokens_container: + if tokens: + cols = [ + {"name": "created", "label": "Created", "field": "created", "align": "left"}, + {"name": "expires", "label": "Expires", "field": "expires", "align": "left"}, + {"name": "status", "label": "Status", "field": "status", "align": "left"}, + {"name": "actions", "label": "", "field": "id", "align": "center"}, + ] + rows = [{ + "id": str(t.id), + "created": str(t.inserted_at)[:19], + "expires": str(t.expires_at)[:19] if t.expires_at else "Never", + "status": "Expired" if t.expires_at and t.expires_at < utcnow() else "Active", + } for t in tokens] + tbl = ui.table(columns=cols, rows=rows, row_key="id").props("dense flat bordered").classes("w-full text-xs") + tbl.add_slot("body-cell-status", r'''''') + tbl.add_slot("body-cell-actions", r'''''') + tbl.on("delete", lambda e: delete_token(e.args)) + else: + ui.label("No API tokens.").classes("text-sm text-grey-7") - async with async_session() as session: - u = await session.get(User, user_id) - if not verify_password(current_pw.value, u.password_hash): - ui.notify("Current password is incorrect", type="negative") - return - u.password_hash = hash_password(new_pw.value) - session.add(u) - await session.commit() - - logger.info("Password changed for {}", user.email) - ui.notify("Password changed", type="positive") - current_pw.value = "" - new_pw.value = "" - confirm_pw.value = "" - - ui.button("Change Password", on_click=change_password).props("color=primary").classes("q-mt-sm") - - # OIDC connections - async with async_session() as session: - oidc_conns = (await session.execute( - select(OIDCConnection).where(OIDCConnection.user_id == user_id) - )).scalars().all() - - if oidc_conns: - with ui.card().classes("w-full q-mt-md"): - ui.label("Connected SSO Providers").classes("text-subtitle1 text-bold") - ui.separator() - for conn in oidc_conns: - with ui.row().classes("w-full items-center justify-between q-pa-xs"): - ui.label(f"{conn.provider}").classes("text-bold") - ui.label(f"Last refreshed: {str(conn.refreshed_at)[:19] if conn.refreshed_at else 'Never'}") - - # === MFA === - with ui.tab_panel(mfa_tab): - await _render_mfa_panel(user_id, user.email) - - # === API Tokens === - with ui.tab_panel(tokens_tab): - await _render_tokens_panel(user_id) - - -async def _render_mfa_panel(user_id: UUID, email: str): - """Render the MFA management tab.""" - async def load_methods(): - async with async_session() as session: - result = await session.execute( - select(MFAMethod).where(MFAMethod.user_id == user_id).order_by(MFAMethod.inserted_at) - ) - return result.scalars().all() - - async def refresh_methods(): - methods = await load_methods() - methods_container.clear() - with methods_container: - if methods: - for m in methods: - with ui.row().classes("w-full items-center justify-between q-pa-xs"): - with ui.row().classes("items-center gap-2"): - ui.icon("security").props("color=primary") - ui.label(m.name).classes("text-bold") - ui.label(f"({m.type})").classes("text-caption text-grey-7") - with ui.row().classes("items-center gap-2"): - ui.label(f"Last used: {str(m.last_used_at)[:19] if m.last_used_at else 'Never'}").classes("text-caption") - ui.button(icon="delete", on_click=lambda mid=m.id: delete_method(mid)).props("flat dense color=negative") - ui.separator() - else: - ui.label("No MFA methods configured.").classes("text-caption text-grey-7 q-pa-sm") - - async def delete_method(method_id): - async with async_session() as session: - m = await session.get(MFAMethod, method_id) - if m and m.user_id == user_id: - await session.delete(m) + async def create_token(): + days = int(token_days.value) if token_days.value else 30 + plaintext, token_hash = generate_api_token() + expires_at = utcnow() + timedelta(days=days) if days > 0 else None + async with async_session() as session: + session.add(ApiToken(token_hash=token_hash, expires_at=expires_at, user_id=user_id)) await session.commit() - logger.info("MFA method deleted for user {}", email) - ui.notify("MFA method removed") - await refresh_methods() + logger.info("API token created (expires in {} days)", days) + token_banner.clear() + with token_banner: + with ui.row().classes("w-full items-center bg-green-1 rounded px-3 py-1.5 gap-2 q-mb-xs text-xs"): + ui.icon("check_circle", color="positive").props("size=xs") + ui.label("Copy now — won't be shown again.").classes("text-weight-medium") + with ui.row().classes("w-full items-center gap-1 q-mb-sm"): + ui.input(value=plaintext).props("readonly outlined dense").classes("w-full font-mono").style("font-size: 0.75rem") + ui.button(icon="content_copy", on_click=lambda: _copy(plaintext)).props("flat dense size=sm") + await refresh_tokens() - # Registration state - registration = {"secret": None} + async def _copy(text): + await ui.run_javascript(f"navigator.clipboard.writeText('{text}')") + ui.notify("Copied", type="positive") - def start_registration(): - secret = generate_totp_secret() - registration["secret"] = secret - uri = get_totp_uri(secret, email) - svg = generate_totp_qr_svg(uri) - - reg_container.clear() - with reg_container: - ui.label("Scan this QR code with your authenticator app:").classes("text-body2") - ui.html(svg).classes("w-64 q-my-sm") - ui.label(f"Or enter this secret manually: {secret}").classes("text-caption font-mono") - reg_name_input = ui.input("Method Name", value="Authenticator").props("outlined dense").classes("w-full") - reg_code_input = ui.input("Verification Code", placeholder="Enter 6-digit code").props("outlined dense maxlength=6").classes("w-full") - - async def verify_and_save(): - code = reg_code_input.value.strip() - name = reg_name_input.value.strip() or "Authenticator" - if not verify_totp_code(registration["secret"], code): - ui.notify("Invalid code — check your authenticator", type="negative") - return - - async with async_session() as session: - method = MFAMethod( - name=name, - type="totp", - payload={"secret": registration["secret"]}, - user_id=user_id, - ) - session.add(method) + async def delete_token(token_id): + async with async_session() as session: + t = await session.get(ApiToken, UUID(token_id)) + if t and t.user_id == user_id: + await session.delete(t) await session.commit() + ui.notify("Token deleted") + await refresh_tokens() - logger.info("MFA TOTP registered for {}", email) - ui.notify("MFA method added!", type="positive") - registration["secret"] = None - reg_container.clear() - await refresh_methods() + await refresh_tokens() + with ui.row().classes("items-center gap-3 q-mt-sm"): + token_days = ui.input("Expires in days", value="30").props("outlined dense").classes("w-36") + ui.button("+ Add API Token", on_click=create_token).props(BTN_OUTLINE) - ui.button("Verify & Save", on_click=verify_and_save).props("color=primary").classes("q-mt-sm") - ui.button("Cancel", on_click=lambda: reg_container.clear()).props("flat") - - with ui.card().classes("w-full"): - ui.label("Two-Factor Authentication Methods").classes("text-subtitle1 text-bold") - ui.separator() + # ===== MFA ===== + _section_header("Multi Factor Authentication", "Your MFA methods are invoked when login with username and password.") methods_container = ui.column().classes("w-full") - await refresh_methods() + reg_container = ui.column().classes("w-full") + registration = {"secret": None} + webauthn_state = {"challenge": None} - with ui.row().classes("q-mt-sm gap-2"): - ui.button("Add TOTP Method", icon="add", on_click=start_registration).props("outline") - ui.button("Add Security Key", icon="key", on_click=lambda: start_webauthn_registration()).props("outline") + async def refresh_methods(): + async with async_session() as session: + methods = (await session.execute( + select(MFAMethod).where(MFAMethod.user_id == user_id).order_by(MFAMethod.inserted_at) + )).scalars().all() + methods_container.clear() + with methods_container: + if methods: + cols = [ + {"name": "name", "label": "Name", "field": "name", "align": "left"}, + {"name": "type", "label": "Type", "field": "type", "align": "left"}, + {"name": "last_used", "label": "Last Used", "field": "last_used", "align": "left"}, + {"name": "actions", "label": "", "field": "id", "align": "center"}, + ] + rows = [{"id": str(m.id), "name": m.name, "type": m.type.upper(), "last_used": str(m.last_used_at)[:19] if m.last_used_at else "Never"} for m in methods] + tbl = ui.table(columns=cols, rows=rows, row_key="id").props("dense flat bordered").classes("w-full text-xs") + tbl.add_slot("body-cell-actions", r'''''') + tbl.on("delete", lambda e: _confirm_del_mfa(e.args)) + else: + ui.label("No MFA methods added.").classes("text-sm text-grey-7") - reg_container = ui.column().classes("w-full q-mt-md") - webauthn_state = {"challenge": None} + async def _confirm_del_mfa(mid): + with ui.dialog(value=True) as dlg: + with ui.card().classes("w-72"): + ui.label("Remove MFA method?").classes("text-subtitle2") + with ui.row().classes("w-full justify-end q-mt-sm gap-2"): + ui.button("Cancel", on_click=dlg.close).props("flat dense size=sm") + ui.button("Remove", on_click=lambda: _del_mfa(mid, dlg)).props("color=negative dense size=sm unelevated") - async def start_webauthn_registration(): - # Get existing webauthn credentials to exclude - existing = [] - async with async_session() as session: - from sqlmodel import select as sel - result = await session.execute( - sel(MFAMethod).where(MFAMethod.user_id == user_id, MFAMethod.type.in_(["native", "portable"])) - ) - for m in result.scalars().all(): - existing.append(m.payload) + async def _del_mfa(mid, dlg): + async with async_session() as session: + m = await session.get(MFAMethod, UUID(mid)) + if m and m.user_id == user_id: + await session.delete(m) + await session.commit() + dlg.close() + ui.notify("Removed") + await refresh_methods() - try: - reg_data = create_registration_options(user_id, email, existing) - except Exception as e: - ui.notify(f"WebAuthn not available: {e}", type="negative") - return + def start_totp(): + secret = generate_totp_secret() + registration["secret"] = secret + svg = generate_totp_qr_svg(get_totp_uri(secret, user.email)) + reg_container.clear() + with reg_container: + ui.separator().classes("q-my-sm") + with ui.row().classes("items-start gap-4"): + ui.html(svg).style("width: 140px; height: 140px") + with ui.column().classes("gap-2"): + ui.label("Scan or enter manually:").classes("text-xs") + ui.label(secret).classes("text-xs font-mono bg-grey-2 px-2 py-1 rounded") + reg_name = ui.input("Name", value="Authenticator").props("outlined dense").classes("w-52").style("font-size: 0.8rem") + reg_code = ui.input("6-digit code").props("outlined dense maxlength=6").classes("w-52").style("font-size: 0.8rem") - webauthn_state["challenge"] = reg_data["challenge"] - options_json = reg_data["options_json"] + async def verify(): + if not verify_totp_code(registration["secret"], reg_code.value.strip()): + ui.notify("Invalid code", type="negative") + return + async with async_session() as session: + session.add(MFAMethod(name=reg_name.value.strip() or "Authenticator", type="totp", payload={"secret": registration["secret"]}, user_id=user_id)) + await session.commit() + ui.notify("TOTP added!", type="positive") + reg_container.clear() + await refresh_methods() - # Call browser's navigator.credentials.create() via JavaScript - js = f""" - async function() {{ - try {{ - const options = JSON.parse('{options_json}'); - // Convert base64url strings to ArrayBuffers - options.challenge = Uint8Array.from(atob(options.challenge.replace(/-/g,'+').replace(/_/g,'/')), c => c.charCodeAt(0)); - options.user.id = Uint8Array.from(atob(options.user.id.replace(/-/g,'+').replace(/_/g,'/')), c => c.charCodeAt(0)); - if (options.excludeCredentials) {{ - options.excludeCredentials = options.excludeCredentials.map(c => ({{ - ...c, - id: Uint8Array.from(atob(c.id.replace(/-/g,'+').replace(/_/g,'/')), ch => ch.charCodeAt(0)) - }})); - }} - const credential = await navigator.credentials.create({{publicKey: options}}); - // Serialize the response - const response = {{ - id: credential.id, - rawId: btoa(String.fromCharCode(...new Uint8Array(credential.rawId))).replace(/\\+/g,'-').replace(/\\//g,'_').replace(/=/g,''), - type: credential.type, - response: {{ - attestationObject: btoa(String.fromCharCode(...new Uint8Array(credential.response.attestationObject))).replace(/\\+/g,'-').replace(/\\//g,'_').replace(/=/g,''), - clientDataJSON: btoa(String.fromCharCode(...new Uint8Array(credential.response.clientDataJSON))).replace(/\\+/g,'-').replace(/\\//g,'_').replace(/=/g,''), - }}, - }}; - return JSON.stringify(response); - }} catch(e) {{ - return JSON.stringify({{"error": e.message}}); - }} - }} - """ - result = await ui.run_javascript(f"({js})()") - await _handle_webauthn_response(result) + with ui.row().classes("gap-2"): + ui.button("Verify & Save", on_click=verify).props(BTN_PRIMARY) + ui.button("Cancel", on_click=lambda: reg_container.clear()).props("flat dense size=sm") - async def _handle_webauthn_response(result_json: str): - try: - result = json.loads(result_json) - except (json.JSONDecodeError, TypeError): - ui.notify("WebAuthn response error", type="negative") - return - - if "error" in result: - ui.notify(f"WebAuthn failed: {result['error']}", type="negative") - return - - challenge = webauthn_state.get("challenge") - if not challenge: - ui.notify("No pending WebAuthn challenge", type="negative") - return - - try: - credential_data = verify_registration(result_json, challenge) - except Exception as e: - ui.notify(f"Verification failed: {e}", type="negative") - return - - async with async_session() as session: - method = MFAMethod( - name="Security Key", - type="portable", - payload=credential_data, - user_id=user_id, - ) - session.add(method) - await session.commit() - - logger.info("WebAuthn key registered for {}", email) - ui.notify("Security key registered!", type="positive") - webauthn_state["challenge"] = None - await refresh_methods() - - -async def _render_tokens_panel(user_id: UUID): - """Render the API tokens tab.""" - async def load_tokens(): - async with async_session() as session: - result = await session.execute( - select(ApiToken).where(ApiToken.user_id == user_id).order_by(ApiToken.inserted_at.desc()) - ) - return result.scalars().all() - - async def refresh_tokens(): - tokens = await load_tokens() - token_table.rows = [ - { - "id": str(t.id), - "created": str(t.inserted_at)[:19], - "expires": str(t.expires_at)[:19] if t.expires_at else "Never", - "status": "Expired" if t.expires_at and t.expires_at < utcnow() else "Active", - } - for t in tokens - ] - token_table.update() - - async def create_token(): - from datetime import timedelta - days = int(token_days.value) if token_days.value else 30 - - plaintext, token_hash = generate_api_token() - expires_at = utcnow() + timedelta(days=days) if days > 0 else None - - async with async_session() as session: - token = ApiToken(token_hash=token_hash, expires_at=expires_at, user_id=user_id) - session.add(token) - await session.commit() - - logger.info("API token created (expires in {} days)", days) - - # Show the token once - with ui.dialog(value=True) as token_dialog: - with ui.card().classes("w-96"): - ui.label("API Token Created").classes("text-h6") - ui.label("Copy this token now — it won't be shown again.").classes("text-caption text-negative") - ui.input(value=plaintext).props("readonly outlined dense").classes("w-full font-mono q-mt-sm") - ui.button("Close", on_click=token_dialog.close).props("flat").classes("w-full q-mt-sm") - - await refresh_tokens() - - async def delete_token(token_id: str): - async with async_session() as session: - t = await session.get(ApiToken, UUID(token_id)) - if t and t.user_id == user_id: - await session.delete(t) + async def start_webauthn(): + existing = [] + async with async_session() as session: + existing = [m.payload for m in (await session.execute( + select(MFAMethod).where(MFAMethod.user_id == user_id, MFAMethod.type.in_(["native", "portable"])) + )).scalars().all()] + try: + reg_data = create_registration_options(user_id, user.email, existing) + except Exception as e: + ui.notify(f"WebAuthn unavailable: {e}", type="negative") + return + webauthn_state["challenge"] = reg_data["challenge"] + js = f"""(async()=>{{try{{const o=JSON.parse('{reg_data["options_json"]}');o.challenge=Uint8Array.from(atob(o.challenge.replace(/-/g,'+').replace(/_/g,'/')),c=>c.charCodeAt(0));o.user.id=Uint8Array.from(atob(o.user.id.replace(/-/g,'+').replace(/_/g,'/')),c=>c.charCodeAt(0));if(o.excludeCredentials)o.excludeCredentials=o.excludeCredentials.map(c=>({{...c,id:Uint8Array.from(atob(c.id.replace(/-/g,'+').replace(/_/g,'/')),h=>h.charCodeAt(0))}}));const cr=await navigator.credentials.create({{publicKey:o}});return JSON.stringify({{id:cr.id,rawId:btoa(String.fromCharCode(...new Uint8Array(cr.rawId))).replace(/\\+/g,'-').replace(/\\//g,'_').replace(/=/g,''),type:cr.type,response:{{attestationObject:btoa(String.fromCharCode(...new Uint8Array(cr.response.attestationObject))).replace(/\\+/g,'-').replace(/\\//g,'_').replace(/=/g,''),clientDataJSON:btoa(String.fromCharCode(...new Uint8Array(cr.response.clientDataJSON))).replace(/\\+/g,'-').replace(/\\//g,'_').replace(/=/g,'')}}}})}}catch(e){{return JSON.stringify({{error:e.message}})}}}})()\n""" + result = await ui.run_javascript(js) + try: + data = json.loads(result) + except Exception: + ui.notify("WebAuthn error", type="negative") + return + if "error" in data: + ui.notify(f"WebAuthn: {data['error']}", type="negative") + return + try: + cred = verify_registration(result, webauthn_state["challenge"]) + except Exception as e: + ui.notify(f"Failed: {e}", type="negative") + return + async with async_session() as session: + session.add(MFAMethod(name="Security Key", type="portable", payload=cred, user_id=user_id)) await session.commit() - ui.notify("Token deleted") - await refresh_tokens() + ui.notify("Security key registered!", type="positive") + await refresh_methods() - with ui.card().classes("w-full"): - ui.label("API Tokens").classes("text-subtitle1 text-bold") - ui.separator() - ui.label("Use API tokens for programmatic access to the REST API.").classes("text-caption text-grey-7") + await refresh_methods() + with ui.row().classes("gap-2 q-mt-xs"): + ui.button("+ Add MFA Method", on_click=start_totp).props(BTN_OUTLINE) + ui.button("+ Add Security Key", on_click=lambda: start_webauthn()).props(BTN_OUTLINE) - token_columns = [ - {"name": "created", "label": "Created", "field": "created", "align": "left"}, - {"name": "expires", "label": "Expires", "field": "expires", "align": "left"}, - {"name": "status", "label": "Status", "field": "status", "align": "left"}, - {"name": "actions", "label": "", "field": "id", "align": "center"}, - ] - token_table = ui.table(columns=token_columns, rows=[], row_key="id").classes("w-full") - token_table.add_slot( - "body-cell-actions", - ''' - - - - ''', + # ===== Danger Zone ===== + _section_header("Danger Zone") + async with async_session() as session: + admin_count = (await session.execute(select(func.count()).select_from(User).where(User.role == "admin"))).scalar() + is_only_admin = user.role == "admin" and admin_count <= 1 + + async def confirm_delete(): + with ui.dialog(value=True) as dlg: + with ui.card().classes("w-80"): + ui.label("Are you sure?").classes("text-subtitle2 text-negative") + ui.label(f"Type {user.email} to confirm:").classes("text-xs q-my-xs") + ci = ui.input().props("outlined dense").classes("w-full").style("font-size: 0.8rem") + async def do_del(): + if ci.value.strip() != user.email: + ui.notify("Email doesn't match", type="negative") + return + async with async_session() as session: + for model in (Device, Rule, MFAMethod, ApiToken, OIDCConnection): + for item in (await session.execute(select(model).where(model.user_id == user_id))).scalars().all(): + await session.delete(item) + u = await session.get(User, user_id) + if u: + await session.delete(u) + await session.commit() + dlg.close() + app.storage.user.clear() + ui.navigate.to("/login") + with ui.row().classes("w-full justify-end q-mt-sm gap-2"): + ui.button("Cancel", on_click=dlg.close).props("flat dense size=sm") + ui.button("Delete", on_click=do_del).props("color=negative dense size=sm unelevated") + + ui.button("Delete Your Account", icon="delete", on_click=confirm_delete).props( + BTN_DANGER + (" disable" if is_only_admin else "") ) - token_table.on("delete", lambda e: delete_token(e.args)) - with ui.row().classes("items-center gap-2 q-mt-sm"): - token_days = ui.input("Expires in (days)", value="30").props("outlined dense").classes("w-40") - ui.button("Create Token", icon="add", on_click=create_token).props("color=primary") - - await refresh_tokens() + # Password dialog + with ui.dialog() as pw_dialog: + with ui.card().classes("w-96"): + ui.label("Change Email or Password").classes("text-subtitle1 text-weight-medium") + ui.separator() + cur = ui.input("Current Password", password=True, password_toggle_button=True).props("outlined dense").classes("w-full") + npw = ui.input("New Password", password=True, password_toggle_button=True).props("outlined dense").classes("w-full") + cpw = ui.input("Confirm Password", password=True, password_toggle_button=True).props("outlined dense").classes("w-full") + async def save_pw(): + if not cur.value or not npw.value: + ui.notify("All fields required", type="negative") + return + if npw.value != cpw.value: + ui.notify("Passwords don't match", type="negative") + return + if len(npw.value) < 8: + ui.notify("Min 8 characters", type="negative") + return + async with async_session() as session: + u = await session.get(User, user_id) + if not verify_password(cur.value, u.password_hash): + ui.notify("Wrong current password", type="negative") + return + u.password_hash = hash_password(npw.value) + session.add(u) + await session.commit() + ui.notify("Password changed", type="positive") + pw_dialog.close() + with ui.row().classes("w-full justify-end q-mt-sm"): + ui.button("Cancel", on_click=pw_dialog.close).props("flat") + ui.button("Save", on_click=save_pw).props("color=primary unelevated") diff --git a/wiregui/tasks/connectivity.py b/wiregui/tasks/connectivity.py index 9cac99b..25366a2 100644 --- a/wiregui/tasks/connectivity.py +++ b/wiregui/tasks/connectivity.py @@ -12,7 +12,7 @@ from wiregui.models.connectivity_check import ConnectivityCheck from wiregui.services import notifications from wiregui.utils.time import utcnow -DEFAULT_URL = "https://ping-dev.firezone.dev" +DEFAULT_URL = "https://one.one.one.one/cdn-cgi/trace" DEFAULT_INTERVAL = 300 # 5 minutes