124 lines
4 KiB
Haskell
Executable file
124 lines
4 KiB
Haskell
Executable file
-- pulling heavily from https://abhinavsarkar.net/posts/static-site-generator-using-shake/
|
|
-- docs:
|
|
-- https://hackage.haskell.org/package/pandoc-3.2.1/docs/doc-index-All.html
|
|
-- https://hackage.haskell.org/package/mustache-2.4.2/docs/doc-index.html
|
|
--
|
|
{-# LANGUAGE ApplicativeDo, DataKinds, DeriveGeneric #-}
|
|
{-# LANGUAGE DerivingVia, TypeApplications #-}
|
|
|
|
module Main where
|
|
|
|
import Control.Monad (forM, void)
|
|
import Data.Aeson.Types (Result (..))
|
|
import Data.List (nub, sortOn)
|
|
import Data.Text (Text)
|
|
import Data.Time (UTCTime, defaultTimeLocale, formatTime, parseTimeM)
|
|
import Deriving.Aeson
|
|
import Deriving.Aeson.Stock (PrefixedSnake)
|
|
import Development.Shake (Action, Rules, (%>), (|%>), (~>))
|
|
import Development.Shake.FilePath ((<.>), (</>))
|
|
import Text.Pandoc (Block (Plain), Meta (..), MetaValue (..), Pandoc (..))
|
|
import qualified Data.Aeson.Types as A
|
|
import qualified Data.HashMap.Strict as HM
|
|
import qualified Data.Ord as Ord
|
|
import qualified Data.Text as T
|
|
import qualified Development.Shake as Shake
|
|
import qualified Development.Shake.FilePath as Shake
|
|
import qualified Text.Mustache as Mus
|
|
import qualified Text.Mustache.Compile as Mus
|
|
import qualified Text.Pandoc as Pandoc
|
|
import Config
|
|
import Utilities
|
|
import Templates
|
|
-- target = thing we want
|
|
-- Rule = pattern of thing being made + actions to produce the thing
|
|
-- Action = actions to produce a thing
|
|
|
|
main :: IO ()
|
|
main = Shake.shakeArgs Shake.shakeOptions $ do
|
|
Shake.withTargetDocs "Build the site" $
|
|
"build" ~> buildSite
|
|
Shake.withTargetDocs "Clean the built site" $
|
|
"clean" ~> Shake.removeFilesAfter outputDir ["//*"]
|
|
|
|
|
|
buildSite :: Action ()
|
|
buildSite = do
|
|
-- static files
|
|
assetPaths <- Shake.getDirectoryFiles "" assetGlobs
|
|
-- path concat each asset path so it's output into the outputDir
|
|
Shake.need $ map (outputDir </>) assetPaths
|
|
|
|
-- take the misc pages which aren't blog posts and make their html files
|
|
Shake.need $ map indexHtmlOutputPath pagePaths
|
|
|
|
-- handle posts
|
|
postPaths <- Shake.getDirectoryFiles "" postGlobs
|
|
Shake.need $ map indexHtmlOutputPath postPaths
|
|
|
|
-- remaining pages, index.xml = rss feed
|
|
Shake.need $ map (outputDir </>) ["index.html", "index.xml"]
|
|
|
|
buildRules :: Rules ()
|
|
buildRules = do
|
|
assets
|
|
pages
|
|
posts
|
|
|
|
-- make a rule of the pattern outputDir/asset_name which copes from outputDir/../pages
|
|
assets :: Rules ()
|
|
assets = map (outputDir </>) assetGlobs |%> \target -> do
|
|
let src = Shake.dropDirectory1 target </> "pages"
|
|
Shake.copyFileChanged src target
|
|
Shake.putInfo $ "Copied " <> target <> " from " <> src
|
|
|
|
data Page = Page {pageTitle :: Text, pageContent :: Text}
|
|
deriving (Show, Generic)
|
|
deriving (ToJSON) via PrefixedSnake "page" Page
|
|
|
|
pages :: Rules ()
|
|
pages = map indexHtmlOutputPath pagePaths |%> \target -> do
|
|
let src = indexHtmlSourcePath target
|
|
(meta, html) <- typstToHtml src
|
|
|
|
let page = Page (meta HM.! "title") html
|
|
applyTemplateAndWrite "default.html" page target
|
|
Shake.putInfo $ "Built " <> target <> " from " <> src
|
|
|
|
data Post = Post
|
|
{ postTitle :: Text,
|
|
postAuthor :: Maybe Text,
|
|
postTags :: [Text],
|
|
postDate :: Maybe Text,
|
|
postContent :: Maybe Text,
|
|
postLink :: Maybe Text
|
|
} deriving (Show, Generic)
|
|
deriving (FromJSON, ToJSON) via PrefixedSnake "post" Post
|
|
|
|
posts :: Rules ()
|
|
posts = map indexHtmlOutputPath postGlobs |%> \target -> do
|
|
let src = indexHtmlSourcePath target
|
|
post <- readPost src
|
|
postHtml <- applyTemplate "post.html" post
|
|
|
|
let page = Page (postTitle post) postHtml
|
|
applyTemplateAndWrite "default.html" page target
|
|
Shake.putInfo $ "Built " <> target <> " from " <> src
|
|
|
|
readPost :: FilePath -> Action Post
|
|
readPost postPath = do
|
|
date <- parseTimeM False defaultTimeLocale "%Y-%-m-%-d"
|
|
. take 10
|
|
. Shake.takeBaseName
|
|
$ postPath
|
|
let formattedDate =
|
|
T.pack $ formatTime @UTCTime defaultTimeLocale "%b %e, %Y" date
|
|
|
|
(post, html) <- typstToHtml postPath
|
|
Shake.putInfo $ "Read " <> postPath
|
|
return $ post
|
|
{ postDate = Just formattedDate,
|
|
postContent = Just html,
|
|
postLink = Just . T.pack $ "/" <> Shake.dropExtension postPath <> "/"
|
|
}
|
|
|