<?php
// 1. 在作者主页封面右下角菜单添加“公开/私密”切换按钮
add_filter('author_header_drop_lists', 'custom_zib_author_privacy_toggle_btn', 20, 2);
function custom_zib_author_privacy_toggle_btn($lists, $author_id) {
    // 安全校验：只有当前登录用户是主页作者本人时才显示该选项
    if (get_current_user_id() !== $author_id) {
        return $lists;
    }

    // 获取当前状态（新逻辑：获取是否已公开。如果用户未设置过，默认为空/false，即私密状态）
    $is_public = get_user_meta($author_id, 'profile_is_public', true);
    
    // 如果是公开状态，按钮显示“设为私密”；如果是私密状态（默认），按钮显示“设为公开”
    $text = $is_public ? '设为私密' : '设为公开';
    $icon = $is_public ? '<i class="fa fa-eye-slash"></i>' : '<i class="fa fa-eye"></i>';

    // 拼接进原有的菜单列表中
    $lists .= '<li><a href="javascript:;" id="toggle-profile-privacy" data-user-id="' . esc_attr($author_id) . '">' . $icon . ' <span class="privacy-text">' . $text . '</span></a></li>';

    return $lists;
}

// 2. 处理 AJAX 请求，保存用户的公开/私密设置
add_action('wp_ajax_toggle_profile_privacy', 'custom_zib_handle_privacy_toggle');
function custom_zib_handle_privacy_toggle() {
    if (!is_user_logged_in()) {
        wp_send_json_error('请先登录');
    }

    $user_id = isset($_POST['user_id']) ? intval($_POST['user_id']) : 0;
    $current_user_id = get_current_user_id();

    // 严格权限校验：防止恶意接口调用
    if ($user_id !== $current_user_id) {
        wp_send_json_error('无权操作');
    }

    $is_public = get_user_meta($user_id, 'profile_is_public', true);
    
    // 切换状态
    if ($is_public) {
        // 当前是公开，点击后设为私密（删除记录恢复默认的私密状态）
        delete_user_meta($user_id, 'profile_is_public'); 
        wp_send_json_success(array(
            'text' => '设为公开', 
            'icon' => '<i class="fa fa-eye"></i>',
            'msg'  => '您的主页已设为私密状态'
        ));
    } else {
        // 当前是私密（默认），点击后主动设为公开
        update_user_meta($user_id, 'profile_is_public', '1'); 
        wp_send_json_success(array(
            'text' => '设为私密', 
            'icon' => '<i class="fa fa-eye-slash"></i>',
            'msg'  => '您的主页已公开展示'
        ));
    }
}

// 3. 前端添加 JavaScript 用于触发无刷新 AJAX 请求
add_action('wp_footer', 'custom_zib_privacy_toggle_js');
function custom_zib_privacy_toggle_js() {
    // 为了节省性能，只在作者主页加载此 JS
    if (!is_author()) return; 
    ?>
    <script>
    jQuery(document).ready(function($) {
        $('body').on('click', '#toggle-profile-privacy', function() {
            var btn = $(this);
            var user_id = btn.data('user-id');
            var textSpan = btn.find('.privacy-text');
            
            $.ajax({
                url: '<?php echo admin_url('admin-ajax.php'); ?>',
                type: 'POST',
                dataType: 'json',
                data: {
                    action: 'toggle_profile_privacy',
                    user_id: user_id
                },
                success: function(res) {
                    if (res.success) {
                        // 动态更新按钮文案和图标
                        textSpan.text(res.data.text);
                        btn.find('i').replaceWith(res.data.icon);
                        
                        // 兼容子比主题内置的弹窗提示体系
                        if (typeof notyf !== 'undefined') {
                            notyf.success(res.data.msg);
                        } else if (typeof layer !== 'undefined') {
                            layer.msg(res.data.msg);
                        } else {
                            alert(res.data.msg);
                        }
                    } else {
                        alert(res.data);
                    }
                },
                error: function() {
                    alert('网络请求失败，请稍后重试');
                }
            });
        });
    });
    </script>
    <?php
}

// 4. 访问拦截逻辑：如果是私密主页，融合主题的 header 和 footer 进行显示
add_action('template_redirect', 'custom_zib_enforce_profile_privacy', 1);
function custom_zib_enforce_profile_privacy() {
    if (is_author()) {
        $author_id = get_queried_object_id();
        
        // 获取是否公开（新逻辑：如果用户从未点过按钮，这里获取不到值，返回 false）
        $is_public = get_user_meta($author_id, 'profile_is_public', true);
        
        // 如果【未公开】（即默认的私密状态）
        if (!$is_public) {
            $current_user_id = get_current_user_id();
            
            // 允许作者本人以及拥有管理员权限（manage_options）的用户查看，其他人拦截
            if ($current_user_id !== $author_id && !current_user_can('manage_options')) {
                
                // 设置 HTTP 状态码为 403，告知搜索引擎此页面无权抓取
                status_header(403);
                
                // 1. 加载主题的导航和头部
                get_header(); 
                
                // 获取子比主题自带的无用户缺省图
                $null_img = get_theme_file_uri('/img/null-user.svg');
                ?>
                
                <main class="container">
                    <div class="content-wrap">
                        <div class="content-layout">
                            <div class="zib-widget box-body radius8 main-bg theme-box" style="text-align:center; padding: 100px 20px; min-height: 50vh; display: flex; flex-direction: column; justify-content: center; align-items: center;">
                                <img src="https://www.zibll.top/wp-content/themes/zibll/img/thumbnail.svg" data-src="<?php echo esc_url($null_img); ?>" alt="私密主页" style="width: 200px; max-width: 100%; margin-bottom: 20px; opacity: 0.8;">
                                <h2 style="font-size: 22px; color: var(--main-color); margin-bottom: 15px; font-weight: bold;">主页未公开</h2>
                                <p style="color: var(--muted-color); margin-bottom: 30px; font-size: 15px;">该用户暂未开启主页公开展示，您目前无法查看其个人资料和动态</p>
                                <a href="<?php echo home_url(); ?>" class="btn btn-primary radius">返回首页</a>
                            </div>
                        </div>
                    </div>
                </main>

                <?php
                // 3. 加载主题的底部及侧边栏悬浮等功能
                get_footer(); 
                
                // 4. 强制中断，防止继续加载原始的 author.php 文件
                exit;
            }
        }
    }
}