site.hs

{-# LANGUAGE OverloadedStrings #-}
import Data.Monoid (mappend)
import Data.List
import Hakyll
import Text.Pandoc
import Text.Pandoc.Walk (walk)
-- import qualified Data.Map as M 

main :: IO ()
main = hakyll $ do
  match "static/*/*"       $ do route idRoute
                                compile copyFileCompiler
  match (fromList tops)    $ crunchWithCtx siteCtx
  match "lectures/*"       $ crunchWithCtxCustom "lecture" postCtx
  match "assignments/*"    $ crunchWithCtx postCtx
  match "discussions/*"    $ crunchWithCtx postCtx
  match "templates/*"      $ compile templateCompiler

crunchWithCtx ctx = do
  route   $ setExtension "html"
  compile $ pandocCompiler
            >>= loadAndApplyTemplate "templates/page.html"    ctx
            >>= loadAndApplyTemplate "templates/default.html" ctx
            >>= relativizeUrls

crunchWithCtxCustom mode ctx = do
  route   $ setExtension "html"
  compile $ pandocCompilerWithTransform
              defaultHakyllReaderOptions
              defaultHakyllWriterOptions
              (walk (toggleMode mode . haskellizeBlock) . walk haskellizeInline)
            >>= loadAndApplyTemplate "templates/page.html"    ctx
            >>= loadAndApplyTemplate "templates/default.html" ctx
            >>= relativizeUrls

-- | Treat an ordered list with uppercase roman numerals as a map:
--   in each item, the first paragraph is the key, and the second is the value;
--   pick the value with key `mode` and discard all other items
-- toggleMode :: String -> Block -> Block
toggleMode mode (OrderedList (_, UpperRoman, _) items) = select items
  where
    select ([Para [Str key], payload] : rest) =
      if key == mode then payload else select rest
    select _ = Null
toggleMode _ b = b

-- | Make inline code Haskell by default
haskellizeInline :: Inline -> Inline
haskellizeInline (Code (ident, [], kvs) str) = Code (ident, ["haskell"], kvs) str
haskellizeInline i = i

-- | Make code blocks Haskell by default
haskellizeBlock :: Block -> Block
haskellizeBlock (CodeBlock (ident, [], kvs) str) = CodeBlock (ident, ["haskell"], kvs) str
haskellizeBlock b = b

--------------------------------------------------------------------------------
postCtx :: Context String
postCtx =
    dateField  "date"       "%B %e, %Y"  `mappend`
    -- constField "headerImg"  "Eiffel.jpg" `mappend`
    siteCtx

siteCtx :: Context String
siteCtx =
    constField "cssUrl"             "https://ucsd-cse230.github.io/sp20"      `mappend`
    constField "baseUrl"            "https://ucsd-cse130.github.io/wi21"      `mappend`
    constField "site_name"          "cse130"                                  `mappend`
    constField "site_description"   "UCSD CSE 130"                            `mappend`
    constField "site_username"      "Ranjit Jhala"                            `mappend`
    constField "twitter_username"   "ranjitjhala"                             `mappend`
    constField "github_username"    "ranjitjhala"                             `mappend`
    constField "google_username"    "rjhala@eng.ucsd.edu"                     `mappend`
    constField "google_userid"      "u/0/104385825850161331469"               `mappend`
    constField "piazza_classid"     "class/kjivoxdgfuc1w2"                    `mappend`
    constField "canvasUrl"          "https://canvas.ucsd.edu/courses/21656"   `mappend`
    defaultContext


tops =
  [ "index.md"
  , "grades.md"
  , "lectures.md"
  , "links.md"
  , "assignments.md"
  , "calendar.md"
  , "contact.md"
  ]

fac  :: Int -> Int 
fac 0 = 1
fac n = n * fac (n-1)