module Hakyll.Preview.Server
( staticServer
) where
import Control.Monad.Trans (liftIO)
import qualified Data.ByteString.Char8 as B
import qualified Snap.Core as Snap
import qualified Snap.Http.Server as Snap
import qualified Snap.Util.FileServe as Snap
import Hakyll.Core.Logger (Logger)
import qualified Hakyll.Core.Logger as Logger
static :: FilePath
-> (FilePath -> IO ())
-> Snap.Snap ()
static directory preServe =
Snap.serveDirectoryWith directoryConfig directory
where
directoryConfig :: Snap.DirectoryConfig Snap.Snap
directoryConfig = Snap.fancyDirectoryConfig
{ Snap.preServeHook = liftIO . preServe
}
staticServer :: Logger
-> FilePath
-> (FilePath -> IO ())
-> String
-> Int
-> IO ()
staticServer logger directory preServe host port = do
Logger.header logger $ "Listening on http://" ++ host ++ ":" ++ show port
Snap.httpServe config $ static directory preServe
where
config = Snap.setBind (B.pack host)
$ Snap.setPort port
$ Snap.setAccessLog Snap.ConfigNoLog
$ Snap.setErrorLog Snap.ConfigNoLog
$ Snap.setVerbose False
$ Snap.emptyConfig