Add account deletion cleanup route

2026-05-09 06:50:22 -04:00
parent a8d2688020
commit 91821a0fc9
7 changed files with 724 additions and 1 deletions
@@ -0,0 +1,373 @@
+import type { AuthContext } from "./auth.js";
+import { authSessionCacheKvKey } from "./auth.js";
+import type { ElyD1PreparedStatement, Env } from "./bindings.js";
+import { StorageObjectError, deleteKnownObject } from "./storage.js";
+
+const ACCOUNT_DELETION_CONFIRMATION = "delete-elydora-account";
+const IDEMPOTENCY_KEY_PATTERN = /^[a-zA-Z0-9._:-]{16,128}$/;
+const ACCOUNT_DELETION_EVENT_QUERY = `
+  SELECT actor_device_id, outcome, subject_id, created_at
+  FROM audit_events
+  WHERE event_id = ? AND event_type = 'account.delete'
+`;
+const ACCOUNT_DELETION_COUNTS_QUERY = `
+  SELECT
+    (SELECT COUNT(*) FROM user_devices WHERE user_id = ?) AS devices,
+    (SELECT COUNT(*) FROM device_approvals WHERE user_id = ?) AS approvals,
+    (SELECT COUNT(*) FROM sync_objects WHERE user_id = ?) AS sync_objects,
+    (SELECT COUNT(*) FROM sync_change_log WHERE user_id = ?) AS sync_changes,
+    (SELECT COUNT(*) FROM sync_snapshots WHERE user_id = ?) AS sync_snapshots,
+    (SELECT COUNT(*) FROM sync_tombstones WHERE user_id = ?) AS sync_tombstones,
+    (SELECT COUNT(*) FROM audit_events WHERE user_id = ?) AS audit_events,
+    (SELECT COUNT(*) FROM better_auth_session_device_context WHERE user_id = ?) AS session_device_contexts,
+    (SELECT COUNT(*) FROM better_auth_session WHERE userId = ?) AS sessions,
+    (SELECT COUNT(*) FROM better_auth_account WHERE userId = ?) AS accounts,
+    (SELECT COUNT(*) FROM better_auth_user WHERE id = ?) AS users
+`;
+const ACCOUNT_DELETION_R2_KEYS_QUERY = `
+  SELECT payload_r2_key AS r2_key
+  FROM sync_objects
+  WHERE user_id = ? AND payload_r2_key IS NOT NULL
+  UNION
+  SELECT r2_key
+  FROM sync_snapshots
+  WHERE user_id = ?
+  ORDER BY r2_key ASC
+`;
+const DELETE_SYNC_CHANGE_LOG_QUERY = "DELETE FROM sync_change_log WHERE user_id = ?";
+const DELETE_SYNC_TOMBSTONES_QUERY = "DELETE FROM sync_tombstones WHERE user_id = ?";
+const DELETE_SYNC_SNAPSHOTS_QUERY = "DELETE FROM sync_snapshots WHERE user_id = ?";
+const DELETE_SYNC_OBJECTS_QUERY = "DELETE FROM sync_objects WHERE user_id = ?";
+const DELETE_DEVICE_APPROVALS_QUERY = "DELETE FROM device_approvals WHERE user_id = ?";
+const DELETE_USER_DEVICES_QUERY = "DELETE FROM user_devices WHERE user_id = ?";
+const DELETE_SESSION_DEVICE_CONTEXTS_QUERY =
+  "DELETE FROM better_auth_session_device_context WHERE user_id = ?";
+const DELETE_BETTER_AUTH_SESSIONS_QUERY = "DELETE FROM better_auth_session WHERE userId = ?";
+const DELETE_BETTER_AUTH_ACCOUNTS_QUERY = "DELETE FROM better_auth_account WHERE userId = ?";
+const DELETE_BETTER_AUTH_USER_QUERY = "DELETE FROM better_auth_user WHERE id = ?";
+const DELETE_USER_AUDIT_EVENTS_QUERY = "DELETE FROM audit_events WHERE user_id = ?";
+const ACCOUNT_DELETION_AUDIT_INSERT_QUERY = `
+  INSERT INTO audit_events (
+    event_id,
+    user_id,
+    actor_device_id,
+    event_type,
+    subject_type,
+    subject_id,
+    outcome,
+    metadata_hash,
+    created_at
+  ) VALUES (?, NULL, ?, 'account.delete', 'account', ?, 'success', ?, ?)
+  ON CONFLICT(event_id) DO NOTHING
+`;
+
+export interface AccountDeletionDocument {
+  version: 1;
+  account_hash: string;
+  device_id: string;
+  idempotency_key: string;
+  deleted_at: number;
+  deleted: AccountDeletionDeletedDocument;
+}
+
+export interface AccountDeletionDeletedDocument {
+  devices: number;
+  approvals: number;
+  sync_objects: number;
+  sync_changes: number;
+  sync_snapshots: number;
+  sync_tombstones: number;
+  audit_events: number;
+  session_device_contexts: number;
+  sessions: number;
+  accounts: number;
+  users: number;
+  r2_objects: number;
+  kv_session_cache: number;
+}
+
+interface AccountDeletionRequest {
+  idempotencyKey: string;
+}
+
+interface AccountDeletionEventRow {
+  actor_device_id: unknown;
+  outcome: unknown;
+  subject_id: unknown;
+  created_at: unknown;
+}
+
+interface AccountDeletionCountsRow {
+  devices: unknown;
+  approvals: unknown;
+  sync_objects: unknown;
+  sync_changes: unknown;
+  sync_snapshots: unknown;
+  sync_tombstones: unknown;
+  audit_events: unknown;
+  session_device_contexts: unknown;
+  sessions: unknown;
+  accounts: unknown;
+  users: unknown;
+}
+
+interface AccountDeletionR2KeyRow {
+  r2_key: unknown;
+}
+
+type RequestBody = Record<string, unknown>;
+
+export class AccountDeletionRequestError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "AccountDeletionRequestError";
+  }
+}
+
+export class AccountDeletionPersistenceError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "AccountDeletionPersistenceError";
+  }
+}
+
+export async function accountDeletionDocument(
+  request: Request,
+  env: Env,
+  context: AuthContext,
+  nowSeconds = Math.floor(Date.now() / 1000),
+): Promise<AccountDeletionDocument> {
+  const deviceId = currentDeviceId(context);
+  const deletion = await accountDeletionRequest(request);
+  const accountHash = await sha256Hex(textBytes(context.userId));
+  const idempotencyHash = await sha256Hex(textBytes(deletion.idempotencyKey));
+  const eventId = accountDeletionEventId(accountHash, idempotencyHash);
+  const existingEvent = await env.ELY_DB.prepare(ACCOUNT_DELETION_EVENT_QUERY)
+    .bind(eventId)
+    .first<AccountDeletionEventRow>();
+  if (existingEvent !== null) {
+    return existingDeletionDocument(accountHash, deviceId, deletion, existingEvent);
+  }
+
+  const counts = await accountDeletionCounts(env, context.userId);
+  const r2Keys = await accountDeletionR2Keys(env, context.userId);
+  for (const key of r2Keys) {
+    await deleteAccountObject(env, key);
+  }
+
+  await env.ELY_DB.batch(
+    accountDeletionStatements(
+      env,
+      context.userId,
+      deviceId,
+      accountHash,
+      idempotencyHash,
+      eventId,
+      nowSeconds,
+    ),
+  );
+  await deleteCurrentSessionCache(env, context.tokenHash);
+
+  return {
+    version: 1,
+    account_hash: accountHash,
+    device_id: deviceId,
+    idempotency_key: deletion.idempotencyKey,
+    deleted_at: nowSeconds,
+    deleted: { ...counts, r2_objects: r2Keys.length, kv_session_cache: 1 },
+  };
+}
+
+function existingDeletionDocument(
+  accountHash: string,
+  deviceId: string,
+  deletion: AccountDeletionRequest,
+  row: AccountDeletionEventRow,
+): AccountDeletionDocument {
+  if (
+    row.actor_device_id !== deviceId ||
+    row.outcome !== "success" ||
+    row.subject_id !== accountHash
+  ) {
+    throw new AccountDeletionRequestError("account_deletion_replay_mismatch");
+  }
+  return {
+    version: 1,
+    account_hash: accountHash,
+    device_id: deviceId,
+    idempotency_key: deletion.idempotencyKey,
+    deleted_at: integer(row.created_at, "created_at"),
+    deleted: emptyDeletedDocument(),
+  };
+}
+
+async function accountDeletionCounts(
+  env: Env,
+  userId: string,
+): Promise<Omit<AccountDeletionDeletedDocument, "r2_objects" | "kv_session_cache">> {
+  const row = await env.ELY_DB.prepare(ACCOUNT_DELETION_COUNTS_QUERY)
+    .bind(userId, userId, userId, userId, userId, userId, userId, userId, userId, userId, userId)
+    .first<AccountDeletionCountsRow>();
+  if (row === null) {
+    throw new AccountDeletionPersistenceError("account_deletion_counts_missing");
+  }
+  return {
+    devices: integer(row.devices, "devices"),
+    approvals: integer(row.approvals, "approvals"),
+    sync_objects: integer(row.sync_objects, "sync_objects"),
+    sync_changes: integer(row.sync_changes, "sync_changes"),
+    sync_snapshots: integer(row.sync_snapshots, "sync_snapshots"),
+    sync_tombstones: integer(row.sync_tombstones, "sync_tombstones"),
+    audit_events: integer(row.audit_events, "audit_events"),
+    session_device_contexts: integer(row.session_device_contexts, "session_device_contexts"),
+    sessions: integer(row.sessions, "sessions"),
+    accounts: integer(row.accounts, "accounts"),
+    users: integer(row.users, "users"),
+  };
+}
+
+async function accountDeletionR2Keys(env: Env, userId: string): Promise<string[]> {
+  const result = await env.ELY_DB.prepare(ACCOUNT_DELETION_R2_KEYS_QUERY)
+    .bind(userId, userId)
+    .all<AccountDeletionR2KeyRow>();
+  return result.results.map(r2Key);
+}
+
+function accountDeletionStatements(
+  env: Env,
+  userId: string,
+  deviceId: string,
+  accountHash: string,
+  idempotencyHash: string,
+  eventId: string,
+  nowSeconds: number,
+): ElyD1PreparedStatement[] {
+  return [
+    env.ELY_DB.prepare(DELETE_SYNC_CHANGE_LOG_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_SYNC_TOMBSTONES_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_SYNC_SNAPSHOTS_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_SYNC_OBJECTS_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_DEVICE_APPROVALS_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_SESSION_DEVICE_CONTEXTS_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_USER_DEVICES_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_BETTER_AUTH_SESSIONS_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_BETTER_AUTH_ACCOUNTS_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_BETTER_AUTH_USER_QUERY).bind(userId),
+    env.ELY_DB.prepare(DELETE_USER_AUDIT_EVENTS_QUERY).bind(userId),
+    env.ELY_DB.prepare(ACCOUNT_DELETION_AUDIT_INSERT_QUERY).bind(
+      eventId,
+      deviceId,
+      accountHash,
+      idempotencyHash,
+      nowSeconds,
+    ),
+  ];
+}
+
+async function deleteAccountObject(env: Env, key: string): Promise<void> {
+  try {
+    await deleteKnownObject(env.ELY_STORAGE, key);
+  } catch (error) {
+    if (error instanceof StorageObjectError) {
+      throw new AccountDeletionPersistenceError(error.message);
+    }
+    throw error;
+  }
+}
+
+function deleteCurrentSessionCache(env: Env, tokenHash: string): Promise<void> {
+  return env.ELY_KV.delete(authSessionCacheKvKey(env.ELY_ENVIRONMENT, tokenHash));
+}
+
+async function accountDeletionRequest(request: Request): Promise<AccountDeletionRequest> {
+  const body = await requestBody(request);
+  assertOnlyFields(body, ["version", "confirmation", "idempotency_key"]);
+  if (body.version !== 1) {
+    throw new AccountDeletionRequestError("version_invalid");
+  }
+  if (body.confirmation !== ACCOUNT_DELETION_CONFIRMATION) {
+    throw new AccountDeletionRequestError("confirmation_invalid");
+  }
+  return { idempotencyKey: idempotencyKey(body.idempotency_key) };
+}
+
+async function requestBody(request: Request): Promise<RequestBody> {
+  let value: unknown;
+  try {
+    value = await request.json();
+  } catch {
+    throw new AccountDeletionRequestError("json_invalid");
+  }
+  if (typeof value !== "object" || value === null || Array.isArray(value)) {
+    throw new AccountDeletionRequestError("body_invalid");
+  }
+  return value as RequestBody;
+}
+
+function assertOnlyFields(value: RequestBody, fields: string[]): void {
+  const allowed = new Set(fields);
+  for (const field of Object.keys(value)) {
+    if (!allowed.has(field)) {
+      throw new AccountDeletionRequestError(`unexpected_field:${field}`);
+    }
+  }
+}
+
+function currentDeviceId(context: AuthContext): string {
+  if (context.deviceId === undefined) {
+    throw new AccountDeletionRequestError("device_context_required");
+  }
+  return context.deviceId;
+}
+
+function idempotencyKey(value: unknown): string {
+  if (typeof value !== "string" || !IDEMPOTENCY_KEY_PATTERN.test(value)) {
+    throw new AccountDeletionRequestError("idempotency_key_invalid");
+  }
+  return value;
+}
+
+function r2Key(row: AccountDeletionR2KeyRow): string {
+  if (typeof row.r2_key !== "string") {
+    throw new AccountDeletionPersistenceError("r2_key_invalid");
+  }
+  return row.r2_key;
+}
+
+function integer(value: unknown, label: string): number {
+  if (typeof value !== "number" || !Number.isSafeInteger(value) || value < 0) {
+    throw new AccountDeletionPersistenceError(`${label}_invalid`);
+  }
+  return value;
+}
+
+function emptyDeletedDocument(): AccountDeletionDeletedDocument {
+  return {
+    devices: 0,
+    approvals: 0,
+    sync_objects: 0,
+    sync_changes: 0,
+    sync_snapshots: 0,
+    sync_tombstones: 0,
+    audit_events: 0,
+    session_device_contexts: 0,
+    sessions: 0,
+    accounts: 0,
+    users: 0,
+    r2_objects: 0,
+    kv_session_cache: 0,
+  };
+}
+
+function accountDeletionEventId(accountHash: string, idempotencyHash: string): string {
+  return `account-delete:${accountHash}:${idempotencyHash}`;
+}
+
+function textBytes(value: string): Uint8Array {
+  return new TextEncoder().encode(value);
+}
+
+async function sha256Hex(payload: Uint8Array): Promise<string> {
+  const digest = await crypto.subtle.digest("SHA-256", payload);
+  return [...new Uint8Array(digest)].map((byte) => byte.toString(16).padStart(2, "0")).join("");
+}
@@ -1,5 +1,6 @@
 export interface ElyKvNamespace {
  get(key: string): Promise<string | null>;
+  delete(key: string): Promise<void>;
 }

 export interface ElyR2Object {
@@ -1,5 +1,14 @@
 import type { Env } from "./bindings.js";
-import { withAuthenticatedApiControls, withPublicApiControls } from "./api_controls.js";
+import {
+  withApprovedDeviceApiControls,
+  withAuthenticatedApiControls,
+  withPublicApiControls,
+} from "./api_controls.js";
+import {
+  AccountDeletionPersistenceError,
+  AccountDeletionRequestError,
+  accountDeletionDocument,
+} from "./account_deletion.js";
 import { handleBetterAuthRoute } from "./better_auth.js";
 import {
  DevicePermissionError,
@@ -178,6 +187,37 @@ export async function handleRequest(request: Request, env: Env): Promise<Respons
      },
    );
  }
+  if (url.pathname === "/api/account/delete") {
+    return withApprovedDeviceApiControls(
+      request,
+      env,
+      "account.delete",
+      ["POST"],
+      async (context) => {
+        try {
+          return jsonResponse(await accountDeletionDocument(request, env, context), 200, {
+            "Cache-Control": "no-store",
+          });
+        } catch (error) {
+          if (error instanceof AccountDeletionRequestError) {
+            return jsonResponse(
+              { error: "invalid_account_deletion" },
+              400,
+              { "Cache-Control": "no-store" },
+            );
+          }
+          if (error instanceof AccountDeletionPersistenceError) {
+            return jsonResponse(
+              { error: "account_deletion_failed" },
+              500,
+              { "Cache-Control": "no-store" },
+            );
+          }
+          throw error;
+        }
+      },
+    );
+  }
  const syncResponse = await handleSyncRoute(request, env, url);
  if (syncResponse !== null) {
    return syncResponse;