场景与挑战
在许多wordpress网站中，特别是那些包含用户个人资料页面的网站，通常会有一个需求：当用户访问自己的个人资料页面时，显示一个“编辑”按钮或特定功能，而当访问其他用户的页面时，则不显示。一个常见的实现方式是将用户的个人资料页面设置为一个自定义文章类型（custom post type），并把用户本人设为该文章的作者。
然而，直接通过简单的PHP变量比较来判断当前用户ID和文章作者ID，可能会导致未定义变量警告，或者更糟糕的是，使按钮在所有页面上都显示，这显然不符合预期。核心挑战在于，如何在WordPress环境中，安全且准确地获取当前页面信息、当前登录用户信息以及当前文章的作者信息，并基于这些信息进行条件判断。
WordPress 解决方案
WordPress提供了一套强大的API和全局变量来处理这类情境。解决此问题的关键在于：
- 确定当前页面是否为单篇文章页面：使用 is_single() 函数。
- 获取当前文章对象：通过 global $post 访问全局的 WP_Post 对象。
- 获取当前登录用户ID：使用 get_current_user_id() 函数。
- 获取当前文章作者ID：从 $post 对象中获取 post_author 属性。
- 条件判断并输出样式：比较两个ID，如果匹配则输出CSS来显示预设的元素。
代码实现
以下代码应放置在您当前活动主题的 functions.php 文件中，或者一个自定义插件中。
<?php
/**
* 根据当前用户是否为文章作者，动态显示前端元素
*/
function current_user_is_post_author_display_element() {
// 确保当前页面是单篇文章页面（包括自定义文章类型）
if ( is_single() ) {
// 访问全局的 $post 对象，它包含了当前文章的所有信息
global $post;
// 获取当前登录用户的ID
$current_user_id = get_current_user_id();
// 获取当前文章的作者ID
$post_author_id = $post->post_author;
// 比较当前用户ID和文章作者ID
if ( $current_user_id == $post_author_id ) {
// 如果ID匹配，则输出CSS来显示具有特定ID的元素
// 这里的CSS会覆盖预设的 display: none; 样式
?>
<style type="text/css">
#profile_edit_button {
display: flex !important; /* 使用 flex 或 block，并添加 !important 确保覆盖 */
}
</style>
<?php
}
}
}
// 将函数挂载到 wp_head 钩子，确保在页面的 <head> 部分输出CSS
add_action( 'wp_head', 'current_user_is_post_author_display_element', 10, 1 );
?>代码解析
- current_user_is_post_author_display_element() 函数：这是一个自定义函数，用于封装我们的逻辑。将其封装在函数中是WordPress开发中的良好实践。
- if ( is_single() )：这个条件判断非常重要。它确保了只有在访问单个文章（包括自定义文章类型）页面时，内部的代码才会执行。这避免了在首页、归档页或其他非单文章页面上不必要的执行。
- global $post;：在WordPress中，当前文章的数据通常存储在一个名为 $post 的全局变量中。为了在函数内部访问它，我们需要使用 global 关键字声明。
- get_current_user_id()：这是一个WordPress内置函数，用于获取当前已登录用户的ID。如果用户未登录，它将返回 0。
- $post->post_author：这是从 $post 对象中获取当前文章作者ID的方式。
- if ( $current_user_id == $post_author_id )：这是核心的逻辑判断。当且仅当当前登录用户的ID与正在查看的文章的作者ID相同时，条件为真。
- ：当条件为真时，这部分PHP代码会直接输出一个
- display: flex !important;：这行CSS会覆盖该元素默认的 display: none; 样式，使其可见。!important 规则确保了这条样式具有最高优先级，可以覆盖其他可能存在的样式。您可以根据需要选择 display: block; 或 display: inline-block; 等。
前端 HTML 和 CSS 设置
为了使上述PHP代码生效，您需要在前端页面模板中设置相应的HTML元素，并默认将其隐藏。
立即学习“前端免费学习笔记（深入）”；
HTML 示例（在您的主题模板文件，如 single-{custom-post-type}.php 中）：
<div id="profile_edit_button">
<a href="/edit-profile">编辑我的资料</a>
</div>初始 CSS 示例（在您的主题样式表，如 style.css 中）：
#profile_edit_button {
display: none; /* 默认隐藏此按钮 */
}当PHP代码检测到当前用户是作者时，它会动态注入的CSS会覆盖 display: none;，使按钮可见。
注意事项与最佳实践
- 安全性：此方法仅控制前端元素的“显示”，而不控制其背后的“访问权限”。如果“编辑资料”按钮链接到一个编辑表单，务必在处理表单提交时，在服务器端再次验证用户权限（即当前用户是否为该资料的作者），以防止未经授权的修改。
- 性能考量：对于少量需要条件显示的元素，直接注入
- 或者，在PHP中根据条件给父元素添加一个特殊的CSS类，然后通过CSS选择器控制子元素的显示。例如：body.is-author-profile #profile_edit_button { display: flex; }。
- 代码位置：将代码放置在主题的 functions.php 文件中是快速实现的方法。但如果您的功能是通用的，不依赖于特定主题，或者您希望在更换主题后仍保留此功能，那么创建一个自定义插件是更好的选择。
- 用户体验：确保隐藏的元素不会在页面加载时短暂闪烁（FOUC - Flash Of Unstyled Content）。通常，将 display: none; 放在主CSS文件中可以避免这个问题，因为浏览器在渲染HTML之前会先解析CSS。
总结
通过利用WordPress的 is_single()、global $post、get_current_user_id() 等核心函数以及 wp_head 动作钩子，我们可以高效且准确地实现根据用户是否为文章作者来动态显示前端元素的功能。这种模式不仅适用于个人资料页面，也可以推广到其他需要基于用户权限或文章作者身份进行前端内容控制的场景。始终记住，前端的显示控制只是第一步，后端的数据处理和权限验证才是确保应用安全的关键。
