From 56a049a4fba10b1abdb1b2b72a47a0f0f1732626 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment=20Hurlin?= <clement.hurlin@tweag.io>
Date: Mon, 4 Dec 2023 11:32:10 +0100
Subject: [PATCH] create-staked: simplify some code

---
 .../src/Cardano/CLI/EraBased/Run/Genesis.hs   | 62 +++++++------------
 1 file changed, 23 insertions(+), 39 deletions(-)

diff --git a/cardano-cli/src/Cardano/CLI/EraBased/Run/Genesis.hs b/cardano-cli/src/Cardano/CLI/EraBased/Run/Genesis.hs
index 92e03ad0fc..86c4289572 100644
--- a/cardano-cli/src/Cardano/CLI/EraBased/Run/Genesis.hs
+++ b/cardano-cli/src/Cardano/CLI/EraBased/Run/Genesis.hs
@@ -167,14 +167,9 @@ runGenesisKeyGenGenesisCmd
     } = do
     skey <- liftIO $ generateSigningKey AsGenesisKey
     let vkey = getVerificationKey skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile signingKeyPath
-      $ textEnvelopeToJSON (Just skeyDesc) skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile verificationKeyPath
-      $ textEnvelopeToJSON (Just Key.genesisVkeyDesc) vkey
+    firstExceptT GenesisCmdGenesisFileError . newExceptT $ do
+      void $ writeLazyByteStringFile signingKeyPath $ textEnvelopeToJSON (Just skeyDesc) skey
+      writeLazyByteStringFile verificationKeyPath $ textEnvelopeToJSON (Just Key.genesisVkeyDesc) vkey
   where
     skeyDesc :: TextEnvelopeDescr
     skeyDesc = "Genesis Signing Key"
@@ -191,21 +186,16 @@ runGenesisKeyGenDelegateCmd
     } = do
     skey <- liftIO $ generateSigningKey AsGenesisDelegateKey
     let vkey = getVerificationKey skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile signingKeyPath
-      $ textEnvelopeToJSON (Just skeyDesc) skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile verificationKeyPath
-      $ textEnvelopeToJSON (Just Key.genesisVkeyDelegateDesc) vkey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile opCertCounterPath
-      $ textEnvelopeToJSON (Just certCtrDesc)
-      $ OperationalCertificateIssueCounter
-          initialCounter
-          (castVerificationKey vkey)  -- Cast to a 'StakePoolKey'
+    firstExceptT GenesisCmdGenesisFileError . newExceptT $ do
+      void $ writeLazyByteStringFile signingKeyPath
+        $ textEnvelopeToJSON (Just skeyDesc) skey
+      void $ writeLazyByteStringFile verificationKeyPath
+        $ textEnvelopeToJSON (Just Key.genesisVkeyDelegateDesc) vkey
+      writeLazyByteStringFile opCertCounterPath
+        $ textEnvelopeToJSON (Just certCtrDesc)
+        $ OperationalCertificateIssueCounter
+            initialCounter
+            (castVerificationKey vkey)  -- Cast to a 'StakePoolKey'
   where
     skeyDesc, certCtrDesc :: TextEnvelopeDescr
     skeyDesc = "Genesis delegate operator key"
@@ -223,14 +213,11 @@ runGenesisKeyGenDelegateVRF ::
 runGenesisKeyGenDelegateVRF vkeyPath skeyPath = do
     skey <- liftIO $ generateSigningKey AsVrfKey
     let vkey = getVerificationKey skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile skeyPath
-      $ textEnvelopeToJSON (Just skeyDesc) skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile vkeyPath
-      $ textEnvelopeToJSON (Just vkeyDesc) vkey
+    firstExceptT GenesisCmdGenesisFileError . newExceptT $ do
+      void $ writeLazyByteStringFile skeyPath
+        $ textEnvelopeToJSON (Just skeyDesc) skey
+      writeLazyByteStringFile vkeyPath
+        $ textEnvelopeToJSON (Just vkeyDesc) vkey
   where
     skeyDesc, vkeyDesc :: TextEnvelopeDescr
     skeyDesc = "VRF Signing Key"
@@ -247,14 +234,11 @@ runGenesisKeyGenUTxOCmd
     } = do
     skey <- liftIO $ generateSigningKey AsGenesisUTxOKey
     let vkey = getVerificationKey skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile signingKeyPath
-      $ textEnvelopeToJSON (Just skeyDesc) skey
-    firstExceptT GenesisCmdGenesisFileError
-      . newExceptT
-      $ writeLazyByteStringFile verificationKeyPath
-      $ textEnvelopeToJSON (Just vkeyDesc) vkey
+    firstExceptT GenesisCmdGenesisFileError . newExceptT $ do
+      void $ writeLazyByteStringFile signingKeyPath
+        $ textEnvelopeToJSON (Just skeyDesc) skey
+      writeLazyByteStringFile verificationKeyPath
+        $ textEnvelopeToJSON (Just vkeyDesc) vkey
   where
     skeyDesc, vkeyDesc :: TextEnvelopeDescr
     skeyDesc = "Genesis Initial UTxO Signing Key"