You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.4 KiB
53 lines
1.4 KiB
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,
|
|
async migrate(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;
|
|
}
|
|
|