Add unsafeToString methods
This commit is contained in:
parent
a0510eaec1
commit
809ffc7745
@ -8,6 +8,7 @@ main =
|
|||||||
doctest
|
doctest
|
||||||
["-isrc"
|
["-isrc"
|
||||||
, "-XOverloadedStrings"
|
, "-XOverloadedStrings"
|
||||||
|
, "-XScopedTypeVariables"
|
||||||
, "src/HPath.hs"
|
, "src/HPath.hs"
|
||||||
]
|
]
|
||||||
|
|
||||||
|
50
src/HPath.hs
50
src/HPath.hs
@ -10,12 +10,14 @@
|
|||||||
-- Support for well-typed paths.
|
-- Support for well-typed paths.
|
||||||
|
|
||||||
|
|
||||||
|
{-# LANGUAGE BangPatterns #-}
|
||||||
{-# LANGUAGE CPP #-}
|
{-# LANGUAGE CPP #-}
|
||||||
{-# LANGUAGE DeriveDataTypeable #-}
|
{-# LANGUAGE DeriveDataTypeable #-}
|
||||||
{-# LANGUAGE EmptyDataDecls #-}
|
{-# LANGUAGE EmptyDataDecls #-}
|
||||||
#if __GLASGOW_HASKELL__ >= 708
|
#if __GLASGOW_HASKELL__ >= 708
|
||||||
{-# LANGUAGE PatternSynonyms #-}
|
{-# LANGUAGE PatternSynonyms #-}
|
||||||
#endif
|
#endif
|
||||||
|
{-# LANGUAGE ScopedTypeVariables #-}
|
||||||
|
|
||||||
module HPath
|
module HPath
|
||||||
(
|
(
|
||||||
@ -39,6 +41,8 @@ module HPath
|
|||||||
,fromAbs
|
,fromAbs
|
||||||
,fromRel
|
,fromRel
|
||||||
,toFilePath
|
,toFilePath
|
||||||
|
,unsafeToString
|
||||||
|
,unsafeToString'
|
||||||
-- * Path Operations
|
-- * Path Operations
|
||||||
,(</>)
|
,(</>)
|
||||||
,basename
|
,basename
|
||||||
@ -53,8 +57,10 @@ module HPath
|
|||||||
)
|
)
|
||||||
where
|
where
|
||||||
|
|
||||||
import Control.Exception (Exception)
|
import Control.Exception (IOException, Exception, catch)
|
||||||
|
import Control.Monad ((<$!>))
|
||||||
import Control.Monad.Catch (MonadThrow(..))
|
import Control.Monad.Catch (MonadThrow(..))
|
||||||
|
import Data.ByteString.Unsafe(unsafeUseAsCStringLen)
|
||||||
#if MIN_VERSION_bytestring(0,10,8)
|
#if MIN_VERSION_bytestring(0,10,8)
|
||||||
import Data.ByteString(ByteString, stripPrefix)
|
import Data.ByteString(ByteString, stripPrefix)
|
||||||
#else
|
#else
|
||||||
@ -65,10 +71,17 @@ import qualified Data.ByteString as BS
|
|||||||
import Data.Data
|
import Data.Data
|
||||||
import Data.Maybe
|
import Data.Maybe
|
||||||
import Data.Word8
|
import Data.Word8
|
||||||
|
import GHC.Foreign(peekCStringLen)
|
||||||
|
import GHC.IO.Encoding(getLocaleEncoding, TextEncoding)
|
||||||
import HPath.Internal
|
import HPath.Internal
|
||||||
|
import System.IO.Unsafe(unsafePerformIO)
|
||||||
import System.Posix.FilePath hiding ((</>))
|
import System.Posix.FilePath hiding ((</>))
|
||||||
|
|
||||||
|
|
||||||
|
-- $setup
|
||||||
|
-- >>> import GHC.IO.Encoding(utf8)
|
||||||
|
|
||||||
|
|
||||||
--------------------------------------------------------------------------------
|
--------------------------------------------------------------------------------
|
||||||
-- Types
|
-- Types
|
||||||
|
|
||||||
@ -232,6 +245,41 @@ fromAbs = toFilePath
|
|||||||
fromRel :: RelC r => Path r -> ByteString
|
fromRel :: RelC r => Path r -> ByteString
|
||||||
fromRel = toFilePath
|
fromRel = toFilePath
|
||||||
|
|
||||||
|
-- | This converts the underlying bytestring of the path to an unsafe
|
||||||
|
-- FilePath by assuming the encoding of the current locale setting. This
|
||||||
|
-- may be utterly wrong, but isn't particularly worse than what the
|
||||||
|
-- base library does. Blows up on decoding errors.
|
||||||
|
--
|
||||||
|
-- >>> unsafeToString (MkPath "/lal/lad")
|
||||||
|
-- "/lal/lad"
|
||||||
|
-- >>> unsafeToString (MkPath "/")
|
||||||
|
-- "/"
|
||||||
|
-- >>> unsafeToString (MkPath "lad")
|
||||||
|
-- "lad"
|
||||||
|
-- >>> catch (Just <$> unsafeToString (MkPath "<22>")) (\(_ :: IOException) -> pure Nothing)
|
||||||
|
-- Nothing
|
||||||
|
unsafeToString :: Path b -> IO FilePath
|
||||||
|
unsafeToString (MkPath p) = do
|
||||||
|
enc <- getLocaleEncoding
|
||||||
|
unsafeUseAsCStringLen p (peekCStringLen enc)
|
||||||
|
|
||||||
|
-- | Same as @unsafeToString@, except requires the encoding
|
||||||
|
-- to be passed explicitly. This uses 'unsafePerformIO' and
|
||||||
|
-- returns 'Nothing' on decoding errors.
|
||||||
|
--
|
||||||
|
-- >>> unsafeToString' (MkPath "/lal/lad") utf8
|
||||||
|
-- Just "/lal/lad"
|
||||||
|
-- >>> unsafeToString' (MkPath "/") utf8
|
||||||
|
-- Just "/"
|
||||||
|
-- >>> unsafeToString' (MkPath "lad") utf8
|
||||||
|
-- Just "lad"
|
||||||
|
-- >>> unsafeToString' (MkPath "<22>") utf8
|
||||||
|
-- Nothing
|
||||||
|
unsafeToString' :: Path b -> TextEncoding -> Maybe FilePath
|
||||||
|
unsafeToString' (MkPath !p) enc =
|
||||||
|
unsafePerformIO $!
|
||||||
|
catch (Just <$!> unsafeUseAsCStringLen p (peekCStringLen enc))
|
||||||
|
(\(_ :: IOException) -> pure Nothing)
|
||||||
|
|
||||||
|
|
||||||
--------------------------------------------------------------------------------
|
--------------------------------------------------------------------------------
|
||||||
|
Loading…
Reference in New Issue
Block a user