Java로 PDF 그리기

//Apache PDFBox implementation 'org.apache.pdfbox:pdfbox:2.0.29'

@GetMapping(value = "/{resumeId}/pdf", produces = MediaType.APPLICATION_PDF_VALUE) public ResponseEntity<ByteArrayResource> createPDF(HttpServletRequest request, @PathVariable Long resumeId) { PdfDTO pdf = resumeService.createResumePDF(request, resumeId); HttpHeaders headers = new HttpHeaders(); headers.add(HttpHeaders.CONTENT_DISPOSITION, "inline; filename=" + URLEncoder.encode(pdf.getFileName(), StandardCharsets.UTF_8)); return new ResponseEntity<>(pdf.getPdf(), headers, HttpStatus.OK); }

@Getter @Setter @NoArgsConstructor public class PdfDTO { private String fileName; private ByteArrayResource pdf; @Builder public PdfDTO(String fileName, ByteArrayResource pdf) { this.fileName = fileName; this.pdf = pdf; } }

public PdfDTO createResumePDF(HttpServletRequest request, Long resumeId) { String accessToken = authenticationProvider.resolveAccessToken(request); Long userId = authenticationProvider.getId(accessToken); User user = userRepository.findById(userId).orElseThrow(() -> new CustomException(ErrorCode.BLANK_MEMBER)); Resume resume = resumeRepository.findResumeById(resumeId).orElseThrow(() -> new CustomException(ErrorCode.BLANK_RESUME)); if (!resume.getUser().getId().equals(user.getId())) { throw new CustomException(ErrorCode.ACCESS_DENIED); } if (resume.getUser().getUserProfile() == null && (resume.getEducations() == null || resume.getEducations().isEmpty())) { throw new CustomException(ErrorCode.BLANK_REQUIRED); } byte[] bytes; try { PDDocument document = pdfUtil.createPDFDocument(); pdfUtil.setDocumentInfo(document, resume); pdfUtil.initFonts(document); int height = (int) (document.getPage(0).getMediaBox().getHeight() - 84); height = pdfUtil.setProfile(document, resume, height); height = pdfUtil.setEducations(document, resume, height); height = pdfUtil.setCareers(document, resume, height); height = pdfUtil.setCertifications(document, resume, height); height = pdfUtil.setFluencies(document, resume, height); height = pdfUtil.setActivities(document, resume, height); pdfUtil.setPageNumber(document); pdfUtil.setWaterMark(document); bytes = pdfUtil.pdfToBytes(document); document.close(); } catch (IOException e) { throw new CustomException(ErrorCode.IO_ERROR); } return PdfDTO.builder() .fileName("[고초대졸닷컴_이력서]_" + resume.getUser().getUserProfile().getName() + "_" + LocalDate.now().format(DateTimeFormatter.ofPattern("yyyyMMdd")) + ".pdf") .pdf(new ByteArrayResource(bytes)) .build(); }

public PDDocument createPDFDocument() { PDDocument document = new PDDocument(); PDPage page = new PDPage(PDRectangle.A4); document.addPage(page); return document; }

private PDPage createPDFPage() { PDPage page = new PDPage(PDRectangle.A4); return page; }

private boolean needNewPage(PDDocument document, int height, int needHeight) { if (height - needHeight < 84) { PDPage newPage = createPDFPage(); document.addPage(newPage); return true; } return false; }

public void initFonts(PDDocument document) throws IOException { Map<String, PDType0Font> map = new HashMap<>(); ClassPathResource extraBoldFontPath = new ClassPathResource("fonts/Pretendard-ExtraBold.ttf"); ClassPathResource mediumFontPath = new ClassPathResource("fonts/Pretendard-Medium.ttf"); ClassPathResource regularFontPath = new ClassPathResource("fonts/Pretendard-Regular.ttf"); map.put("extraBold", PDType0Font.load(document, extraBoldFontPath.getInputStream())); map.put("medium", PDType0Font.load(document, mediumFontPath.getInputStream())); map.put("regular", PDType0Font.load(document, regularFontPath.getInputStream())); fonts = map; }

