ui/js/component/drawer/view.jsx

@@ -7,16 +7,17 @@ const DrawerItem = (props) => {
     currentPage,
     href,
     subPages,
+    badge,
     label,
     linkClick,
     icon,
   } = props
   const isSelected = (
-    currentPage == href.substr(1) ||
+    currentPage == href.substr(0) ||
     (subPages && subPages.indexOf(currentPage) != -1)
   )
 
-  return <Link icon={icon} label={label} onClick={() => linkClick(href)} className={ 'drawer-item ' + (isSelected ? 'drawer-item-selected' : '') } />
+  return <Link icon={icon} badge={badge} label={label} onClick={() => linkClick(href)} className={ 'drawer-item ' + (isSelected ? 'drawer-item-selected' : '') } />
 }
 
 var drawerImageStyle = { //@TODO: remove this, img should be properly scaled once size is settled
@@ -32,9 +33,9 @@ class Drawer extends React.Component {
   componentDidMount() {
     const { updateBalance } = this.props
 
-    this._balanceSubscribeId = lbry.balanceSubscribe(function(balance) {
+    this._balanceSubscribeId = lbry.balanceSubscribe((balance) => {
       updateBalance(balance)
-    }.bind(this));
+    });
   }
   componentWillUnmount() {
     if (this._balanceSubscribeId) {
@@ -46,7 +47,6 @@ class Drawer extends React.Component {
     const {
       closeDrawerClick,
       logoClick,
-      currentPage,
       balance,
     } = this.props
 
@@ -58,7 +58,7 @@ class Drawer extends React.Component {
       <DrawerItem {...this.props} href='discover' label="Discover" icon="icon-search"  />
       <DrawerItem {...this.props} href='publish' label="Publish" icon="icon-upload" />
       <DrawerItem {...this.props} href='downloaded' subPages={['published']} label="My Files" icon='icon-cloud-download' />
-      <DrawerItem {...this.props} href="wallet" subPages={['send', 'receive', 'claim', 'referral']} label="My Wallet" badge={lbry.formatCredits(balance) } icon="icon-bank" />
+      <DrawerItem {...this.props} href="wallet" subPages={['send', 'receive', 'claim']} label="My Wallet" badge={lbry.formatCredits(balance) } icon="icon-bank" />
       <DrawerItem {...this.props} href='settings' label="Settings" icon='icon-gear' />
       <DrawerItem {...this.props} href='help' label="Help" icon='icon-question-circle' />
     </nav>)

ui/js/component/link/view.js

@@ -15,7 +15,7 @@ const Link = (props) => {
     disabled,
   } = props
 
-
+console.log(props);
   const className = (props.className || '') +
     (!props.className && !props.button ? 'button-text' : '') + // Non-button links get the same look as text buttons
     (props.button ? ' button-block button-' + props.button + ' button-set-item' : '') +