WIP

autoHeightWebView/common.js

     subtree: true,
     attributes: true
 });
-`;
+`;
+
+export const getCurrentSize = `
+function getSize(container) {
+  var height = container.clientHeight || document.body.offsetHeight;
+  var width = container.clientWidth || document.body.offsetWidth;
+  return {
+    height,
+    width
+  };
+}
+`;
+

autoHeightWebView/index.android.js

 
 import Immutable from 'immutable';
 
-import { handleSizeUpdated, getWidth, getScript, domMutationObserveScript } from './common.js';
+import { handleSizeUpdated, getWidth, getScript, domMutationObserveScript, getCurrentSize } from './common.js';
 
 const RCTAutoHeightWebView = requireNativeComponent('RCTAutoHeightWebView', AutoHeightWebView, {
   nativeOnly: {
     super(props);
     const { enableAnimation, style } = props;
     enableAnimation && (this.opacityAnimatedValue = new Animated.Value(0));
-    this.webView = React.createRef();
     isBelowKitKat && DeviceEventEmitter.addListener('webViewBridgeMessage', this.listenWebViewBridgeMessage);
     this.state = {
       isChangingSource: false,
   }
 
   onMessage = e => {
+    console.log(e)
     const { height, width } = JSON.parse(isBelowKitKat ? e.nativeEvent.message : e.nativeEvent.data);
     const { height: oldHeight, width: oldWidth } = this.state;
     if ((height && height !== oldHeight) || (width && width !== oldWidth)) {
     onLoadEnd && onLoadEnd(event);
   };
 
+  getWebView = webView => (this.webView = webView);
+
   stopLoading() {
     UIManager.dispatchViewManagerCommand(
       findNodeHandle(this.webView),
 
   render() {
     const { height, width, script, isChangingSource, heightOffset } = this.state;
-    const { scalesPageToFit, enableAnimation, source, customScript, style, enableBaseUrl, scrollEnabled } = this.props;
+    const { scalesPageToFit, enableAnimation, source, style, enableBaseUrl, scrollEnabled } = this.props;
     let webViewSource = source;
     if (enableBaseUrl) {
       webViewSource = Object.assign({}, source, {
             onLoadingStart={this.onLoadingStart}
             onLoadingFinish={this.onLoadingFinish}
             onLoadingError={this.onLoadingError}
-            originWhitelist={['*']}
-            ref={this.webView}
+            originWhitelist={['.*']}
+            ref={this.getWebView}
             style={styles.webView}
             javaScriptEnabled={true}
-            injectedJavaScript={script + customScript}
+            injectedJavaScript={script}
             scalesPageToFit={scalesPageToFit}
             scrollEnabled={!!scrollEnabled}
             source={webViewSource}
   }
 });
 
-const getBaseScript = isBelowKitKat
-  ? function() {
-      return `
-  ; (function () {
+const updateSize = `
+  function updateSize() {
+    if(document.body.offsetHeight !== height || document.body.offsetWidth !== width) {
+      var size = getSize(document.body.firstChild); 
+      height = size.height;
+      width = size.width;
+      AutoHeightWebView.send(JSON.stringify({ width, height }));
+    }
+  }
+`
+
+const commonScript = `
+    ${getCurrentSize}
+    ${updateSize}
     var wrapper = document.createElement('div');
     wrapper.id = 'wrapper';
     while (document.body.firstChild instanceof Node) {
         wrapper.appendChild(document.body.firstChild);
     }
+`;
+
+const getBaseScript = isBelowKitKat
+  ? function(style) {
+      return `
+    ; 
+    var height = 0;
+    var width = ${getWidth(style)};
+    (function () {
+    ${commonScript}
     document.body.appendChild(wrapper);
-    AutoHeightWebView.onMessage = function (message) {
-        var rect = document.body.firstElementChild.getBoundingClientRect().toJSON();
-        var width = Math.round(rect.width);
-        var height = Math.round(rect.height);
-        AutoHeightWebView.send(JSON.stringify({ width, height }));
-    };
+    AutoHeightWebView.onMessage = updateSize;
     ${domMutationObserveScript}
 } ());
   `;
     }
-  : function() {
+  : function(style) {
       return `
-  ; (function () {
-    var wrapper = document.createElement('div');
-    wrapper.id = 'wrapper';
-    while (document.body.firstChild instanceof Node) {
-        wrapper.appendChild(document.body.firstChild);
-    }
+    ; 
+    var height = 0;
+    var width = ${getWidth(style)};
+    (function () {
+    ${commonScript}
     document.body.appendChild(wrapper);
-    document.addEventListener('message', function (e) {
-        var rect = document.body.firstElementChild.getBoundingClientRect().toJSON();
-        var width = Math.round(rect.width);
-        var height = Math.round(rect.height);
-        window.postMessage(JSON.stringify({ width, height }));
-    });
+    document.addEventListener('message', updateSize);
     ${domMutationObserveScript}
 } ());
   `;

autoHeightWebView/index.ios.js

 
 import PropTypes from 'prop-types';
 
-import { getSize, isEqual, setState, getWidth, handleSizeUpdated, domMutationObserveScript } from './common.js';
+import { getSize, isEqual, setState, getWidth, handleSizeUpdated, domMutationObserveScript, getCurrentSize } from './common.js';
 
 import momoize from './momoize';
 
     window.addEventListener('resize', updateSize);
     `;
 
-const getCurrentSize = `
-    function getSize(container) {
-      var height = container.clientHeight || document.body.offsetHeight;
-      var width = container.clientWidth || document.body.offsetWidth;
-      return {
-        height,
-        width
-      };
-    }
-    `;
 
 function getBaseScript(style) {
   return `