Tor: enable multiple ports hidden services with hiddenService'/hiddenServiceAvailable'

1 files changed, 14 insertions, 6 deletions

diff --git a/src/Propellor/Property/Tor.hs b/src/Propellor/Property/Tor.hs
index ea9f39ed..52b38230 100644
--- a/src/Propellor/Property/Tor.hs
+++ b/src/Propellor/Property/Tor.hs
@@ -124,22 +124,30 @@ bandwidthRate' s divby = case readSize dataUnits s of
 -- If used without `hiddenServiceData`, tor will generate a new
 -- private key.
 hiddenService :: HiddenServiceName -> Port -> Property DebianLike
-hiddenService hn (Port port) = ConfFile.adjustSection
-	(unwords ["hidden service", hn, "available on port", show port])
+hiddenService hn port = hiddenService' hn [port]
+
+hiddenService' :: HiddenServiceName -> [Port] -> Property DebianLike
+hiddenService' hn ports = ConfFile.adjustSection
+	(unwords ["hidden service", hn, "available on ports", intercalate "," (map show ports')])
 	(== oniondir)
 	(not . isPrefixOf "HiddenServicePort")
-	(const [oniondir, onionport])
-	(++ [oniondir, onionport])
+	(const (oniondir : onionports))
+	(++ oniondir : onionports)
 	mainConfig
 	`onChange` restarted
   where
 	oniondir = unwords ["HiddenServiceDir", varLib </> hn]
-	onionport = unwords ["HiddenServicePort", show port, "127.0.0.1:" ++ show port]
+	onionports = map onionport ports'
+	ports' = sort ports
+	onionport port = unwords ["HiddenServicePort", show port, "127.0.0.1:" ++ show port]
 
 -- | Same as `hiddenService` but also causes propellor to display
 -- the onion address of the hidden service.
 hiddenServiceAvailable :: HiddenServiceName -> Port -> Property DebianLike
-hiddenServiceAvailable hn port = hiddenServiceHostName $ hiddenService hn port
+hiddenServiceAvailable hn port = hiddenServiceAvailable' hn [port]
+
+hiddenServiceAvailable' :: HiddenServiceName -> [Port] -> Property DebianLike
+hiddenServiceAvailable' hn ports = hiddenServiceHostName $ hiddenService hn ports
   where
 	hiddenServiceHostName p =  adjustPropertySatisfy p $ \satisfy -> do
 		r <- satisfy