|
|
|
@@ -7,13 +7,14 @@ import httpx
|
|
|
|
|
from authlib.integrations.starlette_client import OAuth, OAuthError
|
|
|
|
|
from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException, Request
|
|
|
|
|
from fastapi.responses import JSONResponse, RedirectResponse
|
|
|
|
|
from starlette.requests import HTTPConnection
|
|
|
|
|
from sqlalchemy import delete, func, select
|
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
|
|
|
|
|
|
from app import email as email_mod
|
|
|
|
|
from app import sysconfig
|
|
|
|
|
from app.config import settings
|
|
|
|
|
from app.db import get_db
|
|
|
|
|
from app.db import SessionLocal, get_db
|
|
|
|
|
from app.models import AuthToken, DemoWhitelist, Invite, OAuthToken, User
|
|
|
|
|
from app.ratelimit import RateLimiter
|
|
|
|
|
from app.security import decrypt, encrypt, hash_password, hash_token, verify_password
|
|
|
|
@@ -72,6 +73,8 @@ READ_SCOPE = f"{WRITE_SCOPE}.readonly"
|
|
|
|
|
BASE_SCOPES = "openid email profile"
|
|
|
|
|
READ_SCOPES = f"{BASE_SCOPES} {READ_SCOPE}"
|
|
|
|
|
WRITE_SCOPES = f"{BASE_SCOPES} {READ_SCOPE} {WRITE_SCOPE}"
|
|
|
|
|
# The YouTube grants (read ⊂ write). A sign-in must never silently drop one of these — see _store_token.
|
|
|
|
|
_YOUTUBE_SCOPES = frozenset({READ_SCOPE, WRITE_SCOPE})
|
|
|
|
|
|
|
|
|
|
log = logging.getLogger("siftlode.auth")
|
|
|
|
|
|
|
|
|
@@ -131,6 +134,20 @@ def has_write_scope(user: User) -> bool:
|
|
|
|
|
return WRITE_SCOPE in tok.scopes.split()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def admin_notify_emails(db: Session) -> list[str]:
|
|
|
|
|
"""Who to email about admin events (e.g. a new access request). The ACTUAL admins — every active,
|
|
|
|
|
non-suspended `role=admin` user — UNIONED with the env ADMIN_EMAILS bootstrap list. Using the DB
|
|
|
|
|
roles means an admin promoted through the UI is notified too (env alone would miss them, yet they
|
|
|
|
|
CAN approve requests — the approve UI is role-gated, so the notify set must match); the env union
|
|
|
|
|
keeps a freshly-seeded instance working before any DB admin exists."""
|
|
|
|
|
db_admins = db.execute(
|
|
|
|
|
select(User.email).where(
|
|
|
|
|
User.role == "admin", User.is_active.is_(True), User.is_suspended.is_(False)
|
|
|
|
|
)
|
|
|
|
|
).scalars()
|
|
|
|
|
return sorted({e.lower() for e in db_admins if e} | settings.admin_email_set)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def is_allowed(db: Session, email: str) -> bool:
|
|
|
|
|
"""May this email sign in? An approved Invite is the source of truth; the env
|
|
|
|
|
ALLOWED_EMAILS/ADMIN_EMAILS remain a bootstrap fallback (e.g. a fresh DB)."""
|
|
|
|
@@ -281,12 +298,9 @@ async def callback(
|
|
|
|
|
# land the user on a friendly "request received" screen instead of a raw 403.
|
|
|
|
|
if email:
|
|
|
|
|
inv = upsert_pending_invite(db, email)
|
|
|
|
|
if inv is not None and settings.admin_email_set:
|
|
|
|
|
background.add_task(
|
|
|
|
|
email_mod.send_admin_new_request,
|
|
|
|
|
sorted(settings.admin_email_set),
|
|
|
|
|
email,
|
|
|
|
|
)
|
|
|
|
|
admins = admin_notify_emails(db)
|
|
|
|
|
if inv is not None and admins:
|
|
|
|
|
background.add_task(email_mod.send_admin_new_request, admins, email)
|
|
|
|
|
return RedirectResponse(url="/?access=requested")
|
|
|
|
|
return RedirectResponse(url="/?access=denied")
|
|
|
|
|
|
|
|
|
@@ -375,6 +389,18 @@ def _store_token(db: Session, user: User, token: dict) -> None:
|
|
|
|
|
include_granted_scopes=true means `scope` is the union of everything granted, so this
|
|
|
|
|
reflects read/write upgrades correctly."""
|
|
|
|
|
tok = user.token or OAuthToken(user=user)
|
|
|
|
|
old = set((tok.scopes or "").split())
|
|
|
|
|
new = set((token.get("scope") or "").split())
|
|
|
|
|
# A plain re-sign-in requests only BASE_SCOPES, and Google can hand back a FRESH base-only
|
|
|
|
|
# access+refresh token. Adopting it would DESTROY a previously-granted YouTube read/write grant —
|
|
|
|
|
# can_read/can_write flip to false and the feed demands a reconnect (the exact bug users hit on a
|
|
|
|
|
# logout→login). The OLD refresh token stays valid on Google's side, so when this exchange would
|
|
|
|
|
# NARROW our YouTube scopes, keep the WHOLE existing grant (refresh token, access token, scopes)
|
|
|
|
|
# untouched. The grant only shrinks on a full disconnect (purge_user deletes the token row); an
|
|
|
|
|
# externally-revoked scope surfaces later as a YouTube API 403 → reconnect, not through this write.
|
|
|
|
|
if (old & _YOUTUBE_SCOPES) - new:
|
|
|
|
|
db.add(tok)
|
|
|
|
|
return
|
|
|
|
|
if token.get("refresh_token"):
|
|
|
|
|
tok.refresh_token_enc = encrypt(token["refresh_token"])
|
|
|
|
|
# Encrypt the access token at rest too (it's a live ~1h bearer credential), matching the
|
|
|
|
@@ -382,7 +408,8 @@ def _store_token(db: Session, user: User, token: dict) -> None:
|
|
|
|
|
tok.access_token = encrypt(token.get("access_token"))
|
|
|
|
|
expires_at = token.get("expires_at")
|
|
|
|
|
tok.expiry = datetime.fromtimestamp(expires_at, tz=timezone.utc) if expires_at else None
|
|
|
|
|
tok.scopes = token.get("scope") or BASE_SCOPES
|
|
|
|
|
# Union (never narrow): a broader-or-equal exchange may still list only the newly-added scope.
|
|
|
|
|
tok.scopes = " ".join(sorted(old | new)) or BASE_SCOPES
|
|
|
|
|
db.add(tok)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -413,7 +440,9 @@ def purge_user(db: Session, user: User, background: BackgroundTasks) -> None:
|
|
|
|
|
email = user.email.lower()
|
|
|
|
|
user_id = user.id
|
|
|
|
|
tok = user.token
|
|
|
|
|
google_token = (decrypt(tok.refresh_token_enc) or tok.access_token) if tok else None
|
|
|
|
|
# Both are stored encrypted — decrypt for the revoke call (the access-token fallback was passing
|
|
|
|
|
# ciphertext, so a token row with no refresh token silently failed to revoke).
|
|
|
|
|
google_token = (decrypt(tok.refresh_token_enc) or decrypt(tok.access_token)) if tok else None
|
|
|
|
|
# Raw delete so Postgres applies ON DELETE CASCADE / SET NULL on every dependent table.
|
|
|
|
|
db.execute(delete(User).where(User.id == user_id))
|
|
|
|
|
db.execute(delete(Invite).where(Invite.email == email))
|
|
|
|
@@ -494,10 +523,9 @@ async def request_access(
|
|
|
|
|
if is_allowed(db, email):
|
|
|
|
|
return {"status": "approved"} # already allowed — just sign in
|
|
|
|
|
inv = upsert_pending_invite(db, email)
|
|
|
|
|
if inv is not None and settings.admin_email_set:
|
|
|
|
|
background.add_task(
|
|
|
|
|
email_mod.send_admin_new_request, sorted(settings.admin_email_set), email
|
|
|
|
|
)
|
|
|
|
|
admins = admin_notify_emails(db)
|
|
|
|
|
if inv is not None and admins:
|
|
|
|
|
background.add_task(email_mod.send_admin_new_request, admins, email)
|
|
|
|
|
return {"status": "pending"}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -596,11 +624,24 @@ def register(
|
|
|
|
|
if not _register_limiter.allow(_client_ip(request)):
|
|
|
|
|
return {"status": "ok"} # silently throttle; uniform response
|
|
|
|
|
|
|
|
|
|
existing = db.execute(select(User).where(User.email == email)).scalar_one_or_none()
|
|
|
|
|
if existing is None:
|
|
|
|
|
# Without working SMTP we can't deliver a verification link, so email ownership can't
|
|
|
|
|
# gate sign-in. Admin approval stays the real gate (is_active=False); mark the account
|
|
|
|
|
# verified so the flow still completes on a no-SMTP self-host. See email.email_enabled().
|
|
|
|
|
# Do the existence check + account creation OFF the response path (a background task with its own
|
|
|
|
|
# session). The create path hashes the password + writes several rows + schedules emails; doing it
|
|
|
|
|
# inline would make a NEW email respond measurably slower than an already-registered one (which
|
|
|
|
|
# skips all that) — a timing oracle for enumeration. Off-path, any valid email responds the same.
|
|
|
|
|
background.add_task(_register_account, email, password)
|
|
|
|
|
return {"status": "ok"}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _register_account(email: str, password: str) -> None:
|
|
|
|
|
"""Background worker for /register: create the pending account (+ verification email + admin
|
|
|
|
|
notice) for a genuinely new email; no-op for an already-registered one. Runs after the response
|
|
|
|
|
with its own DB session, so the request's timing never reveals whether the email already exists."""
|
|
|
|
|
with SessionLocal() as db:
|
|
|
|
|
if db.execute(select(User).where(User.email == email)).scalar_one_or_none() is not None:
|
|
|
|
|
return # already registered — nothing to do (the uniform "ok" was already returned)
|
|
|
|
|
# Without working SMTP we can't deliver a verification link, so email ownership can't gate
|
|
|
|
|
# sign-in. Admin approval stays the real gate (is_active=False); mark the account verified so
|
|
|
|
|
# the flow still completes on a no-SMTP self-host. See email.email_enabled().
|
|
|
|
|
email_ok = email_mod.email_enabled()
|
|
|
|
|
user = User(
|
|
|
|
|
email=email,
|
|
|
|
@@ -610,20 +651,14 @@ def register(
|
|
|
|
|
)
|
|
|
|
|
db.add(user)
|
|
|
|
|
db.flush()
|
|
|
|
|
upsert_pending_invite(db, email) # admin-approval gate
|
|
|
|
|
upsert_pending_invite(db, email) # admin-approval gate (commits)
|
|
|
|
|
if email_ok:
|
|
|
|
|
raw = _issue_token(db, user, "verify", VERIFY_TTL)
|
|
|
|
|
# Fragment (#), not a query token: keeps the token out of proxy/access logs + Referer.
|
|
|
|
|
# The SPA reads the fragment and POSTs it to /auth/verify (SB3).
|
|
|
|
|
background.add_task(
|
|
|
|
|
email_mod.send_verify_email, email, f"{_app_base()}/#verify={raw}"
|
|
|
|
|
)
|
|
|
|
|
if settings.admin_email_set:
|
|
|
|
|
background.add_task(
|
|
|
|
|
email_mod.send_admin_new_request, sorted(settings.admin_email_set), email
|
|
|
|
|
)
|
|
|
|
|
# Existing email → do nothing visible (no enumeration). Uniform success either way.
|
|
|
|
|
return {"status": "ok"}
|
|
|
|
|
# Fragment (#), not a query token: keeps the token out of proxy/access logs + Referer (SB3).
|
|
|
|
|
email_mod.send_verify_email(email, f"{_app_base()}/#verify={raw}")
|
|
|
|
|
admins = admin_notify_emails(db)
|
|
|
|
|
if admins:
|
|
|
|
|
email_mod.send_admin_new_request(admins, email)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.post("/verify")
|
|
|
|
@@ -692,21 +727,30 @@ def password_reset_request(
|
|
|
|
|
payload: dict,
|
|
|
|
|
request: Request,
|
|
|
|
|
background: BackgroundTasks,
|
|
|
|
|
db: Session = Depends(get_db),
|
|
|
|
|
) -> dict:
|
|
|
|
|
"""Request a password-reset link. Uniform response regardless of whether the email has a
|
|
|
|
|
password account, so it can't probe for registered emails."""
|
|
|
|
|
"""Request a password-reset link. Uniform response + timing regardless of whether the email has a
|
|
|
|
|
password account (the lookup runs off the response path), so it can't probe for registered emails."""
|
|
|
|
|
if not _reset_limiter.allow(_client_ip(request)):
|
|
|
|
|
return {"status": "ok"}
|
|
|
|
|
email = (payload.get("email") or "").strip().lower()
|
|
|
|
|
# Do the account lookup + token issue + email OFF the response path (a background task with its
|
|
|
|
|
# own session), so the endpoint takes the same time for any valid email whether or not it has a
|
|
|
|
|
# password account — no timing-based enumeration. Any valid email schedules the same task.
|
|
|
|
|
if valid_email(email):
|
|
|
|
|
background.add_task(_send_reset_if_eligible, email)
|
|
|
|
|
return {"status": "ok"}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _send_reset_if_eligible(email: str) -> None:
|
|
|
|
|
"""Background worker for password_reset_request: issue a reset token + email the link, but ONLY
|
|
|
|
|
for a real (non-demo) password account. Runs after the response with its own DB session, so the
|
|
|
|
|
request's timing never reveals whether the account exists. Token rides the URL FRAGMENT (#) so it
|
|
|
|
|
can't leak into proxy/access logs or a Referer (SB3)."""
|
|
|
|
|
with SessionLocal() as db:
|
|
|
|
|
user = db.execute(select(User).where(User.email == email)).scalar_one_or_none()
|
|
|
|
|
if user is not None and user.password_hash and not user.is_demo:
|
|
|
|
|
raw = _issue_token(db, user, "reset", RESET_TTL)
|
|
|
|
|
# Token in the URL FRAGMENT (#), not the query (?): a fragment is never sent to the
|
|
|
|
|
# server, so it can't leak into proxy/access logs or a Referer header (SB3).
|
|
|
|
|
background.add_task(email_mod.send_password_reset, email, f"{_app_base()}/#reset={raw}")
|
|
|
|
|
return {"status": "ok"}
|
|
|
|
|
email_mod.send_password_reset(email, f"{_app_base()}/#reset={raw}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@router.post("/password-reset/confirm")
|
|
|
|
@@ -741,21 +785,22 @@ def password_reset_confirm(payload: dict, db: Session = Depends(get_db)) -> dict
|
|
|
|
|
ACTIVE_ACCOUNT_HEADER = "x-siftlode-account"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def resolved_user_id(request: Request) -> tuple[int | None, bool]:
|
|
|
|
|
"""Which account this request acts as, and whether that's the session's default account.
|
|
|
|
|
def resolved_user_id(conn: HTTPConnection) -> tuple[int | None, bool]:
|
|
|
|
|
"""Which account this connection acts as, and whether that's the session's default account.
|
|
|
|
|
Takes any HTTPConnection — an HTTP Request OR a WebSocket — so the WS push channel shares this
|
|
|
|
|
exact per-tab resolution instead of re-implementing it.
|
|
|
|
|
|
|
|
|
|
The signed cookie holds the *wallet* (`account_ids`, every account signed into this browser)
|
|
|
|
|
plus a default `user_id`. A request may override the default with the X-Siftlode-Account
|
|
|
|
|
header, but ONLY for an account already in the wallet — so a tab can't impersonate an account
|
|
|
|
|
that never authenticated here. Everything else (WebSocket, plain navigations) keeps using the
|
|
|
|
|
cookie default.
|
|
|
|
|
plus a default `user_id`. A caller may override the default with the X-Siftlode-Account header,
|
|
|
|
|
but ONLY for an account already in the wallet — so a tab can't impersonate an account that never
|
|
|
|
|
authenticated here. Everything else (WebSocket, plain navigations) keeps using the cookie default.
|
|
|
|
|
"""
|
|
|
|
|
default_id = request.session.get("user_id")
|
|
|
|
|
wallet = request.session.get("account_ids") or []
|
|
|
|
|
default_id = conn.session.get("user_id")
|
|
|
|
|
wallet = conn.session.get("account_ids") or []
|
|
|
|
|
# Header for normal XHR; ?account= for contexts that can't set headers (WebSocket, and a
|
|
|
|
|
# plain <a> file download). Both are wallet-gated below, so neither can impersonate an
|
|
|
|
|
# account that never signed into this browser.
|
|
|
|
|
hdr = request.headers.get(ACTIVE_ACCOUNT_HEADER) or request.query_params.get("account")
|
|
|
|
|
hdr = conn.headers.get(ACTIVE_ACCOUNT_HEADER) or conn.query_params.get("account")
|
|
|
|
|
if hdr:
|
|
|
|
|
try:
|
|
|
|
|
hid = int(hdr)
|
|
|
|
|