-
Notifications
You must be signed in to change notification settings - Fork 6
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
5e6a558
commit f18f29a
Showing
4 changed files
with
111 additions
and
80 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,80 @@ | ||
| const fs = require('fs') | ||
| const path = require('path') | ||
| const { getNetworkName } = require('common/src/network') | ||
| const similarity = require('string-similarity') | ||
| const storage = require('ethernaut-interact/src/internal/storage') | ||
| const debug = require('common/src/debug') | ||
|
|
||
| module.exports = async function suggestAbi({ abi, hre, address }) { | ||
| try { | ||
| const network = await getNetworkName(hre) | ||
|
|
||
| // Try to complete a partial abi path | ||
| if (abi && !isValidJsonFile(abi)) { | ||
| const abis = storage.readAbiFiles() | ||
| const matches = similarity.findBestMatch( | ||
| abi, | ||
| abis.map((a) => a.name), | ||
| ) | ||
| if (matches) { | ||
| const match = abis.find((a) => a.name === matches.bestMatch.target) | ||
| if (match) { | ||
| debug.log( | ||
| `Matched incoming ABI "${abi}" with known ABI "${match.path}"`, | ||
| 'interact', | ||
| ) | ||
|
|
||
| return match.path | ||
| } | ||
| } | ||
| } | ||
| if (abi) return abi | ||
|
|
||
| // Try to deduce the abi from previous interactions | ||
| // in the current network | ||
| if (address) { | ||
| abi = deduceAbiFromAddress(address, network) | ||
| if (abi) return abi | ||
| } else { | ||
| debug.log('Cannot deduce from address', 'interact') | ||
| } | ||
|
|
||
| return abi | ||
| } catch (err) { | ||
| debug.log(err, 'interact') | ||
| } | ||
| } | ||
|
|
||
| function deduceAbiFromAddress(address, network) { | ||
| const addresses = storage.readAddresses()[network] | ||
| if (!addresses) { | ||
| debug.log(`No addresses found for ${network}`, 'interact') | ||
| return undefined | ||
| } | ||
| debug.log( | ||
| `Found ${Object.keys(addresses).length} addresses on ${network}`, | ||
| 'interact', | ||
| ) | ||
|
|
||
| const abi = addresses[address] | ||
| if (!abi) { | ||
| debug.log(`No address entry found for ${address} on ${network}`, 'interact') | ||
| return | ||
| } | ||
|
|
||
| debug.log(`Found ...${abi.split('/').pop()} for ${address}`, 'interact') | ||
|
|
||
| return abi | ||
| } | ||
|
|
||
| function isValidJsonFile(abi) { | ||
| if (path.extname(abi) !== '.json') { | ||
| return false | ||
| } | ||
|
|
||
| if (!fs.existsSync(abi)) { | ||
| return false | ||
| } | ||
|
|
||
| return true | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters