如何修改WooCommerce自定义函数实现多外部URL对应不同联盟ID

没问题，我帮你调整代码实现这个按目标站点匹配联盟ID的需求，还优化了原有代码的健壮性，避免潜在问题。

完整解决方案代码 #

/**
 * Custom Add Affiliate link to Buy Product based on target site
 */
add_filter( 'woocommerce_product_add_to_cart_url', 'custom_product_add_to_cart_url', 20, 2 );
function custom_product_add_to_cart_url( $add_to_cart_url, $product ){
    // 仅对外部产品生效
    if( !$product->is_type('external') ) {
        return $add_to_cart_url;
    }

    // 解析目标URL，提取域名部分
    $url_parts = parse_url( $add_to_cart_url );
    // 如果解析失败（比如URL格式错误），直接返回原链接
    if( empty( $url_parts['host'] ) ) {
        return $add_to_cart_url;
    }

    // 转成小写，避免大小写敏感问题（比如SiteA.com和sitea.com视为同一个站点）
    $target_host = strtolower( $url_parts['host'] );

    // 配置：域名对应的联盟后缀，后续新增站点直接在这里加行即可
    $affiliate_suffix_map = [
        'www.sitea.com' => 'affiliateID-1',
        'www.siteb.net' => 'affiliateID-2'
        // 示例：'www.sitec.org' => 'affiliateID-3'
    ];

    // 如果当前目标域名在配置中，追加对应的联盟参数
    if( isset( $affiliate_suffix_map[$target_host] ) ) {
        // 自动判断用?还是&：如果原URL已有查询参数，用&拼接，否则用?
        $separator = strpos( $add_to_cart_url, '?' ) !== false ? '&' : '?';
        $add_to_cart_url .= $separator . $affiliate_suffix_map[$target_host];
    }

    return $add_to_cart_url;
}
/**
 * End Custom Add Affiliate link to Buy Product
 */

关键细节说明 #

1. 精准的域名匹配：用parse_url()提取URL的域名部分，比直接字符串匹配更可靠——不管原URL带不带https://、www或者路径，都能准确识别目标站点。
2. 自动处理参数拼接：通过判断原URL是否已有?，自动选择用?还是&追加参数，避免出现www.siteA.com/?foo=bar/?affiliateID-1这种错误格式的链接。
3. 可扩展的配置：所有站点和联盟ID的对应关系都存在$affiliate_suffix_map数组里，后续新增联盟站点只需要在数组里加一行，不用修改核心逻辑，维护起来更方便。
4. 健壮性保障：增加了对非外部产品、URL解析失败的判断，不会意外修改普通产品的购物车链接。

如果你的联盟系统要求用标准的键值对参数（比如?affiliate=ID-1而不是?affiliateID-1），可以把拼接部分改成用WooCommerce内置的add_query_arg()函数，这样更规范：

// 替换原来的拼接逻辑
if( isset( $affiliate_suffix_map[$target_host] ) ) {
    $add_to_cart_url = add_query_arg( 'affiliate', $affiliate_suffix_map[$target_host], $add_to_cart_url );
}
// 同时调整配置数组为：
$affiliate_suffix_map = [
    'www.sitea.com' => 'ID-1',
    'www.siteb.net' => 'ID-2'
];

内容的提问来源于stack exchange，提问作者Gloves