Update: 2026-08-08 14:09:20

backend/ride/invitor/get_driver_referrals.php

@@ -1,17 +1,18 @@
 <?php
 require_once __DIR__ . '/../../connect.php';
+require_once __DIR__ . '/referral_code_helper.php';
 
 // Use JWT token variables provided by connect.php
 if (!$user_id || $role != 'driver') {
     jsonError("Invalid parameters or unauthorized token");
 }
 
-// 1. Get the driver's referral code
-$stmtCode = $con->prepare("SELECT referral_code FROM user_referral_codes WHERE user_id = ? AND user_type = 'driver'");
-$stmtCode->execute([$user_id]);
+// 1. Get the driver's referral code — ونُنشئه إن غاب.
+// ‏الردّ بـ null كان يترك شاشة الدعوة تدور بلا كود إلى الأبد، لأن لا
+// ‏شيء في التطبيق ينادي get_unified_code.php الذي يولّده.
+$referralCode = ensureReferralCode($con, (string) $user_id, 'driver');
 
-if ($stmtCode->rowCount() == 0) {
-    // If no code exists, return empty stats
+if ($referralCode === null) {
     printSuccess([
         "referral_code" => null,
         "total_invited_drivers" => 0,
@@ -21,8 +22,6 @@ if ($stmtCode->rowCount() == 0) {
     exit;
 }
 
-$referralCode = $stmtCode->fetchColumn();
-
 // 2. Fetch all referrals made by this code
 $stmtRefs = $con->prepare("
     SELECT id, invited_user_id, invited_user_type, status, trip_count, is_reward_claimed, created_at