Implementing NetworkId parser

IntersectMBO · Jun 5, 2023 · 23b81f8 · 23b81f8
1 parent 07a74b1
commit 23b81f8
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 2 deletions.
diff --git a/cardano-testnet/src/Parsers/Babbage.hs b/cardano-testnet/src/Parsers/Babbage.hs
@@ -9,7 +9,7 @@ import           Prelude
 import           Options.Applicative
 import qualified Options.Applicative as OA
 
-import           Cardano.CLI.Common.Parsers
+import           Cardano.CLI.Common.Parsers hiding (pNetworkId)
 
 import           Testnet
 import           Testnet.Options
@@ -44,7 +44,7 @@ optsTestnet = BabbageTestnetOptions
       <>  OA.showDefault
       <>  OA.value (babbageSecurityParam defaultTestnetOptions)
       )
-  <*> error "testnetmagic"
+  <*> pNetworkId
   <*> OA.option auto
       (   OA.long "total-balance"
       <>  OA.help "Total balance"

diff --git a/cardano-testnet/src/Testnet/Parsers.hs b/cardano-testnet/src/Testnet/Parsers.hs
@@ -46,6 +46,7 @@ commands =
     , fmap (Help pref opts) (subparser cmdHelp)
     ]
 
+
 runTestnetCmd :: CardanoTestnetCommands -> IO ()
 runTestnetCmd = \case
   StartBabbageTestnet cmdOpts -> runBabbageOptions cmdOpts

diff --git a/cardano-testnet/src/Testnet/Util/Cli.hs b/cardano-testnet/src/Testnet/Util/Cli.hs
@@ -25,6 +25,10 @@ module Testnet.Util.Cli
   , cliByronSigningKeyAddress
   ) where
 
+import           Cardano.Api (NetworkId (..), NetworkMagic (..), bounded)
+
+import           Options.Applicative
+import qualified Options.Applicative as Opt
 import           System.FilePath.Posix
 
 import qualified Hedgehog.Extras.Test.Base as H
@@ -136,3 +140,22 @@ cliByronSigningKeyAddress tmp testnetMagic (File key) destPath = do
       ]
   H.writeFile addrPath addr
   return $ File addrPath
+
+pNetworkId :: Parser NetworkId
+pNetworkId =
+  Opt.flag' Mainnet $ mconcat
+     [ Opt.long "mainnet"
+     , Opt.help $ mconcat
+       [ "Use the mainnet magic id. This overrides the CARDANO_NODE_NETWORK_ID "
+       , "environment variable"
+       ]
+     ]
+  <|>
+  fmap (Testnet . NetworkMagic) $ Opt.option (bounded "TESTNET_MAGIC") $ mconcat
+   [ Opt.long "testnet-magic"
+   , Opt.metavar "NATURAL"
+   , Opt.help $ mconcat
+     [ "Specify a testnet magic id. This overrides the CARDANO_NODE_NETWORK_ID "
+     , "environment variable"
+     ]
+   ]