From a4735ff01cdb3ac784820b0510bf07a33a561f1b Mon Sep 17 00:00:00 2001 From: Alan Pearce Date: Sun, 19 Oct 2014 13:20:46 +0100 Subject: Allow base-url to be overridden by parse argument --- src/microformats/parser.clj | 22 +++++++++++++++++++--- test/microformats/parser_expectations.clj | 6 ++++++ 2 files changed, 25 insertions(+), 3 deletions(-) diff --git a/src/microformats/parser.clj b/src/microformats/parser.clj index 21fcc26..9137eaa 100644 --- a/src/microformats/parser.clj +++ b/src/microformats/parser.clj @@ -387,8 +387,24 @@ [locs] (or (apply merge-with into (map parse-rel (select-rels locs))) {})) +(defprotocol node + (to-node [html])) + +(extend-protocol node + String + (to-node [html] + (some->> html str/trim html/html-snippet))) + +(extend-protocol node + clojure.lang.LazySeq + (to-node [html] + html)) + (defn parse "Parse a HTML string with microformats" - [html] - (let [document (some->> html str/trim html/html-snippet with-base-url (map z/xml-zip) first)] - {:items (some-> document walk) :rels (parse-rels document)})) + ([html] + (let [document (some->> html to-node with-base-url)] + (parse document (get-base-url document)))) + ([html base-url] + (let [document (some->> html to-node first (with-base-url base-url) z/xml-zip)] + {:items (some-> document walk) :rels (parse-rels document)}))) diff --git a/test/microformats/parser_expectations.clj b/test/microformats/parser_expectations.clj index f8c7bfb..dd7b335 100644 --- a/test/microformats/parser_expectations.clj +++ b/test/microformats/parser_expectations.clj @@ -305,6 +305,12 @@