mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-11-24 02:58:09 +01:00
34 lines
1.6 KiB
JavaScript
34 lines
1.6 KiB
JavaScript
|
const mergeMiddleware = require('json-rpc-engine/src/mergeMiddleware')
|
||
|
const createAsyncMiddleware = require('json-rpc-engine/src/createAsyncMiddleware')
|
||
|
const createFetchMiddleware = require('eth-json-rpc-middleware/fetch')
|
||
|
const createBlockRefMiddleware = require('eth-json-rpc-middleware/block-ref')
|
||
|
const createBlockRefMiddleware = require('eth-json-rpc-middleware/block-cache')
|
||
|
const BlockTracker = require('eth-block-tracker')
|
||
|
|
||
|
module.exports = createLocalhostClient
|
||
|
|
||
|
function createLocalhostClient() {
|
||
|
const fetchMiddleware = createFetchMiddleware({ rpcUrl: 'http://localhost:8545/' })
|
||
|
const blockProvider = providerFromMiddleware(fetchMiddleware)
|
||
|
const blockTracker = new BlockTracker({ provider: blockProvider, pollingInterval: 1000 })
|
||
|
|
||
|
const networkMiddleware = mergeMiddleware([
|
||
|
createBlockRefMiddleware({ blockTracker }),
|
||
|
createBlockTrackerInspectorMiddleware({ blockTracker }),
|
||
|
fetchMiddleware,
|
||
|
])
|
||
|
return { networkMiddleware, blockTracker }
|
||
|
}
|
||
|
|
||
|
// inspect if response contains a block ref higher than our latest block
|
||
|
const futureBlockRefRequests = ['eth_getTransactionByHash', 'eth_getTransactionReceipt']
|
||
|
function createBlockTrackerInspectorMiddleware ({ blockTracker }) {
|
||
|
return createAsyncMiddleware(async (req, res, next) => {
|
||
|
if (!futureBlockRefRequests.includes(req.method)) return next()
|
||
|
await next()
|
||
|
const blockNumber = Number.parseInt(res.result.blockNumber, 16)
|
||
|
const currentBlockNumber = Number.parseInt(blockTracker.getCurrentBlock(), 16)
|
||
|
if (blockNumber > currentBlockNumber) await blockTracker.checkForLatestBlock()
|
||
|
})
|
||
|
}
|