diff options
| author | Jasper Van der Jeugt <jaspervdj@gmail.com> | 2011-02-25 14:36:34 +0100 |
|---|---|---|
| committer | Jasper Van der Jeugt <jaspervdj@gmail.com> | 2011-02-25 14:36:34 +0100 |
| commit | c89cfdb456deda5a81b52d9e8516d635e82f70d8 (patch) | |
| tree | 00aaafda1e835bc2028f65de95105809e479bc07 | |
| parent | 730eebe9894e73b7e86a6e5a7546ab5b2484c65d (diff) | |
| download | hakyll-c89cfdb456deda5a81b52d9e8516d635e82f70d8.tar.gz | |
Add `byExtension` compiler
| -rw-r--r-- | src/Hakyll/Core/Compiler.hs | 33 | ||||
| -rw-r--r-- | src/Hakyll/Core/Compiler/Internal.hs | 1 |
2 files changed, 34 insertions, 0 deletions
diff --git a/src/Hakyll/Core/Compiler.hs b/src/Hakyll/Core/Compiler.hs index 908cb55..a3fed7c 100644 --- a/src/Hakyll/Core/Compiler.hs +++ b/src/Hakyll/Core/Compiler.hs @@ -103,6 +103,7 @@ module Hakyll.Core.Compiler , unsafeCompiler , mapCompiler , timedCompiler + , byExtension ) where import Prelude hiding ((.), id) @@ -112,6 +113,7 @@ import Control.Monad.Reader (ask) import Control.Monad.Trans (liftIO) import Control.Category (Category, (.), id) import Data.Maybe (fromMaybe) +import System.FilePath (takeExtension) import Data.Binary (Binary) import Data.Typeable (Typeable) @@ -289,3 +291,34 @@ timedCompiler :: String -- ^ Message timedCompiler msg (Compiler d j) = Compiler d $ \x -> CompilerM $ do logger <- compilerLogger <$> ask timed logger msg $ unCompilerM $ j x + +-- | Choose a compiler by extension +-- +-- Example: +-- +-- > route "css/*" $ setExtension "css" +-- > compile "css/*" $ byExtension (error "Not a (S)CSS file") +-- > [ (".css", compressCssCompiler) +-- > , (".scss", sass) +-- > ] +-- +-- This piece of code will select the @compressCssCompiler@ for @.css@ files, +-- and the @sass@ compiler (defined elsewhere) for @.scss@ files. +-- +byExtension :: Compiler a b -- ^ Default compiler + -> [(String, Compiler a b)] -- ^ Choices + -> Compiler a b -- ^ Resulting compiler +byExtension defaultCompiler choices = Compiler deps job + where + -- Lookup the compiler, give an error when it is not found + lookup' identifier = + let extension = takeExtension $ toFilePath identifier + in fromMaybe defaultCompiler $ lookup extension choices + -- Collect the dependencies of the choice + deps = do + identifier <- dependencyIdentifier <$> ask + compilerDependencies $ lookup' identifier + -- Collect the job of the choice + job x = CompilerM $ do + identifier <- compilerIdentifier <$> ask + unCompilerM $ compilerJob (lookup' identifier) x diff --git a/src/Hakyll/Core/Compiler/Internal.hs b/src/Hakyll/Core/Compiler/Internal.hs index d37c7ef..53df044 100644 --- a/src/Hakyll/Core/Compiler/Internal.hs +++ b/src/Hakyll/Core/Compiler/Internal.hs @@ -3,6 +3,7 @@ {-# LANGUAGE GeneralizedNewtypeDeriving #-} module Hakyll.Core.Compiler.Internal ( Dependencies + , DependencyEnvironment (..) , CompilerEnvironment (..) , CompilerM (..) , Compiler (..) |