module Tonalude.Prelude.IO
  ( withLazyFile
  , readFileBinary
  , writeFileBinary
  , readFileUtf8
  , writeFileUtf8
  , hPutBuilder
  ) where

import RIO hiding
  ( withLazyFile
  , readFileBinary
  , writeFileBinary
  , readFileUtf8
  , writeFileUtf8
  , hPutBuilder
  )
import qualified RIO

import qualified Data.ByteString as B ()
import qualified Data.ByteString.Lazy as BL

-- | Lazily get the contents of a file. Unlike 'BL.readFile', this
-- ensures that if an exception is thrown, the file handle is closed
-- immediately.
withLazyFile :: FilePath -> (BL.ByteString -> RIO env a) -> RIO env a
withLazyFile :: forall env a. FilePath -> (ByteString -> RIO env a) -> RIO env a
withLazyFile = forall (m :: * -> *) a.
MonadUnliftIO m =>
FilePath -> (ByteString -> m a) -> m a
RIO.withLazyFile

-- | Write a file in UTF8 encoding
--
-- This function will use OS-specific line ending handling.
writeFileUtf8 :: FilePath -> Text -> RIO env ()
writeFileUtf8 :: forall env. FilePath -> Text -> RIO env ()
writeFileUtf8 = forall (m :: * -> *). MonadIO m => FilePath -> Text -> m ()
RIO.writeFileUtf8

hPutBuilder :: Handle -> Builder -> RIO env ()
hPutBuilder :: forall env. Handle -> Builder -> RIO env ()
hPutBuilder = forall (m :: * -> *). MonadIO m => Handle -> Builder -> m ()
RIO.hPutBuilder

-- | Same as 'B.readFile', but focused on 'RIO'
readFileBinary :: FilePath -> RIO env ByteString
readFileBinary :: forall env. FilePath -> RIO env ByteString
readFileBinary = forall (m :: * -> *). MonadIO m => FilePath -> m ByteString
RIO.readFileBinary

-- | Same as 'B.writeFile', but focused on 'RIO'
writeFileBinary :: FilePath -> ByteString -> RIO env ()
writeFileBinary :: forall env. FilePath -> ByteString -> RIO env ()
writeFileBinary = forall (m :: * -> *). MonadIO m => FilePath -> ByteString -> m ()
RIO.writeFileBinary

-- | Read a file in UTF8 encoding, throwing an exception on invalid character
-- encoding.
--
-- This function will use OS-specific line ending handling.
readFileUtf8 :: FilePath -> RIO env Text
readFileUtf8 :: forall env. FilePath -> RIO env Text
readFileUtf8 = forall (m :: * -> *). MonadIO m => FilePath -> m Text
RIO.readFileUtf8