From c21bb53382d99365125c9a23db3ae008d7e561d8 Mon Sep 17 00:00:00 2001
From: androidlover5842 <androidlover5842@gmail.com>
Date: Sat, 31 Jan 2026 10:57:09 +0530
Subject: [PATCH] Fix lazy booking load in doc city update

---
 .../trisolarisserver/component/DocumentExtractionService.kt    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt b/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
index 1888581..3048a28 100644
--- a/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
+++ b/src/main/kotlin/com/android/trisolarisserver/component/DocumentExtractionService.kt
@@ -427,7 +427,8 @@ class DocumentExtractionService(
     }
 
     private fun applyBookingCityUpdates(document: GuestDocument, results: Map<String, String>) {
-        val booking = document.booking
+        val bookingId = document.booking?.id ?: return
+        val booking = bookingRepo.findById(bookingId).orElse(null) ?: return
         if (booking.fromCity?.isNotBlank() == true && booking.toCity?.isNotBlank() == true) return
         val pin = cleanedValue(results[DocumentPrompts.PIN_CODE.first]) ?: return
         if (!isValidPin(pin)) return