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

initial watermark support

Browse source
This commit is contained in:
Gered 2017-05-04 21:02:31 -04:00
parent ca7bd70959
commit d883f4fd72
1 changed files with 31 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
src/clj_htmltopdf/core.clj
View file

@ -4,10 +4,14 @@
[hiccup.page :as h] [hiccup.page :as h]
[clj-htmltopdf.options :as o]) [clj-htmltopdf.options :as o])
(:import (:import
[java.io OutputStream] [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.font PDType1Font]
[org.apache.pdfbox.pdmodel.graphics.state PDExtendedGraphicsState]
[org.apache.pdfbox.util Matrix]
[org.jsoup Jsoup] [org.jsoup Jsoup]
[org.jsoup.nodes Document])) [org.jsoup.nodes Document]))
@ -46,18 +50,36 @@
html-doc)) html-doc))
(defn write-pdf! (defn write-pdf!
[^Document html-doc ^String base-uri out] [^Document html-doc ^String base-uri]
(let [builder (PdfRendererBuilder.) (let [builder (PdfRendererBuilder.)]
os (->output-stream out)]
(.withW3cDocument builder (DOMBuilder/jsoup2DOM html-doc) base-uri) (.withW3cDocument builder (DOMBuilder/jsoup2DOM html-doc) base-uri)
(with-open [os os] (let [piped-in (PipedInputStream.)
(.toStream builder os) piped-out (PipedOutputStream. piped-in)]
(.run builder) (future
os))) (with-open [os piped-out]
(.toStream builder os)
(.run builder)))
piped-in)))
(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))))
(.save doc out)
out))
(defn ->pdf (defn ->pdf
[in out & [options]] [in out & [options]]
(let [options (o/get-final-options options) (let [options (o/get-final-options options)
html-doc (prepare-html in options)] html-doc (prepare-html in options)]
(configure-logging! options) (configure-logging! options)
(write-pdf! html-doc (o/->base-uri options) out))) (let [result (write-pdf! html-doc (o/->base-uri options))
out (->output-stream out)]
(if (:watermark options)
(write-watermark! result out options)
(with-open [os out]
(io/copy result os)
os)))))