(ns cryogen-core.compiler (:require [selmer.parser :refer [cache-off! render-file]] [selmer.util :refer [set-custom-resource-path!]] [io.aviso.exception :refer [write-exception]] [clojure.java.io :refer [copy file reader writer]] [clojure.string :as s] [text-decoration.core :refer :all] [net.cgrand.enlive-html :as enlive] [cryogen-core.toc :refer [generate-toc]] [cryogen-core.sass :as sass] [cryogen-core.markup :as m] [cryogen-core.io :refer [get-resource find-assets create-folder create-file-recursive create-file wipe-public-folder copy-resources copy-resources-from-theme path]] [cryogen-core.sitemap :as sitemap] [cryogen-core.rss :as rss]) (:import java.util.Locale)) (cache-off!) (defn root-uri "Creates the uri for posts and pages. Returns root-path by default" [k config] (if-let [uri (k config)] uri (config (-> k (name) (s/replace #"-uri$" "") (keyword))))) (defn re-pattern-from-ext "Creates a properly quoted regex pattern for the given file extension" [ext] (re-pattern (str (s/replace ext "." "\\.") "$"))) (defn find-entries "Returns a list of files under the templates directory according to the implemented Markup protocol and specified root directory. It defaults to looking under the implemented protocol's subdirectory, but fallsback to look at the templates directory." [root mu ignored-files] (let [assets (find-assets (path "templates" (m/dir mu) root) (m/ext mu) ignored-files)] (if (seq assets) assets (find-assets (path "templates" root) (m/ext mu) ignored-files)))) (defn find-posts "Returns a list of markdown files representing posts under the post root." [{:keys [post-root ignored-files]} mu] (find-entries post-root mu ignored-files)) (defn find-pages "Returns a list of markdown files representing pages under the page root." [{:keys [page-root ignored-files]} mu] (find-entries page-root mu ignored-files)) (defn parse-post-date "Parses the post date from the post's file name and returns the corresponding java date object" [^String file-name date-fmt] (let [fmt (java.text.SimpleDateFormat. date-fmt)] (.parse fmt (.substring file-name 0 10)))) (defn page-uri "Creates a URI from file name. `uri-type` is any of the uri types specified in config, e.g., `:post-root-uri`." ([file-name params] (page-uri file-name nil params)) ([file-name uri-type {:keys [blog-prefix clean-urls?] :as params}] (let [page-uri (params uri-type) uri-end (if clean-urls? (s/replace file-name #"(index)?\.html" "/") file-name)] (path "/" blog-prefix page-uri uri-end)))) (defn read-page-meta "Returns the clojure map from the top of a markdown page/post" [page rdr] (try (read rdr) (catch Exception _ (throw (IllegalArgumentException. (str "Malformed metadata on page: " page)))))) (defn page-content "Returns a map with the given page's file-name, metadata and content parsed from the file with the given markup." [^java.io.File page config markup] (with-open [rdr (java.io.PushbackReader. (reader page))] (let [re-root (re-pattern (str "^.*?(" (:page-root config) "|" (:post-root config) ")/")) page-fwd (s/replace (str page) "\\" "/") ;; make it work on Windows page-name (s/replace page-fwd re-root "") file-name (s/replace page-name (re-pattern-from-ext (m/ext markup)) ".html") page-meta (read-page-meta page-name rdr) content ((m/render-fn markup) rdr config)] {:file-name file-name :page-meta page-meta :content content}))) (defn merge-meta-and-content "Merges the page metadata and content maps, adding :toc if necessary." [file-name page-meta content] (merge (update-in page-meta [:layout] #(str (name %) ".html")) {:file-name file-name :content content :toc (if-let [toc (:toc page-meta)] (generate-toc content :list-type toc))})) (defn parse-page "Parses a page/post and returns a map of the content, uri, date etc." [page config markup] (let [{:keys [file-name page-meta content]} (page-content page config markup)] (merge (merge-meta-and-content file-name page-meta content) {:uri (page-uri file-name :page-root-uri config) :page-index (:page-index page-meta)}))) (defn parse-post "Return a map with the given post's information." [page config markup] (let [{:keys [file-name page-meta content]} (page-content page config markup)] (merge (merge-meta-and-content file-name page-meta content) (let [date (if (:date page-meta) (.parse (java.text.SimpleDateFormat. (:post-date-format config)) (:date page-meta)) (parse-post-date file-name (:post-date-format config))) archive-fmt (java.text.SimpleDateFormat. (get config :archive-group-format "yyyy MMMM") (Locale/getDefault)) formatted-group (.format archive-fmt date)] {:date date :formatted-archive-group formatted-group :parsed-archive-group (.parse archive-fmt formatted-group) :uri (page-uri file-name :post-root-uri config) :tags (set (:tags page-meta))})))) (defn read-posts "Returns a sequence of maps representing the data from markdown files of posts. Sorts the sequence by post date." [config] (->> (mapcat (fn [mu] (->> (find-posts config mu) (pmap #(parse-post % config mu)) (remove #(= (:draft? %) true)))) (m/markups)) (sort-by :date) reverse (drop-while #(and (:hide-future-posts? config) (.after (:date %) (java.util.Date.)))))) (defn read-pages "Returns a sequence of maps representing the data from markdown files of pages. Sorts the sequence by post date." [config] (->> (mapcat (fn [mu] (->> (find-pages config mu) (map #(parse-page % config mu)))) (m/markups)) (sort-by :page-index))) (defn tag-post "Adds the uri and title of a post to the list of posts under each of its tags" [tags post] (reduce (fn [tags tag] (update-in tags [tag] (fnil conj []) (select-keys post [:uri :title :content :date :enclosure]))) tags (:tags post))) (defn group-by-tags "Maps all the tags with a list of posts that contain each tag" [posts] (reduce tag-post {} posts)) (defn group-for-archive "Groups the posts by month and year for archive sorting" [posts] (->> posts (map #(select-keys % [:title :uri :date :formatted-archive-group :parsed-archive-group])) (group-by :formatted-archive-group) (map (fn [[group posts]] {:group group :parsed-group (:parsed-archive-group (get posts 0)) :posts (map #(select-keys % [:title :uri :date]) posts)})) (sort-by :parsed-group) reverse)) (defn group-for-author "Groups the posts by author. If no post author if found defaults `default-author`." [posts default-author] (->> posts (map #(select-keys % [:title :uri :date :formatted-archive-group :parsed-archive-group :author])) (map #(update % :author (fn [author] (or author default-author)))) (group-by :author) (map (fn [[author posts]] {:author author :posts posts})))) (defn tag-info "Returns a map containing the name and uri of the specified tag" [config tag] {:name (name tag) :uri (page-uri (str (name tag) ".html") :tag-root-uri config)}) (defn add-prev-next "Adds a :prev and :next key to the page/post data containing the title and uri of the prev/next post/page if it exists" [pages] (map (fn [[prev target next]] (assoc target :prev (if prev (select-keys prev [:title :uri]) nil) :next (if next (select-keys next [:title :uri]) nil))) (partition 3 1 (flatten [nil pages nil])))) (defn group-pages "Separates the pages into links for the navbar and links for the sidebar" [pages] (let [{navbar-pages true sidebar-pages false} (group-by #(boolean (:navbar? %)) pages)] (map (partial sort-by :page-index) [navbar-pages sidebar-pages]))) (defn write-html "When `clean-urls?` is set, appends `/index.html` before spit; otherwise just spits." [file-uri {:keys [clean-urls?]} data] (if clean-urls? (create-file-recursive (path file-uri "index.html") data) (create-file file-uri data))) (defn compile-pages "Compiles all the pages into html and spits them out into the public folder" [{:keys [blog-prefix page-root-uri debug?] :as params} pages] (when-not (empty? pages) (println (blue "compiling pages")) (create-folder (path "/" blog-prefix page-root-uri)) (doseq [{:keys [uri] :as page} pages] (println "\t-->" (cyan uri)) (when debug? (println "\t-->" (cyan page))) (write-html uri params (render-file (str "/html/" (:layout page)) (merge params {:active-page "pages" :home false :servlet-context (path "/" blog-prefix "/") :page page :uri uri})))))) (defn compile-posts "Compiles all the posts into html and spits them out into the public folder" [{:keys [blog-prefix post-root-uri disqus-shortname debug?] :as params} posts] (when-not (empty? posts) (println (blue "compiling posts")) (create-folder (path "/" blog-prefix post-root-uri)) (doseq [post posts] (println "\t-->" (cyan (:uri post))) (println "\t-->" (cyan debug?)) (when debug? (println "\t-->" (cyan post))) (write-html (:uri post) params (render-file (str "/html/" (:layout post)) (merge params {:active-page "posts" :servlet-context (path "/" blog-prefix "/") :post post :disqus-shortname disqus-shortname :uri (:uri post)})))))) (defn compile-tags "Compiles all the tag pages into html and spits them out into the public folder" [{:keys [blog-prefix tag-root-uri] :as params} posts-by-tag] (when-not (empty? posts-by-tag) (println (blue "compiling tags")) (create-folder (path "/" blog-prefix tag-root-uri)) (doseq [[tag posts] posts-by-tag] (let [{:keys [name uri]} (tag-info params tag)] (println "\t-->" (cyan uri)) (write-html uri params (render-file "/html/tag.html" (merge params {:active-page "tags" :servlet-context (path "/" blog-prefix "/") :name name :posts posts :uri uri}))))))) (defn compile-tags-page [{:keys [blog-prefix] :as params}] (println (blue "compiling tags page")) (let [uri (page-uri "tags.html" params)] (write-html uri params (render-file "/html/tags.html" (merge params {:active-page "tags" :servlet-context (path "/" blog-prefix "/") :uri uri}))))) (defn content-until-more-marker "Returns the content until the special comment, closing any unclosed tags. Returns nil if there's no such comment." [content] (when-let [index (s/index-of content "")] (->> (subs content 0 index) enlive/html-snippet enlive/emit* (apply str)))) (defn create-preview "Creates a single post preview" [blocks-per-preview post] (update post :content #(or (content-until-more-marker %) (->> (enlive/html-snippet %) (take blocks-per-preview) enlive/emit* (apply str))))) (defn create-previews "Returns a sequence of vectors, each containing a set of post previews" [posts-per-page blocks-per-preview posts] (->> posts (map #(create-preview blocks-per-preview %)) (partition-all posts-per-page) (map-indexed (fn [i v] {:index (inc i) :posts v})))) (defn create-preview-links "Turn each vector of previews into a map with :prev and :next keys that contain the uri of the prev/next preview page" [previews params] (mapv (fn [[prev target next]] (merge target {:prev (if prev (page-uri (path "p" (str (:index prev) ".html")) params) nil) :next (if next (page-uri (path "p" (str (:index next) ".html")) params) nil)})) (partition 3 1 (flatten [nil previews nil])))) (defn compile-preview-pages "Compiles a series of pages containing 'previews' from each post" [{:keys [blog-prefix posts-per-page blocks-per-preview] :as params} posts] (when-not (empty? posts) (let [previews (-> (create-previews posts-per-page blocks-per-preview posts) (create-preview-links params)) previews (if (> (count previews) 1) (assoc-in previews [1 :prev] (page-uri "index.html" params)) previews)] (create-folder (path "/" blog-prefix "p")) (doseq [{:keys [index posts prev next]} previews :let [index-page? (= 1 index)]] (write-html (if index-page? (page-uri "index.html" params) (page-uri (path "p" (str index ".html")) params)) params (render-file "/html/previews.html" (merge params {:active-page "preview" :home (when index-page? true) :servlet-context (path "/" blog-prefix "/") :posts posts :prev-uri prev :next-uri next}))))))) (defn compile-index "Compiles the index page into html and spits it out into the public folder" [{:keys [disqus?] :as params}] (println (blue "compiling index")) (let [uri (page-uri "index.html" params) debug? (-> params :debug?) home-page (-> params :home-page) meta {:active-page "home" :home true :disqus? disqus? :uri uri :post home-page :page home-page}] (when debug? (println "\t-->" (cyan meta))) (write-html uri params (render-file (str "/html/" (:layout home-page)) (merge params meta))))) (defn compile-archives "Compiles the archives page into html and spits it out into the public folder" [{:keys [blog-prefix] :as params} posts] (println (blue "compiling archives")) (let [uri (page-uri "archives.html" params)] (write-html uri params (render-file "/html/archives.html" (merge params {:active-page "archives" :archives true :groups (group-for-archive posts) :servlet-context (path "/" blog-prefix "/") :uri uri}))))) (defn compile-authors "For each author, creates a page with filtered posts." [{:keys [blog-prefix author-root-uri author] :as params} posts] (println (blue "compiling authors")) (create-folder (path "/" blog-prefix author-root-uri)) ;; if the post author is empty defaults to config's :author (doseq [{:keys [author posts]} (group-for-author posts author)] (let [uri (page-uri (str author ".html") :author-root-uri params)] (println "\t-->" (cyan uri)) (write-html uri params (render-file "/html/author.html" (merge params {:author author :groups (group-for-archive posts) :servlet-context (path "/" blog-prefix "/") :uri uri})))))) (defn tag-posts "Converts the tags in each post into links" [posts config] (map #(update-in % [:tags] (partial map (partial tag-info config))) posts)) (defn- template-dir? "Checks that the dir exists in the templates directory." [dir] (.isDirectory (file (str "resources/templates/" dir)))) (defn- markup-entries [post-root page-root] (let [entries (for [mu (m/markups) t (distinct [post-root page-root])] [(str (m/dir mu) "/" t) t])] (apply concat entries))) (defn copy-resources-from-markup-folders "Copy resources from markup folders. This does not copy the markup entries." [{:keys [post-root page-root] :as config}] (let [folders (->> (markup-entries post-root page-root) (filter template-dir?))] (copy-resources (merge config {:resources folders :ignored-files (map #(re-pattern-from-ext (m/ext %)) (m/markups))})))) (defn read-config "Reads the config file" [] (try (let [config (-> "templates/config.edn" get-resource slurp read-string (update-in [:blog-prefix] (fnil str "")) (update-in [:page-root] (fnil str "")) (update-in [:post-root] (fnil str "")) (update-in [:tag-root-uri] (fnil str "")) (update-in [:rss-name] (fnil str "rss.xml")) (update-in [:rss-filters] (fnil seq [])) (update-in [:sass-src] (fnil str "css")) (update-in [:sass-dest] (fnil str "css")) (update-in [:sass-path] (fnil str "sass")) (update-in [:compass-path] (fnil str "compass")) (update-in [:post-date-format] (fnil str "yyyy-MM-dd")) (update-in [:keep-files] (fnil seq [])) (update-in [:ignored-files] (fnil seq [#"^\.#.*" #".*\.swp$"])))] (merge config {:page-root-uri (root-uri :page-root-uri config) :post-root-uri (root-uri :post-root-uri config)})) (catch Exception _ (throw (IllegalArgumentException. "Failed to parse config.edn"))))) (defn compile-assets "Generates all the html and copies over resources specified in the config" [] (println (green "compiling assets...")) (let [{:keys [^String site-url blog-prefix rss-name recent-posts sass-src sass-dest sass-path compass-path keep-files ignored-files previews? clean-urls? debug? author-root-uri] :as config} (read-config) posts (add-prev-next (read-posts config)) pages (add-prev-next (read-pages config)) home-pages (filter #(boolean (:home? %)) pages) pages-without-home (filter #(boolean (not (:home? %))) pages) [navbar-pages sidebar-pages] (group-pages pages-without-home) posts-by-tag (group-by-tags posts) posts (tag-posts posts config) latest-posts (->> posts (take recent-posts) vec) params (merge config {:today (java.util.Date.) :title (:site-title config) :active-page "home" :tags (map (partial tag-info config) (keys posts-by-tag)) :latest-posts latest-posts :navbar-pages navbar-pages :sidebar-pages sidebar-pages :home-page (if (not-empty home-pages) (first home-pages) (merge (first latest-posts) {:layout "home.html"})) :archives-uri (page-uri "archives.html" config) :index-uri (page-uri "index.html" config) :tags-uri (page-uri "tags.html" config) :rss-uri (path "/" blog-prefix rss-name) :site-url (if (.endsWith site-url "/") (.substring site-url 0 (dec (count site-url))) site-url) :theme-path (str "file:resources/templates/themes/" (:theme config))})] (println (blue "debug info")) (println "\t-->" (cyan navbar-pages)) (set-custom-resource-path! (:theme-path params)) (wipe-public-folder keep-files) (println (blue "copying theme resources")) (copy-resources-from-theme config) (println (blue "copying resources")) (copy-resources config) (copy-resources-from-markup-folders config) (compile-pages params pages-without-home) (compile-posts params posts) (compile-tags params posts-by-tag) (compile-tags-page params) (when previews? (compile-preview-pages params posts)) (when (or (not-empty home-pages) (not previews?)) (compile-index params)) (compile-archives params posts) (when author-root-uri (println (blue "generating authors views")) (compile-authors params posts)) (println (blue "generating site map")) (create-file (path "/" blog-prefix "sitemap.xml") (sitemap/generate site-url ignored-files)) (println (blue "generating main rss")) (create-file (path "/" blog-prefix rss-name) (rss/make-channel config posts)) (println (blue "generating filtered rss")) (rss/make-filtered-channels config posts-by-tag) (println (blue "compiling sass")) (sass/compile-sass->css! {:path-sass sass-path :path-compass compass-path :src-sass sass-src :dest-sass (path ".." "public" blog-prefix sass-dest) :ignored-files ignored-files :base-dir "resources/templates/"}))) (defn compile-assets-timed [] (time (try (compile-assets) (catch Exception e (if (or (instance? IllegalArgumentException e) (instance? clojure.lang.ExceptionInfo e)) (println (red "Error:") (yellow (.getMessage e))) (write-exception e))))))