iframe-test.js

import React, { useEffect, useState } from 'react';
import { get } from 'lodash';

function IframeTest() {
  const [size, setSize] = useState({});

  function receiveMessage(message) {
    const { data } = message;
    if (get(data, 'type') === 'resize') {
      const { width, height } = data;
      setSize({ width, height });
    }
  }

  if (typeof window !== 'undefined') {
    useEffect(() => {
      window.addEventListener('message', receiveMessage, false);
    }, []);
  }

  return (
    <iframe
      style={{ borderBottom: 'dotted red 6px' }}
      title="test"
      src="/"
      width="100%"
      height={size.height}
    />
  );
}

export default IframeTest;