feat: add Uint8Array support

ipld · Jun 12, 2020 · 418eb5f · 418eb5f
1 parent 7059cc8
commit 418eb5f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -16,8 +16,10 @@ const withIs = require('class-is')
  */
 class Block {
   constructor (data, cid) {
-    if (!data || !Buffer.isBuffer(data)) {
-      throw new Error('first argument  must be a buffer')
+    if (!data || !ArrayBuffer.isView(data)) {
+      throw new Error('first argument  must be a buffer or typed array')
+    } else if (!Buffer.isBuffer(data)) {
+      data = Buffer.from(data.buffer, data.byteOffset, data.byteLength)
     }
 
     if (!cid || !CID.isCID(cid)) {

diff --git a/test/index.spec.js b/test/index.spec.js
@@ -27,6 +27,15 @@ describe('block', () => {
     expect(Block.isBlock(b)).to.eql(true)
   })
 
+  it('create with Uint8Array', () => {
+    const b = new Block(
+      new Uint8Array([104, 101, 108, 108, 111]),
+      new CID('QmdfTbBqBPQ7VNxZEYEj14VmRuZBkqFbiwReogJgS1zR1n')
+    )
+
+    expect(Block.isBlock(b)).to.eql(true)
+  })
+
   it('block stays immutable', () => {
     const b = new Block(Buffer.from('hello'), new CID('QmdfTbBqBPQ7VNxZEYEj14VmRuZBkqFbiwReogJgS1zR1n'))