module Hakyll.Util
    ( dropParentRoute
    , modifyDirRoute
    , modifyDirRoute_
    ) where

import Prolog
import Hakyll
import System.FilePath

-- | Drop some number of parent directories
dropParentRoute :: Int -> Routes
dropParentRoute n = modifyDirRoute_ $ drop n

modifyDirRoute :: ([FilePath] -> FilePath) -> Routes
modifyDirRoute f = customRoute $ f . splitDirectories . toFilePath

modifyDirRoute_ :: ([FilePath] -> [FilePath]) -> Routes
modifyDirRoute_ f = modifyDirRoute $ joinPath . f