Merge branch 'master' into encoding-length

Author: jhermsmeier

.travis.yml

@@ -1,8 +1,8 @@
 language: node_js
 node_js:
   - stable
+  - 12
+  - 10
   - 8
-  - 6
-  - 4
 script:
   - npm test

lib/encode.js

@@ -32,7 +32,9 @@ encode._encode = function (buffers, data) {
   switch (getType(data)) {
     case 'buffer': encode.buffer(buffers, data); break
     case 'object': encode.dict(buffers, data); break
+    case 'map': encode.dictMap(buffers, data); break
     case 'array': encode.list(buffers, data); break
+    case 'set': encode.listSet(buffers, data); break
     case 'string': encode.string(buffers, data); break
     case 'number': encode.number(buffers, data); break
     case 'boolean': encode.number(buffers, data); break
@@ -90,6 +92,22 @@ encode.dict = function (buffers, data) {
   buffers.push(buffE)
 }
 
+encode.dictMap = function (buffers, data) {
+  buffers.push(buffD)
+
+  var keys = Array.from(data.keys()).sort()
+
+  for (var key of keys) {
+    if (data.get(key) == null) continue
+    Buffer.isBuffer(key)
+      ? encode._encode(buffers, key)
+      : encode.string(buffers, String(key))
+    encode._encode(buffers, data.get(key))
+  }
+
+  buffers.push(buffE)
+}
+
 encode.list = function (buffers, data) {
   var i = 0
   var c = data.length
@@ -103,4 +121,15 @@ encode.list = function (buffers, data) {
   buffers.push(buffE)
 }
 
+encode.listSet = function (buffers, data) {
+  buffers.push(buffL)
+
+  for (var item of data) {
+    if (item == null) continue
+    encode._encode(buffers, item)
+  }
+
+  buffers.push(buffE)
+}
+
 module.exports = encode

test/encode.test.js

@@ -31,6 +31,16 @@ test('bencode#encode()', function (t) {
     t.equal(bencode.encode(data).toString(), 'd2:1211:Hello World2:34i12345ee')
   })
 
+  t.test('should encode a Map as dictionary', function (t) {
+    t.plan(1)
+    var data = new Map([
+      [ 12, 'Hello World' ],
+      [ '34', 12345 ],
+      [ Buffer.from('buffer key'), Buffer.from('buffer value') ]
+    ])
+    t.equal(bencode.encode(data).toString(), 'd2:1211:Hello World2:34i12345e10:buffer key12:buffer valuee')
+  })
+
   t.test('should be able to encode a positive integer', function (t) {
     t.plan(1)
     t.equal(bencode.encode(123).toString(), 'i123e')
@@ -110,6 +120,11 @@ test('bencode#encode()', function (t) {
     t.equal(bencode.encode([32, 12]).toString(), 'li32ei12ee')
     t.equal(bencode.encode([':asdf:']).toString(), 'l6::asdf:e')
   })
+  t.test('should be able to encode a Set as a list', function (t) {
+    t.plan(2)
+    t.equal(bencode.encode(new Set([32, 12])).toString(), 'li32ei12ee')
+    t.equal(bencode.encode(new Set([':asdf:'])).toString(), 'l6::asdf:e')
+  })
   t.test('should be able to encode an object', function (t) {
     t.plan(3)
     t.equal(bencode.encode({ 'a': 'bc' }).toString(), 'd1:a2:bce')