Use class name for property keywords

1 files changed, 35 insertions, 2 deletions

diff --git a/src/microformats/parser.clj b/src/microformats/parser.clj
index 4071e85..c39e396 100644
--- a/src/microformats/parser.clj
+++ b/src/microformats/parser.clj
@@ -1,12 +1,45 @@
 (ns microformats.parser
-  (:require [net.cgrand.enlive-html :as html]))
+  (:require [net.cgrand.enlive-html :as html]
+            [clojure.core.reducers :as r]
+            [clojure.string :as str]))
 
 ;;; Turn string into stream
 
+(defn mf-names-from-class
+  "Get microformat classnames from a class attribute"
+  [prefix]
+  (r/filter #(.startsWith % prefix)))
+
+(defn remove-mf-prefix
+  "Remove microformats prefixes from a class attribute"
+  [prefix]
+  (r/map #(apply str (drop (count prefix) %))))
+
+(defn- split-classes
+  "Split a whitespace-separated string."
+  [class]
+  (str/split class #"\s+"))
+
+(defn classes-to-props
+  "Convert class list to list of microformat property keywords"
+  [prefix]
+  (comp (r/map keyword)
+        (remove-mf-prefix prefix)
+        (mf-names-from-class prefix)))
+
+(defn element-to-classes
+  "Get list of classes from an element"
+  [el] (-> el
+           :attrs
+           :class
+           split-classes))
+
 (defn parse-p
   "Parse p-* classes within HTML element."
   [element]
-  (hash-map :properties (hash-map :name (first (:content (first (html/select element [(html/attr-starts :class "p-")])))))))
+  (let [el (first (html/select element [(html/attr-starts :class "p-")]))
+        props (into [] ((classes-to-props "p-") (element-to-classes el)))]
+    (hash-map :properties (apply hash-map (first props) (repeat (count props) (first (:content el)))))))
 
 (defn parse-h
   "Parse h-* classes within a HTML document."