1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-23 09:52:26 +01:00
metamask-extension/app/scripts/migrations/037.js

54 lines
1.4 KiB
JavaScript
Raw Normal View History

import { cloneDeep } from 'lodash'
import { util } from '@metamask/controllers'
const version = 37
/**
* 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,
2020-11-03 00:41:28 +01:00
async migrate(originalVersionedData) {
const versionedData = cloneDeep(originalVersionedData)
versionedData.meta.version = version
const state = versionedData.data
versionedData.data = transformState(state)
return versionedData
},
}
2020-11-03 00:41:28 +01:00
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
2020-07-21 23:10:45 +02:00
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()) {
2020-11-03 00:41:28 +01:00
// 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
}