bonus-modules/bonus-material-mall/src/main/java/com/bonus/material/order/controller/OrderController.java

@@ -164,16 +164,19 @@ public class OrderController extends BaseController {
     }
 
     @ApiOperation(value = "租赁协议(查看)")
-    @GetMapping("/leaseAgreement")
+    @PostMapping("/leaseAgreement")
-    public ResponseEntity<byte[]> getleaseAgreement(String orderId, Map<String, String> replacements) throws IOException {
+    public ResponseEntity<byte[]> getleaseAgreement(@RequestParam String orderId,
+                                                     @RequestParam("file") MultipartFile file,
+                                                     @RequestParam Map<String, String> replacements) throws IOException {
         OrderInfoDto orderInfoDto = orderService.getAgreementByOrderId(orderId);
-        BmContract bmContract = new BmContract();
+     /*   BmContract bmContract = new BmContract();
         bmContract.setStatus(1);
         List<BmContract> list = bmContractService.list(bmContract);
         String wordUrl = list.get(0).getBmFileInfoList().get(0).getFileUrl();
         InputStream inputStream = new URL(wordUrl).openStream();
 
-        XWPFDocument document = new XWPFDocument(inputStream);
+        XWPFDocument document = new XWPFDocument(inputStream);*/
+        XWPFDocument document = new XWPFDocument(file.getInputStream());
         //订单日期
         replacements.put("${orderTime}", orderInfoDto.getOrderTime().toString());
         //装备所属公司
@@ -203,19 +206,20 @@ public class OrderController extends BaseController {
         byte[] wordBytes = outputStream.toByteArray();
 
         // Convert Word to Image (simple example using BufferedImage)
-/*        BufferedImage image = new BufferedImage(800, 600, BufferedImage.TYPE_INT_RGB);
+        BufferedImage image = new BufferedImage(800, 600, BufferedImage.TYPE_INT_RGB);
         Graphics2D graphics = image.createGraphics();
         graphics.setPaint(Color.white);
         graphics.fillRect(0, 0, image.getWidth(), image.getHeight());
         graphics.setPaint(Color.black);
         graphics.drawString(new String(wordBytes), 10, 20);
-        graphics.dispose();*/
+        graphics.dispose();
+
+        ByteArrayOutputStream imageOutputStream = new ByteArrayOutputStream();
+        ImageIO.write(image, "png", imageOutputStream);
+        byte[] imageBytes = imageOutputStream.toByteArray();
 
         HttpHeaders headers = new HttpHeaders();
-        headers.setContentType(MediaType.APPLICATION_OCTET_STREAM);
+        headers.setContentType(MediaType.IMAGE_PNG);
-        headers.setContentDispositionFormData("attachment", "contract.docx");
+        return ResponseEntity.ok().headers(headers).body(imageBytes);
-        return ResponseEntity.ok()
-                .headers(headers)
-                .body(wordBytes);
     }
 }