all files / src/modules/ class.ts

40.91% Statements 9/22
0% Branches 0/10
20% Functions 1/5
40% Lines 8/20
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                                                   
import { ElementVNode, EventHandler, Module, VNode } from '../'
 
import { BaseModule } from './BaseModule'
import { emptyVNode } from './emptyVNode'
 
export function createClassModule(): Module {
  return new ClassModule()
}
 
class ClassModule extends BaseModule {
  public create(vNode: ElementVNode) {
    updateClass(emptyVNode, vNode)
  }
 
  public update(formerVNode: ElementVNode, vNode: ElementVNode) {
    updateClass(formerVNode, vNode)
  }
}
 
function updateClass(formerVNode: ElementVNode, vNode: ElementVNode): void {
  const { props: { class: formerClass = {} }, element: formerElement } = formerVNode
  const { props: { class: klass = {} }, element } = vNode
 
  if (formerClass === klass) return
 
  for (const name in formerClass)
    if (!klass[name])
      formerElement.classList.remove(name)
 
  for (const name in klass)
    if (klass[name] !== formerClass[name])
      element.classList.toggle(name)
}