test.hs

   1 {-# LANGUAGE OverloadedStrings #-}
   2
   3 module Main where
   4
   5 import Prelude hiding (readFile)
   6 import System.Environment
   7 import Data.ByteString.Lazy (ByteString)
   8 import Data.ByteString.Lazy.Char8 (readFile)
   9 import System.IO.Error
  10 import Control.Exception
  11
  12 import NmcDom
  13 import NmcTransform
  14
  15 queryOp :: String -> IO (Either String ByteString)
  16 queryOp key = catch (readFile key >>= return . Right)
  17                     (\e -> return (Left (show (e :: IOException))))
  18
  19 main = do
  20         (d:_) <- getArgs
  21         descendNmcDom queryOp []               (seedNmcDom d) >>= print
  22         descendNmcDom queryOp ["_tcp","_smtp"] (seedNmcDom d) >>= print
  23