module.exports = function prefixStreamWrite (writableStream, prefix) {

const oldWrite = writableStream.write

function newWrite (...args) {

const [chunk, encoding] = args

// Prepend the prefix if the chunk is either a string or a Buffer.

// Otherwise, just leave it alone to be safe.

if (typeof chunk === 'string') {

// Only prepend the prefix is the `encoding` is safe or not provided.

// If it's a function, it is third arg `callback` provided as optional second

if (!encoding || encoding === 'utf8' || typeof encoding === 'function') {

args[0] = prefix + chunk

}

} else if (Buffer.isBuffer(chunk)) {

args[0] = Buffer.concat([Buffer.from(prefix), chunk])

}

return oldWrite.apply(this, args)

}

writableStream.write = newWrite

return writableStream

}