add stats

Author: hacdias

package.json

@@ -56,6 +56,7 @@
   "dependencies": {
     "async": "^2.6.0",
     "base32.js": "~0.1.0",
+    "big.js": "^5.0.3",
     "cids": "~0.5.2",
     "datastore-core": "~0.4.0",
     "datastore-fs": "~0.4.2",

src/index.js

@@ -7,6 +7,8 @@ const each = require('async/each')
 const assert = require('assert')
 const path = require('path')
 const debug = require('debug')
+const Big = require('big.js')
+const pull = require('pull-stream')
 
 const backends = require('./backends')
 const version = require('./version')
@@ -17,6 +19,8 @@ const defaultOptions = require('./default-options')
 
 const log = debug('repo')
 
+const noLimit = Number.MAX_SAFE_INTEGER
+
 const lockers = {
   memory: require('./lock-memory'),
   fs: require('./lock')
@@ -201,32 +205,73 @@ class IpfsRepo {
     this.version.exists(callback)
   }
 
+  /**
+   * Get repo status.
+   *
+   * @param {function(Error, Object)} callback
+   * @return {void}
+   */
   stat (callback) {
     waterfall([
-      (cb) => cb(null, {repoPath: this.path}),
-      (stats, cb) => {
-        this.version.get((err, version) => {
-          stats.version = version
-          cb(err, stats)
+      (cb) => cb(null, {
+        repoPath: this.path,
+        repoSize: new Big(0),
+        storageMax: new Big(noLimit)
+      }),
+      (stats, cb) => this.version.get((err, version) => {
+        stats.version = version
+        cb(err, stats)
+      }),
+      (stats, cb) => this.blocks.query({}, (err, list) => {
+        list = list || []
+        stats.numObjects = new Big(list.length)
+
+        list.forEach(block => {
+          stats.repoSize = stats.repoSize
+            .plus(block.value.byteLength)
+            .plus(block.key._buf.byteLength)
         })
-      },
-      (stats, cb) => {
-        this.blocks.query({ keysOnly: true }, (err, list) => {
-          list = list || []  
-          stats.numObjects = list.length
-          cb(err, stats)
-        })  
-      },
-      (stats, cb) => {
-        // TODO:
-        stats.repoSize = null
-        stats.storageMax = null
+
+        cb(err, stats)
+      }),
+      (stats, cb) => getSize(this.datastore, (err, size) => {
+        stats.repoSize = stats.repoSize.plus(size)
+        cb(err, stats)
+      }),
+      (stats, cb) => getSize(this.keys, (err, size) => {
+        stats.repoSize = stats.repoSize.plus(size)
+        cb(err, stats)
+      }),
+      (stats, cb) => this.config.get('Datastore.StorageMax', (err, max) => {
+        if (!err) {
+          stats.storageMax = new Big(max)
+        }
+
         cb(null, stats)
-      }
+      })
     ], callback)
   }
 }
 
+function getSize (queryFn, callback) {
+  pull(
+    queryFn.query({}),
+    pull.collect((err, list) => {
+      if (err) return callback(err, 0)
+
+      let val = new Big(0)
+
+      list.forEach(block => {
+        val = val
+          .plus(block.value.byteLength)
+          .plus(block.key._buf.byteLength)
+      })
+
+      callback(null, val)
+    })
+  )
+}
+
 module.exports = IpfsRepo
 module.exports.repoVersion = repoVersion