metamask-extension/ui/app/components/ui/jazzicon/jazzicon.component.js

import React, { createRef, PureComponent } from 'react'
import PropTypes from 'prop-types'
import jazzicon from 'jazzicon'
import iconFactoryGenerator from '../../../../lib/icon-factory'

const iconFactory = iconFactoryGenerator(jazzicon)

/**
 * Wrapper around the jazzicon library to return a React component, as the library returns an
 * HTMLDivElement which needs to be appended.
 */
export default class Jazzicon extends PureComponent {
  static propTypes = {
    address: PropTypes.string.isRequired,
    className: PropTypes.string,
    diameter: PropTypes.number,
    style: PropTypes.object,
  }

  static defaultProps = {
    diameter: 46,
  }

  container = createRef()

  componentDidMount () {
    this.appendJazzicon()
  }

  componentDidUpdate (prevProps) {
    const { address: prevAddress, diameter: prevDiameter } = prevProps
    const { address, diameter } = this.props

    if (address !== prevAddress || diameter !== prevDiameter) {
      this.removeExistingChildren()
      this.appendJazzicon()
    }
  }

  removeExistingChildren () {
    const { children } = this.container.current

    for (let i = 0; i < children.length; i++) {
      this.container.current.removeChild(children[i])
    }
  }

  appendJazzicon () {
    const { address, diameter } = this.props
    const image = iconFactory.iconForAddress(address, diameter)
    this.container.current.appendChild(image)
  }

  render () {
    const { className, style } = this.props

    return (
      <div
        className={className}
        ref={this.container}
        style={style}
      />
    )
  }
}