在 WooCommerce 中扩展产品搜索以包含自定义数据字段

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

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

    woocommerce_wp_text_input( [
        'id'            => '_pdc_notch',
        'wrapper_class' => '',
        'label'         => 'Notch',
        'desc_tip'      => true,
        'type'          => 'text',
        'description'   => '输入 Notch PDC',
        'value'         => get_post_meta( $post->ID, '_pdc_notch', true )
    ] );

    echo '</div>';
}

/**
 * 保存自定义字段的值
 *
 * @param int $product_id 当前产品的ID
 */
add_action( 'woocommerce_process_product_meta', 'woocommerce_process_product_meta_action', 10, 1 );
function woocommerce_process_product_meta_action( 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', 'filter_posts_search', 10, 2 );
function filter_posts_search( string $search, WP_Query $query ): string {
    global $wpdb;

    // 1. 验证条件：
    //    - 搜索字符串不为空
    //    - 存在搜索查询变量 's' 和 'post_type'
    //    - 搜索类型为 'product'
    //    - 不在后台管理页面
    //    - 是主查询且是搜索页面
    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() || ! is_main_query() ) {
        return $search;
    }

    $product_ids = [];
    $search_term = esc_attr( $query->query_vars['s'] ); // 获取搜索词并进行转义

    // 2. 使用 wc_get_products 查找匹配自定义字段的产品
    //    这允许我们利用 WooCommerce 的 API 进行高效的元数据查询
    $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();
    }

    // 确保产品ID唯一
    $product_ids = array_unique( $product_ids );

    // 3. 修改原始 SQL 搜索字符串
    //    如果找到了匹配自定义字段的产品，则将这些产品的 ID 添加到现有的搜索条件中
    //    通过 str_replace 将 "AND (((" 替换为 "AND (((products.ID IN (...)) OR ("
    //    这样，搜索结果将包含：
    //    (原始搜索条件) OR (自定义字段匹配的产品ID)
    if ( count( $product_ids ) > 0 ) {
        // 构建 ID 列表字符串
        $ids_string = implode( ',', $product_ids );
        // 查找并替换 SQL 字符串中的特定模式
        // 注意：这里的替换模式 'AND (((' 是基于 WordPress 生成的默认搜索查询结构
        $search = str_replace( "AND (((", "AND ((({$wpdb->posts}.ID IN ({$ids_string})) OR (", $search );
    }

    return $search;
}