package com.nq.service.impl;
|
|
import com.nq.common.ServerResponse;
|
import com.nq.pojo.User;
|
import com.nq.service.IUserAgreementService;
|
import com.nq.utils.PropertiesUtil;
|
import org.apache.commons.lang3.StringUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.*;
|
import java.text.SimpleDateFormat;
|
import java.util.Date;
|
|
import com.itextpdf.text.*;
|
import com.itextpdf.text.pdf.*;
|
import com.itextpdf.text.pdf.BaseFont;
|
import com.itextpdf.text.pdf.parser.PdfTextExtractor;
|
import com.itextpdf.text.pdf.parser.SimpleTextExtractionStrategy;
|
|
@Service("iUserAgreementService")
|
public class UserAgreementServiceImpl implements IUserAgreementService {
|
private static final Logger log = LoggerFactory.getLogger(UserAgreementServiceImpl.class);
|
|
@Autowired
|
private com.nq.service.IUserService iUserService;
|
|
@Override
|
public ServerResponse generateAgreementPdf(HttpServletRequest request) {
|
try {
|
User user = iUserService.getCurrentUser(request);
|
if (user == null) {
|
return ServerResponse.createByErrorMsg("用户未登录");
|
}
|
|
if (StringUtils.isBlank(user.getRealName()) || StringUtils.isBlank(user.getIdCard()) || StringUtils.isBlank(user.getRegAddress())) {
|
return ServerResponse.createByErrorMsg("用户信息不完整，请先完成实名认证");
|
}
|
|
// 检查是否已生成PDF
|
String pdfDir = PropertiesUtil.getProperty("agreement.pdf.dir", System.getProperty("user.dir") + "/agreement_pdf");
|
File dir = new File(pdfDir);
|
if (!dir.exists()) {
|
dir.mkdirs();
|
}
|
|
String pdfFileName = "agreement_" + user.getId() + ".pdf";
|
File pdfFile = new File(dir, pdfFileName);
|
if (pdfFile.exists()) {
|
return ServerResponse.createBySuccess(pdfFileName);
|
}
|
|
// 获取模板文件路径
|
String templatePath = this.getClass().getResource("/templates").getPath();
|
if (templatePath.startsWith("/") && System.getProperty("os.name").toLowerCase().contains("windows")) {
|
templatePath = templatePath.substring(1);
|
}
|
templatePath = templatePath.replaceAll("%20", " ");
|
if (templatePath.contains("file:")) {
|
templatePath = templatePath.substring(templatePath.indexOf("file:") + 5);
|
}
|
if (templatePath.contains("!")) {
|
templatePath = templatePath.substring(0, templatePath.indexOf("!"));
|
}
|
|
// 使用PDF模板文件
|
File pdfTemplate = new File(templatePath, "中原证券股票分成协议（电子正式版）.pdf");
|
if (!pdfTemplate.exists()) {
|
return ServerResponse.createByErrorMsg("PDF模板文件不存在");
|
}
|
|
// 从PDF模板生成新的PDF，替换占位符
|
generatePdfFromTemplate(pdfTemplate, pdfFile, user);
|
|
return ServerResponse.createBySuccess(pdfFileName);
|
} catch (Exception e) {
|
log.error("生成用户协议PDF失败", e);
|
return ServerResponse.createByErrorMsg("生成PDF失败：" + e.getMessage());
|
}
|
}
|
|
public static void main(String[] args) {
|
generateAgreementPdf11();
|
}
|
|
public static void generateAgreementPdf11() {
|
try {
|
|
// 检查是否已生成PDF
|
String pdfDir = PropertiesUtil.getProperty("agreement.pdf.dir", System.getProperty("user.dir") + "/agreement_pdf");
|
File dir = new File(pdfDir);
|
if (!dir.exists()) {
|
dir.mkdirs();
|
}
|
|
String pdfFileName = "agreement_111.pdf";
|
File pdfFile = new File(dir, pdfFileName);
|
|
// 获取模板文件路径
|
String templatePath = "D:/work/A-stock/src/main/resources/templates";
|
if (templatePath.startsWith("/") && System.getProperty("os.name").toLowerCase().contains("windows")) {
|
templatePath = templatePath.substring(1);
|
}
|
templatePath = templatePath.replaceAll("%20", " ");
|
if (templatePath.contains("file:")) {
|
templatePath = templatePath.substring(templatePath.indexOf("file:") + 5);
|
}
|
if (templatePath.contains("!")) {
|
templatePath = templatePath.substring(0, templatePath.indexOf("!"));
|
}
|
|
// 使用PDF模板文件
|
File pdfTemplate = new File(templatePath, "中原证券股票分成协议（电子正式版）.pdf");
|
if (!pdfTemplate.exists()) {
|
System.out.println("PDF模板文件不存在");
|
}
|
|
// 从PDF模板生成新的PDF，替换占位符
|
generatePdfFromTemplate1(pdfTemplate, pdfFile);
|
|
|
} catch (Exception e) {
|
log.error("生成用户协议PDF失败", e);
|
|
}
|
}
|
public static void generatePdfFromTemplate1(File templatePdf, File outputPdf) throws Exception {
|
PdfReader reader = new PdfReader(new FileInputStream(templatePdf));
|
PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(outputPdf));
|
|
// 设置中文字体
|
BaseFont baseFont = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
|
Font font = new Font(baseFont, 12, Font.NORMAL);
|
// Font boldFont = new Font(baseFont, 12, Font.BOLD);
|
// BaseFont baseFont = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
|
// Font font = new Font(baseFont, 5, Font.BOLD);
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy年MM月dd日");
|
String currentDate = sdf.format(new Date());
|
|
// 准备替换的数据
|
String realName = "陈大雷";
|
String idCard = "433123198802456635";
|
String address = "广东省清远市三八街道明星之城18栋2单元1506";
|
|
// 获取PDF页数
|
int pageCount = reader.getNumberOfPages();
|
|
// 在每一页上查找并替换占位符
|
for (int pageNum = 1; pageNum <= pageCount; pageNum++) {
|
PdfContentByte canvas = stamper.getOverContent(pageNum);
|
Rectangle pageSize = reader.getPageSize(pageNum);
|
|
// 提取PDF文本查找占位符位置
|
String pageText = PdfTextExtractor.getTextFromPage(reader, pageNum, new SimpleTextExtractionStrategy());
|
|
// 在PDF上查找并替换占位符
|
replacePlaceholdersInPdf1(canvas, pageSize, pageText, realName, idCard, address, currentDate, font,pageNum);
|
// 添加签名
|
if(pageNum == 3){
|
// 添加签名
|
addHeaderImageToPdf(canvas, templatePdf.getParent(), pageSize);
|
}
|
}
|
|
stamper.close();
|
reader.close();
|
}
|
|
/**
|
* 在PDF上替换占位符（使用文本覆盖方式）
|
*/
|
public static void replacePlaceholdersInPdf1(PdfContentByte canvas, Rectangle pageSize, String pageText,
|
String realName, String idCard, String address,
|
String currentDate, Font font,int pageNum) throws Exception {
|
// 查找占位符在文本中的位置，然后估算在PDF中的坐标
|
// 由于PDF坐标系统复杂，这里使用固定位置替换
|
// 实际使用时需要根据PDF模板的具体布局调整坐标
|
|
float margin = 50;
|
float lineHeight = 14;
|
float startY = pageSize.getHeight() - 194;
|
|
if(pageNum == 1){
|
replacePlaceholder1(canvas, "", realName, margin + 75, startY, 0, font);
|
startY -= lineHeight;
|
|
replacePlaceholder1(canvas, "", address, margin + 75, startY-2, 0, font);
|
startY -= lineHeight;
|
|
replacePlaceholder1(canvas, "", idCard, margin + 95, startY-2, 0, font);
|
}
|
if(pageNum == 3){
|
replacePlaceholder1(canvas, "", currentDate, margin + 60, startY+32, 0, font);
|
replacePlaceholder1(canvas, "", currentDate, margin + 60, startY, 0, font);
|
}
|
|
// if (pageText.contains("${jcurrentDate}")) {
|
// replacePlaceholder1(canvas, "${jcurrentDate}", currentDate, margin + 100, startY, 150, font);
|
// startY -= lineHeight;
|
// }
|
//
|
// if (pageText.contains("${ucurrentDate}")) {
|
// replacePlaceholder1(canvas, "${ucurrentDate}", currentDate, margin + 100, startY, 300, font);
|
// }
|
}
|
|
/**
|
* 在PDF左上角添加图片
|
*/
|
public static void addHeaderImageToPdf(PdfContentByte canvas, String templatePath, Rectangle pageSize) throws Exception {
|
// 获取图片路径
|
String imagePath = templatePath + File.separator + "zczq.png";
|
File imageFile = new File(imagePath);
|
// 如果使用相对路径找不到，尝试从resources获取
|
if (!imageFile.exists()) {
|
String resourcePath = "D:/work/A-stock/src/main/resources/templates";
|
if (resourcePath.startsWith("/") && System.getProperty("os.name").toLowerCase().contains("windows")) {
|
resourcePath = resourcePath.substring(1);
|
}
|
resourcePath = resourcePath.replaceAll("%20", " ");
|
if (resourcePath.contains("file:")) {
|
resourcePath = resourcePath.substring(resourcePath.indexOf("file:") + 5);
|
}
|
if (resourcePath.contains("!")) {
|
resourcePath = resourcePath.substring(0, resourcePath.indexOf("!"));
|
}
|
imageFile = new File(resourcePath);
|
}
|
|
if (imageFile.exists()) {
|
Image image = Image.getInstance(imageFile.getAbsolutePath());
|
// 设置图片大小
|
image.scaleToFit(100, 100);
|
// 设置图片位置（左上角）
|
image.setAbsolutePosition(370, pageSize.getHeight() - 194);
|
canvas.addImage(image);
|
} else {
|
log.warn("未找到图片文件: {}", imagePath);
|
}
|
}
|
/**
|
* 替换单个占位符
|
*/
|
public static void replacePlaceholder1(PdfContentByte canvas, String placeholder, String value,
|
float x, float y, float width, Font font) {
|
// 添加白色背景覆盖占位符区域
|
canvas.saveState();
|
canvas.setColorFill(BaseColor.WHITE);
|
canvas.rectangle(x - 5, y - 15, width, 20);
|
canvas.fill();
|
canvas.restoreState();
|
|
// 添加新文本
|
ColumnText.showTextAligned(canvas, Element.ALIGN_LEFT,
|
new Phrase(value, font), x, y, 0);
|
}
|
|
|
/**
|
* 从PDF模板生成新的PDF，替换占位符
|
*/
|
private void generatePdfFromTemplate(File templatePdf, File outputPdf, User user) throws Exception {
|
PdfReader reader = new PdfReader(new FileInputStream(templatePdf));
|
PdfStamper stamper = new PdfStamper(reader, new FileOutputStream(outputPdf));
|
|
// 设置中文字体
|
BaseFont baseFont = BaseFont.createFont("STSong-Light", "UniGB-UCS2-H", BaseFont.NOT_EMBEDDED);
|
Font font = new Font(baseFont, 12, Font.NORMAL);
|
Font boldFont = new Font(baseFont, 12, Font.BOLD);
|
|
SimpleDateFormat sdf = new SimpleDateFormat("yyyy年MM月dd日");
|
String currentDate = sdf.format(new Date());
|
|
// 准备替换的数据
|
String realName = user.getRealName() != null ? user.getRealName() : "";
|
String idCard = user.getIdCard() != null ? user.getIdCard() : "";
|
String address = user.getRegAddress() != null ? user.getRegAddress() : "";
|
|
// 获取PDF页数
|
int pageCount = reader.getNumberOfPages();
|
|
// 在每一页上查找并替换占位符
|
for (int pageNum = 1; pageNum <= pageCount; pageNum++) {
|
PdfContentByte canvas = stamper.getOverContent(pageNum);
|
Rectangle pageSize = reader.getPageSize(pageNum);
|
|
// 提取PDF文本查找占位符位置
|
String pageText = PdfTextExtractor.getTextFromPage(reader, pageNum, new SimpleTextExtractionStrategy());
|
|
// 在PDF上查找并替换占位符
|
replacePlaceholdersInPdf(canvas, pageSize, pageText, realName, idCard, address, currentDate, font,pageNum);
|
}
|
|
stamper.close();
|
reader.close();
|
}
|
|
/**
|
* 在PDF上替换占位符（使用文本覆盖方式）
|
*/
|
private void replacePlaceholdersInPdf(PdfContentByte canvas, Rectangle pageSize, String pageText,
|
String realName, String idCard, String address,
|
String currentDate, Font font,int pageNum) throws Exception {
|
// 查找占位符在文本中的位置，然后估算在PDF中的坐标
|
// 由于PDF坐标系统复杂，这里使用固定位置替换
|
// 实际使用时需要根据PDF模板的具体布局调整坐标
|
float margin = 50;
|
float lineHeight = 14;
|
float startY = pageSize.getHeight() - 194;
|
|
if(pageNum == 1){
|
replacePlaceholder(canvas, "", realName, margin + 75, startY, 0, font);
|
startY -= lineHeight;
|
|
replacePlaceholder(canvas, "", address, margin + 75, startY-2, 0, font);
|
startY -= lineHeight;
|
|
replacePlaceholder(canvas, "", idCard, margin + 95, startY-2, 0, font);
|
}
|
if(pageNum == 3){
|
replacePlaceholder(canvas, "", currentDate, margin + 60, startY+32, 0, font);
|
replacePlaceholder(canvas, "", currentDate, margin + 60, startY, 0, font);
|
}
|
|
}
|
|
/**
|
* 替换单个占位符
|
*/
|
private void replacePlaceholder(PdfContentByte canvas, String placeholder, String value,
|
float x, float y, float width, Font font) {
|
// 添加白色背景覆盖占位符区域
|
canvas.saveState();
|
canvas.setColorFill(BaseColor.WHITE);
|
canvas.rectangle(x - 5, y - 15, width, 20);
|
canvas.fill();
|
canvas.restoreState();
|
|
// 添加新文本
|
ColumnText.showTextAligned(canvas, Element.ALIGN_LEFT,
|
new Phrase(value, font), x, y, 0);
|
}
|
|
@Override
|
public void viewAgreementPdf(HttpServletRequest request, HttpServletResponse response) {
|
try {
|
User user = iUserService.getCurrentUser(request);
|
if (user == null) {
|
response.sendError(HttpServletResponse.SC_UNAUTHORIZED, "用户未登录");
|
return;
|
}
|
|
String pdfDir = PropertiesUtil.getProperty("agreement.pdf.dir", System.getProperty("user.dir") + "/agreement_pdf");
|
String pdfFileName = "agreement_" + user.getId() + ".pdf";
|
File pdfFile = new File(pdfDir, pdfFileName);
|
|
if (!pdfFile.exists()) {
|
response.sendError(HttpServletResponse.SC_NOT_FOUND, "协议文件不存在，请先生成协议");
|
return;
|
}
|
|
response.setContentType("application/pdf");
|
response.setHeader("Content-Disposition", "inline; filename=" + pdfFileName);
|
|
FileInputStream fis = new FileInputStream(pdfFile);
|
OutputStream os = response.getOutputStream();
|
|
byte[] buffer = new byte[4096];
|
int bytesRead;
|
while ((bytesRead = fis.read(buffer)) != -1) {
|
os.write(buffer, 0, bytesRead);
|
}
|
|
fis.close();
|
os.flush();
|
os.close();
|
} catch (Exception e) {
|
log.error("查看用户协议PDF失败", e);
|
try {
|
response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR, "查看PDF失败");
|
} catch (IOException ex) {
|
log.error("发送错误响应失败", ex);
|
}
|
}
|
}
|
}
|
