313 lines
8.5 KiB
JavaScript
313 lines
8.5 KiB
JavaScript
'use strict';
|
|
|
|
var stream = require('stream');
|
|
var util = require('util');
|
|
var Transform = stream.Transform;
|
|
|
|
// expose to the world
|
|
module.exports = {
|
|
encode: encode,
|
|
decode: decode,
|
|
wrap: wrap,
|
|
Encoder: Encoder,
|
|
Decoder: Decoder
|
|
};
|
|
|
|
/**
|
|
* Encodes a Buffer into a Quoted-Printable encoded string
|
|
*
|
|
* @param {Buffer} buffer Buffer to convert
|
|
* @returns {String} Quoted-Printable encoded string
|
|
*/
|
|
function encode(buffer) {
|
|
if (typeof buffer === 'string') {
|
|
buffer = new Buffer(buffer, 'utf-8');
|
|
}
|
|
|
|
// usable characters that do not need encoding
|
|
var ranges = [
|
|
// https://tools.ietf.org/html/rfc2045#section-6.7
|
|
[0x09], // <TAB>
|
|
[0x0A], // <LF>
|
|
[0x0D], // <CR>
|
|
[0x20, 0x3C], // <SP>!"#$%&'()*+,-./0123456789:;
|
|
[0x3E, 0x7E] // >?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\]^_`abcdefghijklmnopqrstuvwxyz{|}
|
|
];
|
|
var result = '';
|
|
var ord;
|
|
|
|
for (var i = 0, len = buffer.length; i < len; i++) {
|
|
ord = buffer[i];
|
|
// if the char is in allowed range, then keep as is, unless it is a ws in the end of a line
|
|
if (checkRanges(ord, ranges) && !((ord === 0x20 || ord === 0x09) && (i === len - 1 || buffer[i + 1] === 0x0a || buffer[i + 1] === 0x0d))) {
|
|
result += String.fromCharCode(ord);
|
|
continue;
|
|
}
|
|
result += '=' + (ord < 0x10 ? '0' : '') + ord.toString(16).toUpperCase();
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
/**
|
|
* Decodes a Quoted-Printable encoded string to a Buffer object
|
|
*
|
|
* @param {String} str Quoted-Printable encoded string
|
|
* @returns {Buffer} Decoded value
|
|
*/
|
|
function decode(str) {
|
|
str = (str || '').toString().
|
|
// remove invalid whitespace from the end of lines
|
|
replace(/[\t ]+$/gm, '').
|
|
// remove soft line breaks
|
|
replace(/\=(?:\r?\n|$)/g, '');
|
|
|
|
var encodedBytesCount = (str.match(/\=[\da-fA-F]{2}/g) || []).length,
|
|
bufferLength = str.length - encodedBytesCount * 2,
|
|
chr, hex,
|
|
buffer = new Buffer(bufferLength),
|
|
bufferPos = 0;
|
|
|
|
for (var i = 0, len = str.length; i < len; i++) {
|
|
chr = str.charAt(i);
|
|
if (chr === '=' && (hex = str.substr(i + 1, 2)) && /[\da-fA-F]{2}/.test(hex)) {
|
|
buffer[bufferPos++] = parseInt(hex, 16);
|
|
i += 2;
|
|
continue;
|
|
}
|
|
buffer[bufferPos++] = chr.charCodeAt(0);
|
|
}
|
|
|
|
return buffer;
|
|
}
|
|
|
|
/**
|
|
* Adds soft line breaks to a Quoted-Printable string
|
|
*
|
|
* @param {String} str Quoted-Printable encoded string that might need line wrapping
|
|
* @param {Number} [lineLength=76] Maximum allowed length for a line
|
|
* @returns {String} Soft-wrapped Quoted-Printable encoded string
|
|
*/
|
|
function wrap(str, lineLength) {
|
|
str = (str || '').toString();
|
|
lineLength = lineLength || 76;
|
|
|
|
if (str.length <= lineLength) {
|
|
return str;
|
|
}
|
|
|
|
var pos = 0,
|
|
len = str.length,
|
|
match, code, line,
|
|
lineMargin = Math.floor(lineLength / 3),
|
|
result = '';
|
|
|
|
// insert soft linebreaks where needed
|
|
while (pos < len) {
|
|
line = str.substr(pos, lineLength);
|
|
if ((match = line.match(/\r\n/))) {
|
|
line = line.substr(0, match.index + match[0].length);
|
|
result += line;
|
|
pos += line.length;
|
|
continue;
|
|
}
|
|
|
|
if (line.substr(-1) === '\n') {
|
|
// nothing to change here
|
|
result += line;
|
|
pos += line.length;
|
|
continue;
|
|
} else if ((match = line.substr(-lineMargin).match(/\n.*?$/))) {
|
|
// truncate to nearest line break
|
|
line = line.substr(0, line.length - (match[0].length - 1));
|
|
result += line;
|
|
pos += line.length;
|
|
continue;
|
|
} else if (line.length > lineLength - lineMargin && (match = line.substr(-lineMargin).match(/[ \t\.,!\?][^ \t\.,!\?]*$/))) {
|
|
// truncate to nearest space
|
|
line = line.substr(0, line.length - (match[0].length - 1));
|
|
} else {
|
|
if (line.match(/\=[\da-f]{0,2}$/i)) {
|
|
|
|
// push incomplete encoding sequences to the next line
|
|
if ((match = line.match(/\=[\da-f]{0,1}$/i))) {
|
|
line = line.substr(0, line.length - match[0].length);
|
|
}
|
|
|
|
// ensure that utf-8 sequences are not split
|
|
while (line.length > 3 && line.length < len - pos && !line.match(/^(?:=[\da-f]{2}){1,4}$/i) && (match = line.match(/\=[\da-f]{2}$/ig))) {
|
|
code = parseInt(match[0].substr(1, 2), 16);
|
|
if (code < 128) {
|
|
break;
|
|
}
|
|
|
|
line = line.substr(0, line.length - 3);
|
|
|
|
if (code >= 0xC0) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (pos + line.length < len && line.substr(-1) !== '\n') {
|
|
if (line.length === lineLength && line.match(/\=[\da-f]{2}$/i)) {
|
|
line = line.substr(0, line.length - 3);
|
|
} else if (line.length === lineLength) {
|
|
line = line.substr(0, line.length - 1);
|
|
}
|
|
pos += line.length;
|
|
line += '=\r\n';
|
|
} else {
|
|
pos += line.length;
|
|
}
|
|
|
|
result += line;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
/**
|
|
* Helper function to check if a number is inside provided ranges
|
|
*
|
|
* @param {Number} nr Number to check for
|
|
* @param {Array} ranges An Array of allowed values
|
|
* @returns {Boolean} True if the value was found inside allowed ranges, false otherwise
|
|
*/
|
|
function checkRanges(nr, ranges) {
|
|
for (var i = ranges.length - 1; i >= 0; i--) {
|
|
if (!ranges[i].length) {
|
|
continue;
|
|
}
|
|
if (ranges[i].length === 1 && nr === ranges[i][0]) {
|
|
return true;
|
|
}
|
|
if (ranges[i].length === 2 && nr >= ranges[i][0] && nr <= ranges[i][1]) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Creates a transform stream for encoding data to Quoted-Printable encoding
|
|
*
|
|
* @constructor
|
|
* @param {Object} options Stream options
|
|
* @param {Number} [options.lineLength=76] Maximum lenght for lines, set to false to disable wrapping
|
|
*/
|
|
function Encoder(options) {
|
|
// init Transform
|
|
this.options = options || {};
|
|
|
|
if (this.options.lineLength !== false) {
|
|
this.options.lineLength = this.options.lineLength || 76;
|
|
}
|
|
|
|
this._curLine = '';
|
|
|
|
this.inputBytes = 0;
|
|
this.outputBytes = 0;
|
|
|
|
Transform.call(this, this.options);
|
|
}
|
|
util.inherits(Encoder, Transform);
|
|
|
|
Encoder.prototype._transform = function(chunk, encoding, done) {
|
|
var qp, _self = this;
|
|
|
|
if (encoding !== 'buffer') {
|
|
chunk = new Buffer(chunk, encoding);
|
|
}
|
|
|
|
if (!chunk || !chunk.length) {
|
|
return done();
|
|
}
|
|
|
|
this.inputBytes += chunk.length;
|
|
|
|
if (this.options.lineLength) {
|
|
qp = this._curLine + encode(chunk);
|
|
qp = wrap(qp, this.options.lineLength);
|
|
qp = qp.replace(/(^|\n)([^\n]*)$/, function(match, lineBreak, lastLine) {
|
|
_self._curLine = lastLine;
|
|
return lineBreak;
|
|
});
|
|
|
|
if (qp) {
|
|
this.outputBytes += qp.length;
|
|
this.push(qp);
|
|
}
|
|
|
|
} else {
|
|
qp = encode(chunk);
|
|
this.outputBytes += qp.length;
|
|
this.push(qp, 'ascii');
|
|
}
|
|
|
|
done();
|
|
};
|
|
|
|
Encoder.prototype._flush = function(done) {
|
|
if (this._curLine) {
|
|
this.outputBytes += this._curLine.length;
|
|
this.push(this._curLine, 'ascii');
|
|
}
|
|
done();
|
|
};
|
|
|
|
/**
|
|
* Creates a transform stream for decoding Quoted-Printable encoded strings
|
|
*
|
|
* @constructor
|
|
* @param {Object} options Stream options
|
|
*/
|
|
function Decoder(options) {
|
|
// init Transform
|
|
this.options = options || {};
|
|
this._curLine = '';
|
|
|
|
this.inputBytes = 0;
|
|
this.outputBytes = 0;
|
|
|
|
Transform.call(this, this.options);
|
|
}
|
|
util.inherits(Decoder, Transform);
|
|
|
|
Decoder.prototype._transform = function(chunk, encoding, done) {
|
|
var qp, buf, _self = this;
|
|
|
|
chunk = chunk.toString('ascii');
|
|
|
|
if (!chunk || !chunk.length) {
|
|
return done();
|
|
}
|
|
|
|
this.inputBytes += chunk.length;
|
|
|
|
qp = (this._curLine + chunk);
|
|
this._curLine = '';
|
|
qp = qp.replace(/=[^\n]?$/, function(lastLine) {
|
|
_self._curLine = lastLine;
|
|
return '';
|
|
});
|
|
|
|
if (qp) {
|
|
buf = decode(qp);
|
|
this.outputBytes += buf.length;
|
|
this.push(buf);
|
|
}
|
|
|
|
done();
|
|
};
|
|
|
|
Decoder.prototype._flush = function(done) {
|
|
var qp, buf;
|
|
if (this._curLine) {
|
|
buf = decode(this._curLine);
|
|
this.outputBytes += buf.length;
|
|
this.push(buf);
|
|
}
|
|
done();
|
|
}; |