X-Git-Url: http://www.average.org/gitweb/?p=pdns-pipe-nmc.git;a=blobdiff_plain;f=NmcTransform.hs;h=3cd37dfe314e46cbdb09918c12fb5f646bf6ef71;hp=de6fc0f3e82172475c5b722968199e9ad5cb81c3;hb=aa91db8940672a57169b1bb400fa3c1a1e9dd335;hpb=1ad7a7ab8f41ac7c10eebac17907fe15b1483d4f diff --git a/NmcTransform.hs b/NmcTransform.hs index de6fc0f..3cd37df 100644 --- a/NmcTransform.hs +++ b/NmcTransform.hs @@ -9,6 +9,7 @@ import Data.Map.Lazy (empty, lookup, delete, size, singleton , foldrWithKey, insert, insertWith) import Control.Monad (foldM) import Data.Aeson (decode) +import Data.Default.Class (def) import NmcDom @@ -42,10 +43,10 @@ mergeIncl queryOp depth base = do base' = mbase {domDelegate = Nothing, domImport = Nothing} -- print base if depth <= 0 then return $ Left "Nesting of imports is too deep" - else case ((domDelegate mbase), (domImport mbase)) of - (Nothing, Nothing ) -> return $ Right base' - (Nothing, Just keys) -> foldM mergeIncl1 (Right base') keys - (Just key, _ ) -> mergeIncl1 (Right emptyNmcDom) key + else case ((domDelegate mbase), (domImport mbase)) of + (Nothing, Nothing ) -> return $ Right base' + (Nothing, Just keys) -> foldM mergeIncl1 (Right base') keys + (Just key, _ ) -> mergeIncl1 (Right def) key where mergeIncl1 (Left err) _ = return $ Left err -- can never happen mergeIncl1 (Right acc) key = do @@ -85,10 +86,10 @@ expandSrv base = where addSrvMx sr acc = sub1 `mergeNmcDom` acc where - sub1 = emptyNmcDom { domMap = Just (singleton proto sub2) + sub1 = def { domMap = Just (singleton proto sub2) , domMx = maybemx} - sub2 = emptyNmcDom { domMap = Just (singleton srvid sub3) } - sub3 = emptyNmcDom { domSrv = Just [srvStr] } + sub2 = def { domMap = Just (singleton srvid sub3) } + sub3 = def { domSrv = Just [srvStr] } proto = "_" ++ (srvProto sr) srvid = "_" ++ (srvName sr) srvStr = (show (srvPrio sr)) ++ "\t" @@ -111,17 +112,18 @@ splitSubdoms base = in case domMap base of Nothing -> base' - Just sdmap -> (emptyNmcDom { domMap = Just sdmap' }) `mergeNmcDom` base' + Just sdmap -> (def { domMap = Just sdmap' }) `mergeNmcDom` base' where sdmap' = foldrWithKey stow empty sdmap stow fqdn sdom acc = insertWith mergeNmcDom fqdn' sdom' acc where (fqdn', sdom') = - nest (map unpack (splitOn (pack ".") (pack fqdn)), sdom) + nest (filter (/= "") (splitOnDots fqdn), sdom) + splitOnDots s = map unpack (splitOn (pack ".") (pack s)) nest ([], v) = (fqdn, v) -- can split result be empty? nest ([k], v) = (k, v) nest (k:ks, v) = - nest (ks, emptyNmcDom { domMap = Just (singleton k v) }) + nest (ks, def { domMap = Just (singleton k v) }) -- | Presence of some elements require removal of some others normalizeDom :: NmcDom -> NmcDom @@ -131,7 +133,7 @@ normalizeDom dom = foldr id dom [ translateNormalizer where nsNormalizer dom = case domNs dom of Nothing -> dom - Just ns -> emptyNmcDom { domNs = domNs dom, domEmail = domEmail dom } + Just ns -> def { domNs = domNs dom, domEmail = domEmail dom } translateNormalizer dom = case domTranslate dom of Nothing -> dom Just tr -> dom { domMap = Nothing } @@ -151,14 +153,14 @@ descendNmcDom queryOp subdom base = do Left err -> return base' Right base'' -> case domMap base'' of - Nothing -> return $ Right emptyNmcDom + Nothing -> return $ Right def Just map -> case lookup d map of - Nothing -> return $ Right emptyNmcDom + Nothing -> return $ Right def Just sub -> descendNmcDom queryOp ds sub -- | Initial NmcDom populated with "import" only, suitable for "descend" seedNmcDom :: String -- ^ domain key (without namespace prefix) -> NmcDom -- ^ resulting seed domain -seedNmcDom dn = emptyNmcDom { domImport = Just (["d/" ++ dn])} +seedNmcDom dn = def { domImport = Just (["d/" ++ dn])}