feat: 支持用户管理

This commit is contained in:
richarjiang
2026-03-15 15:45:09 +08:00
parent 7628768869
commit 3c35f1982f
9 changed files with 738 additions and 82 deletions

View File

@@ -0,0 +1,248 @@
'use client'
import { useState } from 'react'
import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query'
import { Button } from '@/components/ui/button'
import { Header } from '@/components/layout/header'
import { UserDialog } from '@/components/users/user-dialog'
import { Spinner } from '@/components/ui/spinner'
import { User, UserFormData } from '@/types'
import { Plus, Pencil, Trash2 } from 'lucide-react'
export default function UsersPage() {
const queryClient = useQueryClient()
const [isDialogOpen, setIsDialogOpen] = useState(false)
const [editingUser, setEditingUser] = useState<User | null>(null)
const [deleteConfirmId, setDeleteConfirmId] = useState<string | null>(null)
// Fetch users
const { data: users, isLoading, error } = useQuery<User[]>({
queryKey: ['users'],
queryFn: async () => {
const res = await fetch('/api/users')
if (!res.ok) throw new Error('Failed to fetch users')
return res.json()
},
})
// Create user mutation
const createMutation = useMutation({
mutationFn: async (data: UserFormData) => {
const res = await fetch('/api/users', {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify(data),
})
if (!res.ok) {
const error = await res.json()
throw new Error(error.error || 'Failed to create user')
}
return res.json()
},
onSuccess: () => {
queryClient.invalidateQueries({ queryKey: ['users'] })
},
})
// Update user mutation
const updateMutation = useMutation({
mutationFn: async ({ id, data }: { id: string; data: UserFormData }) => {
const res = await fetch('/api/users', {
method: 'PUT',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({ id, ...data }),
})
if (!res.ok) {
const error = await res.json()
throw new Error(error.error || 'Failed to update user')
}
return res.json()
},
onSuccess: () => {
queryClient.invalidateQueries({ queryKey: ['users'] })
},
})
// Delete user mutation
const deleteMutation = useMutation({
mutationFn: async (id: string) => {
const res = await fetch(`/api/users?id=${id}`, {
method: 'DELETE',
})
if (!res.ok) {
const error = await res.json()
throw new Error(error.error || 'Failed to delete user')
}
return res.json()
},
onSuccess: () => {
queryClient.invalidateQueries({ queryKey: ['users'] })
setDeleteConfirmId(null)
},
})
const handleOpenCreate = () => {
setEditingUser(null)
setIsDialogOpen(true)
}
const handleOpenEdit = (user: User) => {
setEditingUser(user)
setIsDialogOpen(true)
}
const handleDelete = (id: string) => {
if (deleteConfirmId === id) {
deleteMutation.mutate(id)
} else {
setDeleteConfirmId(id)
// Reset after 3 seconds
setTimeout(() => setDeleteConfirmId(null), 3000)
}
}
const handleSubmit = async (data: UserFormData) => {
if (editingUser) {
await updateMutation.mutateAsync({ id: editingUser.id, data })
} else {
await createMutation.mutateAsync(data)
}
}
const formatDate = (date: Date | string) => {
return new Date(date).toLocaleDateString('zh-CN', {
year: 'numeric',
month: '2-digit',
day: '2-digit',
hour: '2-digit',
minute: '2-digit',
})
}
if (isLoading) {
return (
<div className="h-screen flex items-center justify-center">
<Spinner size="lg" />
</div>
)
}
if (error) {
return (
<div className="h-screen flex items-center justify-center">
<div className="text-center">
<p className="text-red-600"></p>
<Button
className="mt-4"
onClick={() => queryClient.invalidateQueries({ queryKey: ['users'] })}
>
</Button>
</div>
</div>
)
}
return (
<div className="h-screen flex flex-col">
<Header />
<div className="flex-1 overflow-auto p-6">
<div className="max-w-4xl mx-auto">
<div className="flex items-center justify-between mb-6">
<div>
<h1 className="text-2xl font-bold"></h1>
<p className="text-gray-500 mt-1">
{users?.length || 0}
</p>
</div>
<Button onClick={handleOpenCreate}>
<Plus className="h-4 w-4 mr-2" />
</Button>
</div>
<div className="bg-white rounded-lg shadow overflow-hidden">
<table className="min-w-full divide-y divide-gray-200">
<thead className="bg-gray-50">
<tr>
<th className="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">
</th>
<th className="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">
</th>
<th className="px-6 py-3 text-left text-xs font-medium text-gray-500 uppercase tracking-wider">
</th>
<th className="px-6 py-3 text-right text-xs font-medium text-gray-500 uppercase tracking-wider">
</th>
</tr>
</thead>
<tbody className="bg-white divide-y divide-gray-200">
{users?.map((user) => (
<tr key={user.id} className="hover:bg-gray-50">
<td className="px-6 py-4 whitespace-nowrap">
<div className="flex items-center">
<div className="h-10 w-10 rounded-full bg-gray-200 flex items-center justify-center text-gray-600 font-medium">
{user.email[0]?.toUpperCase() || 'U'}
</div>
<div className="ml-4">
<div className="text-sm font-medium text-gray-900">
{user.name || '未设置'}
</div>
</div>
</div>
</td>
<td className="px-6 py-4 whitespace-nowrap">
<div className="text-sm text-gray-900">{user.email}</div>
</td>
<td className="px-6 py-4 whitespace-nowrap text-sm text-gray-500">
{formatDate(user.createdAt)}
</td>
<td className="px-6 py-4 whitespace-nowrap text-right text-sm font-medium">
<Button
variant="ghost"
size="sm"
onClick={() => handleOpenEdit(user)}
className="text-gray-600 hover:text-gray-900"
>
<Pencil className="h-4 w-4" />
</Button>
<Button
variant="ghost"
size="sm"
onClick={() => handleDelete(user.id)}
className={
deleteConfirmId === user.id
? 'text-red-600 hover:text-red-700'
: 'text-gray-600 hover:text-gray-900'
}
>
<Trash2 className="h-4 w-4" />
</Button>
</td>
</tr>
))}
{users?.length === 0 && (
<tr>
<td colSpan={4} className="px-6 py-12 text-center text-gray-500">
</td>
</tr>
)}
</tbody>
</table>
</div>
</div>
</div>
<UserDialog
open={isDialogOpen}
onOpenChange={setIsDialogOpen}
user={editingUser}
onSubmit={handleSubmit}
/>
</div>
)
}

