/// <reference path="../virtual-dom.d.ts" />

import h = require('virtual-dom/h');
import patch = require('virtual-dom/patch');
import diff = require('virtual-dom/diff');
import VNode = require('virtual-dom/vnode/vnode');
import VText = require('virtual-dom/vnode/vtext');
import createElement = require('virtual-dom/create-element');
import virtualize = require('vdom-virtualize');

export function updateElement(element, vtree) {
  console.assert(!!element);
  if (element.firstChild) {
    const prev = element.firstChild.vtree || virtualize(element.firstChild);
    const patches = diff(prev, vtree);
    patch(element.firstChild, patches);
  } else {
    const node = createElement(vtree);
    element.appendChild(node);
  }
  element.firstChild.vtree = vtree;
}

export function updateElementVtree(element) {
  console.assert(!!element);
  if (element.firstChild) {
    element.firstChild.vtree = virtualize(element.firstChild);
  }
}

import app from '../app';
app.h = (el, props, ...children) => (typeof el === 'string') ?
    h(el, props, children) : el(props, children);

app.createElement = app.h;
/// <reference path="../virtual-dom.d.ts" />

import h = require('virtual-dom/h');
import patch = require('virtual-dom/patch');
import diff = require('virtual-dom/diff');
import VNode = require('virtual-dom/vnode/vnode');
import VText = require('virtual-dom/vnode/vtext');
import createElement = require('virtual-dom/create-element');
import virtualize = require('vdom-virtualize');

export function updateElement(element, vtree) {
  console.assert(!!element);
  if (element.firstChild) {
    const prev = element.firstChild.vtree || virtualize(element.firstChild);
    const patches = diff(prev, vtree);
    patch(element.firstChild, patches);
  } else {
    const node = createElement(vtree);
    element.appendChild(node);
  }
  element.firstChild.vtree = vtree;
}

export function updateElementVtree(element) {
  console.assert(!!element);
  if (element.firstChild) {
    element.firstChild.vtree = virtualize(element.firstChild);
  }
}

import app from '../app';
app.h = (el, props, ...children) => h(el, props, children);
app.createElement = app.h;
--- 
+++ 
@@ -29,5 +29,7 @@
 }
 
 import app from '../app';
-app.h = (el, props, ...children) => h(el, props, children);
+app.h = (el, props, ...children) => (typeof el === 'string') ?
+    h(el, props, children) : el(props, children);
+
 app.createElement = app.h;
Support custom element in jsx/tsx
