mirror of https://github.com/lukechilds/node.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
123 lines
3.0 KiB
123 lines
3.0 KiB
var chownr = require('chownr')
|
|
var dezalgo = require('dezalgo')
|
|
var fs = require('graceful-fs')
|
|
var inflight = require('inflight')
|
|
var log = require('npmlog')
|
|
var mkdirp = require('mkdirp')
|
|
|
|
// memoize the directories created by this step
|
|
var stats = {}
|
|
var effectiveOwner
|
|
module.exports = function correctMkdir (path, cb) {
|
|
cb = dezalgo(cb)
|
|
cb = inflight('correctMkdir:' + path, cb)
|
|
if (!cb) {
|
|
return log.verbose('correctMkdir', path, 'correctMkdir already in flight; waiting')
|
|
} else {
|
|
log.verbose('correctMkdir', path, 'correctMkdir not in flight; initializing')
|
|
}
|
|
|
|
if (stats[path]) return cb(null, stats[path])
|
|
|
|
fs.stat(path, function (er, st) {
|
|
if (er) return makeDirectory(path, cb)
|
|
|
|
if (!st.isDirectory()) {
|
|
log.error('correctMkdir', 'invalid dir %s', path)
|
|
return cb(er)
|
|
}
|
|
|
|
var ownerStats = calculateOwner()
|
|
// there's always a chance the permissions could have been frobbed, so fix
|
|
if (st.uid !== ownerStats.uid) {
|
|
stats[path] = ownerStats
|
|
setPermissions(path, ownerStats, cb)
|
|
} else {
|
|
stats[path] = st
|
|
cb(null, stats[path])
|
|
}
|
|
})
|
|
}
|
|
|
|
function calculateOwner () {
|
|
if (!effectiveOwner) {
|
|
effectiveOwner = { uid: 0, gid: 0 }
|
|
|
|
// Pretty much only on windows
|
|
if (!process.getuid) {
|
|
return effectiveOwner
|
|
}
|
|
|
|
effectiveOwner.uid = +process.getuid()
|
|
effectiveOwner.gid = +process.getgid()
|
|
|
|
if (effectiveOwner.uid === 0) {
|
|
if (process.env.SUDO_UID) effectiveOwner.uid = +process.env.SUDO_UID
|
|
if (process.env.SUDO_GID) effectiveOwner.gid = +process.env.SUDO_GID
|
|
}
|
|
}
|
|
|
|
return effectiveOwner
|
|
}
|
|
|
|
function makeDirectory (path, cb) {
|
|
cb = inflight('makeDirectory:' + path, cb)
|
|
if (!cb) {
|
|
return log.verbose('makeDirectory', path, 'creation already in flight; waiting')
|
|
} else {
|
|
log.verbose('makeDirectory', path, 'creation not in flight; initializing')
|
|
}
|
|
|
|
var owner = calculateOwner()
|
|
|
|
if (!process.getuid) {
|
|
return mkdirp(path, function (er) {
|
|
log.verbose('makeCacheDir', 'UID & GID are irrelevant on', process.platform)
|
|
|
|
stats[path] = owner
|
|
return cb(er, stats[path])
|
|
})
|
|
}
|
|
|
|
if (owner.uid !== 0 || !process.env.HOME) {
|
|
log.silly(
|
|
'makeDirectory', path,
|
|
'uid:', owner.uid,
|
|
'gid:', owner.gid
|
|
)
|
|
stats[path] = owner
|
|
mkdirp(path, afterMkdir)
|
|
} else {
|
|
fs.stat(process.env.HOME, function (er, st) {
|
|
if (er) {
|
|
log.error('makeDirectory', 'homeless?')
|
|
return cb(er)
|
|
}
|
|
|
|
log.silly(
|
|
'makeDirectory', path,
|
|
'uid:', st.uid,
|
|
'gid:', st.gid
|
|
)
|
|
stats[path] = st
|
|
mkdirp(path, afterMkdir)
|
|
})
|
|
}
|
|
|
|
function afterMkdir (er, made) {
|
|
if (er || !stats[path] || isNaN(stats[path].uid) || isNaN(stats[path].gid)) {
|
|
return cb(er, stats[path])
|
|
}
|
|
|
|
if (!made) return cb(er, stats[path])
|
|
|
|
setPermissions(made, stats[path], cb)
|
|
}
|
|
}
|
|
|
|
function setPermissions (path, st, cb) {
|
|
chownr(path, st.uid, st.gid, function (er) {
|
|
if (er && er.code === 'ENOENT') return cb(null, st)
|
|
return cb(er, st)
|
|
})
|
|
}
|
|
|