1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-11-27 12:56:01 +01:00
metamask-extension/app/scripts/migrations/037.js
2020-07-21 18:40:45 -02:30

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
Object.values(ab).forEach((v) => {
chainIds.add(v.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
}