mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-12-23 09:52:26 +01:00
247659ca65
* Nix notification for Share Address * Add Connections settings tab in place of privacy mode toggle * Split ProviderApprovalController into two stores * Remove privacyMode feature flag altogether * Add migration to remove privacyMode feature flag
160 lines
5.5 KiB
JavaScript
160 lines
5.5 KiB
JavaScript
const ObservableStore = require('obs-store')
|
|
const SafeEventEmitter = require('safe-event-emitter')
|
|
const createAsyncMiddleware = require('json-rpc-engine/src/createAsyncMiddleware')
|
|
|
|
/**
|
|
* A controller that services user-approved requests for a full Ethereum provider API
|
|
*/
|
|
class ProviderApprovalController extends SafeEventEmitter {
|
|
/**
|
|
* Creates a ProviderApprovalController
|
|
*
|
|
* @param {Object} [config] - Options to configure controller
|
|
*/
|
|
constructor ({ closePopup, initState, keyringController, openPopup, preferencesController } = {}) {
|
|
super()
|
|
this.closePopup = closePopup
|
|
this.keyringController = keyringController
|
|
this.openPopup = openPopup
|
|
this.preferencesController = preferencesController
|
|
this.memStore = new ObservableStore({
|
|
providerRequests: [],
|
|
})
|
|
|
|
const defaultState = { approvedOrigins: {} }
|
|
this.store = new ObservableStore(Object.assign(defaultState, initState))
|
|
}
|
|
|
|
/**
|
|
* Called when a user approves access to a full Ethereum provider API
|
|
*
|
|
* @param {object} opts - opts for the middleware contains the origin for the middleware
|
|
*/
|
|
createMiddleware ({ origin, getSiteMetadata }) {
|
|
return createAsyncMiddleware(async (req, res, next) => {
|
|
// only handle requestAccounts
|
|
if (req.method !== 'eth_requestAccounts') return next()
|
|
// if already approved or privacy mode disabled, return early
|
|
const isUnlocked = this.keyringController.memStore.getState().isUnlocked
|
|
if (this.shouldExposeAccounts(origin) && isUnlocked) {
|
|
res.result = [this.preferencesController.getSelectedAddress()]
|
|
return
|
|
}
|
|
// register the provider request
|
|
const metadata = await getSiteMetadata(origin)
|
|
this._handleProviderRequest(origin, metadata.name, metadata.icon)
|
|
// wait for resolution of request
|
|
const approved = await new Promise(resolve => this.once(`resolvedRequest:${origin}`, ({ approved }) => resolve(approved)))
|
|
if (approved) {
|
|
res.result = [this.preferencesController.getSelectedAddress()]
|
|
} else {
|
|
throw new Error('User denied account authorization')
|
|
}
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Called when a tab requests access to a full Ethereum provider API
|
|
*
|
|
* @param {string} origin - Origin of the window requesting full provider access
|
|
* @param {string} siteTitle - The title of the document requesting full provider access
|
|
* @param {string} siteImage - The icon of the window requesting full provider access
|
|
*/
|
|
_handleProviderRequest (origin, siteTitle, siteImage) {
|
|
const { providerRequests } = this.memStore.getState()
|
|
this.memStore.updateState({
|
|
providerRequests: [
|
|
...providerRequests,
|
|
{ origin, siteTitle, siteImage },
|
|
],
|
|
})
|
|
const isUnlocked = this.keyringController.memStore.getState().isUnlocked
|
|
const { approvedOrigins } = this.store.getState()
|
|
const originAlreadyHandled = approvedOrigins[origin]
|
|
if (originAlreadyHandled && isUnlocked) {
|
|
return
|
|
}
|
|
this.openPopup && this.openPopup()
|
|
}
|
|
|
|
/**
|
|
* Called when a user approves access to a full Ethereum provider API
|
|
*
|
|
* @param {string} origin - origin of the domain that had provider access approved
|
|
*/
|
|
approveProviderRequestByOrigin (origin) {
|
|
if (this.closePopup) {
|
|
this.closePopup()
|
|
}
|
|
|
|
const { approvedOrigins } = this.store.getState()
|
|
const { providerRequests } = this.memStore.getState()
|
|
const providerRequest = providerRequests.find((request) => request.origin === origin)
|
|
const remainingProviderRequests = providerRequests.filter(request => request.origin !== origin)
|
|
this.store.updateState({
|
|
approvedOrigins: {
|
|
...approvedOrigins,
|
|
[origin]: {
|
|
siteTitle: providerRequest ? providerRequest.siteTitle : null,
|
|
siteImage: providerRequest ? providerRequest.siteImage : null,
|
|
},
|
|
},
|
|
})
|
|
this.memStore.updateState({ providerRequests: remainingProviderRequests })
|
|
this.emit(`resolvedRequest:${origin}`, { approved: true })
|
|
}
|
|
|
|
/**
|
|
* Called when a tab rejects access to a full Ethereum provider API
|
|
*
|
|
* @param {string} origin - origin of the domain that had provider access approved
|
|
*/
|
|
rejectProviderRequestByOrigin (origin) {
|
|
if (this.closePopup) {
|
|
this.closePopup()
|
|
}
|
|
|
|
const { approvedOrigins } = this.store.getState()
|
|
const { providerRequests } = this.memStore.getState()
|
|
const remainingProviderRequests = providerRequests.filter(request => request.origin !== origin)
|
|
|
|
// We're cloning and deleting keys here because we don't want to keep unneeded keys
|
|
const _approvedOrigins = Object.assign({}, approvedOrigins)
|
|
delete _approvedOrigins[origin]
|
|
|
|
this.store.putState({ approvedOrigins: _approvedOrigins })
|
|
this.memStore.putState({ providerRequests: remainingProviderRequests })
|
|
this.emit(`resolvedRequest:${origin}`, { approved: false })
|
|
}
|
|
|
|
/**
|
|
* Clears any approvals for user-approved origins
|
|
*/
|
|
clearApprovedOrigins () {
|
|
this.store.updateState({
|
|
approvedOrigins: {},
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Determines if a given origin should have accounts exposed
|
|
*
|
|
* @param {string} origin - Domain origin to check for approval status
|
|
* @returns {boolean} - True if the origin has been approved
|
|
*/
|
|
shouldExposeAccounts (origin) {
|
|
return Boolean(this.store.getState().approvedOrigins[origin])
|
|
}
|
|
|
|
/**
|
|
* Returns a merged state representation
|
|
* @return {object}
|
|
* @private
|
|
*/
|
|
_getMergedState () {
|
|
return Object.assign({}, this.memStore.getState(), this.store.getState())
|
|
}
|
|
}
|
|
|
|
module.exports = ProviderApprovalController
|