package com.ruoyi.order.controller;


import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import com.ruoyi.account.api.dto.UploadPdfDTO;
import com.ruoyi.account.api.feignClient.AppUserClient;
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.common.security.annotation.Logical;
import com.ruoyi.common.security.annotation.RequiresPermissions;
import com.ruoyi.common.security.service.TokenService;
import com.ruoyi.common.security.utils.SecurityUtils;
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.*;
import com.ruoyi.order.export.OrderInvoiceExport;
import com.ruoyi.order.export.InvoiceApplicationWorkbookBuilder;
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.time.LocalDateTime;
import java.util.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author xiaochen
 * @since 2024-08-07
 */
@Api(tags = "订单开票")
@RestController
@RequestMapping("/t-order-invoice")
public class TOrderInvoiceController {

	@Resource
	private TOrderInvoiceService orderInvoiceService;
	
	@Resource
	private TokenService tokenService;
	@Resource
	private AppUserClient appUserClient;

	
	
	@ResponseBody
	@PostMapping("/addOrderInvoice")
	@ApiOperation(value = "添加开票申请", tags = {"小程序-充电发票"})
	public AjaxResult addOrderInvoice(@RequestBody AddOrderInvoice addOrderInvoice){
		return orderInvoiceService.addOrderInvoice(addOrderInvoice);
	}
	
	
	@RequiresPermissions(value = {"/invoiceManagement"}, logical = Logical.OR)
	@PostMapping("/pageList")
	@ApiOperation(value = "查询开票分页列表", tags = {"管理后台-发票管理"})
	public AjaxResult<PageInfo<TOrderInvoiceVO>> pageList(@RequestBody TOrderInvoiceQuery query){
		return AjaxResult.success(orderInvoiceService.pageList(query));
	}

	@RequiresPermissions(value = {"/invoiceManagement"}, logical = Logical.OR)
	@PostMapping("/invoicingCompanyOptions")
	@ApiOperation(value = "查询开票公司选项", tags = {"管理后台-发票管理"})
	public AjaxResult<List<String>> invoicingCompanyOptions() {
		return AjaxResult.success(orderInvoiceService.invoicingCompanyOptions());
	}


	
	@RequiresPermissions(value = {"/invoiceManagement/accept_hear_case"}, logical = Logical.OR)
	@PostMapping("/uploadPdf")
	@ApiOperation(value = "上传发票", tags = {"管理后台-发票管理"})
	public AjaxResult<String> uploadPdf(@RequestBody UploadPdfDTO dto){
		TOrderInvoice orderInvoice = orderInvoiceService.getById(dto.getId());
		orderInvoice.setInvoiceUrl(dto.getInvoiceUrl());
		orderInvoice.setStatus(3);
		orderInvoice.setBillingTime(LocalDateTime.now());
		Long userId = SecurityUtils.getUserId();
		orderInvoice.setBillingUserId(userId == null ? null : userId.intValue());
		orderInvoiceService.updateById(orderInvoice);
		dto.setMailBox(orderInvoice.getMailbox());
		appUserClient.uploadPdf(dto);
		return AjaxResult.success();
	}

	@RequiresPermissions(value = {"/invoiceManagement/accept_hear_case"}, logical = Logical.OR)
	@PostMapping("/cancel")
	@ApiOperation(value = "取消开票申请", tags = {"管理后台-发票管理"})
	public AjaxResult<String> cancel(@RequestBody TOrderInvoice dto){
		TOrderInvoice orderInvoice = orderInvoiceService.getById(dto.getId());
		if (orderInvoice == null) {
			return AjaxResult.error("开票申请不存在");
		}
		if (Integer.valueOf(3).equals(orderInvoice.getStatus())) {
			return AjaxResult.error("已完成开票不能取消");
		}
		orderInvoice.setStatus(4);
		orderInvoice.setCancelTime(LocalDateTime.now());
		orderInvoiceService.updateById(orderInvoice);
		return AjaxResult.success();
	}
	
	
	@RequiresPermissions(value = {"/invoiceManagement"}, logical = Logical.OR)
	@PostMapping("/statusCount")
	@ApiOperation(value = "查询开票状态数量", tags = {"管理后台-发票管理"})
	public AjaxResult<Map<String,Integer>> statusCount(@RequestBody TOrderInvoiceQuery query){
		return AjaxResult.success(orderInvoiceService.statusCount(query));
	}

