初次提交

2026-09-06 08:19:02 +08:00
commit 4ce4649aaf
1061 changed files with 337258 additions and 0 deletions
@@ -0,0 +1,17 @@
+import { generateAuthenticationOptions } from '@simplewebauthn/server'
+import { setWebAuthnChallenge } from '~~/server/utils/webauthn-token'
+import { getWebAuthnConfig } from '~~/server/utils/webauthn-config'
+
+export default defineEventHandler(async (event) => {
+  const { rpID } = getWebAuthnConfig(event)
+
+  const options = await generateAuthenticationOptions({
+    rpID,
+    userVerification: 'preferred',
+    allowCredentials: [], 
+  })
+
+  setWebAuthnChallenge(event, options.challenge, 'authentication')
+
+  return options
+})
@@ -0,0 +1,106 @@
+import { verifyAuthenticationResponse } from '@simplewebauthn/server'
+import { getWebAuthnChallenge, clearWebAuthnChallenge } from '~~/server/utils/webauthn-token'
+import { getWebAuthnConfig } from '~~/server/utils/webauthn-config'
+import { db, eq, and, userIdentities } from '~/drizzle/db'
+import { createError, defineEventHandler, setCookie } from 'h3'
+import { JWTEnhanced } from '~~/server/utils/jwt-enhanced'
+
+export default defineEventHandler(async (event) => {
+  const body = await readBody(event)
+  const challengeData = getWebAuthnChallenge(event)
+
+  if (!challengeData) {
+    throw createError({ statusCode: 400, message: 'Challenge 已失效' })
+  }
+
+  // 通过 credentialID 查找对应的用户身份
+  const credentialID = body.id
+  if (!credentialID) {
+    throw createError({ statusCode: 400, message: '缺少 Credential ID' })
+  }
+
+  const identity = await db.query.userIdentities.findFirst({
+    where: and(eq(userIdentities.provider, 'webauthn'), eq(userIdentities.providerUserId, credentialID)),
+    with: { user: true }
+  })
+
+  if (!identity) {
+    throw createError({ statusCode: 400, message: '未找到该 Passkey 关联的账号' })
+  }
+
+  // 检查用户账号状态
+  if (identity.user.status !== 'active') {
+    throw createError({ statusCode: 403, message: '账号已被禁用或注销' })
+  }
+
+  // 解析存储的凭证数据
+  let credentialData
+  try {
+    credentialData = JSON.parse(identity.providerUsername)
+  } catch {
+    throw createError({ statusCode: 500, message: '凭证数据损坏' })
+  }
+
+  const { publicKey, counter } = credentialData
+  
+  // 验证公钥存在
+  if (!publicKey) {
+     throw createError({ statusCode: 500, message: '凭证数据缺失 publicKey' })
+  }
+
+  const { rpID, origin } = getWebAuthnConfig(event)
+
+  // 构造 WebAuthn 凭证对象
+  const credential = {
+    id: credentialID,
+    publicKey: Buffer.from(publicKey, 'base64url'),
+    counter: Number(counter || 0),
+    transports: credentialData.transports
+  }
+
+  try {
+    const verification = await verifyAuthenticationResponse({
+      response: body,
+      expectedChallenge: challengeData.challenge,
+      expectedOrigin: origin,
+      expectedRPID: rpID,
+      credential
+    })
+
+    if (verification.verified) {
+      const newCounter = verification.authenticationInfo.newCounter
+      const currentCounter = Number(counter || 0)
+
+      // 防重放攻击：检查计数器是否递增
+      if (currentCounter > 0 && newCounter <= currentCounter) {
+        console.error('WebAuthn 计数器未递增:', { currentCounter, newCounter, credentialID })
+        throw createError({ statusCode: 400, message: '认证无效：凭证计数器未递增' })
+      }
+
+      credentialData.counter = newCounter
+      
+      await db.update(userIdentities)
+        .set({ providerUsername: JSON.stringify(credentialData) })
+        .where(eq(userIdentities.id, identity.id))
+
+      // 签发登录 Token
+      const token = JWTEnhanced.generateToken(identity.user.id, identity.user.role)
+      setCookie(event, 'auth-token', token, {
+        httpOnly: true,
+        secure: process.env.NODE_ENV === 'production',
+        sameSite: 'lax',
+        maxAge: 60 * 60 * 24 * 7,
+        path: '/'
+      })
+
+      clearWebAuthnChallenge(event)
+      return { success: true, redirect: '/' }
+    } else {
+      throw createError({ statusCode: 400, message: '验证失败' })
+    }
+  } catch (error) {
+    console.error('WebAuthn 登录验证失败:', error)
+    const message = error instanceof Error ? error.message : '验证失败'
+    throw createError({ statusCode: 400, message })
+  }
+})
@@ -0,0 +1,60 @@
+import { generateRegistrationOptions } from '@simplewebauthn/server'
+import { isoUint8Array } from '@simplewebauthn/server/helpers'
+import { setWebAuthnChallenge } from '~~/server/utils/webauthn-token'
+import { getWebAuthnConfig } from '~~/server/utils/webauthn-config'
+import { db, eq, and, userIdentities } from '~/drizzle/db'
+
+export default defineEventHandler(async (event) => {
+  const user = event.context.user
+  if (!user) {
+    throw createError({ statusCode: 401, message: '未授权访问' })
+  }
+
+  // 获取该用户已绑定的 WebAuthn 凭证
+  const existingCredentials = await db.query.userIdentities.findMany({
+    where: and(eq(userIdentities.userId, user.id), eq(userIdentities.provider, 'webauthn')),
+    columns: {
+      providerUserId: true,
+      providerUsername: true
+    }
+  })
+
+  // 防止重复注册相同凭证，使用存储的实际 transports 信息
+  const excludeCredentials = existingCredentials.map(cred => {
+    let transports: string[] | undefined
+    try {
+      if (cred.providerUsername) {
+        const data = JSON.parse(cred.providerUsername)
+        transports = data.transports
+      }
+    } catch (e) {
+      console.error('解析 WebAuthn 凭证数据失败:', e)
+    }
+    return {
+      id: cred.providerUserId,
+      type: 'public-key' as const,
+      transports
+    }
+  })
+
+  const { rpID } = getWebAuthnConfig(event)
+  const rpName = 'VoiceHub'
+
+  const options = await generateRegistrationOptions({
+    rpName,
+    rpID,
+    userID: isoUint8Array.fromUTF8String(user.id.toString()),
+    userName: user.username || user.email,
+    attestationType: 'none',
+    excludeCredentials,
+    authenticatorSelection: {
+      residentKey: 'required',
+      userVerification: 'preferred',
+    },
+  })
+
+  // 存储注册挑战到 Cookie
+  setWebAuthnChallenge(event, options.challenge, user.id.toString())
+
+  return options
+})
@@ -0,0 +1,83 @@
+import { verifyRegistrationResponse } from '@simplewebauthn/server'
+import { getWebAuthnChallenge, clearWebAuthnChallenge } from '~~/server/utils/webauthn-token'
+import { getWebAuthnConfig } from '~~/server/utils/webauthn-config'
+import { db, userIdentities } from '~/drizzle/db'
+
+const VALID_TRANSPORTS = ['internal', 'hybrid', 'usb', 'nfc', 'ble']
+
+function sanitizeTransports(transports: unknown): string[] {
+  if (!Array.isArray(transports)) return []
+  return transports.filter((t): t is string => 
+    typeof t === 'string' && VALID_TRANSPORTS.includes(t)
+  )
+}
+
+export default defineEventHandler(async (event) => {
+  const user = event.context.user
+  if (!user) {
+    throw createError({ statusCode: 401, message: '未授权访问' })
+  }
+
+  const body = await readBody(event)
+  const challengeData = getWebAuthnChallenge(event)
+
+  if (!challengeData || challengeData.userId !== user.id.toString()) {
+    throw createError({ statusCode: 400, message: 'Challenge 已失效或不匹配' })
+  }
+
+  const { rpID, origin } = getWebAuthnConfig(event)
+
+  try {
+    const verification = await verifyRegistrationResponse({
+      response: body,
+      expectedChallenge: challengeData.challenge,
+      expectedOrigin: origin,
+      expectedRPID: rpID,
+    })
+
+    if (verification.verified && verification.registrationInfo) {
+      const { credential } = verification.registrationInfo
+      
+      // 提取凭证数据
+      const { id: credentialID, publicKey: credentialPublicKey, counter, transports } = credential
+
+      if (!credentialID || !credentialPublicKey) {
+        throw new Error('注册信息缺少凭证 ID 或公钥')
+      }
+
+      // 转换为 Base64URL 格式存储
+      const credentialIDBase64 = credentialID
+      const publicKeyBase64 = Buffer.from(credentialPublicKey).toString('base64url')
+
+      if (!publicKeyBase64) {
+        throw new Error('生成的公钥为空，注册失败')
+      }
+
+      // 构造凭证数据
+      const credentialData = {
+        label: body.label || 'WebAuthn 设备',
+        publicKey: publicKeyBase64,
+        counter: Number(counter),
+        transports: sanitizeTransports(transports)
+      }
+
+      // 存储到数据库
+      await db.insert(userIdentities).values({
+        userId: user.id,
+        provider: 'webauthn',
+        providerUserId: credentialIDBase64,
+        providerUsername: JSON.stringify(credentialData),
+        createdAt: new Date()
+      })
+
+      clearWebAuthnChallenge(event)
+      return { success: true }
+    } else {
+      throw createError({ statusCode: 400, message: '验证失败' })
+    }
+  } catch (error) {
+    console.error('WebAuthn 验证失败:', error)
+    const message = error instanceof Error ? error.message : '验证失败'
+    throw createError({ statusCode: 400, message })
+  }
+})
@@ -0,0 +1,51 @@
+import { db, eq, and, userIdentities, sql } from '~/drizzle/db'
+
+export default defineEventHandler(async (event) => {
+  const user = event.context.user
+  if (!user) {
+    throw createError({ statusCode: 401, message: '未授权访问' })
+  }
+
+  interface RenameRequestBody {
+    id: string
+    name: string
+  }
+  const body = await readBody<RenameRequestBody>(event)
+  const { id, name: rawName } = body
+  const name = typeof rawName === 'string' ? rawName.trim() : ''
+
+  if (!id || !name) {
+    throw createError({ statusCode: 400, message: '参数错误' })
+  }
+
+  if (name.length > 50) {
+    throw createError({ statusCode: 400, message: '设备名称过长' })
+  }
+
+  // 查找对应的 Passkey
+  const identity = await db.query.userIdentities.findFirst({
+    where: and(
+      eq(userIdentities.id, id),
+      eq(userIdentities.userId, user.id),
+      eq(userIdentities.provider, 'webauthn')
+    )
+  })
+
+  if (!identity) {
+    throw createError({ statusCode: 404, message: '未找到该设备' })
+  }
+
+  try {
+    // 使用 jsonb_set 原子更新 label，避免覆盖 concurrent login 产生的 counter 更新
+    await db.update(userIdentities)
+      .set({ 
+        providerUsername: sql`jsonb_set(${userIdentities.providerUsername}::jsonb, '{label}', ${JSON.stringify(name)}::jsonb)::text` 
+      })
+      .where(eq(userIdentities.id, id))
+
+    return { success: true }
+  } catch (e) {
+    console.error('更新设备名称失败:', e)
+    throw createError({ statusCode: 500, message: '更新失败' })
+  }
+})