encode: preserve 'created by' and 'comment' keys

Closes webtorrent/create-torrent#36
webtorrent · Apr 28, 2016 · 6fa5d60 · 6fa5d60
1 parent 600ac91
commit 6fa5d60
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/index.js b/index.js
@@ -107,18 +107,26 @@ function encodeTorrentFile (parsed) {
     info: parsed.info
   }
 
-  torrent['announce-list'] = parsed.announce.map(function (url) {
+  torrent['announce-list'] = (parsed.announce || []).map(function (url) {
     if (!torrent.announce) torrent.announce = url
     url = new Buffer(url, 'utf8')
     return [ url ]
   })
 
+  torrent['url-list'] = parsed.urlList || []
+
   if (parsed.created) {
     torrent['creation date'] = (parsed.created.getTime() / 1000) | 0
   }
-  if (parsed.urlList) {
-    torrent['url-list'] = parsed.urlList
+
+  if (parsed.createdBy) {
+    torrent['created by'] = parsed.createdBy
   }
+
+  if (parsed.comment) {
+    torrent.comment = parsed.comment
+  }
+
   return bencode.encode(torrent)
 }
 

diff --git a/test/encode.js b/test/encode.js
@@ -7,20 +7,16 @@ test('encode', function (t) {
   var buf = parseTorrentFile.encode(parsedTorrent)
   var doubleParsedTorrent = parseTorrentFile(buf)
 
-  t.deepEqual(doubleParsedTorrent.infoBuffer, parsedTorrent.infoBuffer)
-  t.equal(doubleParsedTorrent.created.getDate(), parsedTorrent.created.getDate())
-  t.deepEqual(doubleParsedTorrent.announce, parsedTorrent.announce)
-
+  t.deepEqual(doubleParsedTorrent, parsedTorrent)
   t.end()
 })
 
 test('encode w/ comment field', function (t) {
   var parsedTorrent = parseTorrentFile(fixtures.leaves.torrent)
   parsedTorrent.comment = 'hi there!'
-
   var buf = parseTorrentFile.encode(parsedTorrent)
   var doubleParsedTorrent = parseTorrentFile(buf)
 
-  t.equal(doubleParsedTorrent.comment, parsedTorrent.comment)
+  t.deepEqual(doubleParsedTorrent, parsedTorrent)
   t.end()
 })