WebSVN – SmartDukaan – Diff – /trunk/profitmandi-fofo/src/main/java/com/spice/profitmandi/web/controller/OrderController.java

 import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;
 import com.spice.profitmandi.common.model.*;
 import com.spice.profitmandi.common.solr.SolrService;
 import com.spice.profitmandi.common.util.FileUtil;
 import com.spice.profitmandi.common.util.PdfUtils;
-import com.spice.profitmandi.common.util.PrinterType;
 import com.spice.profitmandi.common.util.StringUtils;
 import com.spice.profitmandi.common.web.util.ResponseSender;
 import com.spice.profitmandi.dao.entity.auth.AuthUser;
 import com.spice.profitmandi.dao.entity.catalog.CustomerOffer;
 import com.spice.profitmandi.dao.entity.catalog.CustomerOfferItem;
+    }
     @RequestMapping(value = "/generateInvoice")
     public ResponseEntity<?> generateInvoice(HttpServletRequest request, HttpServletResponse response,
+                                             @RequestParam(name = ProfitMandiConstants.ORDER_ID) int orderId) throws ProfitMandiBusinessException {
+        LOGGER.info("Request received at url {} with params [{}={}] ", request.getRequestURI(),
+                ProfitMandiConstants.ORDER_ID, orderId);
+        LoginDetails fofoDetails = cookiesProcessor.getCookiesObject(request);
+        InvoicePdfModel pdfModel = null;
+        if (roleManager.isAdmin(fofoDetails.getRoleIds())) {
+            pdfModel = orderService.getInvoicePdfModel(orderId);
+        } else {
+            pdfModel = orderService.getInvoicePdfModel(fofoDetails.getFofoId(), orderId);
+        }
+        FofoOrder fofoOrder = fofoOrderRepository.selectByOrderId(orderId);
+        List<InsurancePolicy> insurancePolicies = insurancePolicyRepository.selectByRetailerIdInvoiceNumber(fofoOrder.getInvoiceNumber());
+        // Step 1: Generate invoice PDF
+        ByteArrayOutputStream invoiceOutput = new ByteArrayOutputStream();
+        PdfUtils.generateAndWrite(Arrays.asList(pdfModel), invoiceOutput);
+        byte[] invoicePdf = invoiceOutput.toByteArray();
+        // Step 2: Load all policy certificate PDFs
+        List<byte[]> pdfFiles = new ArrayList<>();
+        pdfFiles.add(invoicePdf); // first add invoice
+        for (InsurancePolicy insurancePolicy : insurancePolicies) {
+            if (insurancePolicy.getProviderId() == 6) {
+                String policyNumber = insurancePolicy.getPolicyNumber();
+                String safePolicyNo = policyNumber.replace("/", "-");
+                String filePath = "/uploads/policy-certificate-" + safePolicyNo + ".pdf";
+                File file = new File(filePath);
+                if (file.exists()) {
+                    try {
+                        byte[] policyPdf = Files.readAllBytes(file.toPath());
+                        pdfFiles.add(policyPdf);
+                    } catch (IOException e) {
+                        LOGGER.error("Failed to read policy PDF: {}", filePath, e);
+                    }
+                } else {
+                    LOGGER.warn("Policy PDF not found: {}", filePath);
+                }
+            }
+        }
+        // Step 3: Merge all PDFs
+        byte[] mergedPdf;
+        try {
+            mergedPdf = PdfUtils.mergePdfFiles(pdfFiles);
+        } catch (Exception e) {
+            LOGGER.error("Error merging PDFs", e);
+            throw new ProfitMandiBusinessException("Failed to generate merged PDF", "", "");
+        }
+        // Step 4: Return merged PDF as response
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_PDF);
+        headers.setContentDispositionFormData("inline", "invoice-with-policies-" + pdfModel.getInvoiceNumber() + ".pdf");
+        headers.setContentLength(mergedPdf.length);
+        InputStreamResource resource = new InputStreamResource(new ByteArrayInputStream(mergedPdf));
+        return new ResponseEntity<>(resource, headers, HttpStatus.OK);
+    }
+   /* @RequestMapping(value = "/generateInvoice")
+    public ResponseEntity<?> generateInvoice(HttpServletRequest request, HttpServletResponse response,
                                              @RequestParam(name = ProfitMandiConstants.ORDER_ID) int orderId, @RequestParam(required = false) PrinterType printerType) throws ProfitMandiBusinessException {
         LOGGER.info("Request received at url {} with params [{}={}] ", request.getRequestURI(),
                 ProfitMandiConstants.ORDER_ID, orderId);
         LoginDetails fofoDetails = cookiesProcessor.getCookiesObject(request);
         InvoicePdfModel pdfModel = null;
         headers.setContentDispositionFormData("inline", "invoice-with-policies-" + pdfModel.getInvoiceNumber() + ".pdf");
         headers.setContentLength(mergedPdf.length);
         InputStreamResource resource = new InputStreamResource(new ByteArrayInputStream(mergedPdf));
         return new ResponseEntity<>(resource, headers, HttpStatus.OK);
-    }
+    }*/
     @RequestMapping(value = "/generateInvoices")
     public ResponseEntity<?> generateInvoice(HttpServletRequest request, HttpServletResponse response,
                                              @RequestParam LocalDateTime startDate, @RequestParam LocalDateTime endDate, @RequestParam int partnerId)
             throws ProfitMandiBusinessException {

Subversion Repositories SmartDukaan

(root)/trunk/profitmandi-fofo/src/main/java/com/spice/profitmandi/web/controller/OrderController.java – Rev 35076 → 35081