149 lines
3.5 KiB
JavaScript
149 lines
3.5 KiB
JavaScript
#!/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();
|
||
}
|
||
|
||
}());
|