private int setText(PDDocument document, PDType0Font font, int fontSize, int r, int g, int b, int x, int y, String text, int maxLength) throws IOException { PDPageContentStream contentStream = new PDPageContentStream(document, document.getPage(document.getPages().getCount() - 1), PDPageContentStream.AppendMode.APPEND, true); if (needNewPage(document, y, 22)) { contentStream.close(); contentStream = new PDPageContentStream(document, document.getPage(document.getPages().getCount() - 1), PDPageContentStream.AppendMode.APPEND, true); y = (int) (document.getPage(document.getPages().getCount() - 1).getMediaBox().getHeight() - 84); } contentStream.setFont(font, fontSize); contentStream.setNonStrokingColor(r, g, b); contentStream.beginText(); contentStream.newLineAtOffset(x, y); if (maxLength != 0 && text.length() > maxLength) { for (int i = 0; i < text.length(); i += maxLength) { int endIndex = Math.min(i + maxLength, text.length()); contentStream.showText(text.substring(i, endIndex)); contentStream.newLineAtOffset(0, -18); y = i != maxLength ? y - 18 : y; } } else { contentStream.showText(text); } contentStream.endText(); contentStream.close(); return y; }

private void setImage(PDDocument document, int x, int y, int width, int height, BufferedImage bufferedImage) throws IOException { PDPageContentStream contentStream = new PDPageContentStream(document, document.getPage(document.getPages().getCount() - 1), PDPageContentStream.AppendMode.APPEND, true); if (needNewPage(document, y, height)) { contentStream.close(); contentStream = new PDPageContentStream(document, document.getPage(document.getPages().getCount() - 1), PDPageContentStream.AppendMode.APPEND, true); y = (int) (document.getPage(document.getPages().getCount() - 1).getMediaBox().getHeight() - 84); } PDImageXObject image = LosslessFactory.createFromImage(document, bufferedImage); contentStream.drawImage(image, x, y, width, height); contentStream.close(); }

private BufferedImage getImageByURL(String imagePath) throws IOException { InputStream inputStream = new ByteArrayInputStream(s3Util.getObject(imagePath.substring(imagePath.indexOf("/", 8) + 1)));; BufferedImage image = ImageIO.read(inputStream); inputStream.close(); return image; }

private BufferedImage getImageByResource(ClassPathResource resource) throws IOException { InputStream inputStream = resource.getInputStream(); BufferedImage image = ImageIO.read(inputStream); inputStream.close(); return image; }

private int setHorizontalLine(PDDocument document, int lineWidth, int r, int g, int b, int startX, int startY, int endX, int endY) throws IOException { PDPageContentStream contentStream = new PDPageContentStream(document, document.getPage(document.getPages().getCount() - 1), PDPageContentStream.AppendMode.APPEND, true); if (needNewPage(document, startY, 1)) { contentStream.close(); contentStream = new PDPageContentStream(document, document.getPage(document.getPages().getCount() - 1), PDPageContentStream.AppendMode.APPEND, true); startY = (int) (document.getPage(document.getPages().getCount() - 1).getMediaBox().getHeight() - 84); endY = (int) (document.getPage(document.getPages().getCount() - 1).getMediaBox().getHeight() - 84); } contentStream.setLineWidth(lineWidth); contentStream.setStrokingColor(r, g, b); contentStream.moveTo(startX, startY); contentStream.lineTo(endX, endY); contentStream.stroke(); contentStream.close(); return startY - 34; }

private int calculatePixel(String fontName, int fontSize, String text) throws IOException { PDType0Font font = fonts.get(fontName); float pixel = font.getStringWidth(text); return (int) ((pixel / 1000 * fontSize) + 8); }

public byte[] pdfToBytes(PDDocument document) throws IOException { ByteArrayOutputStream outputStream = new ByteArrayOutputStream(); document.save(outputStream); document.close(); return outputStream.toByteArray(); }

