setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
$pdo->setAttribute(PDO::ATTR_DEFAULT_FETCH_MODE, PDO::FETCH_ASSOC);
$pdo->exec("CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
username VARCHAR(50) NOT NULL UNIQUE,
password VARCHAR(255) NOT NULL,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
)");
} catch (PDOException $e) {
die('数据库初始化失败: ' . $e->getMessage());
}
// 处理动作
$action = $_GET['action'] ?? 'default';
$error = '';
$success = '';
// 注销
if ($action === 'logout') {
$_SESSION = array();
session_destroy();
header('Location: ?action=default');
exit;
}
// 登录处理
if ($action === 'login' && $_SERVER['REQUEST_METHOD'] === 'POST') {
$username = trim($_POST['username'] ?? '');
$password = $_POST['password'] ?? '';
if (empty($username) || empty($password)) {
$error = '用户名和密码不能为空';
} else {
$stmt = $pdo->prepare("SELECT * FROM users WHERE username = ?");
$stmt->execute([$username]);
$user = $stmt->fetch();
if ($user && password_verify($password, $user['password'])) {
$_SESSION['user_id'] = $user['id'];
$_SESSION['username'] = $user['username'];
header('Location: ?action=document');
exit;
} else {
$error = '用户名或密码错误';
}
}
$action = 'default';
}
// 注册处理
if ($action === 'register' && $_SERVER['REQUEST_METHOD'] === 'POST') {
$username = trim($_POST['username'] ?? '');
$password = $_POST['password'] ?? '';
$confirm = $_POST['confirm_password'] ?? '';
if (empty($username) || empty($password) || empty($confirm)) {
$error = '所有字段都必须填写';
} elseif ($password !== $confirm) {
$error = '两次输入的密码不一致';
} elseif (strlen($password) < 3) {
$error = '密码至少需要3个字符';
} else {
$stmt = $pdo->prepare("SELECT id FROM users WHERE username = ?");
$stmt->execute([$username]);
if ($stmt->fetch()) {
$error = '用户名已存在';
} else {
$hash = password_hash($password, PASSWORD_DEFAULT);
$stmt = $pdo->prepare("INSERT INTO users (username, password) VALUES (?, ?)");
$stmt->execute([$username, $hash]);
$success = '注册成功！请登录。';
$action = 'default';
}
}
}
// 文档页面检查登录
if ($action === 'document' && !isset($_SESSION['user_id'])) {
header('Location: ?action=default&need_login=1');
exit;
}
?>
<?php echo APP_NAME; ?>
📄 文档
欢迎回来，
🌞
今天有太阳
—— 这是您登录后才能看到的内容。
📝 注册
创建新账户，轻松查阅文档
🔐 登录
登录后即可查看专属文档
请先登录后查看文档
⚡ 简易文档系统 · 演示