3
0
Fork 0
mirror of https://gitea.com/actions/setup-python.git synced 2024-11-29 21:49:32 +01:00
setup-python/node_modules/cross-spawn/lib/util/readShebang.js
Danny McCormick 39c08a0eaa Initial pass
2019-06-26 21:12:00 -04:00

32 lines
740 B
JavaScript

'use strict';
const fs = require('fs');
const shebangCommand = require('shebang-command');
function readShebang(command) {
// Read the first 150 bytes from the file
const size = 150;
let buffer;
if (Buffer.alloc) {
// Node.js v4.5+ / v5.10+
buffer = Buffer.alloc(size);
} else {
// Old Node.js API
buffer = new Buffer(size);
buffer.fill(0); // zero-fill
}
let fd;
try {
fd = fs.openSync(command, 'r');
fs.readSync(fd, buffer, 0, size, 0);
fs.closeSync(fd);
} catch (e) { /* Empty */ }
// Attempt to extract shebang (null is returned if not a shebang)
return shebangCommand(buffer.toString());
}
module.exports = readShebang;