Simplify anyAddressInEra

IntersectMBO · Oct 28, 2023 · 93d71ad · 93d71ad
1 parent c8ed7de
commit 93d71ad
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/cardano-api/internal/Cardano/Api/Address.hs b/cardano-api/internal/Cardano/Api/Address.hs
@@ -489,13 +489,13 @@ anyAddressInShelleyBasedEra sbe = \case
 anyAddressInEra :: CardanoEra era
                 -> AddressAny
                 -> Either String (AddressInEra era)
-anyAddressInEra _ (AddressByron addr) =
+anyAddressInEra era = \case
+  AddressByron addr ->
     Right (AddressInEra ByronAddressInAnyEra addr)
-
-anyAddressInEra era (AddressShelley addr) =
-    case cardanoEraStyle era of
-      LegacyByronEra       -> Left "Expected Byron based era address"
-      ShelleyBasedEra era' -> Right (AddressInEra (ShelleyAddressInEra era') addr)
+  AddressShelley addr ->
+    forEraInEon era
+      (Left "Expected Byron based era address")
+      (\sbe -> Right (AddressInEra (ShelleyAddressInEra sbe) addr))
 
 toAddressAny :: Address addr -> AddressAny
 toAddressAny a@ShelleyAddress{} = AddressShelley a