mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-11-27 12:56:01 +01:00
55 lines
1.4 KiB
JavaScript
55 lines
1.4 KiB
JavaScript
const version = 37
|
|
import { cloneDeep } from 'lodash'
|
|
import { util } from '@metamask/controllers'
|
|
|
|
/**
|
|
* The purpose of this migration is to update the address book state
|
|
* to the new schema with chainId as a key.
|
|
* and to add the isEns flag to all entries
|
|
*/
|
|
export default {
|
|
version,
|
|
migrate: async function (originalVersionedData) {
|
|
const versionedData = cloneDeep(originalVersionedData)
|
|
versionedData.meta.version = version
|
|
const state = versionedData.data
|
|
versionedData.data = transformState(state)
|
|
return versionedData
|
|
},
|
|
}
|
|
|
|
function transformState (state) {
|
|
|
|
if (state.AddressBookController) {
|
|
const ab = state.AddressBookController.addressBook
|
|
|
|
const chainIds = new Set()
|
|
const newAddressBook = {}
|
|
|
|
// add all of the chainIds to a set
|
|
for (const item in ab) {
|
|
chainIds.add(ab[item].chainId)
|
|
}
|
|
|
|
// fill the chainId object with the entries with the matching chainId
|
|
for (const id of chainIds.values()) {
|
|
// make an empty object entry for each chainId
|
|
newAddressBook[id] = {}
|
|
for (const address in ab) {
|
|
if (ab[address].chainId === id) {
|
|
|
|
ab[address].isEns = false
|
|
if (util.normalizeEnsName(ab[address].name)) {
|
|
ab[address].isEns = true
|
|
}
|
|
newAddressBook[id][address] = ab[address]
|
|
}
|
|
}
|
|
}
|
|
|
|
state.AddressBookController.addressBook = newAddressBook
|
|
}
|
|
|
|
return state
|
|
}
|