module Main where

--------------------------------------------------------------------------------
{-# LANGUAGE OverloadedStrings #-}
import           Data.Monoid
import           Control.Monad (forM_)
import           Hakyll
import           HakyllBibTex
import           System.Process
import           Text.Pandoc.Options

config :: Configuration
config = defaultConfiguration
    {   deployCommand = "echo `pwd`; ./scripts/deploy.sh" }

--------------------------------------------------------------------------------
bibHeaders :: [(String, [String])]
bibHeaders = [ ("draft"  , [ "-c", "$type='UNPUBLISHED'"   ] ) ,
               ("thesis" , [ "-c", "$type='PHDTHESIS'"     ] ) ,
               ("pubs"   , [ "-c", "$type='INPROCEEDINGS' or $type='ARTICLE'" ] ) ]

bibSources :: [String]
bibSources = [ "bibs/header.bib"
             , "bibs/myrefs.bib"
             ]

contentSections :: [String]
contentSections = [ "contacts", "about", "news", "biblio" ]
--------------------------------------------------------------------------------
compass :: Compiler (Item String)
compass =
    getResourceString >>=
    withItemBody (unixFilter "sass" ["-s", "--scss", "--compass"])

b2bflags :: [String]
b2bflags = [ "--expand"
           , "-c"
           , "jh:'yes'"
           , "-s"
           , "$date"
           , "-r"
           ]


bib2bib :: [String]
           -> Compiler (Item String)
bib2bib flags = (unsafeCompiler $
    readProcess "bib2bib" (bibSources ++ b2bflags ++ flags) "")
        >>= makeItem

bib2bibParse :: Item String
                -> Compiler [Item BibEntry]
bib2bibParse file =
    let (BibFile bf) = parseBibFile (itemBody file) in
    sequence $ fmap makeItem bf


--------------------------------------------------------------------------------
main :: IO ()
main = hakyllWith config $ do
    match (fromGlob "files/**") $ do
        route   idRoute
        compile copyFileCompiler

    match (fromGlob "images/**") $ do
        route   idRoute
        compile copyFileCompiler

    match (fromGlob "css/*.scss") $ do
        route $ setExtension "css"
        compile $ compass

    forM_ bibHeaders $ \(nick, flags) ->
        create [fromCapture (fromGlob "bibs/*.html") nick] $ do
            route idRoute
            compile $
                let biblioCtx = mconcat
                        [ field "bibitems" $ return . itemBody ] in do
                        bibTpl <- loadBody $ fromFilePath "templates/bibitemtpl.html"
                        (bib2bib flags)
                            >>= bib2bibParse
                            >>= applyTemplateList bibTpl bibEntryContext
                            >>= makeItem
                            >>= loadAndApplyTemplate (fromFilePath "templates/biblisttpl.html") biblioCtx
                            >>= relativizeUrls

    match (fromGlob "content/biblio.md") $ do
        route $ setExtension "html"
        compile $ do
            let makeBibCtx (nick, _) =
                    field nick (\_ -> loadBody $ fromFilePath ("bibs/" ++ nick ++ ".html"))
                secDiv    = defaultHakyllWriterOptions { writerSectionDivs = True }
                biblioCtx = mconcat $ map makeBibCtx bibHeaders in do
                    pandocCompilerWith defaultHakyllReaderOptions secDiv
                        >>= applyAsTemplate biblioCtx
                        >>= relativizeUrls

    match (fromGlob "content/*.md") $ do
        route $ setExtension "html"
        compile $ do
            let secDiv = defaultHakyllWriterOptions { writerSectionDivs = True }
            pandocCompilerWith defaultHakyllReaderOptions secDiv
                >>= relativizeUrls

    match (fromGlob "index.html") $ do
        route idRoute
        compile $ do
            let loadSec sec  = field sec
                  (\_ -> loadBody (fromFilePath $ "content/" ++ sec ++ ".md"))
                mainContext  = mconcat $ fmap loadSec contentSections
                indexContext =
                  field "footer" (\_ -> loadBody $ fromFilePath "content/footer.md")
                  `mappend`
                  defaultContext
            getResourceBody
                >>= applyAsTemplate mainContext
                >>= loadAndApplyTemplate (fromFilePath "templates/main.html") indexContext
                >>= relativizeUrls

    match (fromGlob "templates/*") $ compile templateCompiler