@@ -19,6 +19,7 @@
19module Data.FileEmbed
19module Data.FileEmbed
20 ( -- * Embed at compile time
20 ( -- * Embed at compile time
22 , embedOneFileOf
23 , embedOneFileOf
23 , embedDir
24 , embedDir
24 , embedDirListing
25 , embedDirListing
@@ -38,6 +39,7 @@ module Data.FileEmbed
38 , injectFileWith
39 , injectFileWith
39 -- * Relative path manipulation
40 -- * Relative path manipulation
40 , makeRelativeToProject
41 , makeRelativeToProject
42+ , makeRelativeToLocationPredicate
41 -- * Internal
43 -- * Internal
42 , stringToBs
44 , stringToBs
@@ -60,17 +62,20 @@ import qualified Data.ByteString.Internal as B
61import System.Directory (doesDirectoryExist, doesFileExist,
63import System.Directory (doesDirectoryExist, doesFileExist,
62 getDirectoryContents, canonicalizePath)
64 getDirectoryContents, canonicalizePath)
63−import Control.Exception (throw, ErrorCall(..))
65+import Control.Exception (throw, tryJust, ErrorCall(..))
64−import Control.Monad (filterM)
66+import Control.Monad (filterM, guard)
65import qualified Data.ByteString as B
67import qualified Data.ByteString as B
66import qualified Data.ByteString.Char8 as B8
68import qualified Data.ByteString.Char8 as B8
67import Control.Arrow ((&&&), second)
69import Control.Arrow ((&&&), second)
68import Control.Applicative ((<$>))
70import Control.Applicative ((<$>))
69import Data.ByteString.Unsafe (unsafePackAddressLen)
71import Data.ByteString.Unsafe (unsafePackAddressLen)
72+import System.IO.Error (isDoesNotExistError)
70import System.IO.Unsafe (unsafePerformIO)
73import System.IO.Unsafe (unsafePerformIO)
71import System.FilePath ((</>), takeDirectory, takeExtension)
74import System.FilePath ((</>), takeDirectory, takeExtension)
72import Data.String (fromString)
75import Data.String (fromString)
73import Prelude as P
76import Prelude as P
77+import Data.List (sortBy)
78+import Data.Ord (comparing)
75-- | Embed a single file in your source code.
80-- | Embed a single file in your source code.
@@ -85,6 +90,33 @@ embedFile fp =
86 (runIO $ B.readFile fp) >>= bsToExp
91 (runIO $ B.readFile fp) >>= bsToExp
93+-- | Maybe embed a single file in your source code depending on whether or not file exists.
95+-- Warning: When a build is compiled with the file missing, a recompile when the file exists might not trigger an embed of the file.
96+-- You might try to fix this by doing a clean build.
98+-- > import qualified Data.ByteString
100+-- > maybeMyFile :: Maybe Data.ByteString.ByteString
101+-- > maybeMyFile = $(embedFileIfExists "dirName/fileName")
104+embedFileIfExists :: FilePath -> Q Exp
105+embedFileIfExists fp = do
106+ mbs <- runIO maybeFile
108+ Nothing -> [| Nothing |]
110+#if MIN_VERSION_template_haskell(2,7,0)
111+ qAddDependentFile fp
113+ [| Just $(bsToExp bs) |]
115+ maybeFile :: IO (Maybe B.ByteString)
117+ either (const Nothing) Just <$>
118+ tryJust (guard . isDoesNotExistError) (B.readFile fp)
88-- | Embed a single existing file in your source code
120-- | Embed a single existing file in your source code
89-- out of list a list of paths supplied.
121-- out of list a list of paths supplied.
@@ -234,7 +266,7 @@ fileList' realTop top = do
234 mapM (liftPair2 . second B.readFile)
266 mapM (liftPair2 . second B.readFile)
235 dirs <- filterM (doesDirectoryExist . snd) all' >>=
267 dirs <- filterM (doesDirectoryExist . snd) all' >>=
236 mapM (fileList' realTop . fst)
268 mapM (fileList' realTop . fst)
237− return $ concat $ files : dirs
269+ return $ sortBy (comparing fst) $ concat $ files : dirs
239liftPair2 :: Monad m => (a, m b) -> m (a, b)
271liftPair2 :: Monad m => (a, m b) -> m (a, b)
240liftPair2 (a, b) = b >>= \b' -> return (a, b')
272liftPair2 (a, b) = b >>= \b' -> return (a, b')
@@ -377,7 +409,19 @@ available, you can use the non-@With@ variants.
378-- @since 0.0.10
410-- @since 0.0.10
379makeRelativeToProject :: FilePath -> Q FilePath
411makeRelativeToProject :: FilePath -> Q FilePath
380−makeRelativeToProject rel = do
412+makeRelativeToProject = makeRelativeToLocationPredicate $ (==) ".cabal" . takeExtension
414+-- | Take a predicate to infer the project root and a relative file path, the given file path is then attached to the inferred project root
416+-- This function looks at the source location of the Haskell file calling it,
417+-- finds the first parent directory with a file matching the given predicate, and uses that as the
418+-- root directory for fixing the relative path.
420+-- @$(makeRelativeToLocationPredicate ((==) ".cabal" . takeExtension) "data/foo.txt" >>= embedFile)@
423+makeRelativeToLocationPredicate :: (FilePath -> Bool) -> FilePath -> Q FilePath
424+makeRelativeToLocationPredicate isTargetFile rel = do
381 loc <- qLocation
425 loc <- qLocation
382 runIO $ do
426 runIO $ do
383 srcFP <- canonicalizePath $ loc_filename loc
427 srcFP <- canonicalizePath $ loc_filename loc
@@ -392,8 +436,6 @@ makeRelativeToProject rel = do
392 then return Nothing
436 then return Nothing
394 contents <- getDirectoryContents dir
438 contents <- getDirectoryContents dir
395− if any isCabalFile contents
439+ if any isTargetFile contents
396 then return (Just dir)
440 then return (Just dir)
397 else findProjectDir dir
441 else findProjectDir dir
399− isCabalFile fp = takeExtension fp == ".cabal"