获取 MemberPress 非订阅交易的企业账户 ID：一个健壮的解决方案

add_action('mepr-event-transaction-completed', 'my_custom_corporate_id_handler');

function my_custom_corporate_id_handler($event) {
  $transaction = $event->get_data();
  $membership_type_ids = array(1, 2, 4); // 假设这些是需要处理的会员类型ID

  if (in_array($transaction->product_id, $membership_type_ids) && $transaction->txn_type == 'payment') {
    $org_id = $transaction->corporate_account_id;
    // my_custom_function($org_id); // 在这里执行自定义逻辑
  }
}

/**
 * 处理 MemberPress 交易完成事件，确保获取正确的企业账户 ID。
 * 针对非订阅交易的企业 ID 延迟设置问题，采用延迟处理机制。
 */
add_action('mepr-event-transaction-completed', 'single_transaction_corporate_id_fix');

function single_transaction_corporate_id_fix($event) {
  $transaction = $event->get_data();
  $membership_type_ids = array(1, 2, 4); // 根据实际情况替换为需要监控的会员产品ID

  // 仅处理特定会员类型和支付交易
  if (!in_array($transaction->product_id, $membership_type_ids) || $transaction->txn_type !== 'payment') {
    return;
  }

  // 检查 corporate_account_id 是否有效
  // MemberPress 在此事件中可能返回字符串 "0" 或整数 0
  if ($transaction->corporate_account_id !== "0" && $transaction->corporate_account_id !== 0) {
    // 如果 corporate_account_id 已经正确设置，则立即执行自定义逻辑
    // 例如：my_custom_function($transaction->corporate_account_id);
    // write_log('Corporate ID already set for transaction: ' . $transaction->trans_num . ' ID: ' . $transaction->corporate_account_id);
  } else {
    // 如果 corporate_account_id 为 0，说明是 MemberPress 的已知问题，需要延迟处理
    // 安排一个单次事件，在 2 分钟后执行，以等待 MemberPress 更新数据库
    $cron_hook = 'send_fix_for_zero_corporate_id_transaction';
    $args = array($transaction->trans_num); // 传递交易编号，因为完整的 $transaction 对象可能太大或序列化问题

    // 避免重复调度相同的事件
    if (!wp_next_scheduled($cron_hook, $args)) {
      wp_schedule_single_event(strtotime("+2 minutes"), $cron_hook, $args);
      // 可选：记录日志以便调试
      // write_log('Scheduled cron for transaction ' . $transaction->trans_num . ' due to zero corporate ID.');
    }
  }
}

/**
 * 定时任务回调函数，用于在 MemberPress 更新数据库后获取正确的企业账户 ID。
 * 注意：此函数仅在 MemberPress 存在相关 Bug 且 corporate_id 未即时返回时被调用。
 */
add_action('send_fix_for_zero_corporate_id_transaction', 'process_delayed_corporate_transaction');

function process_delayed_corporate_transaction($trans_num) {
  // 从数据库重新加载完整的交易对象，此时 corporate_account_id 应该已经更新
  $full_transaction = MeprTransaction::get_one_by_trans_num($trans_num);

  if ($full_transaction && $full_transaction->corporate_account_id !== "0" && $full_transaction->corporate_account_id !== 0) {
    $correct_corporate_id = $full_transaction->corporate_account_id;
    // 在这里执行您的自定义逻辑，例如：
    // my_custom_function($correct_corporate_id);
    // write_log('Successfully retrieved corporate ID ' . $correct_corporate_id . ' for transaction ' . $trans_num . ' via delayed cron.');
  } else {
    // 如果延迟后仍然无法获取，可能需要进一步的错误处理或重试机制
    // write_log('Error: Still unable to retrieve corporate ID for transaction ' . $trans_num . ' after delay.');
  }
}

// 辅助函数：简单的日志记录（如果您的环境没有提供，可以自行实现）
// if (!function_exists('write_log')) {
//     function write_log($log) {
//         if (is_array($log) || is_object($log)) {
//             error_log(print_r($log, true));
//         } else {
//             error_log($log);
//         }
//     }
// }