|
12 | 12 | (:require (clojure.data.xml |
13 | 13 | [name :refer [qname-uri qname-local separate-xmlns gen-prefix *gen-prefix-counter*]] |
14 | 14 | [pu-map :as pu] |
| 15 | + [protocols :refer [AsXmlString xml-str]] |
| 16 | + [impl :refer [extend-protocol-fns b64-encode compile-if]] |
15 | 17 | event) |
16 | 18 | [clojure.string :as str]) |
17 | 19 | (:import (java.io OutputStreamWriter Writer StringWriter) |
18 | 20 | (java.nio.charset Charset) |
19 | 21 | (java.util.logging Logger Level) |
20 | | - (javax.xml.namespace NamespaceContext) |
| 22 | + (javax.xml.namespace NamespaceContext QName) |
21 | 23 | (javax.xml.stream XMLStreamWriter XMLOutputFactory) |
22 | 24 | (javax.xml.transform OutputKeys Transformer |
23 | 25 | TransformerFactory) |
24 | | - (clojure.data.xml.event StartElementEvent EndElementEvent CharsEvent CDataEvent CommentEvent))) |
| 26 | + (clojure.data.xml.event StartElementEvent EndElementEvent CharsEvent CDataEvent CommentEvent QNameEvent) |
| 27 | + (clojure.lang BigInt) |
| 28 | + (java.net URI URL) |
| 29 | + (java.util Date) |
| 30 | + (java.text DateFormat SimpleDateFormat))) |
25 | 31 |
|
26 | 32 | (def logger (Logger/getLogger "clojure.data.xml")) |
27 | 33 |
|
|
36 | 42 | {:stream-encoding (.getEncoding stream) |
37 | 43 | :declared-encoding xml-encoding})))) |
38 | 44 |
|
| 45 | +(defn- prefix-for [qn pu] |
| 46 | + (or (pu/get-prefix pu (qname-uri qn)) |
| 47 | + (throw (ex-info "Auto-generating prefixes is not supported for content-qnames. Please declare all URIs used in content qnames." |
| 48 | + {:qname qn |
| 49 | + :uri (qname-uri qn)})))) |
| 50 | + |
| 51 | +(defn- attr-str [value pu] |
| 52 | + (if (or (keyword? value) (instance? QName value)) |
| 53 | + (str (prefix-for value pu) ":" (qname-local value)) |
| 54 | + (xml-str value))) |
| 55 | + |
39 | 56 | (defn- emit-attrs [^XMLStreamWriter writer pu attrs] |
40 | 57 | (reduce-kv |
41 | 58 | (fn [_ attr value] |
42 | 59 | (let [uri (qname-uri attr) |
43 | 60 | local (qname-local attr)] |
44 | 61 | (if (str/blank? uri) |
45 | | - (.writeAttribute writer local value) |
46 | | - (.writeAttribute writer (pu/get-prefix pu uri) uri local value))) |
| 62 | + (.writeAttribute writer local (attr-str value pu)) |
| 63 | + (.writeAttribute writer (pu/get-prefix pu uri) uri local (attr-str value pu)))) |
47 | 64 | _) |
48 | 65 | nil attrs)) |
49 | 66 |
|
|
129 | 146 | CDataEvent |
130 | 147 | (emit-event [{:keys [str]} writer s] (emit-cdata str writer) s) |
131 | 148 | CommentEvent |
132 | | - (emit-event [{:keys [str]} writer s] (.writeComment writer str) s)) |
| 149 | + (emit-event [{:keys [str]} writer s] (.writeComment writer str) s) |
| 150 | + QNameEvent |
| 151 | + (emit-event [{:keys [qn]} writer pu-stack] |
| 152 | + (.writeCharacters writer (prefix-for qn (first pu-stack))) |
| 153 | + (.writeCharacters writer ":") |
| 154 | + (.writeCharacters writer (qname-local qn)) |
| 155 | + pu-stack)) |
| 156 | + |
| 157 | +(def ^:private ^ThreadLocal thread-local-utc-date-format |
| 158 | + ;; SimpleDateFormat is not thread-safe, so we use a ThreadLocal proxy for access. |
| 159 | + ;; http://bugs.sun.com/bugdatabase/view_bug.do?bug_id=4228335 |
| 160 | + (proxy [ThreadLocal] [] |
| 161 | + (initialValue [] |
| 162 | + (doto (SimpleDateFormat. "yyyy-MM-dd'T'HH:mm:ss.SSS-00:00") |
| 163 | + ;; RFC3339 says to use -00:00 when the timezone is unknown (+00:00 implies a known GMT) |
| 164 | + (.setTimeZone (java.util.TimeZone/getTimeZone "GMT")))))) |
| 165 | + |
| 166 | +(extend-protocol-fns |
| 167 | + AsXmlString |
| 168 | + String {:xml-str identity} |
| 169 | + (Boolean Byte Character Short Integer Long Float Double |
| 170 | + BigInteger BigDecimal BigInt URI URL nil) {:xml-str str}) |
| 171 | + |
| 172 | +(extend-protocol AsXmlString |
| 173 | + (Class/forName "[B") |
| 174 | + (xml-str [ba] (b64-encode ba)) |
| 175 | + Date |
| 176 | + (xml-str [d] (let [^DateFormat utc-format (.get thread-local-utc-date-format)] |
| 177 | + (.format utc-format d))) |
| 178 | + clojure.lang.Ratio |
| 179 | + (xml-str [r] (str (.decimalValue r)))) |
| 180 | + |
| 181 | +(compile-if |
| 182 | + (Class/forName "java.time.Instant") |
| 183 | + (extend-protocol AsXmlString |
| 184 | + java.time.Instant |
| 185 | + (xml-str [i] (xml-str (Date/from i)))) |
| 186 | + nil) |
133 | 187 |
|
134 | 188 | ;; Writers |
135 | 189 |
|
|
0 commit comments