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

add_action( 'woocommerce_product_options_general_product_data', 'woocommerce_product_options_general_product_data_action', 9999, 0 );

/**
 * 在WooCommerce产品编辑页面的“常规”选项卡中添加自定义字段
 */
function woocommerce_product_options_general_product_data_action(): 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>';
}

add_action( 'woocommerce_process_product_meta', 'woocommerce_process_product_meta_action', 10, 1 );

/**
 * 保存自定义产品数据字段
 *
 * @param int $product_id 当前产品的ID
 */
function woocommerce_process_product_meta_action( int $product_id ): void {
    // 从POST请求中获取字段值，并进行安全清理
    $pdc_notch = sanitize_text_field( wp_unslash( $_POST['_pdc_notch'] ?? null ) );

    // 更新或添加产品元数据
    update_post_meta( $product_id, '_pdc_notch', $pdc_notch );
}

add_filter( 'posts_search', 'filter_posts_search_by_custom_field', 10, 2 );

/**
 * 扩展产品搜索以使用自定义字段进行搜索
 *
 * @param string $search 原始的SQL搜索字符串
 * @param WP_Query $query 当前的WP_Query对象
 * @return string 修改后的SQL搜索字符串
 */
function filter_posts_search_by_custom_field( string $search, WP_Query $query ): string {
    global $wpdb;

    // 1. 验证条件：
    //    - 搜索字符串不为空
    //    - 存在搜索关键字 `s` 且不为空
    //    - 搜索类型为 `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() || 
         ! $query->is_main_query() ) {
        return $search; // 不符合条件，返回原始搜索字符串
    }

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

    // 2. 使用 wc_get_products 查找匹配自定义字段的产品
    //    - 'limit' => -1 获取所有匹配产品
    //    - 'meta_key' 和 'meta_value' 指定要搜索的自定义字段及其值
    //    - 'meta_compare' 可以是 'LIKE' (模糊匹配) 或 '=' (精确匹配)
    $products = wc_get_products( [
        'post_type'    => 'product',
        'limit'        => -1,
        'meta_key'     => '_pdc_notch',
        'meta_value'   => $search_term,
        'meta_compare' => 'LIKE' // 根据需求选择 'LIKE' 或 '='
    ] );

    // 3. 收集所有匹配产品的ID
    /** @var WC_Product $product */
    foreach ( $products as $product ) {
        $product_ids[] = $product->get_id();
    }

    $product_ids = array_unique( $product_ids ); // 去重

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

        // 将此SQL片段注入到原始搜索查询的WHERE子句中
        // 这里的替换逻辑是为了确保我们的自定义字段搜索条件与原有搜索条件以 OR 关系连接
        // 原始搜索通常是 `AND (((...))`，我们将其修改为 `AND ((({$id_in_clause}) OR (...))`
        $search = str_replace( 'AND (((', "AND (({$id_in_clause}) OR ((", $search );
    }

    return $search; // 返回修改后的搜索字符串
}