2015-12-19 07:05:16 +01:00
|
|
|
const Duplex = require('readable-stream').Duplex
|
|
|
|
const inherits = require('util').inherits
|
|
|
|
|
|
|
|
module.exports = PortDuplexStream
|
|
|
|
|
|
|
|
inherits(PortDuplexStream, Duplex)
|
|
|
|
|
2016-06-21 22:18:32 +02:00
|
|
|
function PortDuplexStream (port) {
|
2015-12-19 07:05:16 +01:00
|
|
|
Duplex.call(this, {
|
|
|
|
objectMode: true,
|
|
|
|
})
|
|
|
|
this._port = port
|
|
|
|
port.onMessage.addListener(this._onMessage.bind(this))
|
2016-01-17 10:27:25 +01:00
|
|
|
port.onDisconnect.addListener(this._onDisconnect.bind(this))
|
2015-12-19 07:05:16 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// private
|
|
|
|
|
2016-06-21 22:18:32 +02:00
|
|
|
PortDuplexStream.prototype._onMessage = function (msg) {
|
2016-03-10 03:33:30 +01:00
|
|
|
if (Buffer.isBuffer(msg)) {
|
|
|
|
delete msg._isBuffer
|
|
|
|
var data = new Buffer(msg)
|
|
|
|
// console.log('PortDuplexStream - saw message as buffer', data)
|
|
|
|
this.push(data)
|
|
|
|
} else {
|
|
|
|
// console.log('PortDuplexStream - saw message', msg)
|
|
|
|
this.push(msg)
|
|
|
|
}
|
2015-12-19 07:05:16 +01:00
|
|
|
}
|
|
|
|
|
2016-06-21 22:18:32 +02:00
|
|
|
PortDuplexStream.prototype._onDisconnect = function () {
|
2016-01-17 10:27:25 +01:00
|
|
|
try {
|
2016-09-13 21:07:29 +02:00
|
|
|
this.push(null)
|
2016-06-21 22:18:32 +02:00
|
|
|
} catch (err) {
|
2016-01-17 10:27:25 +01:00
|
|
|
this.emit('error', err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-12-19 07:05:16 +01:00
|
|
|
// stream plumbing
|
|
|
|
|
|
|
|
PortDuplexStream.prototype._read = noop
|
|
|
|
|
2016-06-21 22:18:32 +02:00
|
|
|
PortDuplexStream.prototype._write = function (msg, encoding, cb) {
|
2016-02-10 20:46:13 +01:00
|
|
|
try {
|
2016-03-10 03:33:30 +01:00
|
|
|
if (Buffer.isBuffer(msg)) {
|
|
|
|
var data = msg.toJSON()
|
|
|
|
data._isBuffer = true
|
|
|
|
// console.log('PortDuplexStream - sent message as buffer', data)
|
|
|
|
this._port.postMessage(data)
|
|
|
|
} else {
|
|
|
|
// console.log('PortDuplexStream - sent message', msg)
|
|
|
|
this._port.postMessage(msg)
|
|
|
|
}
|
2016-02-10 20:46:13 +01:00
|
|
|
cb()
|
2016-06-21 22:18:32 +02:00
|
|
|
} catch (err) {
|
2016-05-06 01:04:43 +02:00
|
|
|
console.error(err)
|
2016-02-10 20:46:13 +01:00
|
|
|
// this.emit('error', err)
|
|
|
|
cb(new Error('PortDuplexStream - disconnected'))
|
|
|
|
}
|
2015-12-19 07:05:16 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// util
|
|
|
|
|
2016-06-21 22:18:32 +02:00
|
|
|
function noop () {}
|