WordPress中setcookie即时读取失效：理解与解决方案

/**
 * 在WordPress初始化时设置'origin' Cookie。
 * 从GET参数中获取值，并进行安全处理。
 */
function custom_set_origin_cookie() {
    // 获取站点URL的路径和主机，确保Cookie作用范围正确
    $site_url_parts = parse_url( get_option('siteurl') );
    $path = isset($site_url_parts['path']) ? $site_url_parts['path'] : '/';
    $host = isset($site_url_parts['host']) ? $site_url_parts['host'] : $_SERVER['HTTP_HOST'];

    // 设置Cookie过期时间（例如10小时）
    $expiry = time() + 36000;

    // 从GET参数中获取'origin'值，并进行安全清理
    // 使用sanitize_text_field确保输入安全
    $origin_from_get = isset($_GET['origin']) ? sanitize_text_field($_GET['origin']) : null;

    // 如果GET参数中存在'origin'，则设置Cookie
    if ( $origin_from_get !== null ) {
        // setcookie() 函数必须在任何输出发送到浏览器之前调用
        setcookie( 'origin', $origin_from_get, [
            'expires' => $expiry,
            'path'    => $path,
            'domain'  => $host,
            'secure'  => is_ssl(), // 仅在HTTPS连接时发送Cookie
            'httponly' => true,    // 阻止JavaScript访问Cookie，增加安全性
            'samesite' => 'Lax',   // 增强CSRF保护
        ]);
    }
}
add_action( 'init', 'custom_set_origin_cookie' );

/**
 * 获取并显示用户的'origin'地址。
 * 优先从GET参数获取，然后从Cookie获取。
 */
function display_user_origin_address() {
    $displayed_origin = null;

    // 步骤1: 优先从当前请求的GET参数中获取'origin'
    // 如果存在，说明这是表单提交后的首次页面加载
    if ( isset( $_GET['origin'] ) ) {
        $displayed_origin = sanitize_text_field( $_GET['origin'] );
    }
    // 步骤2: 如果GET参数中没有'origin'，则尝试从已设置的COOKIE中获取
    // 这适用于用户刷新页面或访问其他页面时
    else if ( isset( $_COOKIE['origin'] ) ) {
        $displayed_origin = sanitize_text_field( $_COOKIE['origin'] );
    }

    // 如果获取到了值，则安全地输出
    if ( $displayed_origin !== null ) {
        echo '<p>您的地址：' . esc_html( $displayed_origin ) . '</p>';
    } else {
        echo '<p>未检测到您的地址信息。</p>';
    }
}

// 可以在你的主题模板文件（如 single.php, page.php, search.php 等）中直接调用此函数
// 例如：
// if ( is_search() ) {
//     display_user_origin_address();
// }

// 或者，你可以创建一个短代码，让用户在内容中插入
add_shortcode( 'show_origin_address', 'display_user_origin_address' );
// 然后在页面或文章中使用 [show_origin_address]