27 lines
734 B
JavaScript
27 lines
734 B
JavaScript
'use strict';
|
|
|
|
module.exports = function (renderer, prototype, styles, block) {
|
|
var render_ = prototype.render;
|
|
var className = '';
|
|
|
|
prototype.render = function () {
|
|
var element = render_.call(this);
|
|
|
|
if (element) {
|
|
if (!className) {
|
|
className = renderer.rule(styles, block);
|
|
}
|
|
|
|
if (process.env.NODE_ENV === 'production') {
|
|
element.props.className = (element.props.className || '') + className;
|
|
} else {
|
|
element = require('react').cloneElement(element, {
|
|
className: (element.props.className || '') + className,
|
|
});
|
|
}
|
|
}
|
|
|
|
return element;
|
|
};
|
|
};
|