扩展WooCommerce产品搜索以集成自定义数据字段

/**
 * 在产品编辑页面的“常规”选项卡中添加自定义文本字段
 */
add_action( 'woocommerce_product_options_general_product_data', 'add_custom_product_data_field', 9999, 0 );
function add_custom_product_data_field(): void {
    global $post;

    echo '<div class="options_group">';

    woocommerce_wp_text_input( [
        'id'            => '_pdc_notch', // 自定义字段的ID
        'wrapper_class' => '',
        'label'         => 'Notch', // 字段标签
        'desc_tip'      => true,
        'type'          => 'text',
        'description'   => '输入PDC Notch信息', // 字段描述
        'value'         => get_post_meta( $post->ID, '_pdc_notch', true ) // 获取并显示已保存的值
    ] );

    echo '</div>';
}

/**
 * 保存自定义产品数据字段的值
 *
 * @param int $product_id 当前产品的ID
 */
add_action( 'woocommerce_process_product_meta', 'save_custom_product_data_field', 10, 1 );
function save_custom_product_data_field( int $product_id ): void {
    // 检查并清理输入数据
    $pdc_notch = sanitize_text_field( wp_unslash( $_POST['_pdc_notch'] ?? null ) );

    // 更新或添加自定义字段的值
    update_post_meta( $product_id, '_pdc_notch', $pdc_notch );
}

/**
 * 扩展产品搜索，使其包含自定义字段值
 *
 * @param string   $search 原始的SQL搜索子句
 * @param WP_Query $query  当前的WP_Query对象
 * @return string 修改后的SQL搜索子句
 */
add_filter( 'posts_search', 'extend_product_search_with_custom_field', 10, 2 );
function extend_product_search_with_custom_field( string $search, WP_Query $query ): string {
    global $wpdb;

    // 1. 验证搜索条件：
    //    - 搜索字符串不为空
    //    - 是产品搜索 ('product' post type)
    //    - 不是后台管理页面
    //    - 是前端搜索结果页
    //    - 是主查询
    if ( empty( $search ) || 
         ! ( isset( $query->query_vars['s'], $query->query_vars['post_type'] ) && 
             ! empty( $query->query_vars['s'] ) && 
             $query->query_vars['post_type'] === 'product' ) || 
         is_admin() || 
         ! is_search() || 
         ! $query->is_main_query() ) {
        return $search;
    }

    $product_ids = [];
    $search_term = esc_attr( $query->query_vars['s'] ); // 获取并清理搜索词

    // 2. 使用 wc_get_products 预先查询匹配自定义字段的产品
    $products = wc_get_products( [
        'post_type'    => 'product',
        'limit'        => - 1, // 获取所有匹配的产品
        'meta_key'     => '_pdc_notch', // 自定义字段的键
        'meta_value'   => $search_term, // 搜索词
        'meta_compare' => 'LIKE' // 使用 LIKE 进行模糊匹配，也可以是 '=' 进行精确匹配
    ] );

    /**
     * @var WC_Product $product
     */
    foreach ( $products as $product ) {
        $product_ids[] = $product->get_id();
    }

    $product_ids = array_unique( $product_ids ); // 去重产品ID

    // 3. 如果找到匹配自定义字段的产品，修改主搜索的SQL语句
    if ( count( $product_ids ) > 0 ) {
        // 构建一个SQL片段，用于检查产品ID是否在匹配列表中
        $ids_in_clause = $wpdb->posts . ".ID IN (" . implode( ',', $product_ids ) . ")";

        // 将新的条件插入到原始搜索SQL语句中
        // 目标是将原始的 "AND (((" 替换为 "AND (($ids_in_clause) OR (("
        // 这样，主搜索就会包含自定义字段的匹配结果
        $search = str_replace( 'AND (((', "AND (({$ids_in_clause}) OR (", $search );
    }

    return $search;
}