package cn.iocoder.yudao.module.crm.service.quotation;
|
|
import cn.iocoder.yudao.module.crm.controller.admin.quotation.vo.CrmSaleQuotationRespVO;
|
import org.apache.poi.ss.usermodel.Cell;
|
import org.apache.poi.ss.usermodel.Workbook;
|
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
|
import org.springframework.core.io.ClassPathResource;
|
|
import java.io.ByteArrayOutputStream;
|
import java.io.IOException;
|
import java.math.BigDecimal;
|
import java.time.format.DateTimeFormatter;
|
|
/** 使用客户提供的 Excel 模板生成单张销售报价单。 */
|
public final class CrmSaleQuotationExcelRenderer {
|
|
private static final String TEMPLATE = "templates/quotation/sale-quotation.xlsx";
|
private static final String DEFAULT_COMPANY = "苏珩线缆南通有限公司";
|
|
private CrmSaleQuotationExcelRenderer() {
|
}
|
|
public static byte[] render(CrmSaleQuotationRespVO quotation, String companyName,
|
String address, String contactName, String contactPhone) throws IOException {
|
if (quotation == null || quotation.getItems() == null || quotation.getItems().isEmpty()) {
|
throw new IllegalArgumentException("报价单不存在或没有产品明细，无法生成报价文件");
|
}
|
String company = companyName == null || companyName.isBlank() ? DEFAULT_COMPANY : companyName;
|
try (var input = new ClassPathResource(TEMPLATE).getInputStream();
|
var workbook = new XSSFWorkbook(input);
|
var output = new ByteArrayOutputStream()) {
|
var sheet = workbook.getSheetAt(0);
|
int extraRows = quotation.getItems().size() - 1;
|
if (extraRows > 0) {
|
sheet.shiftRows(11, sheet.getLastRowNum(), extraRows, true, false);
|
}
|
var templateRow = sheet.getRow(10);
|
for (int i = 0; i < quotation.getItems().size(); i++) {
|
var row = i == 0 ? templateRow : sheet.createRow(10 + i);
|
if (i > 0) {
|
row.setHeight(templateRow.getHeight());
|
for (int col = 0; col < 8; col++) {
|
row.createCell(col).setCellStyle(templateRow.getCell(col).getCellStyle());
|
}
|
}
|
var item = quotation.getItems().get(i);
|
string(row.getCell(0), item.getItemName() == null || item.getItemName().isBlank()
|
? item.getItemCode() : item.getItemName());
|
string(row.getCell(1), item.getItemSpecification());
|
string(row.getCell(2), item.getItemUnitName());
|
number(row.getCell(3), item.getCount());
|
number(row.getCell(4), item.getQuotationPrice());
|
BigDecimal amount = item.getTotalPrice();
|
if (amount == null && item.getCount() != null && item.getQuotationPrice() != null) {
|
amount = item.getCount().multiply(item.getQuotationPrice());
|
}
|
number(row.getCell(5), amount);
|
string(row.getCell(6), null); // 调价没有对应的系统字段
|
string(row.getCell(7), item.getRemark());
|
}
|
string(sheet.getRow(0).getCell(0), company);
|
string(sheet.getRow(2).getCell(0), "编号：" + text(quotation.getNo()));
|
string(sheet.getRow(4).getCell(1), quotation.getCustomerName());
|
string(sheet.getRow(5).getCell(1), address);
|
string(sheet.getRow(6).getCell(1), contactName);
|
string(sheet.getRow(7).getCell(1), contactPhone);
|
string(sheet.getRow(13 + extraRows).getCell(6), "报价单位：" + company);
|
string(sheet.getRow(14 + extraRows).getCell(6), "联 系 人：" + text(quotation.getOwnerUserName()));
|
string(sheet.getRow(15 + extraRows).getCell(6), "报价日期：" + (quotation.getQuotationTime() == null
|
? "" : quotation.getQuotationTime().format(DateTimeFormatter.ofPattern("yyyy年MM月dd日"))));
|
workbook.setPrintArea(0, 0, 7, 0, 17 + extraRows);
|
|
int picture = workbook.addPicture(CrmSaleQuotationDocumentRenderer.renderSeal(company), Workbook.PICTURE_TYPE_PNG);
|
var anchor = workbook.getCreationHelper().createClientAnchor();
|
anchor.setCol1(3);
|
anchor.setCol2(5);
|
anchor.setRow1(13 + extraRows);
|
anchor.setRow2(17 + extraRows);
|
sheet.createDrawingPatriarch().createPicture(anchor, picture);
|
workbook.write(output);
|
return output.toByteArray();
|
}
|
}
|
|
private static void string(Cell cell, String value) {
|
cell.setCellValue(text(value));
|
}
|
|
private static void number(Cell cell, BigDecimal value) {
|
if (value == null) {
|
cell.setBlank();
|
} else {
|
cell.setCellValue(value.doubleValue());
|
}
|
}
|
|
private static String text(String value) {
|
return value == null ? "" : value;
|
}
|
}
|
