PDO::ERRMODE_EXCEPTION, PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC ] ); } catch (PDOException $e) { die("数据库连接失败: " . $e->getMessage()); } } return $pdo; } // 检查登录状态 function checkLogin() { if (!isset($_SESSION['user_id'])) { header('Location: index.php'); exit; } } // 获取当前用户信息 function getCurrentUser() { if (!isset($_SESSION['user_id'])) { return null; } $db = getDB(); $stmt = $db->prepare("SELECT * FROM users WHERE id = ?"); $stmt->execute([$_SESSION['user_id']]); return $stmt->fetch(); } // 获取用户文档目录 function getUserDocDir($username) { return __DIR__ . '/' . $username; }