module Network.DNS.Lookup (
lookupA, lookupAAAA
, lookupMX, lookupAviaMX, lookupAAAAviaMX
, lookupTXT
) where
import Control.Applicative
import qualified Data.ByteString.Lazy.Char8 as L
import Data.IP
import Data.Maybe
import Network.DNS.Resolver as DNS
import Network.DNS.Types
lookupA :: Resolver -> Domain -> IO (Maybe [IPv4])
lookupA rlv dom = toV4 <$> DNS.lookup rlv dom A
where
toV4 = maybe Nothing (Just . map unTag)
unTag (RD_A x) = x
unTag _ = error "lookupA"
lookupAAAA :: Resolver -> Domain -> IO (Maybe [IPv6])
lookupAAAA rlv dom = toV6 <$> DNS.lookup rlv dom AAAA
where
toV6 = maybe Nothing (Just . map unTag)
unTag (RD_AAAA x) = x
unTag _ = error "lookupAAAA"
lookupMX :: Resolver -> Domain -> IO (Maybe [(Domain,Int)])
lookupMX rlv dom = toMX <$> DNS.lookup rlv dom MX
where
toMX = maybe Nothing (Just . map unTag)
unTag (RD_MX pr dm) = (dm,pr)
unTag _ = error "lookupMX"
lookupAviaMX :: Resolver -> Domain -> IO (Maybe [IPv4])
lookupAviaMX rlv dom = lookupXviaMX rlv dom (lookupA rlv)
lookupAAAAviaMX :: Resolver -> Domain -> IO (Maybe [IPv6])
lookupAAAAviaMX rlv dom = lookupXviaMX rlv dom (lookupAAAA rlv)
lookupXviaMX :: Resolver -> Domain -> (Domain -> IO (Maybe [a])) -> IO (Maybe [a])
lookupXviaMX rlv dom func = do
mdps <- lookupMX rlv dom
maybe (return Nothing) lookup' mdps
where
lookup' dps = do
as <- catMaybes <$> mapM (func . fst) dps
case as of
[] -> return Nothing
ass -> return $ Just (concat ass)
lookupTXT :: Resolver -> Domain -> IO (Maybe [L.ByteString])
lookupTXT rlv dom = toTXT <$> DNS.lookup rlv dom TXT
where
toTXT = maybe Nothing (Just . map unTag)
unTag (RD_TXT x) = x
unTag _ = error "lookupTXT"