#!/usr/bin/env node
|
(function() {
|
|
var fs = require('fs');
|
var he = require('../he.js');
|
var strings = process.argv.splice(2);
|
var stdin = process.stdin;
|
var data;
|
var timeout;
|
var action;
|
var options = {};
|
var log = console.log;
|
|
var main = function() {
|
var option = strings[0];
|
var count = 0;
|
|
if (/^(?:-h|--help|undefined)$/.test(option)) {
|
log(
|
'he v%s - https://mths.be/he',
|
he.version
|
);
|
log([
|
'\nUsage:\n',
|
'\the [--escape] string',
|
'\the [--encode] [--use-named-refs] [--everything] [--allow-unsafe] [--decimal] string',
|
'\the [--decode] [--attribute] [--strict] string',
|
'\the [-v | --version]',
|
'\the [-h | --help]',
|
'\nExamples:\n',
|
'\the --escape \\<img\\ src\\=\\\'x\\\'\\ onerror\\=\\"prompt\\(1\\)\\"\\>',
|
'\techo \'© 𝌆\' | he --decode'
|
].join('\n'));
|
return process.exit(option ? 0 : 1);
|
}
|
|
if (/^(?:-v|--version)$/.test(option)) {
|
log('v%s', he.version);
|
return process.exit(0);
|
}
|
|
strings.forEach(function(string) {
|
// Process options
|
if (string == '--escape') {
|
action = 'escape';
|
return;
|
}
|
if (string == '--encode') {
|
action = 'encode';
|
return;
|
}
|
if (string == '--use-named-refs') {
|
action = 'encode';
|
options.useNamedReferences = true;
|
return;
|
}
|
if (string == '--everything') {
|
action = 'encode';
|
options.encodeEverything = true;
|
return;
|
}
|
if (string == '--allow-unsafe') {
|
action = 'encode';
|
options.allowUnsafeSymbols = true;
|
return;
|
}
|
if (string == '--decimal') {
|
action = 'encode';
|
options.decimal = true;
|
return;
|
}
|
if (string == '--decode') {
|
action = 'decode';
|
return;
|
}
|
if (string == '--attribute') {
|
action = 'decode';
|
options.isAttributeValue = true;
|
return;
|
}
|
if (string == '--strict') {
|
action = 'decode';
|
options.strict = true;
|
return;
|
}
|
// Process string(s)
|
var result;
|
if (!action) {
|
log('Error: he requires at least one option and a string argument.');
|
log('Try `he --help` for more information.');
|
return process.exit(1);
|
}
|
try {
|
result = he[action](string, options);
|
log(result);
|
count++;
|
} catch(error) {
|
log(error.message + '\n');
|
log('Error: failed to %s.', action);
|
log('If you think this is a bug in he, please report it:');
|
log('https://github.com/mathiasbynens/he/issues/new');
|
log(
|
'\nStack trace using he@%s:\n',
|
he.version
|
);
|
log(error.stack);
|
return process.exit(1);
|
}
|
});
|
if (!count) {
|
log('Error: he requires a string argument.');
|
log('Try `he --help` for more information.');
|
return process.exit(1);
|
}
|
// Return with exit status 0 outside of the `forEach` loop, in case
|
// multiple strings were passed in.
|
return process.exit(0);
|
};
|
|
if (stdin.isTTY) {
|
// handle shell arguments
|
main();
|
} else {
|
// Either the script is called from within a non-TTY context, or `stdin`
|
// content is being piped in.
|
if (!process.stdout.isTTY) {
|
// The script was called from a non-TTY context. This is a rather uncommon
|
// use case we don’t actively support. However, we don’t want the script
|
// to wait forever in such cases, so…
|
timeout = setTimeout(function() {
|
// …if no piped data arrived after a whole minute, handle shell
|
// arguments instead.
|
main();
|
}, 60000);
|
}
|
data = '';
|
stdin.on('data', function(chunk) {
|
clearTimeout(timeout);
|
data += chunk;
|
});
|
stdin.on('end', function() {
|
strings.push(data.trim());
|
main();
|
});
|
stdin.resume();
|
}
|
|
}());
|
