Pu Zhibing
2024-10-16 c4664502dfdaffff555b532e65b51a57ac8b29c2
ruoyi-service/ruoyi-order/src/main/java/com/ruoyi/order/controller/TOrderInvoiceController.java
@@ -1,8 +1,41 @@
package com.ruoyi.order.controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import com.ruoyi.common.core.utils.DateUtils;
import com.ruoyi.common.core.utils.WebUtils;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.common.core.web.page.PageInfo;
import com.ruoyi.common.log.annotation.Log;
import com.ruoyi.common.log.enums.BusinessType;
import com.ruoyi.order.api.model.TOrderInvoice;
import com.ruoyi.order.api.query.TOrderInvoiceQuery;
import com.ruoyi.order.api.vo.TOrderInvoiceVO;
import com.ruoyi.order.dto.AddOrderInvoice;
import com.ruoyi.order.dto.GetOrderInvoiceList;
import com.ruoyi.order.dto.MyOrderInvoiceInfo;
import com.ruoyi.order.dto.OrderInvoiceList;
import com.ruoyi.order.export.OrderInvoiceExport;
import com.ruoyi.order.service.TOrderInvoiceService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.commons.compress.utils.IOUtils;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.BeanUtils;
import org.springframework.web.bind.annotation.*;
import javax.activation.DataHandler;
import javax.activation.FileDataSource;
import javax.annotation.Resource;
import javax.mail.*;
import javax.mail.internet.*;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URL;
import java.net.URLEncoder;
import java.util.*;
/**
 * <p>
@@ -12,9 +45,201 @@
 * @author xiaochen
 * @since 2024-08-07
 */
