WooCommerce订阅：实现单活跃订阅策略并允许套餐切换

<?php
/**
 * 限制用户同时只能有一个活跃的WooCommerce订阅，但允许升级/降级。
 */

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 返回 true 表示用户已订阅该产品（可能是升级/降级目标）
            // 所以，如果用户有活跃订阅 AND 即将添加的产品不是用户已订阅的，则阻止
            if (! has_woocommerce_subscription('', $product_id, '')) {
                wc_clear_notices(); // 清除WooCommerce可能添加的其他通知
                wc_add_notice(__('您已有一个活跃的订阅。如需更改套餐，请在“我的账户”页面进行切换。', 'woocommerce'), 'error');
                return false; // 阻止产品添加到购物车
            }
        }
    }
    return $valid; // 允许添加，或非订阅产品则继续验证
}

/**
 * 检查当前用户是否有任何活跃的WooCommerce订阅。
 *
 * @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);
}

/**
 * 检查指定用户是否订阅了特定产品。
 *
 * @param int    $the_user_id    用户ID，可选，默认为当前用户。
 * @param int    $the_product_id 产品ID。
 * @param string $the_status     订阅状态，可选。
 * @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;
    }

    // 确保 WC_Subscriptions_Manager 类存在，避免在插件未激活时报错
    if (class_exists('WC_Subscriptions_Manager') && WC_Subscriptions_Manager::user_has_subscription( $the_user_id, $the_product_id, $the_status)) {
        return true;
    }
    return false;
}