1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-11-28 05:12:18 +01:00
metamask-extension/app/scripts/migrations/037.js

55 lines
1.4 KiB
JavaScript
Raw Normal View History

const version = 37
import { cloneDeep } from 'lodash'
import { util } from 'gaba'
/**
* 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
}