	/**
	 * 发票管理导出
	 */
	@RequiresPermissions(value = {"/invoiceManagement/export"}, logical = Logical.OR)
	@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.getCreateTime()!=null){
				orderInvoiceExport.setApplyTime(DateUtils.localDateTimeToString(orderInvoiceVO.getCreateTime()));
			}
			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();
			}
		}
	}

	/**
	 * 下载开票申请，单个返回 Excel，多个返回 zip。
	 */
	@RequiresPermissions(value = {"/invoiceManagement/export"}, logical = Logical.OR)
	@ApiOperation(value = "下载开票申请", tags = {"管理后台-发票管理"})
	@Log(title = "下载开票申请", businessType = BusinessType.EXPORT)
	@PutMapping("/downloadApplication")
	public void downloadApplication(@RequestBody TOrderInvoiceQuery query) {
		List<TOrderInvoiceVO> list = orderInvoiceService.pageList(query).getRecords();
		HttpServletResponse response = WebUtils.response();
		try {
			if (list.isEmpty()) {
				response.sendError(HttpServletResponse.SC_NOT_FOUND, "暂无可下载的开票申请");
				return;
			}
			if (list.size() == 1) {
				writeInvoiceWorkbook(response, list.get(0), buildInvoiceFileName(list.get(0)));
				return;
			}
			response.setCharacterEncoding("utf-8");
			response.setContentType("application/zip");
			response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("开票申请.zip", "utf-8"));
			try (ZipOutputStream zipOutputStream = new ZipOutputStream(response.getOutputStream())) {
				for (TOrderInvoiceVO invoice : list) {
					Workbook workbook = buildInvoiceWorkbook(invoice);
					ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
					workbook.write(outputStream);
					workbook.close();
					zipOutputStream.putNextEntry(new ZipEntry(buildInvoiceFileName(invoice)));
					zipOutputStream.write(outputStream.toByteArray());
					zipOutputStream.closeEntry();
				}
			}
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	private void writeInvoiceWorkbook(HttpServletResponse response, TOrderInvoiceVO invoice, String fileName) throws IOException {
		response.setContentType("application/vnd.ms-excel;charset=UTF-8");
		response.setCharacterEncoding("utf-8");
		response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "utf-8"));
		response.setHeader("Pragma", "no-cache");
		response.setHeader("Cache-Control", "no-cache");
		Workbook workbook = buildInvoiceWorkbook(invoice);
		try (ServletOutputStream outputStream = response.getOutputStream()) {
			workbook.write(outputStream);
		}
		workbook.close();
	}

	private Workbook buildInvoiceWorkbook(TOrderInvoiceVO invoice) {
		return InvoiceApplicationWorkbookBuilder.build(invoice);
	}

	private String buildInvoiceFileName(TOrderInvoiceVO invoice) {
		String title = invoice.getName() == null ? "开票申请" : invoice.getName();
		String phone = invoice.getUserPhone() == null ? "" : invoice.getUserPhone();
		return title + phone + ".xls";
	}
	
	@ResponseBody
	@GetMapping("/getMyOrderInvoiceList")
	@ApiOperation(value = "获取开票记录列表", tags = {"小程序-充电发票"})
	public AjaxResult<List<OrderInvoiceList>> getMyOrderInvoiceList(GetOrderInvoiceList query){
		List<OrderInvoiceList> myOrderInvoiceList = orderInvoiceService.getMyOrderInvoiceList(query);
		return AjaxResult.success(myOrderInvoiceList);
	}
	@ResponseBody
	@PostMapping("/cancelMyOrderInvoice/{id}")
	@ApiOperation(value = "取消我的开票申请", tags = {"小程序-充电发票"})
	public AjaxResult<String> cancelMyOrderInvoice(@PathVariable Long id){
		TOrderInvoice orderInvoice = orderInvoiceService.getById(id);
		if (orderInvoice == null) {
			return AjaxResult.error("开票申请不存在");
		}
		Long userId = tokenService.getLoginUserApplet().getUserId();
		if(!orderInvoice.getAppUserId().equals(userId)){
			return AjaxResult.error("权限不足");
		}
		if (!Integer.valueOf(1).equals(orderInvoice.getStatus())) {
			return AjaxResult.error("当前开票申请不能取消");
		}
		orderInvoice.setStatus(4);
		orderInvoice.setCancelTime(LocalDateTime.now());
		orderInvoiceService.updateById(orderInvoice);
		return AjaxResult.success();
	}
	

	
	
	
	@ResponseBody
	@GetMapping("/getMyOrderInvoiceInfo/{id}")
	@ApiOperation(value = "获取开票记录详情", tags = {"小程序-充电发票"})
	public AjaxResult<MyOrderInvoiceInfo> getMyOrderInvoiceInfo(@PathVariable String id){
		TOrderInvoice orderInvoice = orderInvoiceService.getById(id);
		if (orderInvoice == null) {
			return AjaxResult.error("开票申请不存在");
		}
		Long userId = tokenService.getLoginUserApplet().getUserId();
		if(!orderInvoice.getAppUserId().equals(userId)){
			return AjaxResult.error("权限不足");
		}
		MyOrderInvoiceInfo myOrderInvoiceInfo = orderInvoiceService.getMyOrderInvoiceInfo(id);
		return AjaxResult.success(myOrderInvoiceInfo);
	}
}