feat: 切换为MariaDB数据库存储
This commit is contained in:
@@ -1,31 +1,6 @@
|
||||
import { NextRequest, NextResponse } from 'next/server';
|
||||
import fs from 'fs/promises';
|
||||
import path from 'path';
|
||||
import crypto from 'crypto';
|
||||
|
||||
const DATA_DIR = path.join(process.cwd(), 'data');
|
||||
const USERS_FILE = path.join(DATA_DIR, 'users.json');
|
||||
|
||||
interface User {
|
||||
id: string;
|
||||
username: string;
|
||||
passwordHash: string;
|
||||
token: string;
|
||||
createdAt: string;
|
||||
}
|
||||
|
||||
async function readUsers(): Promise<User[]> {
|
||||
try {
|
||||
const data = await fs.readFile(USERS_FILE, 'utf-8');
|
||||
return JSON.parse(data);
|
||||
} catch {
|
||||
return [];
|
||||
}
|
||||
}
|
||||
|
||||
async function writeUsers(users: User[]): Promise<void> {
|
||||
await fs.writeFile(USERS_FILE, JSON.stringify(users, null, 2));
|
||||
}
|
||||
import pool from '@/lib/db';
|
||||
|
||||
function hashPassword(password: string): string {
|
||||
return crypto.createHash('sha256').update(password).digest('hex');
|
||||
@@ -36,6 +11,7 @@ function generateToken(): string {
|
||||
}
|
||||
|
||||
export async function POST(request: NextRequest) {
|
||||
let connection;
|
||||
try {
|
||||
const { username, password } = await request.json();
|
||||
|
||||
@@ -47,23 +23,19 @@ export async function POST(request: NextRequest) {
|
||||
return NextResponse.json({ error: '用户名至少3位,密码至少6位' }, { status: 400 });
|
||||
}
|
||||
|
||||
const users = await readUsers();
|
||||
connection = await pool.getConnection();
|
||||
|
||||
if (users.find(u => u.username === username)) {
|
||||
const [rows] = await connection.query<any[]>('SELECT id FROM users WHERE username = ?', [username]);
|
||||
|
||||
if (rows.length > 0) {
|
||||
return NextResponse.json({ error: '用户名已存在' }, { status: 400 });
|
||||
}
|
||||
|
||||
const token = generateToken();
|
||||
const newUser: User = {
|
||||
id: crypto.randomUUID(),
|
||||
username,
|
||||
passwordHash: hashPassword(password),
|
||||
token,
|
||||
createdAt: new Date().toISOString()
|
||||
};
|
||||
|
||||
users.push(newUser);
|
||||
await writeUsers(users);
|
||||
await connection.query(
|
||||
'INSERT INTO users (id, username, password_hash, token, created_at) VALUES (?, ?, ?, ?, ?)',
|
||||
[crypto.randomUUID(), username, hashPassword(password), token, new Date()]
|
||||
);
|
||||
|
||||
const response = NextResponse.json({ success: true, username });
|
||||
response.cookies.set('auth_token', token, {
|
||||
@@ -78,5 +50,7 @@ export async function POST(request: NextRequest) {
|
||||
} catch (error) {
|
||||
console.error('Register error:', error);
|
||||
return NextResponse.json({ error: '注册失败' }, { status: 500 });
|
||||
} finally {
|
||||
if (connection) connection.release();
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user