<?php
function smart_chinese_segment($text) {// 中文分词主函数
    if (empty(trim($text))) {
        return $text;
    }
    
    // 清理文本
    $text = trim($text);
    
    // 处理标点符号和特殊字符
    $punctuations = [
        '，', '。', '；', '！', '？', '、', ' ', 
        ',', '.', ';', '!', '?', ':', '：', 
        '(', ')', '（', '）', '[', ']', '【', '】',
        '"', "'", '“', '”', '‘', '’', '《', '》'
    ];
    $text = str_replace($punctuations, ' ', $text);
    
    // 处理中英文混合情况
    $text = preg_replace('/([a-zA-Z0-9]+)([\x{4e00}-\x{9fa5}])/u', '$1 $2', $text);
    $text = preg_replace('/([\x{4e00}-\x{9fa5}])([a-zA-Z0-9]+)/u', '$1 $2', $text);
    
    // 中文按字符分割（简单分词）
    if (preg_match('/[\x{4e00}-\x{9fa5}]/u', $text)) {
        $chars = preg_split('/(?<!^)(?!$)/u', $text);
        $text = implode(' ', $chars);
    }
    
    // 清理多余空格
    $text = preg_replace('/\s+/', ' ', $text);
    
    return trim($text);
}

// 替换原插件的搜索分析函数
function dahuzi_search_analysis($s) {
    return smart_chinese_segment($s);
}

// 搜索查询替换 - 修改WordPress搜索查询
function custom_replace_search_query($query) {
    // 只在主搜索查询且不是后台时工作
    if (!is_admin() && $query->is_main_query() && $query->is_search()) {
        $search_term = $query->get('s');
        if (!empty($search_term)) {
            $new_search_term = dahuzi_search_analysis($search_term);
            // 只有当分词结果不同时才替换
            if ($new_search_term != $search_term) {
                $query->set('s', $new_search_term);
            }
        }
    }
}
add_action('pre_get_posts', 'custom_replace_search_query');

// 修改搜索页面显示的搜索词（可选）
function custom_search_query_display($search_query) {
    if (!is_admin() && is_search()) {
        return dahuzi_search_analysis($search_query);
    }
    return $search_query;
}
add_filter('get_search_query', 'custom_search_query_display');

// 保存搜索历史（兼容原插件功能）
function custom_save_search_history($search_term) {
    if (empty($search_term)) return;
    
    $search_history = get_option('custom_search_history', []);
    $current_search = trim(strip_tags($search_term));
    
    if (!empty($current_search) && !in_array($current_search, $search_history)) {
        // 只保留最近20个搜索记录
        array_unshift($search_history, $current_search);
        $search_history = array_slice($search_history, 0, 20);
        update_option('custom_search_history', $search_history);
    }
}

// 更新搜索关键词（兼容原插件功能）
function custom_update_search_keywords($search_term) {
    if (empty($search_term)) return;
    
    $search_keywords = get_option('custom_search_keywords', []);
    $current_search = trim(strip_tags($search_term));
    
    if (!empty($current_search)) {
        if (isset($search_keywords[$current_search])) {
            $search_keywords[$current_search]++;
        } else {
            $search_keywords[$current_search] = 1;
        }
        
        // 只保留前100个热门关键词
        arsort($search_keywords);
        $search_keywords = array_slice($search_keywords, 0, 100, true);
        
        update_option('custom_search_keywords', $search_keywords);
    }
}

// 挂钩到搜索完成时保存历史
function handle_search_completion() {
    if (is_search() && !is_admin() && get_search_query()) {
        $search_term = get_search_query();
        custom_save_search_history($search_term);
        
        // 只在第一页更新热门关键词
        global $paged;
        if (!$paged || $paged == 1) {
            custom_update_search_keywords($search_term);
        }
    }
}
add_action('wp_head', 'handle_search_completion');

// 禁用原插件的问题API调用（通过移除原插件钩子）
function disable_original_plugin_functions() {
    // 尝试移除可能存在的原插件钩子
    remove_action('parse_query', 'dahuzi_replace_search');
    remove_filter('get_search_query', 'dahuzi_get_search_query');
}
add_action('init', 'disable_original_plugin_functions');

// 测试函数 - 仅在URL参数debug=1时显示
function debug_chinese_segment() {
    if (!isset($_GET['debug']) || $_GET['debug'] != 1 || !current_user_can('edit_posts')) {
        return;
    }
    
    $test_cases = [
        '格式工厂下载',
        '北京欢迎你',
        '这是一段测试文本',
        'hello world',
        '中文和English混合',
        'WordPress主题开发',
        '2023年最新版软件'
    ];
    
    echo '<div style="background:#f0f0f1;border:1px solid #ccc;padding:20px;margin:20px 0;">';
    echo '<h3>中文分词测试结果</h3>';
    
    foreach ($test_cases as $text) {
        $result = smart_chinese_segment($text);
        echo "<p><strong>原文本:</strong> <code>{$text}</code><br>";
        echo "<strong>分词结果:</strong> <code>{$result}</code></p>";
    }
    
    echo '</div>';
}
add_action('wp_footer', 'debug_chinese_segment');

?>