src/Hakyll/Web/Preview/Server.hs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

-- | Implements a basic static file server for previewing options
--
{-# LANGUAGE OverloadedStrings #-}
module Hakyll.Web.Preview.Server
    ( staticServer
    ) where

import Control.Monad.Trans (liftIO)

import Snap.Types (Snap)
import Snap.Util.FileServe ( DirectoryConfig (..), fancyDirectoryConfig
                           , serveDirectoryWith
                           )
import Snap.Http.Server ( httpServe, setAccessLog, setErrorLog
                        , setPort, emptyConfig
                        )

-- | Serve a given directory
--
static :: FilePath             -- ^ Directory to serve
       -> (FilePath -> IO ())  -- ^ Pre-serve hook
       -> Snap ()
static directory preServe =
    serveDirectoryWith directoryConfig directory
  where
    directoryConfig :: DirectoryConfig Snap
    directoryConfig = fancyDirectoryConfig
        { preServeHook = liftIO . preServe
        }

-- | Main method, runs a static server in the given directory
--
staticServer :: FilePath             -- ^ Directory to serve
             -> (FilePath -> IO ())  -- ^ Pre-serve hook
             -> Int                  -- ^ Port to listen on
             -> IO ()                -- ^ Blocks forever
staticServer directory preServe port =
    httpServe config $ static directory preServe
  where
    -- Snap server config
    config = setPort port
           $ setAccessLog Nothing
           $ setErrorLog Nothing
           $ emptyConfig