gered/clj-htmltopdf
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

add configuration-based text watermark support for simpler use-cases

Browse source
This commit is contained in:
Gered 2017-05-05 20:49:21 -04:00
parent d883f4fd72
commit aa4a9fa302
1 changed files with 50 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
src/clj_htmltopdf/core.clj
View file

@ -4,12 +4,14 @@
[hiccup.page :as h] [hiccup.page :as h]
[clj-htmltopdf.options :as o]) [clj-htmltopdf.options :as o])
(:import (:import
[java.awt.geom Point2D$Float]
[java.io InputStream OutputStream PipedInputStream PipedOutputStream] [java.io InputStream OutputStream PipedInputStream PipedOutputStream]
[com.openhtmltopdf DOMBuilder] [com.openhtmltopdf DOMBuilder]
[com.openhtmltopdf.pdfboxout PdfRendererBuilder] [com.openhtmltopdf.pdfboxout PdfRendererBuilder]
[com.openhtmltopdf.util XRLog] [com.openhtmltopdf.util XRLog]
[org.apache.pdfbox.pdmodel PDDocument PDPage PDPageContentStream PDPageContentStream$AppendMode] [org.apache.pdfbox.pdmodel PDDocument PDPage PDPageContentStream PDPageContentStream$AppendMode]
[org.apache.pdfbox.pdmodel.font PDType1Font] [org.apache.pdfbox.pdmodel.common PDRectangle]
[org.apache.pdfbox.pdmodel.font PDType1Font PDFont]
[org.apache.pdfbox.pdmodel.graphics.state PDExtendedGraphicsState] [org.apache.pdfbox.pdmodel.graphics.state PDExtendedGraphicsState]
[org.apache.pdfbox.util Matrix] [org.apache.pdfbox.util Matrix]
[org.jsoup Jsoup] [org.jsoup Jsoup]
@ -61,13 +63,59 @@
(.run builder))) (.run builder)))
piped-in))) piped-in)))
(def watermark-fonts
{"times-roman" PDType1Font/TIMES_ROMAN
"times-bold" PDType1Font/TIMES_BOLD
"times-italic" PDType1Font/TIMES_ITALIC
"times-bolditalic" PDType1Font/TIMES_BOLD_ITALIC
"helvetica" PDType1Font/HELVETICA
"helvetica-bold" PDType1Font/HELVETICA_BOLD
"helvetica-oblique" PDType1Font/HELVETICA_OBLIQUE
"helvetica-boldoblique" PDType1Font/HELVETICA_BOLD_OBLIQUE
"courier" PDType1Font/COURIER
"courier-bold" PDType1Font/COURIER_BOLD
"courier-oblique" PDType1Font/COURIER_OBLIQUE
"courier-boldoblique" PDType1Font/COURIER_BOLD_OBLIQUE})
(defn render-watermark!
[^PDPage page ^PDPageContentStream cs options]
(let [font (or (get watermark-fonts (:font options))
(get watermark-fonts "helvetica-bold"))
font-size (float (or (:font-size options) 36.0))
font-color (or (:color options) [0 0 0])
text (:text options)
text-width (/ (* (.getStringWidth font text) font-size) 1000.0)
text-height (/ (* (.getHeight (.getFontBoundingBox (.getFontDescriptor ^PDFont font))) font-size) 1000.0)
rotation (float (or (:rotation options) 0))
page-size (.getMediaBox page)
page-width (.getWidth page-size)
page-height (.getHeight page-size)
x (if (= :center (:x options)) (/ page-width 2) (float (:x options)))
y (if (= :center (:y options)) (/ page-height 2) (float (:y options)))
transform (doto (Matrix.)
(.translate x y)
(.rotate (Math/toRadians rotation))
(.translate (- (/ text-width 2)) (- (/ text-height 2))))]
(when (:opacity options)
(let [r0 (PDExtendedGraphicsState.)]
(.setNonStrokingAlphaConstant r0 (float (:opacity options)))
(.setGraphicsStateParameters cs r0)))
(.beginText cs)
(.setFont cs font font-size)
(.setNonStrokingColor cs (int (nth font-color 0)) (int (nth font-color 1)) (int (nth font-color 2)))
(.setTextMatrix cs transform)
(.showText cs text)
(.endText cs)))
(defn write-watermark! (defn write-watermark!
[^InputStream pdf ^OutputStream out {:keys [watermark] :as options}] [^InputStream pdf ^OutputStream out {:keys [watermark] :as options}]
(with-open [doc (PDDocument/load pdf)] (with-open [doc (PDDocument/load pdf)]
(doseq [^PDPage page (.getPages doc)] (doseq [^PDPage page (.getPages doc)]
(let [cs (PDPageContentStream. doc page PDPageContentStream$AppendMode/APPEND true true)] (let [cs (PDPageContentStream. doc page PDPageContentStream$AppendMode/APPEND true true)]
(with-open [cs cs] (with-open [cs cs]
(watermark page cs)))) (if (map? watermark)
(render-watermark! page cs watermark)
(watermark page cs)))))
(.save doc out) (.save doc out)
out)) out))