@Api(tags = "订单开票")
@RestController
@RequestMapping("/t-order-invoice")
public class TOrderInvoiceController {
   @Resource
   private TOrderInvoiceService orderInvoiceService;
   @ResponseBody
   @PostMapping("/addOrderInvoice")
   @ApiOperation(value = "添加开票申请", tags = {"小程序-充电发票"})
   public AjaxResult addOrderInvoice(@RequestBody AddOrderInvoice addOrderInvoice){
      return orderInvoiceService.addOrderInvoice(addOrderInvoice);
   }
   @PostMapping("/pageList")
   @ApiOperation(value = "查询开票分页列表", tags = {"管理后台-发票管理"})
   public AjaxResult<PageInfo<TOrderInvoiceVO>> pageList(@RequestBody TOrderInvoiceQuery query){
      return AjaxResult.success(orderInvoiceService.pageList(query));
   }
   @GetMapping("/uploadPdf")
   @ApiOperation(value = "上传发票", tags = {"管理后台-发票管理"})
   public AjaxResult<String> uploadPdf(@RequestParam("id") Long id,
                              @RequestParam("invoiceUrl") String invoiceUrl){
      TOrderInvoice orderInvoice = orderInvoiceService.getById(id);
      orderInvoice.setInvoiceUrl(invoiceUrl);
      orderInvoice.setStatus(3);
      orderInvoiceService.updateById(orderInvoice);
      // 发送邮箱
      // 收件人电子邮箱,TODO 换成自己的收件箱
      String to = orderInvoice.getMailbox();
      // 发件人电子邮箱,TODO 换成自己的发件箱
      String from = "13281306557@163.com";
      // 指定发送邮件的主机为
      String host = "smtp.163.com";
      // 获取系统属性
      Properties properties = new Properties();
      // 设置邮件服务器
      properties.setProperty("mail.smtp.host", host);
      // 邮件发送协议
      properties.setProperty("mail.transport.protocol", "smtp");
      //是否启用调试模式(启用调试模式可打印客户端与服务器交互过程时一问一答的响应消息)
      properties.setProperty("mail.debug","true");
      properties.setProperty("mail.smtp.auth", "true");
      // 获取默认session对象
      Session session = Session.getDefaultInstance(properties,new Authenticator(){
         @Override
         public PasswordAuthentication getPasswordAuthentication()
         {
            //发件人邮件用户名、授权码,换成自己的发件箱及授权码
            return new PasswordAuthentication("13281306557@163.com", "NUSdxDQqadYvVek2");
         }
      });
      try{
         // 创建默认的 MimeMessage 对象
         MimeMessage message = new MimeMessage(session);
         // Set From,设置发件人
         InternetAddress fromMail = new InternetAddress(from);
         //设置发件人名称,TODO 换成自己的发件箱
         fromMail.setPersonal(MimeUtility.encodeText("明星电力<13281306557@163.com>"));
         message.setFrom(fromMail);
         // Set To: 设置收件人
         InternetAddress toMail = new InternetAddress(to);
         // TODO 换成自己的收件箱
         InternetAddress toMail2 = new InternetAddress(to);
         //发多个邮箱
         Address[] allRecipients = {toMail, toMail2};
         message.setRecipients(Message.RecipientType.TO, allRecipients);
         // Set Subject: 邮件主体
         message.setSubject("明星电力");
         // 设置消息体
         message.setSentDate(new Date());
         // 指定为混合关系
         MimeMultipart msgMultipart = new MimeMultipart("mixed");
         message.setContent(msgMultipart);
         // 邮件信息组装
         //组装的顺序非常重要,一定要先组装文本域,再组装文件
         MimeBodyPart htmlPart = new MimeBodyPart();
         // 组装内容
         htmlPart.setContent("开票", "text/html;charset=UTF-8");
         msgMultipart.addBodyPart(htmlPart);
         // 组装附件
         MimeBodyPart filePart = new MimeBodyPart();
         String imageUrl = invoiceUrl;
         try {
            // 下载数据
            URL url = new URL(imageUrl);
            InputStream inputStream = url.openStream();
            byte[] imageBytes = IOUtils.toByteArray(inputStream);
            // 创建临时文件
            // 截取imageUrl后缀名
            String fileExtension = imageUrl.substring(imageUrl.lastIndexOf("."));
            File tempFile = File.createTempFile("tempImage", fileExtension);
            try (FileOutputStream fos = new FileOutputStream(tempFile)) {
               fos.write(imageBytes);
            }
            // 创建 FileDataSource
            FileDataSource fileDataSource = new FileDataSource(tempFile);
            System.out.println("FileDataSource created: " + fileDataSource.getName());
            // 如果需要,可以使用 DataHandler 进行进一步处理
            DataHandler dh = new DataHandler(fileDataSource);
            // 清理:删除临时文件(如果不再需要)
            tempFile.deleteOnExit(); // 可根据需要保留或删除
            filePart.setDataHandler(dh);
            // 附件区别内嵌内容的一个特点是有文件名,为防止中文乱码要编码
            filePart.setFileName(MimeUtility.encodeText(dh.getName()));
            msgMultipart.addBodyPart(filePart);
            message.saveChanges();
            //发送
            //Transport.send(message, message.getAllRecipients());
            Transport.send(message);
            System.out.println("发送成功");
         } catch (Exception e) {
            e.printStackTrace();
         }
      }catch (MessagingException | UnsupportedEncodingException mex) {
         mex.printStackTrace();
      } catch (Exception e) {
         e.printStackTrace();
      }
      return AjaxResult.success();
   }
   @PostMapping("/statusCount")
   @ApiOperation(value = "查询开票状态数量", tags = {"管理后台-发票管理"})
   public AjaxResult<Map<String,Integer>> statusCount(@RequestBody TOrderInvoiceQuery query){
      return AjaxResult.success(orderInvoiceService.statusCount(query));
   }
   /**
    * 发票管理导出
    */
   @ApiOperation(value = "发票管理导出", tags = {"管理后台-发票管理"})
   @Log(title = "发票管理导出", businessType = BusinessType.EXPORT)
   @PutMapping("/export")
   public void export(@RequestBody TOrderInvoiceQuery query)
   {
      List<TOrderInvoiceVO> list = orderInvoiceService.export(query);
      List<OrderInvoiceExport> orderInvoiceExports = new ArrayList<>();
      for (TOrderInvoiceVO orderInvoiceVO : list) {
         OrderInvoiceExport orderInvoiceExport = new OrderInvoiceExport();
         BeanUtils.copyProperties(orderInvoiceVO,orderInvoiceExport);
         if (orderInvoiceVO.getBillingTime()!=null){
            orderInvoiceExport.setBillingTime(DateUtils.localDateTimeToString(orderInvoiceVO.getBillingTime()));
         }
         orderInvoiceExports.add(orderInvoiceExport);
      }
      Workbook workbook = ExcelExportUtil.exportExcel(new ExportParams(), OrderInvoiceExport.class, orderInvoiceExports);
      HttpServletResponse response = WebUtils.response();
      response.setContentType("application/vnd.ms-excel");
      response.setCharacterEncoding("utf-8");
      ServletOutputStream outputStream = null;
      try {
         String fileName = URLEncoder.encode("发票导出.xls", "utf-8");
         response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
         response.setContentType("application/vnd.ms-excel;charset=UTF-8");
         response.setHeader("Pragma", "no-cache");
         response.setHeader("Cache-Control", "no-cache");
         outputStream = response.getOutputStream();
         workbook.write(outputStream);
      } catch (IOException e) {
         e.printStackTrace();
      } finally {
         try {
            outputStream.close();
         } catch (IOException e) {
            e.printStackTrace();
         }
      }
   }
   @ResponseBody
   @GetMapping("/getMyOrderInvoiceList")
   @ApiOperation(value = "获取开票记录列表", tags = {"小程序-充电发票"})
   public AjaxResult<List<OrderInvoiceList>> getMyOrderInvoiceList(GetOrderInvoiceList query){
      List<OrderInvoiceList> myOrderInvoiceList = orderInvoiceService.getMyOrderInvoiceList(query);
      return AjaxResult.success(myOrderInvoiceList);
   }
   @ResponseBody
   @GetMapping("/getMyOrderInvoiceInfo/{id}")
   @ApiOperation(value = "获取开票记录详情", tags = {"小程序-充电发票"})
   public AjaxResult<MyOrderInvoiceInfo> getMyOrderInvoiceInfo(@PathVariable String id){
      MyOrderInvoiceInfo myOrderInvoiceInfo = orderInvoiceService.getMyOrderInvoiceInfo(id);
      return AjaxResult.success(myOrderInvoiceInfo);
   }
}