Versions: mapWithVersion

ouroboros-consensus will use `Versions` which contain only hot peer
protocols (without warm protocols).  To preserve this interface,
we can  use `mapWithVersions` to add warm peer protocols.  This will be
only useful for warm peer applications which do not need to consult
version data, otherwise we will need to push warm peer protocols down to
consensus.

ouroboros-network-framework/src/Ouroboros/Network/Protocol/Handshake/Version.hs

@@ -23,6 +23,7 @@ module Ouroboros.Network.Protocol.Handshake.Version
   , simpleSingletonVersions
   , foldMapVersions
   , combineVersions
+  , mapWithVersion
   ) where
 
 import           Data.Foldable (toList)
@@ -70,6 +71,22 @@ instance Functor (Versions vNum extra) where
       where
         fmapSigma (Sigma t (Version (Application app) extra)) = Sigma t (Version (Application $ \x y -> f (app x y)) extra)
 
+
+mapWithVersion
+    :: forall vNum extra a b.
+       (vNum -> a -> b)
+    -> Versions vNum extra a
+    -> Versions vNum extra b
+mapWithVersion f (Versions vs) = Versions $ Map.mapWithKey g vs
+  where
+    g :: vNum -> Sigma (Version extra a) -> Sigma (Version extra b)
+    g vNum (Sigma vData (Version (Application app) extra)) =
+      Sigma vData
+            (Version
+              (Application $ \vData' vData'' -> f vNum (app vData' vData''))
+              extra)
+
+
 data Sigma f where
   Sigma :: !t -> !(f t) -> Sigma f