all files / src/modules/ styles.ts

27.38% Statements 23/84
0% Branches 0/37
6.25% Functions 1/16
27.16% Lines 22/81
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 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142                                                                                                                                                                                                                                               
import { ElementVNode, Module } from '../'
 
import { BaseModule } from './BaseModule'
import { emptyVNode } from './emptyVNode'
 
export function createStylesModule(): Module {
  return new StylesModule()
}
 
class StylesModule extends BaseModule {
  public pre() {
    setRequestAnimationFrame()
  }
 
  public create(vNode: ElementVNode) {
    updateStyle(emptyVNode, vNode)
  }
 
  public update(formerVNode: ElementVNode, vNode: ElementVNode) {
    updateStyle(formerVNode, vNode)
  }
 
  public remove(vNode: ElementVNode, removeElement: Function) {
    applyRemoveStyle(vNode, removeElement)
  }
 
  public destroy(vNode: ElementVNode) {
    applyDestroyStyle(vNode)
  }
}
 
let requestAnimationFrame: any
 
function setRequestAnimationFrame() {
  if (!requestAnimationFrame)
    requestAnimationFrame =
      (typeof window !== 'undefined' && window.requestAnimationFrame) || setTimeout
}
 
function nextFrame(fn: any) {
  requestAnimationFrame(function() {
    requestAnimationFrame(fn)
  })
}
 
function setValueOnNextFrame(obj: any, prop: string, value: any) {
  nextFrame(function() {
    obj[prop] = value
  })
}
 
function updateStyle(formerVNode: ElementVNode, vNode: ElementVNode): void {
  let styleValue: any
  const element: any = vNode.element
  let formerStyle: any = formerVNode.props.style
  let style: any = vNode.props.style
 
  if (!formerStyle && !style) return
 
  formerStyle = formerStyle || {}
  style = style || {}
 
  const formerHasDelayedProperty: boolean =
    !!formerStyle.delayed
 
  // tslint:disable-next-line:prefer-const
  for (let key in formerStyle)
    if (!style[key])
      element.style[key] = ''
 
  for (const key in style) {
    styleValue = style[key]
 
    if (key === 'delayed') {
      // tslint:disable-next-line:prefer-const
      for (let delayKey in style.delayed) {
        styleValue = style.delayed[delayKey]
 
        if (!formerHasDelayedProperty || styleValue !== formerStyle.delayed[delayKey])
          setValueOnNextFrame((element as any).style, delayKey, styleValue)
      }
    } else if (key !== 'remove') {
      element.style[key] = styleValue
    }
  }
}
 
function applyDestroyStyle(vNode: ElementVNode) {
  const element: any = vNode.element
  const style: any = vNode.props.style
 
  if (!style || !style.destroy) return
 
  const destroy: any = style.destroy
 
  for (const key in destroy)
    element.style[key] = destroy[key]
}
 
function applyRemoveStyle(vNode: ElementVNode, callback: Function) {
  const style: any = vNode.props.style
 
  if (!style || !style.remove)
    return callback()
 
  const element: any = vNode.element
  let index = 0
  let computedStyle: any
  let listenerCount = 0
  const appliedStyles: Array<string> = []
 
  for (const key in style) {
    appliedStyles.push(key)
    element.style[key] = style[key]
  }
 
  computedStyle = getComputedStyle(element)
 
  const transitionProperties: Array<string> =
    computedStyle['transition-property'].split(', ')
 
  for (; index < transitionProperties.length; ++index)
    if (appliedStyles.indexOf(transitionProperties[index]) !== -1)
      listenerCount++
 
  element.addEventListener('transitionend', function(event: TransitionEvent) {
    if (event.target === element)
      --listenerCount
 
    if (listenerCount === 0)
      callback()
  })
}
 
export {
  setRequestAnimationFrame as pre,
  updateStyle as create,
  updateStyle as update,
  applyDestroyStyle as destroy,
  applyRemoveStyle as remove,
}