#!/usr/bin/env python3
"""One-shot patch: add user accounts + email auth to netdesk_server.py."""
import re
p = "server/netdesk_server.py"
s = open(p, encoding="utf-8").read()
ANCHOR = "TUNNELS: dict[int, web.WebSocketResponse] = {} # tid -> viewer-side socket\nTUNNEL_TID = [1]"
assert ANCHOR in s, "tunnel anchor missing"
BLOCK = '''
# ---------------------------------------------------------------- user accounts (email sign-in for viewers)
import sqlite3
import smtplib
from email.message import EmailMessage as _MailMsg
ACCOUNTS_DB = os.path.join(_server_root, "accounts.db")
USER_SESSIONS = {} # token -> email
def _acct_db():
conn = sqlite3.connect(ACCOUNTS_DB)
conn.execute(
"CREATE TABLE IF NOT EXISTS users(id INTEGER PRIMARY KEY AUTOINCREMENT,"
" email TEXT UNIQUE NOT NULL, pw_salt TEXT NOT NULL, pw_hash TEXT NOT NULL,"
" verified INTEGER NOT NULL DEFAULT 0, verify_token TEXT,"
" reset_token TEXT, reset_exp INTEGER, created INTEGER NOT NULL)")
return conn
def _pbkdf(password, salt):
import hashlib as _h
return _h.pbkdf2_hmac("sha256", password.encode(), salt, 200_000).hex()
def _mail_html(title, body_html, link=None, link_text=None):
link_html = ""
if link:
link_html = (
''
'' + link_text + "
"
'Or paste this link: ' + link + "
")
logo = (''
'')
return (
''
'
' + logo +
'NetDesk
'
'
' + title + "
"
'
' + body_html + "
"
+ link_html +
'
'
"If you did not request this, ignore this email.