@leofcoin/peernet 0.9.8 → 0.9.9

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@leofcoin/peernet",
-  "version": "0.9.8",
+  "version": "0.9.9",
   "description": "",
   "main": "dist/commonjs/peernet.js",
   "module": "dist/module/peernet.js",

package/src/client.js

@@ -37,6 +37,10 @@ export default class PeernetClient {
 
     this.p2p.on('peerclose', (peer) => {
       // TODO: close peernetPeer
+      const peernetPeer = connections.get(peer.id)
+      if (peernetPeer) {
+        peernetPeer.close()
+      }
       connections.delete(peer.id)
       pubsub.publish('peer:disconnected', peer)
     })

package/src/peer.js

@@ -1,5 +1,6 @@
 export default class PeernetPeer {
   constructor(id, connection) {
+    this._events = {}
     this.id = id
     this.connection = connection
 
@@ -33,11 +34,25 @@ export default class PeernetPeer {
   }
 
   on(event = 'peernet.data', cb) {
+    this._events[event] = cb
     pubsub.subscribe(event, cb)
     // this.connection.on(event, cb)
   }
 
   removeListener(event = 'data', cb) {
+    delete this._events[event]
     pubsub.unsubscribe(event, cb)
   }
+
+  close() {
+    for (const event of Object.keys(this._events)) {
+      pubsub.unsubscribe(event, this._events[event])
+    }
+    this._events = []
+
+    for (const event of this.connection._events.data) {
+      this.connection.removeListener('data', event)
+    }
+    this.connection.destroy()
+  }
 }

package/src/peernet.js

@@ -345,7 +345,7 @@ export default class Peernet {
         }
 
         if (data) {
-          data = new DataMessageResponse({hash, data: Buffer.from(data)})
+          data = new DataMessageResponse({hash, data: data.decoded ? Buffer.from(JSON.stringify(data)) : Buffer.from(data)});
 
           const node = await this.prepareMessage(from, data.encoded)
           peer.write(Buffer.from(JSON.stringify({id, data: node.encoded})))