diff --git a/browser/components/StorageItem.js b/browser/components/StorageItem.js
index 801b1d5d..603523b0 100644
--- a/browser/components/StorageItem.js
+++ b/browser/components/StorageItem.js
@@ -4,6 +4,7 @@
import React, { PropTypes } from 'react'
import styles from './StorageItem.styl'
import CSSModules from 'browser/lib/CSSModules'
+import { isNumber } from 'lodash'
/**
* @param {boolean} isActive
@@ -33,7 +34,7 @@ const StorageItem = ({
>
{isFolded ? folderName.substring(0, 1) : folderName}
- {!isFolded &&
+ {(!isFolded && isNumber(noteCount)) &&
{noteCount}
}
{isFolded &&
@@ -51,7 +52,7 @@ StorageItem.propTypes = {
folderName: PropTypes.string.isRequired,
folderColor: PropTypes.string,
isFolded: PropTypes.bool.isRequired,
- noteCount: PropTypes.number.isRequired,
+ noteCount: PropTypes.number,
}
export default CSSModules(StorageItem, styles)
diff --git a/browser/finder/StorageSection.js b/browser/finder/StorageSection.js
index 3dfe072a..a8ed2e5e 100644
--- a/browser/finder/StorageSection.js
+++ b/browser/finder/StorageSection.js
@@ -1,6 +1,7 @@
import React, { PropTypes } from 'react'
import CSSModules from 'browser/lib/CSSModules'
import styles from './StorageSection.styl'
+import StorageItem from 'browser/components/StorageItem'
class StorageSection extends React.Component {
constructor (props) {
@@ -30,20 +31,17 @@ class StorageSection extends React.Component {
render () {
let { storage, filter } = this.props
let folderList = storage.folders
- .map((folder) => {
- return (
-
- )
- })
+ .map(folder => (
+ this.handleFolderClick(e, folder)}
+ folderName={folder.name}
+ folderColor={folder.color}
+ isFolded={false}
+ />
+ ))
+
return (