1 files changed, 43 insertions, 0 deletions
diff --git a/src/Hakyll/Preview/Server.hs b/src/Hakyll/Preview/Server.hs
new file mode 100644
index 0000000..14cf377
--- /dev/null
+++ b/src/Hakyll/Preview/Server.hs
@@ -0,0 +1,43 @@
+--------------------------------------------------------------------------------
+-- | Implements a basic static file server for previewing options
+{-# LANGUAGE OverloadedStrings #-}
+module Hakyll.Preview.Server
+    ( staticServer
+    ) where
+
+
+--------------------------------------------------------------------------------
+import           Control.Monad.Trans (liftIO)
+import qualified Snap.Core           as Snap
+import qualified Snap.Http.Server    as Snap
+import qualified Snap.Util.FileServe as Snap
+
+
+--------------------------------------------------------------------------------
+-- | Serve a given directory
+static :: FilePath             -- ^ Directory to serve
+       -> (FilePath -> IO ())  -- ^ Pre-serve hook
+       -> Snap.Snap ()
+static directory preServe =
+    Snap.serveDirectoryWith directoryConfig directory
+  where
+    directoryConfig :: Snap.DirectoryConfig Snap.Snap
+    directoryConfig = Snap.fancyDirectoryConfig
+        { Snap.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 =
+    Snap.httpServe config $ static directory preServe
+  where
+    -- Snap server config
+    config = Snap.setPort      port
+           $ Snap.setAccessLog Snap.ConfigNoLog
+           $ Snap.setErrorLog  Snap.ConfigNoLog
+           $ Snap.emptyConfig