Decode pdf from base64 and add controller endpoint to display report

Author: JebronLames32

src/main/java/org/mifos/loanrisk/external/bsa/arya/AryaBankStatementAnalysisService.java

@@ -2,8 +2,8 @@
 
 import java.util.Base64;
 import java.util.UUID;
-import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
 import io.r2dbc.postgresql.codec.Json;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
@@ -66,15 +66,27 @@ private Mono<Void> sendRequest(String base64, DocumentDataV1 document, Aggregato
                 .header("token", token)
                 .bodyValue(body)
                 .retrieve()
-                .bodyToMono(String.class)
+                .bodyToMono(AryaBsaResponse.class)
                 .flatMap(resp -> saveResult(resp, aggregator))
                 .doOnError(err -> log.error("Arya BSA request failed", err))
                 .then(updateAggregatorStatus(aggregator));
     }
 
-    private Mono<BankStatementAnalysisResult> saveResult(String resp, Aggregator aggregator) {
-            BankStatementAnalysisResult result = new BankStatementAnalysisResult(null, aggregator.getLoanId(), Json.of(resp));
-            return resultRepository.save(result);
+    private Mono<BankStatementAnalysisResult> saveResult(AryaBsaResponse resp, Aggregator aggregator) {
+        String reportKey = "bsa/report-" + resp.reqId() + ".pdf";
+        byte[] pdf = Base64.getDecoder().decode(resp.data());
+        return storageClient.put(pdf, reportKey)
+                .then(Mono.fromCallable(() -> {
+                    ObjectNode node = mapper.createObjectNode();
+                    node.put("req_id", resp.reqId());
+                    node.put("success", resp.success());
+                    if (resp.errorMessage() != null) {
+                        node.put("error_message", resp.errorMessage());
+                    }
+                    node.put("reportKey", reportKey);
+                    return new BankStatementAnalysisResult(null, aggregator.getLoanId(), Json.of(node.toString()));
+                }))
+                .flatMap(resultRepository::save);
     }
 
     private Mono<Void> updateAggregatorStatus(Aggregator aggregator) {

src/main/java/org/mifos/loanrisk/external/bsa/arya/AryaBsaResponse.java

@@ -1,7 +1,6 @@
 package org.mifos.loanrisk.external.bsa.arya;
 
 import com.fasterxml.jackson.annotation.JsonProperty;
-import java.util.Map;
 
 /**
  * DTO for Arya's bank statement analysis response.
@@ -10,6 +9,6 @@ public record AryaBsaResponse(
         @JsonProperty("req_id") String reqId,
         @JsonProperty("success") Boolean success,
         @JsonProperty("error_message") String errorMessage,
-        @JsonProperty("data") Map<String, Object> data) {
+        @JsonProperty("data") String data) {
 }
 

src/main/java/org/mifos/loanrisk/external/controller/AnalysisResultUIController.java

@@ -5,11 +5,14 @@
 import com.fasterxml.jackson.databind.node.ObjectNode;
 import io.r2dbc.postgresql.codec.Json;
 import lombok.AllArgsConstructor;
+import org.mifos.loanrisk.document.storage.ObjectStorageClient;
 import org.mifos.loanrisk.external.bsa.domain.BankStatementAnalysisResult;
 import org.mifos.loanrisk.external.bsa.repository.BankStatementAnalysisResultRepository;
 import org.mifos.loanrisk.external.cb.domain.CreditBureauResult;
 import org.mifos.loanrisk.external.cb.repository.CreditBureauResultRepository;
+import org.springframework.http.HttpHeaders;
 import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RestController;
@@ -25,6 +28,7 @@ public class AnalysisResultUIController {
     private final BankStatementAnalysisResultRepository bsaRepository;
     private final CreditBureauResultRepository cbRepository;
     private final ObjectMapper objectMapper;
+    private final ObjectStorageClient storageClient;
 
     @GetMapping(value = "/bsa", produces = MediaType.TEXT_HTML_VALUE)
     public Mono<String> getAllBsaResults() {
@@ -42,6 +46,30 @@ public Mono<String> getBsaResultsByLoan(@PathVariable("loanId") Long loanId) {
                 .map(this::asHtml);
     }
 
+    @GetMapping(value = "/bsa/report/{id}", produces = MediaType.APPLICATION_PDF_VALUE)
+    public Mono<ResponseEntity<byte[]>> getBsaReport(@PathVariable("id") Long id) {
+        return bsaRepository.findById(id)
+                .flatMap(result -> {
+                    if (result.getAttributes() == null) {
+                        return Mono.empty();
+                    }
+                    try {
+                        String reportKey = objectMapper.readTree(result.getAttributes().asString())
+                                .path("reportKey").asText(null);
+                        if (reportKey == null || reportKey.isBlank()) {
+                            return Mono.empty();
+                        }
+                        return storageClient.get(reportKey)
+                                .map(data -> ResponseEntity.ok()
+                                        .header(HttpHeaders.CONTENT_DISPOSITION, "inline; filename=\"bsa-report-" + id + ".pdf\"")
+                                        .contentType(MediaType.APPLICATION_PDF)
+                                        .body(data));
+                    } catch (JsonProcessingException e) {
+                        return Mono.error(e);
+                    }
+                });
+    }
+
     @GetMapping(value = "/cb", produces = MediaType.TEXT_HTML_VALUE)
     public Mono<String> getAllCbResults() {
         return cbRepository.findAll()