Made document root configurable.

2 files changed, 6 insertions, 5 deletions

diff --git a/src/Network/Hakyll/SimpleServer.hs b/src/Network/Hakyll/SimpleServer.hs
index 63006f2..80c0ff1 100644
--- a/src/Network/Hakyll/SimpleServer.hs
+++ b/src/Network/Hakyll/SimpleServer.hs
@@ -159,14 +159,15 @@ respond handle = do
           putHeader (key, value) = B.hPutStr handle $ key `B.append` B.pack ": "
                                                         `B.append` value `B.append` B.pack "\r\n"
 
--- | Start a simple http server on the given 'PortNumber'.
-simpleServer :: PortNumber -> IO ()
-simpleServer port = do
+-- | Start a simple http server on the given 'PortNumber', serving the given
+--   directory.
+simpleServer :: PortNumber -> FilePath -> IO ()
+simpleServer port root = do
     putStrLn $ "Starting hakyll server on port " ++ show port ++ "..."
     socket <- listenOn (PortNumber port)
     forever (listen socket)
     where -- A default configuration.
-          config = ServerConfig { documentRoot = "_site"
+          config = ServerConfig { documentRoot = root
                                 , portNumber = port
                                 }
 
diff --git a/src/Text/Hakyll.hs b/src/Text/Hakyll.hs
index 9d2f86c..bafe4a8 100644
--- a/src/Text/Hakyll.hs
+++ b/src/Text/Hakyll.hs
@@ -45,4 +45,4 @@ help = do
              ++ name ++ " server [port]   Run a local test server.\n"
 
 server :: Integer -> IO ()
-server p = do simpleServer (fromIntegral $ p)
+server p = do simpleServer (fromIntegral $ p) "_site"