forked from zhurui/management
247 lines
7.8 KiB
JavaScript
247 lines
7.8 KiB
JavaScript
|
'use strict'
|
||
|
|
||
|
const fs = require('graceful-fs')
|
||
|
const path = require('path')
|
||
|
const mkdirp = require('../mkdirs').mkdirs
|
||
|
const pathExists = require('../path-exists').pathExists
|
||
|
const utimes = require('../util/utimes').utimesMillis
|
||
|
|
||
|
const notExist = Symbol('notExist')
|
||
|
|
||
|
function copy (src, dest, opts, cb) {
|
||
|
if (typeof opts === 'function' && !cb) {
|
||
|
cb = opts
|
||
|
opts = {}
|
||
|
} else if (typeof opts === 'function') {
|
||
|
opts = {filter: opts}
|
||
|
}
|
||
|
|
||
|
cb = cb || function () {}
|
||
|
opts = opts || {}
|
||
|
|
||
|
opts.clobber = 'clobber' in opts ? !!opts.clobber : true // default to true for now
|
||
|
opts.overwrite = 'overwrite' in opts ? !!opts.overwrite : opts.clobber // overwrite falls back to clobber
|
||
|
|
||
|
// Warn about using preserveTimestamps on 32-bit node
|
||
|
if (opts.preserveTimestamps && process.arch === 'ia32') {
|
||
|
console.warn(`fs-extra: Using the preserveTimestamps option in 32-bit node is not recommended;\n
|
||
|
see https://github.com/jprichardson/node-fs-extra/issues/269`)
|
||
|
}
|
||
|
|
||
|
checkPaths(src, dest, (err, destStat) => {
|
||
|
if (err) return cb(err)
|
||
|
if (opts.filter) return handleFilter(checkParentDir, destStat, src, dest, opts, cb)
|
||
|
return checkParentDir(destStat, src, dest, opts, cb)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function checkParentDir (destStat, src, dest, opts, cb) {
|
||
|
const destParent = path.dirname(dest)
|
||
|
pathExists(destParent, (err, dirExists) => {
|
||
|
if (err) return cb(err)
|
||
|
if (dirExists) return startCopy(destStat, src, dest, opts, cb)
|
||
|
mkdirp(destParent, err => {
|
||
|
if (err) return cb(err)
|
||
|
return startCopy(destStat, src, dest, opts, cb)
|
||
|
})
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function handleFilter (onInclude, destStat, src, dest, opts, cb) {
|
||
|
Promise.resolve(opts.filter(src, dest)).then(include => {
|
||
|
if (include) {
|
||
|
if (destStat) return onInclude(destStat, src, dest, opts, cb)
|
||
|
return onInclude(src, dest, opts, cb)
|
||
|
}
|
||
|
return cb()
|
||
|
}, error => cb(error))
|
||
|
}
|
||
|
|
||
|
function startCopy (destStat, src, dest, opts, cb) {
|
||
|
if (opts.filter) return handleFilter(getStats, destStat, src, dest, opts, cb)
|
||
|
return getStats(destStat, src, dest, opts, cb)
|
||
|
}
|
||
|
|
||
|
function getStats (destStat, src, dest, opts, cb) {
|
||
|
const stat = opts.dereference ? fs.stat : fs.lstat
|
||
|
stat(src, (err, srcStat) => {
|
||
|
if (err) return cb(err)
|
||
|
|
||
|
if (srcStat.isDirectory()) return onDir(srcStat, destStat, src, dest, opts, cb)
|
||
|
else if (srcStat.isFile() ||
|
||
|
srcStat.isCharacterDevice() ||
|
||
|
srcStat.isBlockDevice()) return onFile(srcStat, destStat, src, dest, opts, cb)
|
||
|
else if (srcStat.isSymbolicLink()) return onLink(destStat, src, dest, opts, cb)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function onFile (srcStat, destStat, src, dest, opts, cb) {
|
||
|
if (destStat === notExist) return copyFile(srcStat, src, dest, opts, cb)
|
||
|
return mayCopyFile(srcStat, src, dest, opts, cb)
|
||
|
}
|
||
|
|
||
|
function mayCopyFile (srcStat, src, dest, opts, cb) {
|
||
|
if (opts.overwrite) {
|
||
|
fs.unlink(dest, err => {
|
||
|
if (err) return cb(err)
|
||
|
return copyFile(srcStat, src, dest, opts, cb)
|
||
|
})
|
||
|
} else if (opts.errorOnExist) {
|
||
|
return cb(new Error(`'${dest}' already exists`))
|
||
|
} else return cb()
|
||
|
}
|
||
|
|
||
|
function copyFile (srcStat, src, dest, opts, cb) {
|
||
|
if (typeof fs.copyFile === 'function') {
|
||
|
return fs.copyFile(src, dest, err => {
|
||
|
if (err) return cb(err)
|
||
|
return setDestModeAndTimestamps(srcStat, dest, opts, cb)
|
||
|
})
|
||
|
}
|
||
|
return copyFileFallback(srcStat, src, dest, opts, cb)
|
||
|
}
|
||
|
|
||
|
function copyFileFallback (srcStat, src, dest, opts, cb) {
|
||
|
const rs = fs.createReadStream(src)
|
||
|
rs.on('error', err => cb(err)).once('open', () => {
|
||
|
const ws = fs.createWriteStream(dest, { mode: srcStat.mode })
|
||
|
ws.on('error', err => cb(err))
|
||
|
.on('open', () => rs.pipe(ws))
|
||
|
.once('close', () => setDestModeAndTimestamps(srcStat, dest, opts, cb))
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function setDestModeAndTimestamps (srcStat, dest, opts, cb) {
|
||
|
fs.chmod(dest, srcStat.mode, err => {
|
||
|
if (err) return cb(err)
|
||
|
if (opts.preserveTimestamps) {
|
||
|
return utimes(dest, srcStat.atime, srcStat.mtime, cb)
|
||
|
}
|
||
|
return cb()
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function onDir (srcStat, destStat, src, dest, opts, cb) {
|
||
|
if (destStat === notExist) return mkDirAndCopy(srcStat, src, dest, opts, cb)
|
||
|
if (destStat && !destStat.isDirectory()) {
|
||
|
return cb(new Error(`Cannot overwrite non-directory '${dest}' with directory '${src}'.`))
|
||
|
}
|
||
|
return copyDir(src, dest, opts, cb)
|
||
|
}
|
||
|
|
||
|
function mkDirAndCopy (srcStat, src, dest, opts, cb) {
|
||
|
fs.mkdir(dest, err => {
|
||
|
if (err) return cb(err)
|
||
|
copyDir(src, dest, opts, err => {
|
||
|
if (err) return cb(err)
|
||
|
return fs.chmod(dest, srcStat.mode, cb)
|
||
|
})
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function copyDir (src, dest, opts, cb) {
|
||
|
fs.readdir(src, (err, items) => {
|
||
|
if (err) return cb(err)
|
||
|
return copyDirItems(items, src, dest, opts, cb)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function copyDirItems (items, src, dest, opts, cb) {
|
||
|
const item = items.pop()
|
||
|
if (!item) return cb()
|
||
|
return copyDirItem(items, item, src, dest, opts, cb)
|
||
|
}
|
||
|
|
||
|
function copyDirItem (items, item, src, dest, opts, cb) {
|
||
|
const srcItem = path.join(src, item)
|
||
|
const destItem = path.join(dest, item)
|
||
|
checkPaths(srcItem, destItem, (err, destStat) => {
|
||
|
if (err) return cb(err)
|
||
|
startCopy(destStat, srcItem, destItem, opts, err => {
|
||
|
if (err) return cb(err)
|
||
|
return copyDirItems(items, src, dest, opts, cb)
|
||
|
})
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function onLink (destStat, src, dest, opts, cb) {
|
||
|
fs.readlink(src, (err, resolvedSrc) => {
|
||
|
if (err) return cb(err)
|
||
|
|
||
|
if (opts.dereference) {
|
||
|
resolvedSrc = path.resolve(process.cwd(), resolvedSrc)
|
||
|
}
|
||
|
|
||
|
if (destStat === notExist) {
|
||
|
return fs.symlink(resolvedSrc, dest, cb)
|
||
|
} else {
|
||
|
fs.readlink(dest, (err, resolvedDest) => {
|
||
|
if (err) {
|
||
|
// dest exists and is a regular file or directory,
|
||
|
// Windows may throw UNKNOWN error. If dest already exists,
|
||
|
// fs throws error anyway, so no need to guard against it here.
|
||
|
if (err.code === 'EINVAL' || err.code === 'UNKNOWN') return fs.symlink(resolvedSrc, dest, cb)
|
||
|
return cb(err)
|
||
|
}
|
||
|
if (opts.dereference) {
|
||
|
resolvedDest = path.resolve(process.cwd(), resolvedDest)
|
||
|
}
|
||
|
if (isSrcSubdir(resolvedSrc, resolvedDest)) {
|
||
|
return cb(new Error(`Cannot copy '${resolvedSrc}' to a subdirectory of itself, '${resolvedDest}'.`))
|
||
|
}
|
||
|
|
||
|
// do not copy if src is a subdir of dest since unlinking
|
||
|
// dest in this case would result in removing src contents
|
||
|
// and therefore a broken symlink would be created.
|
||
|
if (destStat.isDirectory() && isSrcSubdir(resolvedDest, resolvedSrc)) {
|
||
|
return cb(new Error(`Cannot overwrite '${resolvedDest}' with '${resolvedSrc}'.`))
|
||
|
}
|
||
|
return copyLink(resolvedSrc, dest, cb)
|
||
|
})
|
||
|
}
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function copyLink (resolvedSrc, dest, cb) {
|
||
|
fs.unlink(dest, err => {
|
||
|
if (err) return cb(err)
|
||
|
return fs.symlink(resolvedSrc, dest, cb)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// return true if dest is a subdir of src, otherwise false.
|
||
|
function isSrcSubdir (src, dest) {
|
||
|
const srcArray = path.resolve(src).split(path.sep)
|
||
|
const destArray = path.resolve(dest).split(path.sep)
|
||
|
return srcArray.reduce((acc, current, i) => acc && destArray[i] === current, true)
|
||
|
}
|
||
|
|
||
|
function checkStats (src, dest, cb) {
|
||
|
fs.stat(src, (err, srcStat) => {
|
||
|
if (err) return cb(err)
|
||
|
fs.stat(dest, (err, destStat) => {
|
||
|
if (err) {
|
||
|
if (err.code === 'ENOENT') return cb(null, {srcStat, destStat: notExist})
|
||
|
return cb(err)
|
||
|
}
|
||
|
return cb(null, {srcStat, destStat})
|
||
|
})
|
||
|
})
|
||
|
}
|
||
|
|
||
|
function checkPaths (src, dest, cb) {
|
||
|
checkStats(src, dest, (err, stats) => {
|
||
|
if (err) return cb(err)
|
||
|
const {srcStat, destStat} = stats
|
||
|
if (destStat.ino && destStat.ino === srcStat.ino) {
|
||
|
return cb(new Error('Source and destination must not be the same.'))
|
||
|
}
|
||
|
if (srcStat.isDirectory() && isSrcSubdir(src, dest)) {
|
||
|
return cb(new Error(`Cannot copy '${src}' to a subdirectory of itself, '${dest}'.`))
|
||
|
}
|
||
|
return cb(null, destStat)
|
||
|
})
|
||
|
}
|
||
|
|
||
|
module.exports = copy
|