40285aa · typeable/file-embed

git.iko.soy/typeable/file-embed

on master help sign in

Add embedStringFile and embedOneStringFileOf

40285aaOleg Grenrus11 years ago

Data/FileEmbed.hs+47

compare with master

unifiedsplit

@@ -22,6 +22,9 @@ module Data.FileEmbed

22 , embedOneFileOf

23 , embedDir

24 , getDir

25+ -- * Embed as a IsString

26+ , embedStringFile

27+ , embedOneStringFileOf

25 -- * Inject into an executable

28 -- * Inject into an executable

26 -- $inject

29 -- $inject

27#if MIN_VERSION_template_haskell(2,5,0)

30#if MIN_VERSION_template_haskell(2,5,0)

@@ -35,6 +38,7 @@ module Data.FileEmbed

35 -- * Internal

38 -- * Internal

36 , stringToBs

39 , stringToBs

37 , bsToExp

40 , bsToExp

41+ , strToExp

38 ) where

42 ) where

40import Language.Haskell.TH.Syntax

44import Language.Haskell.TH.Syntax

@@ -61,6 +65,8 @@ import Control.Applicative ((<$>))

61import Data.ByteString.Unsafe (unsafePackAddressLen)

65import Data.ByteString.Unsafe (unsafePackAddressLen)

62import System.IO.Unsafe (unsafePerformIO)

66import System.IO.Unsafe (unsafePerformIO)

63import System.FilePath ((</>))

67import System.FilePath ((</>))

68+import Data.String (fromString)

69+import Prelude as P

65-- | Embed a single file in your source code.

71-- | Embed a single file in your source code.

66--

72--

@@ -144,6 +150,47 @@ bsToExp bs = do

144stringToBs :: String -> B.ByteString

150stringToBs :: String -> B.ByteString

145stringToBs = B8.pack

151stringToBs = B8.pack

146

152

153+-- | Embed a single file in your source code.

154+--

155+-- > import Data.String

156+-- >

157+-- > myFile :: IsString a => a

158+-- > myFile = $(embedStringFile "dirName/fileName")

159+embedStringFile :: FilePath -> Q Exp

160+embedStringFile fp =

161+#if MIN_VERSION_template_haskell(2,7,0)

162+ qAddDependentFile fp >>

163+#endif

164+ (runIO $ P.readFile fp) >>= strToExp

165+

166+-- | Embed a single existing string file in your source code

167+-- out of list a list of paths supplied.

168+embedOneStringFileOf :: [FilePath] -> Q Exp

169+embedOneStringFileOf ps =

170+ (runIO $ readExistingFile ps) >>= \ ( path, content ) -> do

171+#if MIN_VERSION_template_haskell(2,7,0)

172+ qAddDependentFile path

173+#endif

174+ strToExp content

175+ where

176+ readExistingFile :: [FilePath] -> IO ( FilePath, String )

177+ readExistingFile xs = do

178+ ys <- filterM doesFileExist xs

179+ case ys of

180+ (p:_) -> P.readFile p >>= \ c -> return ( p, c )

181+ _ -> throw $ ErrorCall "Cannot find file to embed as resource"

182+

183+strToExp :: String -> Q Exp

184+#if MIN_VERSION_template_haskell(2, 5, 0)

185+strToExp s =

186+ return $ VarE 'fromString

187+ `AppE` LitE (StringL s)

188+#else

189+strToExp s = do

190+ helper <- [| fromString |]

191+ return $! AppE helper $! LitE $! StringL s

192+#endif

193+

147notHidden :: FilePath -> Bool

194notHidden :: FilePath -> Bool

148notHidden ('.':_) = False

195notHidden ('.':_) = False

149notHidden _ = True

196notHidden _ = True