PHP实现用户注册后自动登录与页面重定向教程

<?php
include 'main.php';
// ... (之前的验证和数据库查询逻辑，确保所有输入都已验证) ...

// 检查用户名或邮箱是否已存在
$stmt = $pdo->prepare('SELECT id, password FROM accounts WHERE username = ? OR email = ?');
$stmt->execute([ $_POST['username'], $_POST['email'] ]);
$account = $stmt->fetch(PDO::FETCH_ASSOC);

if ($account) {
    // 用户名或邮箱已存在，提示错误
    echo '<div class="error form">Username and/or email exists!</div>';
} else {
    // 用户名不存在，插入新账户
    $stmt = $pdo->prepare('INSERT INTO accounts (username, password, email, activation_code) VALUES (?, ?, ?, ?)');
    // 对密码进行哈希处理
    $password = password_hash($_POST['password'], PASSWORD_DEFAULT);
    // 根据是否需要账户激活生成激活码
    $uniqid = account_activation ? uniqid() : 'activated';
    $stmt->execute([ $_POST['username'], $password, $_POST['email'], $uniqid ]);

    // 获取新注册用户的ID，用于设置会话变量
    $newUserId = $pdo->lastInsertId();

    // 执行自动登录操作：设置会话变量
    session_regenerate_id(); // 重新生成会话ID，提高安全性
    $_SESSION['loggedin'] = TRUE;
    $_SESSION['name'] = $_POST['username'];
    $_SESSION['id'] = $newUserId;
    // 假设新注册用户的默认角色为'member'，可根据实际情况调整或从数据库读取
    $_SESSION['role'] = 'member';

    if (account_activation) {
        // 如果需要账户激活，发送激活邮件
        send_activation_email($_POST['email'], $uniqid);
        echo 'Please check your email to activate your account!';
        // 此时不立即重定向，等待用户激活
    } else {
        // 如果不需要账户激活，直接重定向到主页
        header('Location: home.php');
        exit; // 确保重定向后脚本停止执行
    }
}
?>

<?php
// main.php 中的 check_loggedin 函数示例
// ... (其他配置和数据库连接代码) ...

function check_loggedin($pdo, $redirect_file = 'index.php') {
    // 检查记住我 cookie 和登录会话变量
    if (isset($_COOKIE['rememberme']) && !empty($_COOKIE['rememberme']) && !isset($_SESSION['loggedin'])) {
        // 如果存在记住我cookie但未登录，尝试通过cookie登录
        $stmt = $pdo->prepare('SELECT * FROM accounts WHERE rememberme = ?');
        $stmt->execute([ $_COOKIE['rememberme'] ]);
        $account = $stmt->fetch(PDO::FETCH_ASSOC);
        if ($account) {
            // 找到匹配项，更新会话变量并保持用户登录
            session_regenerate_id();
            $_SESSION['loggedin'] = TRUE;
            $_SESSION['name'] = $account['username'];
            $_SESSION['id'] = $account['id'];
            $_SESSION['role'] = $account['role'];
        } else {
            // 如果记住我cookie无效，重定向到登录页
            header('Location: ' . $redirect_file);
            exit;
        }
    } else if (!isset($_SESSION['loggedin']) || !isset($_SESSION['id'])) { // 建议同时检查 $_SESSION['id']
        // 如果用户未登录（没有会话或会话无效），重定向到登录页面
        header('Location: ' . $redirect_file);
        exit;
    }
    // 如果已登录，函数正常返回，允许继续访问
}
// ... (其他函数，如send_activation_email) ...
?>