all files / src/modules/ attributes.ts

35% Statements 14/40
0% Branches 0/20
20% Functions 1/5
34.21% Lines 13/38
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78                                        44×                                                                                          
import { ElementVNode, Module } from '../'
 
import { BaseModule } from './BaseModule'
import { emptyVNode } from './emptyVNode'
 
export function createAttributesModule(): Module {
  return new AttributesModule()
}
 
const NAMESPACE_URIS = {
  xlink: 'http://www.w3.org/1999/xlink',
}
 
const booleanAttributes: Array<string> = [
  'allowfullscreen', 'async', 'autofocus', 'autoplay', 'checked', 'compact',
  'controls', 'declare', 'default', 'defaultchecked', 'defaultmuted',
  'defaultselected', 'defer', 'disabled', 'draggable', 'enabled',
  'formnovalidate', 'hidden', 'indeterminate', 'inert', 'ismap', 'itemscope',
  'loop', 'multiple', 'muted', 'nohref', 'noresize', 'noshade', 'novalidate',
  'nowrap', 'open', 'pauseonexit', 'readonly', 'required', 'reversed', 'scoped',
  'seamless', 'selected', 'sortable', 'spellcheck', 'translate', 'truespeed',
  'typemustmatch', 'visible',
]
 
const booleanAttributeDictionary: any = Object.create(null)
 
for (let i = 0, count = booleanAttributes.length; i < count; i++)
  booleanAttributeDictionary[booleanAttributes[i]] = true
 
// attributes module
class AttributesModule extends BaseModule {
  public create(vNode: ElementVNode) {
    updateAttributes(emptyVNode, vNode)
  }
 
  public update(formerVNode: ElementVNode, vNode: ElementVNode) {
    updateAttributes(formerVNode, vNode)
  }
}
 
function updateAttributes(formerVNode: ElementVNode, vNode: ElementVNode) {
  let attributeValue: any
  let formerAttributeValue: any
  const element: Element = vNode.element
  let formerAttributes: any = formerVNode.props.attrs
  let attributes: any = vNode.props.attrs
  let attributeParts: Array<string>
 
  if (!formerAttributes && !attributes) return
 
  formerAttributes = formerAttributes || {}
  attributes = attributes || {}
 
  for (const key in attributes) {
    attributeValue = attributes[key]
    formerAttributeValue = formerAttributes[key]
 
    if (formerAttributeValue !== attributeValue) {
      if (!attributeValue && booleanAttributeDictionary[key])
        element.removeAttribute(key)
 
      else {
        attributeParts = key.split(':')
 
        if (attributeParts.length > 1 && NAMESPACE_URIS.hasOwnProperty(attributeParts[0]))
          element.setAttributeNS((NAMESPACE_URIS as any)[attributeParts[0]], key, attributeValue)
 
        else
          element.setAttribute(key, attributeValue)
      }
    }
  }
 
  for (const key in formerAttributes)
    if (!(key in attributes))
      element.removeAttribute(key)
}