Simplify prompts

packages/ethernaut-interact-ui/src/index.js

@@ -4,5 +4,10 @@ require('ethernaut-ui/src/index')
 require('ethernaut-interact/src/index')
 
 extendEnvironment((hre) => {
-  require('./prompts/tasks/contract')(hre)
+  const contract = hre.scopes.interact.tasks.contract
+  contract.paramDefinitions.abi.prompt = require('./prompts/abi')
+  contract.paramDefinitions.address.prompt = require('./prompts/address')
+  contract.paramDefinitions.fn.prompt = require('./prompts/fn')
+  contract.paramDefinitions.params.prompt = require('./prompts/params')
+  contract.paramDefinitions.value.prompt = require('./prompts/value')
 })

packages/ethernaut-network-ui/src/index.js

@@ -1,13 +1,32 @@
 const { extendEnvironment } = require('hardhat/config')
 
+const promptUrl = require('./prompts/url')
+const promptAlias = require('./prompts/alias')
+const promptFork = require('./prompts/fork')
+
 require('ethernaut-ui/src/index')
 require('ethernaut-network/src/index')
 
 extendEnvironment((hre) => {
-  require('./prompts/tasks/activate')(hre)
-  require('./prompts/tasks/add')(hre)
-  require('./prompts/tasks/edit')(hre)
-  require('./prompts/tasks/info')(hre)
-  require('./prompts/tasks/node')(hre)
-  require('./prompts/tasks/remove')(hre)
+  const add = hre.scopes.network.tasks.add
+  add.paramDefinitions.url.prompt = promptUrl
+
+  const activate = hre.scopes.network.tasks.activate
+  activate.positionalParamDefinitions.find((p) => p.name === 'alias').prompt =
+    promptAlias('Select a network to activate')
+
+  const edit = hre.scopes.network.tasks.edit
+  edit.positionalParamDefinitions.find((p) => p.name === 'alias').prompt =
+    promptAlias('Select a network to edit')
+
+  const remove = hre.scopes.network.tasks.remove
+  remove.positionalParamDefinitions.find((p) => p.name === 'alias').prompt =
+    promptAlias('Select a network')
+
+  const node = hre.scopes.network.tasks.node
+  node.paramDefinitions.fork.prompt = promptFork
+
+  const info = hre.scopes.network.tasks.info
+  info.positionalParamDefinitions.find((p) => p.name === 'alias').prompt =
+    promptAlias('Select a network')
 })

packages/ethernaut-util-ui/src/index.js

@@ -1,8 +1,11 @@
 const { extendEnvironment } = require('hardhat/config')
+const promptUnit = require('./prompts/unit')
 
 require('ethernaut-ui/src/index')
 require('ethernaut-util/src/index')
 
 extendEnvironment((hre) => {
-  require('./prompts/tasks/unit')(hre)
+  const unit = hre.scopes.util.tasks.unit
+  unit.paramDefinitions['from'].prompt = promptUnit
+  unit.paramDefinitions['to'].prompt = promptUnit
 })

packages/ethernaut-util-ui/src/prompts/tasks/unit.js → packages/ethernaut-util-ui/src/prompts/unit.js

@@ -1,14 +1,13 @@
 const { units } = require('ethernaut-util/src/tasks/unit')
 const prompt = require('common/src/prompt')
 
-module.exports = function setup(hre) {
-  const unit = hre.scopes.util.tasks.unit
-
-  unit.paramDefinitions['from'].prompt = promptUnit
-  unit.paramDefinitions['to'].prompt = promptUnit
-}
-
-async function promptUnit({ paramName, paramDefault, description, from, to }) {
+module.exports = async function promptUnit({
+  paramName,
+  paramDefault,
+  description,
+  from,
+  to,
+}) {
   const valueProvided =
     paramName === 'from' ? from !== undefined : to !== undefined
   const isDefault =

packages/ethernaut-wallet-ui/src/index.js

@@ -1,10 +1,19 @@
 const { extendEnvironment } = require('hardhat/config')
+const promptAlias = require('./prompts/alias')
 
 require('ethernaut-ui/src/index')
 require('ethernaut-wallet/src/index')
 
 extendEnvironment((hre) => {
-  require('./prompts/tasks/activate')(hre)
-  require('./prompts/tasks/info')(hre)
-  require('./prompts/tasks/remove')(hre)
+  const activate = hre.scopes.wallet.tasks.activate
+  activate.positionalParamDefinitions.find((p) => p.name === 'alias').prompt =
+    promptAlias('Select a wallet to activate')
+
+  const info = hre.scopes.wallet.tasks.info
+  info.positionalParamDefinitions.find((p) => p.name === 'alias').prompt =
+    promptAlias('Select a wallet')
+
+  const remove = hre.scopes.wallet.tasks.remove
+  remove.positionalParamDefinitions.find((p) => p.name === 'alias').prompt =
+    promptAlias('Select a wallet to remove')
 })