- Badge now has hooks

Willie01 · Aug 13, 2019 · 72a9e9f · 72a9e9f
1 parent 4c18042
commit 72a9e9f
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/src/components/Badge/Badge.js b/src/components/Badge/Badge.js
@@ -3,12 +3,15 @@ import React from "react";
 import PropTypes from "prop-types";
 
 // @material-ui/core components
-import withStyles from "@material-ui/core/styles/withStyles";
+import { makeStyles } from "@material-ui/core/styles";
 
-import badgeStyle from "assets/jss/material-kit-react/components/badgeStyle.js";
+import styles from "assets/jss/material-kit-react/components/badgeStyle.js";
 
-function Badge({ ...props }) {
-  const { classes, color, children } = props;
+const useStyles = makeStyles(styles);
+
+export default function Badge(props) {
+  const classes = useStyles();
+  const { color, children } = props;
   return (
     <span className={classes.badge + " " + classes[color]}>{children}</span>
   );
@@ -31,5 +34,3 @@ Badge.propTypes = {
   ]),
   children: PropTypes.node
 };
-
-export default withStyles(badgeStyle)(Badge);