128 lines
5.5 KiB
Haskell
128 lines
5.5 KiB
Haskell
--------------------------------------------------------------------------------
|
|
-- | Module providing the main hakyll function and command-line argument parsing
|
|
{-# LANGUAGE CPP #-}
|
|
{-# LANGUAGE DeriveDataTypeable #-}
|
|
module Hakyll.Main
|
|
( hakyll
|
|
, hakyllWith
|
|
, hakyllWithExitCode
|
|
) where
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
import System.Console.CmdArgs
|
|
import qualified System.Console.CmdArgs.Explicit as CA
|
|
import System.Environment (getProgName)
|
|
import System.IO.Unsafe (unsafePerformIO)
|
|
import System.Exit (ExitCode(ExitSuccess), exitWith)
|
|
|
|
--------------------------------------------------------------------------------
|
|
import qualified Hakyll.Check as Check
|
|
import qualified Hakyll.Commands as Commands
|
|
import qualified Hakyll.Core.Configuration as Config
|
|
import qualified Hakyll.Core.Logger as Logger
|
|
import Hakyll.Core.Rules
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- | This usualy is the function with which the user runs the hakyll compiler
|
|
hakyll :: Rules a -> IO ()
|
|
hakyll = hakyllWith Config.defaultConfiguration
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- | A variant of 'hakyll' which allows the user to specify a custom
|
|
-- configuration
|
|
hakyllWith :: Config.Configuration -> Rules a -> IO ()
|
|
hakyllWith conf rules = hakyllWithExitCode conf rules >>= exitWith
|
|
|
|
hakyllWithExitCode :: Config.Configuration -> Rules a -> IO ExitCode
|
|
hakyllWithExitCode conf rules = do
|
|
args' <- cmdArgs (hakyllArgs conf)
|
|
|
|
let verbosity' = if verbose args' then Logger.Debug else Logger.Message
|
|
check' =
|
|
if internal_links args' then Check.InternalLinks else Check.All
|
|
|
|
logger <- Logger.new verbosity'
|
|
case args' of
|
|
Build _ -> Commands.build conf logger rules
|
|
Check _ _ -> Commands.check conf logger check' >> ok
|
|
Clean _ -> Commands.clean conf logger >> ok
|
|
Deploy _ -> Commands.deploy conf
|
|
Help _ -> showHelp >> ok
|
|
Preview _ p -> Commands.preview conf logger rules p >> ok
|
|
Rebuild _ -> Commands.rebuild conf logger rules
|
|
Server _ _ _ -> Commands.server conf logger (host args') (port args') >> ok
|
|
Watch _ _ p s -> Commands.watch conf logger (host args') p (not s) rules >> ok
|
|
where
|
|
ok = return ExitSuccess
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- | Show usage information.
|
|
showHelp :: IO ()
|
|
showHelp = print $ CA.helpText [] CA.HelpFormatOne $ cmdArgsMode (hakyllArgs Config.defaultConfiguration)
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
data HakyllArgs
|
|
= Build {verbose :: Bool}
|
|
| Check {verbose :: Bool, internal_links :: Bool}
|
|
| Clean {verbose :: Bool}
|
|
| Deploy {verbose :: Bool}
|
|
| Help {verbose :: Bool}
|
|
| Preview {verbose :: Bool, port :: Int}
|
|
| Rebuild {verbose :: Bool}
|
|
| Server {verbose :: Bool, host :: String, port :: Int}
|
|
| Watch {verbose :: Bool, host :: String, port :: Int, no_server :: Bool }
|
|
deriving (Data, Typeable, Show)
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
hakyllArgs :: Config.Configuration -> HakyllArgs
|
|
hakyllArgs conf = modes
|
|
[ (Build $ verboseFlag def) &= help "Generate the site"
|
|
, (Check (verboseFlag def) (False &= help "Check internal links only")) &=
|
|
help "Validate the site output"
|
|
, (Clean $ verboseFlag def) &= help "Clean up and remove cache"
|
|
, (Deploy $ verboseFlag def) &= help "Upload/deploy your site"
|
|
, (Help $ verboseFlag def) &= help "Show this message" &= auto
|
|
, (Preview (verboseFlag def) (portFlag defaultPort)) &=
|
|
help "[Deprecated] Please use the watch command"
|
|
, (Rebuild $ verboseFlag def) &= help "Clean and build again"
|
|
, (Server (verboseFlag def) (hostFlag defaultHost) (portFlag defaultPort)) &=
|
|
help "Start a preview server"
|
|
, (Watch (verboseFlag def) (hostFlag defaultHost) (portFlag defaultPort) (noServerFlag False) &=
|
|
help "Autocompile on changes and start a preview server. You can watch and recompile without running a server with --no-server.")
|
|
] &= help "Hakyll static site compiler" &= program progName
|
|
where
|
|
defaultHost = Config.previewHost conf
|
|
defaultPort = Config.previewPort conf
|
|
|
|
--------------------------------------------------------------------------------
|
|
verboseFlag :: Data a => a -> a
|
|
verboseFlag x = x &= help "Run in verbose mode"
|
|
{-# INLINE verboseFlag #-}
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
noServerFlag :: Data a => a -> a
|
|
noServerFlag x = x &= help "Disable the built-in web server"
|
|
{-# INLINE noServerFlag #-}
|
|
|
|
--------------------------------------------------------------------------------
|
|
hostFlag :: Data a => a -> a
|
|
hostFlag x = x &= help "Host to bind on"
|
|
{-# INLINE hostFlag #-}
|
|
|
|
--------------------------------------------------------------------------------
|
|
portFlag :: Data a => a -> a
|
|
portFlag x = x &= help "Port to listen on"
|
|
{-# INLINE portFlag #-}
|
|
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- | This is necessary because not everyone calls their program the same...
|
|
progName :: String
|
|
progName = unsafePerformIO getProgName
|
|
{-# NOINLINE progName #-}
|