[Haskell] grepコマンド自作

gistfile1.txt

--[grep] 標準入力について、文字列パターンに一致する部分を持つ行を出力する。
-- [Usage] grep {pattern(regex)}
import System
import Text.Regex.Posix
main = do cs <- getContents
          args <- getArgs
          putStr $ unlines $ filter (¥str -> contain (head args) str) $ lines cs
contain :: String -> String -> Bool
contain cs cs' = cs' =‾ cs :: Bool

--[grep2] カレントディレクトリで、ファイル名パターンに一致したファイルについて、文字列パターンに一致する部分を持つ行を出力する。”ファイル名: 行”の形で出力する。
-- [Usage] grep2 {pattern(regex)} {file name pattern(regex)}
import Directory
import Control.Monad
import Data.List
import Text.Regex.Posix
import System

fileNameWidth = 35

main = do args <- getArgs
          do getCurrentDirectory
             >>= getDirectoryContents
             >>= mapM (grep (head args)) . filter (¥n -> n =~ (head (tail args)) :: Bool)
             >>= return . (¥_ -> "")
             >>= putStrLn

grep :: String -> String -> IO ()
grep tg f = flip catch
                   (return . const ())
                   (do cs <- readFile f
                       let ls = filter (¥str -> str =~ tg :: Bool) $ lines cs
                       putStr $ unlines $ map (¥str -> f ++ (replicate (fileNameWidth - length f) ' ') ++ ": " ++ str) ls
                       return ())