RootContainer.js

var PropTypes = require('prop-types');
var React = require('react');
var {
	ensureRootElementWithContainer,
	getRootElementAttributes,
} = require('./RootElement');

class RootContainer extends React.Component { // eslint-disable-line react/require-render-return
	render() {
		throw new Error("RootContainers can't go in non-RootContainers!");
	}
}

module.exports = RootContainer;

RootContainer.propTypes = {
	listen: PropTypes.func,
	when: PropTypes.object, // A promise.
	_isRootContainer: PropTypes.bool,
}

RootContainer.defaultProps = {
	_isRootContainer: true,
}

RootContainer.flattenForRender = function(element) {
	return [{containerOpen: getRootElementAttributes(element)}]
		.concat(prepChildren(element))
		.concat([{containerClose: true}])
		.reduce((m,v) => m.concat(Array.isArray(v)?v:[v]), [])
}

RootContainer.isRootContainer = function(element) {
	return element && element.props && element.props._isRootContainer;
}

function prepChildren (element) {
	return React.Children.toArray(element.props.children).map(
		child => RootContainer.isRootContainer(child)
			?RootContainer.flattenForRender(child)
			:ensureRootElementWithContainer(child, element)
	)
}