@@ -3,7 +3,7 @@
3{-# LANGUAGE OverloadedStrings #-}
3{-# LANGUAGE OverloadedStrings #-}
4-- | This module uses template Haskell. Following is a simplified explanation of usage for those unfamiliar with calling Template Haskell functions.
4-- | This module uses template Haskell. Following is a simplified explanation of usage for those unfamiliar with calling Template Haskell functions.
6−-- The function @embedFile@ in this modules embeds a file into the executable
6+-- The function @embedFile@ in this modules embeds a file into the exceutable
7-- that you can use it at runtime. A file is represented as a @ByteString@.
7-- that you can use it at runtime. A file is represented as a @ByteString@.
8-- However, as you can see below, the type signature indicates a value of type
8-- However, as you can see below, the type signature indicates a value of type
9-- @Q Exp@ will be returned. In order to convert this into a @ByteString@, you
9-- @Q Exp@ will be returned. In order to convert this into a @ByteString@, you
@@ -19,14 +19,9 @@
19module Data.FileEmbed
19module Data.FileEmbed
20 ( -- * Embed at compile time
20 ( -- * Embed at compile time
23 , embedOneFileOf
22 , embedOneFileOf
24 , embedDir
23 , embedDir
27− -- * Embed as a IsString
29− , embedOneStringFileOf
30 -- * Inject into an executable
25 -- * Inject into an executable
31 -- $inject
26 -- $inject
32#if MIN_VERSION_template_haskell(2,5,0)
27#if MIN_VERSION_template_haskell(2,5,0)
@@ -37,45 +32,35 @@ module Data.FileEmbed
37 , injectFile
32 , injectFile
38 , injectWith
33 , injectWith
39 , injectFileWith
34 , injectFileWith
40− -- * Relative path manipulation
41− , makeRelativeToProject
42− , makeRelativeToLocationPredicate
43 -- * Internal
35 -- * Internal
44 , stringToBs
36 , stringToBs
49import Language.Haskell.TH.Syntax
40import Language.Haskell.TH.Syntax
50 ( Exp (AppE, ListE, LitE, TupE, SigE, VarE)
41 ( Exp (AppE, ListE, LitE, TupE, SigE, VarE)
51− , Lit (..)
42+#if MIN_VERSION_template_haskell(2,5,0)
43+ , Lit (StringL, StringPrimL, IntegerL)
45+ , Lit (StringL, IntegerL)
54− , qLocation, loc_filename
55#if MIN_VERSION_template_haskell(2,7,0)
49#if MIN_VERSION_template_haskell(2,7,0)
56 , Quasi(qAddDependentFile)
50 , Quasi(qAddDependentFile)
59−#if MIN_VERSION_template_haskell(2,16,0)
60−import Language.Haskell.TH ( mkBytes, bytesPrimL )
61−import qualified Data.ByteString.Internal as B
63import System.Directory (doesDirectoryExist, doesFileExist,
53import System.Directory (doesDirectoryExist, doesFileExist,
64− getDirectoryContents, canonicalizePath)
54+ getDirectoryContents)
65−import Control.Exception (throw, tryJust, ErrorCall(..))
55+import Control.Exception (throw, ErrorCall(..))
66−import Control.Monad (filterM, guard)
56+import Control.Monad (filterM)
67import qualified Data.ByteString as B
57import qualified Data.ByteString as B
68import qualified Data.ByteString.Char8 as B8
58import qualified Data.ByteString.Char8 as B8
69import Control.Arrow ((&&&), second)
59import Control.Arrow ((&&&), second)
70import Control.Applicative ((<$>))
60import Control.Applicative ((<$>))
71import Data.ByteString.Unsafe (unsafePackAddressLen)
61import Data.ByteString.Unsafe (unsafePackAddressLen)
72−import System.IO.Error (isDoesNotExistError)
73import System.IO.Unsafe (unsafePerformIO)
62import System.IO.Unsafe (unsafePerformIO)
74−import System.FilePath ((</>), takeDirectory, takeExtension)
63+import System.FilePath ((</>))
75−import Data.String (fromString)
77−import Data.List (sortBy)
78−import Data.Ord (comparing)
80-- | Embed a single file in your source code.
65-- | Embed a single file in your source code.
@@ -90,40 +75,13 @@ embedFile fp =
91 (runIO $ B.readFile fp) >>= bsToExp
76 (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)
120-- | Embed a single existing file in your source code
78-- | Embed a single existing file in your source code
121-- out of list a list of paths supplied.
79-- out of list a list of paths supplied.
123-- > import qualified Data.ByteString
81-- > import qualified Data.ByteString
125-- > myFile :: Data.ByteString.ByteString
83-- > myFile :: Data.ByteString.ByteString
126−-- > myFile = $(embedOneFileOf [ "dirName/fileName", "src/dirName/fileName" ])
84+-- > myFile = $(embedFile' [ "dirName/fileName", "src/dirName/fileName" ])
127embedOneFileOf :: [FilePath] -> Q Exp
85embedOneFileOf :: [FilePath] -> Q Exp
128embedOneFileOf ps =
86embedOneFileOf ps =
129 (runIO $ readExistingFile ps) >>= \ ( path, content ) -> do
87 (runIO $ readExistingFile ps) >>= \ ( path, content ) -> do
@@ -151,18 +109,6 @@ embedDir fp = do
151 e <- ListE <$> ((runIO $ fileList fp) >>= mapM (pairToExp fp))
109 e <- ListE <$> ((runIO $ fileList fp) >>= mapM (pairToExp fp))
152 return $ SigE e typ
110 return $ SigE e typ
154−-- | Embed a directory listing recursively in your source code.
156−-- > myFiles :: [FilePath]
157−-- > myFiles = $(embedDirListing "dirName")
160−embedDirListing :: FilePath -> Q Exp
161−embedDirListing fp = do
162− typ <- [t| [FilePath] |]
163− e <- ListE <$> ((runIO $ fmap fst <$> fileList fp) >>= mapM strToExp)
166-- | Get a directory tree in the IO monad.
112-- | Get a directory tree in the IO monad.
168-- This is the workhorse of 'embedDir'
114-- This is the workhorse of 'embedDir'
@@ -175,11 +121,7 @@ pairToExp _root (path, bs) = do
175 qAddDependentFile $ _root ++ '/' : path
121 qAddDependentFile $ _root ++ '/' : path
177 exp' <- bsToExp bs
123 exp' <- bsToExp bs
179−#if MIN_VERSION_template_haskell(2,16,0)
182− [LitE $ StringL path, exp']
124+ return $! TupE [LitE $ StringL path, exp']
184bsToExp :: B.ByteString -> Q Exp
126bsToExp :: B.ByteString -> Q Exp
185#if MIN_VERSION_template_haskell(2, 5, 0)
127#if MIN_VERSION_template_haskell(2, 5, 0)
@@ -187,11 +129,7 @@ bsToExp bs =
187 return $ VarE 'unsafePerformIO
129 return $ VarE 'unsafePerformIO
188 `AppE` (VarE 'unsafePackAddressLen
130 `AppE` (VarE 'unsafePackAddressLen
189 `AppE` LitE (IntegerL $ fromIntegral $ B8.length bs)
131 `AppE` LitE (IntegerL $ fromIntegral $ B8.length bs)
190−#if MIN_VERSION_template_haskell(2, 16, 0)
132+#if MIN_VERSION_template_haskell(2, 8, 0)
191− `AppE` LitE (bytesPrimL (
192− let B.PS ptr off sz = bs
193− in mkBytes ptr (fromIntegral off) (fromIntegral sz))))
194−#elif MIN_VERSION_template_haskell(2, 8, 0)
195 `AppE` LitE (StringPrimL $ B.unpack bs))
133 `AppE` LitE (StringPrimL $ B.unpack bs))
197 `AppE` LitE (StringPrimL $ B8.unpack bs))
135 `AppE` LitE (StringPrimL $ B8.unpack bs))
@@ -206,51 +144,6 @@ bsToExp bs = do
206stringToBs :: String -> B.ByteString
144stringToBs :: String -> B.ByteString
207stringToBs = B8.pack
145stringToBs = B8.pack
209−-- | Embed a single file in your source code.
211−-- > import Data.String
213−-- > myFile :: IsString a => a
214−-- > myFile = $(embedStringFile "dirName/fileName")
217−embedStringFile :: FilePath -> Q Exp
219−#if MIN_VERSION_template_haskell(2,7,0)
220− qAddDependentFile fp >>
222− (runIO $ P.readFile fp) >>= strToExp
224−-- | Embed a single existing string file in your source code
225−-- out of list a list of paths supplied.
228−embedOneStringFileOf :: [FilePath] -> Q Exp
229−embedOneStringFileOf ps =
230− (runIO $ readExistingFile ps) >>= \ ( path, content ) -> do
231−#if MIN_VERSION_template_haskell(2,7,0)
232− qAddDependentFile path
236− readExistingFile :: [FilePath] -> IO ( FilePath, String )
237− readExistingFile xs = do
238− ys <- filterM doesFileExist xs
240− (p:_) -> P.readFile p >>= \ c -> return ( p, c )
241− _ -> throw $ ErrorCall "Cannot find file to embed as resource"
243−strToExp :: String -> Q Exp
244−#if MIN_VERSION_template_haskell(2, 5, 0)
246− return $ VarE 'fromString
247− `AppE` LitE (StringL s)
250− helper <- [| fromString |]
251− return $! AppE helper $! LitE $! StringL s
254notHidden :: FilePath -> Bool
147notHidden :: FilePath -> Bool
255notHidden ('.':_) = False
148notHidden ('.':_) = False
256notHidden _ = True
149notHidden _ = True
@@ -266,7 +159,7 @@ fileList' realTop top = do
266 mapM (liftPair2 . second B.readFile)
159 mapM (liftPair2 . second B.readFile)
267 dirs <- filterM (doesDirectoryExist . snd) all' >>=
160 dirs <- filterM (doesDirectoryExist . snd) all' >>=
268 mapM (fileList' realTop . fst)
161 mapM (fileList' realTop . fst)
269− return $ sortBy (comparing fst) $ concat $ files : dirs
162+ return $ concat $ files : dirs
271liftPair2 :: Monad m => (a, m b) -> m (a, b)
164liftPair2 :: Monad m => (a, m b) -> m (a, b)
272liftPair2 (a, b) = b >>= \b' -> return (a, b')
165liftPair2 (a, b) = b >>= \b' -> return (a, b')
@@ -392,50 +285,3 @@ the magic key. If you know for certain that there will only be one dummy space
392available, you can use the non-@With@ variants.
285available, you can use the non-@With@ variants.
396−-- | Take a relative file path and attach it to the root of the current
399−-- The idea here is that, when building with Stack, the build will always be
400−-- executed with a current working directory of the root of the project (where
401−-- your .cabal file is located). However, if you load up multiple projects with
402−-- @stack ghci@, the working directory may be something else entirely.
404−-- This function looks at the source location of the Haskell file calling it,
405−-- finds the first parent directory with a .cabal file, and uses that as the
406−-- root directory for fixing the relative path.
408−-- @$(makeRelativeToProject "data/foo.txt" >>= embedFile)@
411−makeRelativeToProject :: FilePath -> Q FilePath
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
427− srcFP <- canonicalizePath $ loc_filename loc
428− mdir <- findProjectDir srcFP
430− Nothing -> error $ "Could not find .cabal file for path: " ++ srcFP
431− Just dir -> return $ dir </> rel
433− findProjectDir x = do
434− let dir = takeDirectory x
438− contents <- getDirectoryContents dir
439− if any isTargetFile contents
440− then return (Just dir)
441− else findProjectDir dir