text / tests / benchmarks / Replace.hs

The default branch has multiple heads

Full commit
{-# LANGUAGE BangPatterns #-}
module Main (main) where

import System.Environment (getArgs)
import qualified Data.Text.Lazy as LT
import qualified Data.Text.Lazy.IO as LT
import qualified Data.ByteString.Lazy.Search as LB
import qualified Data.ByteString.Lazy.Char8 as LB
import qualified Data.ByteString.Char8 as B

lazyText file pat sub =
  LT.readFile file >>= LT.putStr . LT.replace (LT.pack pat) (LT.pack sub)

lazyBS file pat sub =
  LB.readFile file >>= LB.putStr . LB.replace (B.pack pat) (LB.pack sub)

main = do
  (kind : file : pat : sub : _) <- getArgs
  case kind of
    "lazyText" -> lazyText file pat sub
    "lazyBS" -> lazyBS file pat sub