mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-12-23 09:52:26 +01:00
273 lines
8.3 KiB
JavaScript
273 lines
8.3 KiB
JavaScript
const ObservableStore = require('obs-store')
|
|
const normalizeAddress = require('eth-sig-util').normalize
|
|
const extend = require('xtend')
|
|
|
|
class PreferencesController {
|
|
|
|
/**
|
|
*
|
|
* @typedef {Object} PreferencesController
|
|
* @param {object} opts Overrides the defaults for the initial state of this.store
|
|
* @property {object} store The stored object containing a users preferences, stored in local storage
|
|
* @property {array} store.frequentRpcList A list of custom rpcs to provide the user
|
|
* @property {string} store.currentAccountTab Indicates the selected tab in the ui
|
|
* @property {array} store.tokens The tokens the user wants display in their token lists
|
|
* @property {boolean} store.useBlockie The users preference for blockie identicons within the UI
|
|
* @property {object} store.featureFlags A key-boolean map, where keys refer to features and booleans to whether the
|
|
* user wishes to see that feature
|
|
* @property {string} store.currentLocale The preferred language locale key
|
|
* @property {string} store.selectedAddress A hex string that matches the currently selected address in the app
|
|
*
|
|
*/
|
|
constructor (opts = {}) {
|
|
const initState = extend({
|
|
frequentRpcList: [],
|
|
currentAccountTab: 'history',
|
|
tokens: [],
|
|
useBlockie: false,
|
|
featureFlags: {},
|
|
currentLocale: opts.initLangCode,
|
|
identities: {},
|
|
}, opts.initState)
|
|
this.store = new ObservableStore(initState)
|
|
}
|
|
// PUBLIC METHODS
|
|
|
|
/**
|
|
* Setter for the `useBlockie` property
|
|
*
|
|
* @param {boolean} val Whether or not the user prefers blockie indicators
|
|
*
|
|
*/
|
|
setUseBlockie (val) {
|
|
this.store.updateState({ useBlockie: val })
|
|
}
|
|
|
|
/**
|
|
* Getter for the `useBlockie` property
|
|
*
|
|
* @returns {boolean} this.store.useBlockie
|
|
*
|
|
*/
|
|
getUseBlockie () {
|
|
return this.store.getState().useBlockie
|
|
}
|
|
|
|
/**
|
|
* Setter for the `currentLocale` property
|
|
*
|
|
* @param {string} key he preferred language locale key
|
|
*
|
|
*/
|
|
setCurrentLocale (key) {
|
|
this.store.updateState({ currentLocale: key })
|
|
}
|
|
|
|
/**
|
|
* Setter for the `selectedAddress` property
|
|
*
|
|
* @param {string} _address A new hex address for an account
|
|
* @returns {Promise<void>} Promise resolves with undefined
|
|
*
|
|
*/
|
|
setSelectedAddress (_address) {
|
|
return new Promise((resolve, reject) => {
|
|
const address = normalizeAddress(_address)
|
|
this.store.updateState({ selectedAddress: address })
|
|
resolve()
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Getter for the `selectedAddress` property
|
|
*
|
|
* @returns {string} The hex address for the currently selected account
|
|
*
|
|
*/
|
|
getSelectedAddress () {
|
|
return this.store.getState().selectedAddress
|
|
}
|
|
|
|
/**
|
|
* Contains data about tokens users add to their account.
|
|
* @typedef {Object} AddedToken
|
|
* @property {string} address - The hex address for the token contract. Will be all lower cased and hex-prefixed.
|
|
* @property {string} symbol - The symbol of the token, usually 3 or 4 capitalized letters
|
|
* {@link https://github.com/ethereum/EIPs/blob/master/EIPS/eip-20.md#symbol}
|
|
* @property {boolean} decimals - The number of decimals the token uses.
|
|
* {@link https://github.com/ethereum/EIPs/blob/master/EIPS/eip-20.md#decimals}
|
|
*/
|
|
|
|
/**
|
|
* Adds a new token to the token array, or updates the token if passed an address that already exists.
|
|
* Modifies the existing tokens array from the store. All objects in the tokens array array AddedToken objects.
|
|
* @see AddedToken {@link AddedToken}
|
|
*
|
|
* @param {string} rawAddress Hex address of the token contract. May or may not be a checksum address.
|
|
* @param {string} symbol The symbol of the token
|
|
* @param {number} decimals The number of decimals the token uses.
|
|
* @returns {Promise<array>} Promises the new array of AddedToken objects.
|
|
*
|
|
*/
|
|
async addToken (rawAddress, symbol, decimals) {
|
|
const address = normalizeAddress(rawAddress)
|
|
const newEntry = { address, symbol, decimals }
|
|
|
|
const tokens = this.store.getState().tokens
|
|
const previousEntry = tokens.find((token, index) => {
|
|
return token.address === address
|
|
})
|
|
const previousIndex = tokens.indexOf(previousEntry)
|
|
|
|
if (previousEntry) {
|
|
tokens[previousIndex] = newEntry
|
|
} else {
|
|
tokens.push(newEntry)
|
|
}
|
|
|
|
this.store.updateState({ tokens })
|
|
|
|
return Promise.resolve(tokens)
|
|
}
|
|
|
|
/**
|
|
* Removes a specified token from the tokens array.
|
|
*
|
|
* @param {string} rawAddress Hex address of the token contract to remove.
|
|
* @returns {Promise<array>} The new array of AddedToken objects
|
|
*
|
|
*/
|
|
removeToken (rawAddress) {
|
|
const tokens = this.store.getState().tokens
|
|
|
|
const updatedTokens = tokens.filter(token => token.address !== rawAddress)
|
|
|
|
this.store.updateState({ tokens: updatedTokens })
|
|
return Promise.resolve(updatedTokens)
|
|
}
|
|
|
|
/**
|
|
* A getter for the `tokens` property
|
|
*
|
|
* @returns {array} The current array of AddedToken objects
|
|
*
|
|
*/
|
|
getTokens () {
|
|
return this.store.getState().tokens
|
|
}
|
|
|
|
/**
|
|
* Sets a custom label for an account
|
|
* @param {string} account the account to set a label for
|
|
* @param {string} label the custom label for the account
|
|
* @return {Promise<string>}
|
|
*/
|
|
setAccountLabel (account, label) {
|
|
const address = normalizeAddress(account)
|
|
const {identities} = this.store.getState()
|
|
identities[address] = identities[address] || {}
|
|
identities[address].name = label
|
|
this.store.updateState({ identities })
|
|
return Promise.resolve(label)
|
|
}
|
|
|
|
/**
|
|
* Gets an updated rpc list from this.addToFrequentRpcList() and sets the `frequentRpcList` to this update list.
|
|
*
|
|
* @param {string} _url The the new rpc url to add to the updated list
|
|
* @returns {Promise<void>} Promise resolves with undefined
|
|
*
|
|
*/
|
|
updateFrequentRpcList (_url) {
|
|
return this.addToFrequentRpcList(_url)
|
|
.then((rpcList) => {
|
|
this.store.updateState({ frequentRpcList: rpcList })
|
|
return Promise.resolve()
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Setter for the `currentAccountTab` property
|
|
*
|
|
* @param {string} currentAccountTab Specifies the new tab to be marked as current
|
|
* @returns {Promise<void>} Promise resolves with undefined
|
|
*
|
|
*/
|
|
setCurrentAccountTab (currentAccountTab) {
|
|
return new Promise((resolve, reject) => {
|
|
this.store.updateState({ currentAccountTab })
|
|
resolve()
|
|
})
|
|
}
|
|
|
|
/**
|
|
* Returns an updated rpcList based on the passed url and the current list.
|
|
* The returned list will have a max length of 2. If the _url currently exists it the list, it will be moved to the
|
|
* end of the list. The current list is modified and returned as a promise.
|
|
*
|
|
* @param {string} _url The rpc url to add to the frequentRpcList.
|
|
* @returns {Promise<array>} The updated frequentRpcList.
|
|
*
|
|
*/
|
|
addToFrequentRpcList (_url) {
|
|
const rpcList = this.getFrequentRpcList()
|
|
const index = rpcList.findIndex((element) => { return element === _url })
|
|
if (index !== -1) {
|
|
rpcList.splice(index, 1)
|
|
}
|
|
if (_url !== 'http://localhost:8545') {
|
|
rpcList.push(_url)
|
|
}
|
|
if (rpcList.length > 2) {
|
|
rpcList.shift()
|
|
}
|
|
return Promise.resolve(rpcList)
|
|
}
|
|
|
|
/**
|
|
* Getter for the `frequentRpcList` property.
|
|
*
|
|
* @returns {array<string>} An array of one or two rpc urls.
|
|
*
|
|
*/
|
|
getFrequentRpcList () {
|
|
return this.store.getState().frequentRpcList
|
|
}
|
|
|
|
/**
|
|
* Updates the `featureFlags` property, which is an object. One property within that object will be set to a boolean.
|
|
*
|
|
* @param {string} feature A key that corresponds to a UI feature.
|
|
* @param {boolean} activated Indicates whether or not the UI feature should be displayed
|
|
* @returns {Promise<object>} Promises a new object; the updated featureFlags object.
|
|
*
|
|
*/
|
|
setFeatureFlag (feature, activated) {
|
|
const currentFeatureFlags = this.store.getState().featureFlags
|
|
const updatedFeatureFlags = {
|
|
...currentFeatureFlags,
|
|
[feature]: activated,
|
|
}
|
|
|
|
this.store.updateState({ featureFlags: updatedFeatureFlags })
|
|
|
|
return Promise.resolve(updatedFeatureFlags)
|
|
}
|
|
|
|
/**
|
|
* A getter for the `featureFlags` property
|
|
*
|
|
* @returns {object} A key-boolean map, where keys refer to features and booleans to whether the
|
|
* user wishes to see that feature
|
|
*
|
|
*/
|
|
getFeatureFlags () {
|
|
return this.store.getState().featureFlags
|
|
}
|
|
//
|
|
// PRIVATE METHODS
|
|
//
|
|
}
|
|
|
|
module.exports = PreferencesController
|