217
app/api/users/route.ts Normal file
View File

@@ -0,0 +1,217 @@
import { NextRequest, NextResponse } from 'next/server'
import { prisma } from '@/lib/prisma'
import { auth } from '@/lib/auth'
import { hashPassword } from 'better-auth/crypto'
import { v4 as uuidv4 } from 'uuid'
// GET /api/users - Get all users
export async function GET(request: NextRequest) {
try {
const session = await auth.api.getSession({
headers: request.headers,
})
if (!session) {
return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
}
const users = await prisma.user.findMany({
orderBy: { createdAt: 'desc' },
select: {
id: true,
email: true,
emailVerified: true,
name: true,
image: true,
createdAt: true,
updatedAt: true,
},
})
return NextResponse.json(users)
} catch (error) {
console.error('Error fetching users:', error)
return NextResponse.json(
{ error: 'Failed to fetch users' },
{ status: 500 }
)
}
}
// POST /api/users - Create a new user
export async function POST(request: NextRequest) {
try {
const session = await auth.api.getSession({
headers: request.headers,
})
if (!session) {
return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
}
const body = await request.json()
const { email, password, name } = body
if (!email || !password) {
return NextResponse.json(
{ error: 'email and password are required' },
{ status: 400 }
)
}
// Check if user already exists
const existingUser = await prisma.user.findUnique({
where: { email },
})
if (existingUser) {
return NextResponse.json(
{ error: '该邮箱已被注册' },
{ status: 400 }
)
}
// Hash password
const hashedPassword = await hashPassword(password)
const userId = uuidv4()
const accountId = uuidv4()
// Create user and account in transaction
const user = await prisma.$transaction(async (tx) => {
const newUser = await tx.user.create({
data: {
id: userId,
email,
name: name || null,
},
})
await tx.account.create({
data: {
id: accountId,
accountId: userId,
providerId: 'credential',
userId: userId,
password: hashedPassword,
},
})
return newUser
})
return NextResponse.json(user, { status: 201 })
} catch (error) {
console.error('Error creating user:', error)
return NextResponse.json(
{ error: 'Failed to create user' },
{ status: 500 }
)
}
}
// PUT /api/users - Update a user
export async function PUT(request: NextRequest) {
try {
const session = await auth.api.getSession({
headers: request.headers,
})
if (!session) {
return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
}
const body = await request.json()
const { id, email, password, name } = body
if (!id) {
return NextResponse.json({ error: 'id is required' }, { status: 400 })
}
// Check if email is taken by another user
if (email) {
const existingUser = await prisma.user.findFirst({
where: {
email,
NOT: { id },
},
})
if (existingUser) {
return NextResponse.json(
{ error: '该邮箱已被其他用户使用' },
{ status: 400 }
)
}
}
// Update user and optionally password
const user = await prisma.$transaction(async (tx) => {
const updatedUser = await tx.user.update({
where: { id },
data: {
email,
name: name || null,
},
})
if (password) {
const hashedPassword = await hashPassword(password)
await tx.account.updateMany({
where: { userId: id, providerId: 'credential' },
data: { password: hashedPassword },
})
}
return updatedUser
})
return NextResponse.json(user)
} catch (error) {
console.error('Error updating user:', error)
return NextResponse.json(
{ error: 'Failed to update user' },
{ status: 500 }
)
}
}
// DELETE /api/users - Delete a user
export async function DELETE(request: NextRequest) {
try {
const session = await auth.api.getSession({
headers: request.headers,
})
if (!session) {
return NextResponse.json({ error: 'Unauthorized' }, { status: 401 })
}
const { searchParams } = new URL(request.url)
const id = searchParams.get('id')
if (!id) {
return NextResponse.json({ error: 'id is required' }, { status: 400 })
}
// Prevent deleting yourself
if (id === session.user.id) {
return NextResponse.json(
{ error: '不能删除自己的账户' },
{ status: 400 }
)
}
await prisma.user.delete({
where: { id },
})
return NextResponse.json({ success: true })
} catch (error) {
console.error('Error deleting user:', error)
return NextResponse.json(
{ error: 'Failed to delete user' },
{ status: 500 }
)
}
}