Use ToJSON instance for getting era in query tip command

IntersectMBO · Mar 31, 2021 · 1340955 · 1340955
1 parent 6732b28
commit 1340955
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/cardano-api/src/Cardano/Api/Eras.hs b/cardano-api/src/Cardano/Api/Eras.hs
@@ -201,6 +201,9 @@ instance Bounded AnyCardanoEra where
     minBound = AnyCardanoEra ByronEra
     maxBound = AnyCardanoEra MaryEra
 
+instance ToJSON AnyCardanoEra where
+   toJSON (AnyCardanoEra era) = toJSON era
+
 -- | Like the 'AnyCardanoEra' constructor but does not demand a 'IsCardanoEra'
 -- class constraint.
 --

diff --git a/cardano-cli/src/Cardano/CLI/Shelley/Run/Query.hs b/cardano-cli/src/Cardano/CLI/Shelley/Run/Query.hs
@@ -156,24 +156,13 @@ runQueryTip (AnyConsensusModeParams cModeParams) network mOutFile = do
   mEpoch <- mEpochQuery anyEra consensusMode  localNodeConnInfo
   tip <- liftIO $ getLocalChainTip localNodeConnInfo
   let output = encodePretty
-        . toObject "era" (Just (eraString anyEra))
+        . toObject "era" (Just (toJSON anyEra))
         . toObject "epoch" mEpoch
         $ toJSON tip
   case mOutFile of
     Just (OutputFile fpath) -> liftIO $ LBS.writeFile fpath output
     Nothing                 -> liftIO $ LBS.putStrLn        output
   where
-    eraString
-      :: AnyCardanoEra
-      -> String
-    eraString (AnyCardanoEra era) =
-      case cardanoEraStyle era of
-        LegacyByronEra -> "byron"
-        ShelleyBasedEra sbe -> case sbe of
-          ShelleyBasedEraShelley -> "shelley"
-          ShelleyBasedEraAllegra -> "allegra"
-          ShelleyBasedEraMary -> "mary"
-
     mEpochQuery
       :: AnyCardanoEra
       -> ConsensusMode mode