-
Notifications
You must be signed in to change notification settings - Fork 2.9k
/
fix-bin.js
44 lines (37 loc) · 1.3 KB
/
fix-bin.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
// make sure that bins are executable, and that they don't have
// windows line-endings on the hashbang line.
const fs = require('fs')
const { promisify } = require('util')
const execMode = 0o777 & (~process.umask())
const writeFileAtomic = require('write-file-atomic')
const open = promisify(fs.open)
const close = promisify(fs.close)
const read = promisify(fs.read)
const chmod = promisify(fs.chmod)
const readFile = promisify(fs.readFile)
const isWindowsHashBang = buf =>
buf[0] === '#'.charCodeAt(0) &&
buf[1] === '!'.charCodeAt(0) &&
/^#![^\n]+\r\n/.test(buf.toString())
const isWindowsHashbangFile = file => {
const FALSE = () => false
return open(file, 'r').then(fd => {
const buf = Buffer.alloc(2048)
return read(fd, buf, 0, 2048, 0)
.then(
() => {
const isWHB = isWindowsHashBang(buf)
return close(fd).then(() => isWHB, () => isWHB)
},
// don't leak FD if read() fails
() => close(fd).then(FALSE, FALSE)
)
}, FALSE)
}
const dos2Unix = file =>
readFile(file, 'utf8').then(content =>
writeFileAtomic(file, content.replace(/^(#![^\n]+)\r\n/, '$1\n')))
const fixBin = (file, mode = execMode) => chmod(file, mode)
.then(() => isWindowsHashbangFile(file))
.then(isWHB => isWHB ? dos2Unix(file) : null)
module.exports = fixBin