⚗ Upadate Badge theme extension to .module.scss

Author: KRRISH96

lib/badge/index.js

@@ -4,7 +4,7 @@ import PropTypes from 'prop-types';
 import cx from 'classnames';
 import { themr } from 'react-css-themr';
 import { transitionEndEventName } from '../globals/helper';
-import defaultTheme from './theme.scss';
+import defaultTheme from './theme.module.scss';
 
 const getNode = findDOMNode;
 
@@ -19,19 +19,21 @@ class Badge extends React.Component {
 
   componentDidUpdate(prevProps) {
     if (this.props.badgeCount !== prevProps.badgeCount) {
+      const { theme } = this.props;
       const countEle = getNode(this.animRef);
       countEle.addEventListener(transitionEndEventName(), this.updateCount, false);
-      countEle.classList.add('fadeOut');
+      countEle.classList.add(theme.fadeOut);
       // eslint-disable-next-line react/no-did-update-set-state
     }
   }
 
   updateCount = () => {
+    const { theme } = this.props;
     const countEle = getNode(this.animRef);
     this.setState({
       count: this.props.badgeCount,
     });
-    countEle.classList.remove('fadeOut');
+    countEle.classList.remove(theme.fadeOut);
   };
 
   render() {
@@ -58,7 +60,7 @@ class Badge extends React.Component {
             </span>
           </span>
         </div>
-        <div className="clearfix" />
+        <div className={`${theme.clearfix} clearfix`} />
       </React.Fragment>
     );
   }