29 lines
744 B
Clojure
29 lines
744 B
Clojure
|
(ns cryogen.toc
|
||
|
|
(:require [crouton.html :as html]
|
||
|
|
[hiccup.core :as hiccup]))
|
||
|
|
|
||
|
|
(defn get-headings [content]
|
||
|
|
(reduce
|
||
|
|
(fn [headings {:keys [tag attrs content] :as elm}]
|
||
|
|
(if (some #{tag} [:h1 :h2 :h3])
|
||
|
|
(conj headings elm)
|
||
|
|
(if-let [more-headings (get-headings content)]
|
||
|
|
(into headings more-headings)
|
||
|
|
headings)))
|
||
|
|
[] content))
|
||
|
|
|
||
|
|
(defn make-links [headings]
|
||
|
|
(into [:ol.contents]
|
||
|
|
(for [{[{{name :name} :attrs} title] :content} headings]
|
||
|
|
[:li [:a {:href (str "#" name)} title]])))
|
||
|
|
|
||
|
|
(defn generate-toc [html]
|
||
|
|
(-> html
|
||
|
|
(.getBytes)
|
||
|
|
(java.io.ByteArrayInputStream.)
|
||
|
|
(html/parse)
|
||
|
|
:content
|
||
|
|
(get-headings)
|
||
|
|
(make-links)
|
||
|
|
(hiccup/html)))
|
