Add command-line args
This commit is contained in:
parent
26c95402d8
commit
5705bb8f88
1 changed files with 72 additions and 2 deletions
|
@ -5,17 +5,87 @@ module Hakyll.Main
|
||||||
, hakyllWith
|
, hakyllWith
|
||||||
) where
|
) where
|
||||||
|
|
||||||
|
import Control.Monad (when)
|
||||||
|
import System.Environment (getProgName, getArgs)
|
||||||
|
import System.Directory (doesDirectoryExist, removeDirectoryRecursive)
|
||||||
|
|
||||||
import Hakyll.Core.Configuration
|
import Hakyll.Core.Configuration
|
||||||
import Hakyll.Core.Run
|
import Hakyll.Core.Run
|
||||||
import Hakyll.Core.Rules
|
import Hakyll.Core.Rules
|
||||||
|
import Hakyll.Network.Server
|
||||||
|
|
||||||
-- | This usualy is the function with which the user runs the hakyll compiler
|
-- | This usualy is the function with which the user runs the hakyll compiler
|
||||||
--
|
--
|
||||||
hakyll :: Rules -> IO ()
|
hakyll :: Rules -> IO ()
|
||||||
hakyll = run defaultHakyllConfiguration
|
hakyll = hakyllWith defaultHakyllConfiguration
|
||||||
|
|
||||||
-- | A variant of 'hakyll' which allows the user to specify a custom
|
-- | A variant of 'hakyll' which allows the user to specify a custom
|
||||||
-- configuration
|
-- configuration
|
||||||
--
|
--
|
||||||
hakyllWith :: HakyllConfiguration -> Rules -> IO ()
|
hakyllWith :: HakyllConfiguration -> Rules -> IO ()
|
||||||
hakyllWith = run
|
hakyllWith configuration rules = do
|
||||||
|
args <- getArgs
|
||||||
|
case args of
|
||||||
|
["build"] -> build configuration rules
|
||||||
|
["clean"] -> clean configuration
|
||||||
|
["help"] -> help
|
||||||
|
["preview"] -> putStrLn "Not implemented"
|
||||||
|
["preview", p] -> putStrLn "Not implemented"
|
||||||
|
["rebuild"] -> rebuild configuration rules
|
||||||
|
["server"] -> server configuration 8000
|
||||||
|
["server", p] -> server configuration (read p)
|
||||||
|
_ -> help
|
||||||
|
|
||||||
|
-- | Build the site
|
||||||
|
--
|
||||||
|
build :: HakyllConfiguration -> Rules -> IO ()
|
||||||
|
build = run
|
||||||
|
|
||||||
|
-- | Remove the output directories
|
||||||
|
--
|
||||||
|
clean :: HakyllConfiguration -> IO ()
|
||||||
|
clean configuration = do
|
||||||
|
remove $ destinationDirectory configuration
|
||||||
|
remove $ storeDirectory configuration
|
||||||
|
where
|
||||||
|
remove dir = do
|
||||||
|
putStrLn $ "Removing " ++ dir ++ "..."
|
||||||
|
exists <- doesDirectoryExist dir
|
||||||
|
when exists $ removeDirectoryRecursive dir
|
||||||
|
|
||||||
|
-- | Show usage information.
|
||||||
|
--
|
||||||
|
help :: IO ()
|
||||||
|
help = do
|
||||||
|
name <- getProgName
|
||||||
|
mapM_ putStrLn
|
||||||
|
[ "ABOUT"
|
||||||
|
, ""
|
||||||
|
, "This is a Hakyll site generator program. You should always"
|
||||||
|
, "run it from the project root directory."
|
||||||
|
, ""
|
||||||
|
, "USAGE"
|
||||||
|
, ""
|
||||||
|
, name ++ " build Generate the site"
|
||||||
|
, name ++ " clean Clean up and remove cache"
|
||||||
|
, name ++ " help Show this message"
|
||||||
|
, name ++ " preview [port] Run a server and autocompile"
|
||||||
|
, name ++ " rebuild Clean up and build again"
|
||||||
|
, name ++ " server [port] Run a local test server"
|
||||||
|
]
|
||||||
|
|
||||||
|
-- | Rebuild the site
|
||||||
|
--
|
||||||
|
rebuild :: HakyllConfiguration -> Rules -> IO ()
|
||||||
|
rebuild configuration rules = do
|
||||||
|
clean configuration
|
||||||
|
build configuration rules
|
||||||
|
|
||||||
|
-- | Start a server
|
||||||
|
--
|
||||||
|
server :: HakyllConfiguration -> Int -> IO ()
|
||||||
|
server configuration port = do
|
||||||
|
let destination = destinationDirectory configuration
|
||||||
|
staticServer destination preServeHook port
|
||||||
|
where
|
||||||
|
preServeHook _ = return ()
|
||||||
|
|
Loading…
Reference in a new issue