diff --git a/src/clj_htmltopdf/core.clj b/src/clj_htmltopdf/core.clj
index 9b991a9..1aa1504 100644
--- a/src/clj_htmltopdf/core.clj
+++ b/src/clj_htmltopdf/core.clj
@@ -4,12 +4,14 @@
[hiccup.page :as h]
[clj-htmltopdf.options :as o])
(:import
+ [java.awt.geom Point2D$Float]
[java.io InputStream OutputStream PipedInputStream PipedOutputStream]
[com.openhtmltopdf DOMBuilder]
[com.openhtmltopdf.pdfboxout PdfRendererBuilder]
[com.openhtmltopdf.util XRLog]
[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.util Matrix]
[org.jsoup Jsoup]
@@ -61,13 +63,59 @@
(.run builder)))
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!
[^InputStream pdf ^OutputStream out {:keys [watermark] :as options}]
(with-open [doc (PDDocument/load pdf)]
(doseq [^PDPage page (.getPages doc)]
(let [cs (PDPageContentStream. doc page PDPageContentStream$AppendMode/APPEND true true)]
(with-open [cs cs]
- (watermark page cs))))
+ (if (map? watermark)
+ (render-watermark! page cs watermark)
+ (watermark page cs)))))
(.save doc out)
out))