astroturf/runtime/helpers.js

"use strict";

function _extends() { _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; }; return _extends.apply(this, arguments); }

/* eslint-disable react/no-multi-comp */
var React = require('react'); // eslint-disable-line import/no-extraneous-dependencies


var getName = function getName(name, C) {
  return name + "(" + (C.displayName || C.name || C) + ")";
};

function mapProps(mapper) {
  return function (Component) {
    return Object.assign(React.forwardRef(function (p, ref) {
      return React.createElement(Component, _extends({
        ref: ref
      }, mapper(p)));
    }), {
      isAstroturf: true,
      displayName: getName('mapProps', Component)
    });
  };
}

function withProps(objOrMapper) {
  return function (Component) {
    return Object.assign(React.forwardRef(function (p, ref) {
      return React.createElement(Component, _extends({
        ref: ref
      }, p, typeof objOrMapper === 'function' ? objOrMapper(p) : objOrMapper));
    }), {
      isAstroturf: true,
      displayName: getName('withProps', Component)
    });
  };
}

module.exports = {
  mapProps: mapProps,
  withProps: withProps
};