fix: don't break backwards compatibility on the Block API

ipfs · Oct 29, 2016 · 3674b8e · 3674b8e
1 parent 2eeea35
commit 3674b8e
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/core/components/block.js b/src/core/components/block.js
@@ -19,6 +19,12 @@ module.exports = function block (self) {
         block = new Block(block)
       }
 
+      if (typeof cid === 'function') {
+        // legacy (without CID)
+        callback = cid
+        cid = new CID(block.key('sha2-256'))
+      }
+
       self._blockService.put({
         block: block,
         cid: cid
@@ -47,8 +53,10 @@ module.exports = function block (self) {
 }
 
 function cleanCid (cid) {
-  if (typeof cid === 'string') {
-    return new CID(cid)
+  if (cid.constructor.name === 'CID') {
+    return cid
   }
-  return cid
+
+  // CID constructor knows how to do the cleaning :)
+  return new CID(cid)
 }