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 (