diff --git a/src/Rect/index.js b/src/Rect/index.js index e8e34ca..2974611 100644 --- a/src/Rect/index.js +++ b/src/Rect/index.js @@ -16,9 +16,11 @@ const zoomableMap = { export default class Rect extends PureComponent { static propTypes = { - styles: PropTypes.object, - zoomable: PropTypes.string, - rotatable: PropTypes.bool, + className: PropTypes.string, + children: PropTypes.oneOfType([ + PropTypes.arrayOf(PropTypes.node), + PropTypes.node + ]), onResizeStart: PropTypes.func, onResize: PropTypes.func, onResizeEnd: PropTypes.func, @@ -28,7 +30,14 @@ export default class Rect extends PureComponent { onDragStart: PropTypes.func, onDrag: PropTypes.func, onDragEnd: PropTypes.func, - parentRotateAngle: PropTypes.number + parentRotateAngle: PropTypes.number, + rotatable: PropTypes.bool, + styles: PropTypes.object, + zoomable: PropTypes.string + } + + static defaultProps = { + className: '' } setElementRef = (ref) => { this.$element = ref } @@ -133,6 +142,8 @@ export default class Rect extends PureComponent { render () { const { + children, + className, styles: { position: { centerX, centerY }, size: { width, height }, @@ -155,7 +166,7 @@ export default class Rect extends PureComponent { { @@ -187,6 +198,7 @@ export default class Rect extends PureComponent { ) }) } + {children} ) } diff --git a/src/index.js b/src/index.js index 8def077..d0a34bf 100644 --- a/src/index.js +++ b/src/index.js @@ -5,6 +5,11 @@ import { centerToTL, tLToCenter, getNewStyle, degToRadian } from './utils' export default class ResizableRect extends Component { static propTypes = { + className: PropTypes.string, + children: PropTypes.oneOfType([ + PropTypes.arrayOf(PropTypes.node), + PropTypes.node + ]), left: PropTypes.number.isRequired, top: PropTypes.number.isRequired, width: PropTypes.number.isRequired, @@ -80,7 +85,7 @@ export default class ResizableRect extends Component { render () { const { - top, left, width, height, rotateAngle, parentRotateAngle, zoomable, rotatable, + children, className, top, left, width, height, rotateAngle, parentRotateAngle, zoomable, rotatable, onRotate, onResizeStart, onResizeEnd, onRotateStart, onRotateEnd, onDragStart, onDragEnd } = this.props @@ -88,6 +93,8 @@ export default class ResizableRect extends Component { return (