nifty-wallet/ui/lib/icon-factory.js

67 lines
1.9 KiB
JavaScript

let iconFactory
const isValidAddress = require('ethereumjs-util').isValidAddress
const { checksumAddress } = require('../app/util')
const contractMap = require('eth-contract-metadata')
const colors = require('../../colors')
module.exports = function (jazzicon) {
if (!iconFactory) {
iconFactory = new IconFactory(jazzicon)
}
return iconFactory
}
function IconFactory (jazzicon) {
jazzicon.setColorsPalette(colors)
this.jazzicon = jazzicon
this.cache = {}
}
IconFactory.prototype.iconForAddress = function (address, diameter) {
const addr = checksumAddress(address)
if (iconExistsFor(addr)) {
return imageElFor(addr)
}
return this.generateIdenticonSvg(address, diameter)
}
// returns svg dom element
IconFactory.prototype.generateIdenticonSvg = function (address, diameter) {
const cacheId = `${address}:${diameter}`
// check cache, lazily generate and populate cache
const identicon = this.cache[cacheId] || (this.cache[cacheId] = this.generateNewIdenticon(address, diameter))
// create a clean copy so you can modify it
const cleanCopy = identicon.cloneNode(true)
return cleanCopy
}
// creates a new identicon
IconFactory.prototype.generateNewIdenticon = function (address, diameter) {
const numericRepresentation = jsNumberForAddress(address)
const identicon = this.jazzicon.generateIdenticon(diameter, numericRepresentation)
return identicon
}
// util
function iconExistsFor (address) {
return contractMap[address] && isValidAddress(address) && contractMap[address].logo
}
function imageElFor (address) {
const contract = contractMap[address]
const fileName = contract.logo
const path = `images/contract/${fileName}`
const img = document.createElement('img')
img.src = path
img.style.width = '100%'
return img
}
function jsNumberForAddress (address) {
const addr = address.slice(2, 10)
const seed = parseInt(addr, 16)
return seed
}