public int setFluencies(PDDocument document, Resume resume, int height) throws IOException { if (resume.getFluencies() == null || resume.getFluencies().isEmpty()) { return height; } PDPage page = document.getPage(document.getPages().getCount() - 1); int width = (int) page.getMediaBox().getWidth(); resume.getFluencies().sort((c1, c2) -> c2.getAcquisitionDate().compareTo(c1.getAcquisitionDate())); height = setText(document, fonts.get("extraBold"), 14, blueGray400[0], blueGray400[1], blueGray400[2], margin, height, "어학", 0); for (int i = 0; i < resume.getFluencies().size(); i++) { height = setText(document, fonts.get("medium"), 14, black[0], black[1], black[2], margin + padding, height, resume.getFluencies().get(i).getName(), 0); height = setText(document, fonts.get("regular"), 12, black[0], black[1], black[2], margin + padding + calculatePixel("medium", 14, resume.getFluencies().get(i).getName()), height, resume.getFluencies().get(i).getLanguageType().getName(), 0); height = setText(document, fonts.get("regular"), 12, black[0], black[1], black[2], width - margin - calculatePixel("regular", 12, resume.getFluencies().get(i).getAcquisitionDate().format(DateTimeFormatter.ofPattern("yyyy.MM"))), height, resume.getFluencies().get(i).getAcquisitionDate().format(DateTimeFormatter.ofPattern("yyyy.MM")), 0); height -= 26; height = setText(document, fonts.get("regular"), 12, black[0], black[1], black[2], margin + padding, height, resume.getFluencies().get(i).getGrade(), 0); if (i != resume.getFluencies().size() - 1) { height -= 21; height = setHorizontalLine(document, 1, gray200[0], gray200[1], gray200[2], margin + padding, height, width - margin, height); } } height -= 21; height = setHorizontalLine(document, 1, gray200[0], gray200[1], gray200[2], margin, height, width - margin, height); return height; }

@Slf4j @Component public class PDFUtil { private int margin = 52; private int padding = 67; private Map<String, PDType0Font> fonts; private int[] black = {28, 28, 28}; private int[] blueGray400 = {107, 118, 132}; private int[] gray200 = {226, 228, 230}; private int[] gray300 = {155, 160, 169}; private int[] blue200 = {232, 243, 255}; private int[] blue300 = {41, 97, 205}; private ClassPathResource logo = new ClassPathResource("images/logo.png"); private ClassPathResource uturn = new ClassPathResource("images/uturn.png");

private BufferedImage getImageByResource(ClassPathResource resource) throws IOException { InputStream inputStream = resource.getInputStream(); BufferedImage image = ImageIO.read(inputStream); inputStream.close(); return image; }

if(file.getSize() > 1024 * 1024 * 5) { throw new CustomException(ErrorCode.EXCEED_SIZE_FILE); } try { try(InputStream inputStream = new ByteArrayInputStream(file.getBytes())) { BufferedImage buff = ImageIO.read(inputStream); float width = buff.getWidth(); float height = buff.getHeight(); if (width > 450) { int resizedWidth = 450; int resizedHeight = (int) (height / (width / 450)); file = imageUtil.imageResize(file, randomName, extension, resizedWidth, resizedHeight); } } } catch (IOException e) { throw new CustomException(ErrorCode.IO_ERROR); }

public MultipartFile imageResize(MultipartFile file, String fileName, String fileFormat, int width, int height) { ByteArrayOutputStream outputStream = null; try { int orientation = getOrientation(file); //MultipartFile to BufferedImage BufferedImage originalImage = ImageIO.read(file.getInputStream()); //Rotate Image if (orientation != 1) { originalImage = rotateImage(originalImage, orientation); } //Resize Image BufferedImage resizedImage = Scalr.resize(originalImage, Scalr.Method.AUTOMATIC, Scalr.Mode.FIT_TO_WIDTH, width, height); outputStream = new ByteArrayOutputStream(); ImageIO.write(resizedImage, fileFormat, outputStream); outputStream.flush(); } catch (Exception e) { throw new CustomException(ErrorCode.IO_ERROR); } return new MockMultipartFile(fileName, fileName, file.getContentType(), outputStream.toByteArray()); }