WooCommerce 订阅：实现单一活跃订阅并支持升级/降级功能

add_filter('woocommerce_add_to_cart_validation', 'check_num_of_subscriptions', 10, 2);

function check_num_of_subscriptions( $valid, $product_id ) {
    $product_to_add = wc_get_product( $product_id );

    if ( $product_to_add instanceof WC_Product_Subscription || $product_to_add instanceof WC_Product_Variable_Subscription) {
        if ( has_active_subscription() ) {
            wc_clear_notices();
            wc_add_notice(__('您已有一个活跃的订阅。', 'woocommerce'), 'error');
            return false;
        } else {
            return true;
        }
    }
    return $valid;
}

function has_active_subscription() {
    $user_id = get_current_user_id();
    $active_subscriptions = get_posts(array(
        'numberposts' => -1,
        'meta_key' => '_customer_user',
        'meta_value' => $user_id,
        'post_type' => 'shop_subscription',
        'post_status' => 'wc-active'
    ));
    return !empty($active_subscriptions);
}

<?php
/**
 * 限制用户只能拥有一份活跃订阅，同时允许订阅升级/降级。
 * 将此代码添加到主题的 functions.php 文件或自定义插件中。
 */

add_filter('woocommerce_add_to_cart_validation', 'check_num_of_subscriptions', 10, 2);

/**
 * 检查用户是否可以添加订阅产品到购物车。
 * 阻止新订阅，但允许现有订阅的升级/降级。
 *
 * @param bool $valid      当前验证结果。
 * @param int  $product_id 尝试添加到购物车的商品ID。
 * @return bool
 */
function check_num_of_subscriptions( $valid, $product_id ){
    $product_to_add = wc_get_product( $product_id );

    // 仅对订阅产品或可变订阅产品进行处理
    if ($product_to_add instanceof WC_Product_Subscription || $product_to_add instanceof WC_Product_Variable_Subscription){
        // 如果用户已有活跃订阅，并且当前添加的商品不是用户已有订阅的商品（即非升级/降级）
        if ( has_active_subscription() && (! has_woocommerce_subscription('', $product_id, '')) ) {
            wc_clear_notices();
            wc_add_notice(__('您已有一个活跃的订阅。', 'woocommerce'), 'error');
            return false; // 阻止添加到购物车
        } else {
            return true; // 允许添加到购物车（要么没有活跃订阅，要么是现有订阅的切换）
        }
    }
    return $valid; // 非订阅产品，按原逻辑处理
}

/**
 * 检查当前用户是否拥有任何活跃的订阅。
 *
 * @return bool 如果用户有活跃订阅则返回 true，否则返回 false。
 */
function has_active_subscription(){
    $user_id = get_current_user_id();
    if ( ! $user_id ) {
        return false; // 未登录用户不可能有活跃订阅
    }

    $active_subscriptions = get_posts(array(
        'numberposts' => -1,
        'meta_key'    => '_customer_user',
        'meta_value'  => $user_id,
        'post_type'   => 'shop_subscription',
        'post_status' => 'wc-active' // 只查询活跃状态的订阅
    ));
    return !empty($active_subscriptions);
}

/**
 * 检查指定用户是否订阅了某个特定的 WooCommerce 产品。
 *
 * @param int|string $the_user_id    用户ID，如果为空则使用当前登录用户ID。
 * @param int        $the_product_id 产品ID。
 * @param string     $the_status     订阅状态（例如 'active', 'on-hold'），空字符串表示任何状态。
 * @return bool 如果用户订阅了该产品则返回 true，否则返回 false。
 */
function has_woocommerce_subscription($the_user_id, $the_product_id, $the_status) {
    $current_user = wp_get_current_user();
    if (empty($the_user_id)) {
        $the_user_id = $current_user->ID;
    }

    // 使用 WooCommerce Subscriptions 官方管理器方法进行检查
    if (WC_Subscriptions_Manager::user_has_subscription( $the_user_id, $the_product_id, $the_status)) {
        return true;
    }
    return false;
}