feat: implement .stat function

Author: hacdias

README.md

@@ -282,6 +282,20 @@ Sets the API address.
 
 * `value` should be a [Multiaddr](https://github.com/multiformats/js-multiaddr) or a String representing a valid one.
 
+### `repo.stat ([options], callback)`
+
+Gets the repo status. 
+
+`options` is an object which might contain the key `human`, which is a boolean indicating whether or not the `repoSize` should be displayed in MiB or not.
+
+`callback` is a function with the signature `function (err, stats)`, where `stats` is an Object with the following keys:
+
+- `numObjects`
+- `repoPath`
+- `repoSize`
+- `version`
+- `storageMax`
+
 ## Notes
 
 - [Explanation of how repo is structured](https://github.com/ipfs/js-ipfs-repo/pull/111#issuecomment-279948247)

package.json

@@ -56,20 +56,22 @@
   "dependencies": {
     "async": "^2.6.0",
     "base32.js": "~0.1.0",
+    "big.js": "^5.0.3",
     "cids": "~0.5.2",
-    "interface-datastore": "~0.4.2",
     "datastore-core": "~0.4.0",
     "datastore-fs": "~0.4.2",
     "datastore-level": "~0.7.0",
     "debug": "^3.1.0",
+    "interface-datastore": "~0.4.2",
     "ipfs-block": "~0.6.1",
     "level-js": "timkuijsten/level.js#idbunwrapper",
     "leveldown": "^1.7.2",
     "lock-me": "^1.0.3",
     "lodash.get": "^4.4.2",
     "lodash.has": "^4.5.2",
     "lodash.set": "^4.3.2",
-    "multiaddr": "^3.0.1"
+    "multiaddr": "^3.0.1",
+    "pull-stream": "^3.6.1"
   },
   "license": "MIT",
   "contributors": [

src/blockstore.js

@@ -8,6 +8,7 @@ const Block = require('ipfs-block')
 const setImmediate = require('async/setImmediate')
 const reject = require('async/reject')
 const CID = require('cids')
+const pull = require('pull-stream')
 
 /**
  * Transform a raw buffer to a base32 encoded key.
@@ -49,6 +50,19 @@ function maybeWithSharding (filestore, options, callback) {
 
 function createBaseStore (store) {
   return {
+    /**
+     * Query the store.
+     *
+     * @param {object} query
+     * @param {function(Error, Array)} callback
+     * @return {void}
+     */
+    query (query, callback) {
+      pull(
+        store.query(query),
+        pull.collect(callback)
+      )
+    },
     /**
      * Get a single block by CID.
      *

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')
@@ -200,6 +204,87 @@ class IpfsRepo {
   exists (callback) {
     this.version.exists(callback)
   }
+
+  /**
+   * Get repo status.
+   *
+   * @param {Object}  options
+   * @param {Boolean} options.human
+   * @param {function(Error, Object)} callback
+   * @return {void}
+   */
+  stat (options, callback) {
+    if (typeof options === 'function') {
+      callback = options
+      options = {}
+    }
+
+    options = Object.assign({}, {human: false}, options)
+
+    parallel([
+      (cb) => cb(null, this.path),
+      (cb) => this.config.get('Datastore.StorageMax', (err, max) => {
+        if (err) {
+          cb(null, new Big(noLimit))
+        } else {
+          cb(null, new Big(max))
+        }
+      }),
+      (cb) => this.version.get(cb),
+      (cb) => waterfall([
+        (cb) => cb(null, {
+          repoSize: new Big(0)
+        }),
+        (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)
+          })
+
+          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) => {
+          if (options.human) {
+            stats.repoSize = stats.repoSize.div(1048576)
+          }
+
+          cb(null, stats)
+        }
+      ], cb)
+    ], (err, results) => {
+      if (err) return callback(err)
+
+      callback(null, {
+        repoPath: results[0],
+        storageMax: results[1],
+        version: results[2],
+        repoSize: results[3].repoSize,
+        numObjects: results[3].numObjects
+      })
+    })
+  }
+}
+
+function getSize (queryFn, callback) {
+  pull(
+    queryFn.query({}),
+    pull.reduce((sum, block) => {
+      return sum
+        .plus(block.value.byteLength)
+        .plus(block.key._buf.byteLength)
+    }, new Big(0), callback))
 }
 
 module.exports = IpfsRepo

test/repo-test.js

@@ -99,4 +99,32 @@ module.exports = (repo) => {
       })
     })
   })
+
+  describe('stat', () => {
+    it('get stats', (done) => {
+      repo.stat((err, stats) => {
+        expect(err).to.not.exist()
+        expect(stats).to.exist()
+        expect(stats).to.have.property('numObjects')
+        expect(stats).to.have.property('version')
+        expect(stats).to.have.property('repoPath')
+        expect(stats).to.have.property('repoSize')
+        expect(stats).to.have.property('storageMax')
+        done()
+      })
+    })
+
+    it('get human stats', (done) => {
+      repo.stat({human: true}, (err, stats) => {
+        expect(err).to.not.exist()
+        expect(stats).to.exist()
+        expect(stats).to.have.property('numObjects')
+        expect(stats).to.have.property('version')
+        expect(stats).to.have.property('repoPath')
+        expect(stats).to.have.property('repoSize')
+        expect(stats).to.have.property('storageMax')
+        done()
+      })
+    })
+  })
 }