tx-generator: push Byron incompatibility upwards

Author: smelc

bench/tx-generator/src/Cardano/Benchmarking/Script/Core.hs

@@ -162,19 +162,23 @@ cancelBenchmark = do
 getLocalConnectInfo :: ActionM LocalNodeConnectInfo
 getLocalConnectInfo = makeLocalConnectInfo <$> getEnvNetworkId <*> getEnvSocketPath
 
-queryEra :: ActionM AnyCardanoEra
+queryEra :: ActionM AnyShelleyBasedEra
 queryEra = do
   localNodeConnectInfo <- getLocalConnectInfo
   chainTip  <- getLocalChainTip localNodeConnectInfo
-  mapExceptT liftIO .
+  AnyCardanoEra era <- mapExceptT liftIO .
     modifyError (Env.TxGenError . TxGenError . show) $
       queryNodeLocalState localNodeConnectInfo (SpecificPoint $ chainTipToChainPoint chainTip) QueryCurrentEra
+  caseByronOrShelleyBasedEra
+    (liftTxGenError $ TxGenError "queryEra Byron not supported")
+    (return . AnyShelleyBasedEra)
+    era
 
 queryRemoteProtocolParameters :: ActionM ProtocolParameters
 queryRemoteProtocolParameters = do
   localNodeConnectInfo <- getLocalConnectInfo
   chainTip  <- liftIO $ getLocalChainTip localNodeConnectInfo
-  AnyCardanoEra era <- queryEra
+  AnyShelleyBasedEra sbe <- queryEra
   let
     callQuery :: forall era.
                  QueryInEra era (Ledger.PParams (ShelleyLedgerEra era))
@@ -187,18 +191,15 @@ queryRemoteProtocolParameters = do
       liftIO $ BSL.writeFile pparamsFile $ prettyPrintOrdered pp'
       traceDebug $ "queryRemoteProtocolParameters : query result saved in: " ++ pparamsFile
       return pp'
-  caseByronOrShelleyBasedEra
-    (liftTxGenError $ TxGenError "queryRemoteProtocolParameters Byron not supported")
-    (\sbe -> callQuery $ QueryInShelleyBasedEra sbe QueryProtocolParameters)
-    era
+  callQuery $ QueryInShelleyBasedEra sbe QueryProtocolParameters
 
 getProtocolParameters :: ActionM ProtocolParameters
 getProtocolParameters = do
   getProtoParamMode  >>= \case
     ProtocolParameterQuery -> queryRemoteProtocolParameters
     ProtocolParameterLocal parameters -> return parameters
 
-waitForEra :: AnyCardanoEra -> ActionM ()
+waitForEra :: AnyShelleyBasedEra -> ActionM ()
 waitForEra era = do
   currentEra <- queryEra
   if currentEra == era

bench/tx-generator/src/Cardano/Benchmarking/Script/Types.hs

@@ -117,7 +117,7 @@ data Action where
   -- with testing and quick fixes.
   Reserved           :: [String] -> Action
   -- 'WaitForEra' loops doing delays/sleeps until the current era matches.
-  WaitForEra         :: !AnyCardanoEra -> Action
+  WaitForEra         :: !AnyShelleyBasedEra -> Action
   -- | 'SetProtocolParameters' has one option to read from a file and
   -- another to pass directly and just sets a state variable for
   -- the @protoParams@ field of 'Cardano.Benchmarking.Script.Env.Env'.