Allow guest document file token access
All checks were successful
build-and-deploy / build-deploy (push) Successful in 1m36s

This commit is contained in:
androidlover5842
2026-01-30 15:58:02 +05:30
parent 8ba3c2f2bd
commit 174652e889
2 changed files with 8 additions and 21 deletions

View File

@@ -27,7 +27,6 @@ import java.time.OffsetDateTime
import java.nio.file.Files
import java.nio.file.Paths
import java.util.UUID
import java.util.Base64
@RestController
@RequestMapping("/properties/{propertyId}/guests/{guestId}/documents")
@@ -44,7 +43,9 @@ class GuestDocuments(
private val llamaClient: LlamaClient,
private val objectMapper: ObjectMapper,
@org.springframework.beans.factory.annotation.Value("\${storage.documents.publicBaseUrl}")
private val publicBaseUrl: String
private val publicBaseUrl: String,
@org.springframework.beans.factory.annotation.Value("\${storage.documents.aiBaseUrl:\${storage.documents.publicBaseUrl}}")
private val aiBaseUrl: String
) {
@PostMapping
@@ -142,7 +143,9 @@ class GuestDocuments(
extractionQueue.enqueue {
val document = guestDocumentRepo.findById(documentId).orElse(null) ?: return@enqueue
try {
val imageUrl = buildImageDataUrl(document) ?: return@enqueue
val token = tokenService.createToken(document.id.toString())
val imageUrl =
"${aiBaseUrl}/properties/$propertyId/guests/$guestId/documents/${document.id}/file?token=$token"
val results = linkedMapOf<String, String>()
results["hasAadhar"] = llamaClient.ask(imageUrl, "CONTAINS AADHAAR? Answer YES or NO only.")
@@ -200,24 +203,6 @@ class GuestDocuments(
}
}
private fun buildImageDataUrl(document: GuestDocument): String? {
val contentType = document.contentType?.trim()?.lowercase()
if (contentType == null || !contentType.startsWith("image/")) {
return null
}
val path = Paths.get(document.storagePath)
if (!Files.exists(path)) {
return null
}
return try {
val bytes = Files.readAllBytes(path)
val encoded = Base64.getEncoder().encodeToString(bytes)
"data:$contentType;base64,$encoded"
} catch (_: Exception) {
null
}
}
}
data class GuestDocumentResponse(

View File

@@ -12,6 +12,7 @@ internal object PublicEndpoints {
private val iconPngFile = Regex("^/icons/png/[^/]+$")
private val payuWebhook = Regex("^/properties/[^/]+/payu/webhook$")
private val payuReturn = Regex("^/properties/[^/]+/payu/return/(success|failure)$")
private val guestDocumentFile = Regex("^/properties/[^/]+/guests/[^/]+/documents/[^/]+/file$")
fun isPublic(request: HttpServletRequest): Boolean {
val path = request.requestURI
@@ -30,5 +31,6 @@ internal object PublicEndpoints {
|| (path == "/image-tags" && method == "GET")
|| path == "/icons/png"
|| iconPngFile.matches(path)
|| guestDocumentFile.matches(path)
}
}