Dns.primary wrote, not quite ready

3 files changed, 34 insertions, 6 deletions

diff --git a/Propellor/Attr.hs b/Propellor/Attr.hs
index 37ed1bad..a4d7a958 100644
--- a/Propellor/Attr.hs
+++ b/Propellor/Attr.hs
@@ -4,7 +4,6 @@ module Propellor.Attr where
 
 import Propellor.Types
 import Propellor.Types.Attr
-import Propellor.Types.Dns
 
 import "mtl" Control.Monad.Reader
 import qualified Data.Set as S
diff --git a/Propellor/Property/Dns.hs b/Propellor/Property/Dns.hs
index 131079ea..a9a8619c 100644
--- a/Propellor/Property/Dns.hs
+++ b/Propellor/Property/Dns.hs
@@ -1,8 +1,10 @@
 module Propellor.Property.Dns (
 	module Propellor.Types.Dns,
+	primary,
 	secondary,
 	servingZones,
 	mkSOA,
+	rootAddressesFrom,
 	writeZoneFile,
 	nextSerialNumber,
 	adjustSerialNumber,
@@ -22,6 +24,23 @@ import qualified Data.Map as M
 import qualified Data.Set as S
 import Data.List
 
+-- | Primary dns server for a domain.
+--
+-- TODO: Does not yet add it to named.conf.local.
+primary :: [Host] -> Domain -> SOA -> Property
+primary hosts domain soa = withwarnings (check needupdate baseprop)
+	`requires` Apt.serviceInstalledRunning "bind9"
+	`onChange` Service.reloaded "bind9"
+  where
+	(zone, warnings) = genZone hosts domain soa
+	zonefile = "/etc/bind/propellor/db." ++ domain
+	needupdate = (/= Just zone) <$> readZonePropellorFile zonefile
+	baseprop = property ("dns primary for " ++ domain) $ makeChange $ do
+		writeZoneFile zone zonefile
+	withwarnings p = adjustProperty p $ \satisfy -> do
+		mapM_ warningMessage warnings
+		satisfy
+
 namedconf :: FilePath
 namedconf = "/etc/bind/named.conf.local"
 
@@ -56,7 +75,7 @@ confStanza c =
 		(map (\ip -> "\t\t" ++ fromIPAddr ip ++ ";") (confMasters c)) ++
 		[ "\t};" ]
 
--- | Rewrites the whole named.conf.local file to serve the specificed
+-- | Rewrites the whole named.conf.local file to serve the specified
 -- zones.
 servingZones :: [NamedConf] -> Property
 servingZones zs = hasContent namedconf (concatMap confStanza zs)
@@ -66,6 +85,10 @@ servingZones zs = hasContent namedconf (concatMap confStanza zs)
 
 -- | Generates a SOA with some fairly sane numbers in it.
 --
+-- The Domain is the domain to use in the SOA record. Typically
+-- something like ns1.example.com. Not the domain that this is the SOA
+-- record for.
+--
 -- The SerialNumber can be whatever serial number was used by the domain
 -- before propellor started managing it. Or 0 if the domain has only ever
 -- been managed by propellor.
@@ -73,19 +96,22 @@ servingZones zs = hasContent namedconf (concatMap confStanza zs)
 -- You do not need to increment the SerialNumber when making changes!
 -- Propellor will automatically add the number of commits in the git
 -- repository to the SerialNumber.
-mkSOA :: Domain -> SerialNumber -> [Record] -> SOA
-mkSOA d sn rs = SOA
+mkSOA :: Domain -> SerialNumber -> [Record] -> [Record] -> SOA
+mkSOA d sn rs1 rs2 = SOA
 	{ sDomain = AbsDomain d
 	, sSerial = sn
 	, sRefresh = hours 4
 	, sRetry = hours 1
 	, sExpire = 2419200 -- 4 weeks
 	, sTTL = hours 8
-	, sRecord = rs
+	, sRecord = rs1 ++ rs2
 	}
   where
 	hours n = n * 60 * 60
 
+rootAddressesFrom :: [Host] -> HostName -> [Record]
+rootAddressesFrom hosts hn = map Address (hostAddresses hn hosts)
+
 dValue :: BindDomain -> String
 dValue (RelDomain d) = d
 dValue (AbsDomain d) = d ++ "."
@@ -137,7 +163,8 @@ writeZoneFile z f = do
 	offset <- serialNumberOffset
 	let z' = nextSerialNumber
 		(adjustSerialNumber z (+ offset))
-		(succ oldserial)
+		oldserial
+	createDirectoryIfMissing True (takeDirectory f)
 	writeFile f (genZoneFile z')
 	writeZonePropellorFile f z'
 
diff --git a/Propellor/Types.hs b/Propellor/Types.hs
index ad822a8b..0e412e82 100644
--- a/Propellor/Types.hs
+++ b/Propellor/Types.hs
@@ -21,6 +21,7 @@ module Propellor.Types
 	, GpgKeyId
 	, SshKeyType(..)
 	, module Propellor.Types.OS
+	, module Propellor.Types.Dns
 	) where
 
 import Data.Monoid
@@ -31,6 +32,7 @@ import "MonadCatchIO-transformers" Control.Monad.CatchIO
 
 import Propellor.Types.Attr
 import Propellor.Types.OS
+import Propellor.Types.Dns
 
 data Host = Host [Property] SetAttr