361 lines
		
	
	
		
			8.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			361 lines
		
	
	
		
			8.7 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
const assert = require("assert")
 | 
						|
const path = require("path")
 | 
						|
const fs = require("fs")
 | 
						|
let glob = undefined
 | 
						|
try {
 | 
						|
  glob = require("glob")
 | 
						|
} catch (_err) {
 | 
						|
  // treat glob as optional.
 | 
						|
}
 | 
						|
 | 
						|
const defaultGlobOpts = {
 | 
						|
  nosort: true,
 | 
						|
  silent: true
 | 
						|
}
 | 
						|
 | 
						|
// for EMFILE handling
 | 
						|
let timeout = 0
 | 
						|
 | 
						|
const isWindows = (process.platform === "win32")
 | 
						|
 | 
						|
const defaults = options => {
 | 
						|
  const methods = [
 | 
						|
    'unlink',
 | 
						|
    'chmod',
 | 
						|
    'stat',
 | 
						|
    'lstat',
 | 
						|
    'rmdir',
 | 
						|
    'readdir'
 | 
						|
  ]
 | 
						|
  methods.forEach(m => {
 | 
						|
    options[m] = options[m] || fs[m]
 | 
						|
    m = m + 'Sync'
 | 
						|
    options[m] = options[m] || fs[m]
 | 
						|
  })
 | 
						|
 | 
						|
  options.maxBusyTries = options.maxBusyTries || 3
 | 
						|
  options.emfileWait = options.emfileWait || 1000
 | 
						|
  if (options.glob === false) {
 | 
						|
    options.disableGlob = true
 | 
						|
  }
 | 
						|
  if (options.disableGlob !== true && glob === undefined) {
 | 
						|
    throw Error('glob dependency not found, set `options.disableGlob = true` if intentional')
 | 
						|
  }
 | 
						|
  options.disableGlob = options.disableGlob || false
 | 
						|
  options.glob = options.glob || defaultGlobOpts
 | 
						|
}
 | 
						|
 | 
						|
const rimraf = (p, options, cb) => {
 | 
						|
  if (typeof options === 'function') {
 | 
						|
    cb = options
 | 
						|
    options = {}
 | 
						|
  }
 | 
						|
 | 
						|
  assert(p, 'rimraf: missing path')
 | 
						|
  assert.equal(typeof p, 'string', 'rimraf: path should be a string')
 | 
						|
  assert.equal(typeof cb, 'function', 'rimraf: callback function required')
 | 
						|
  assert(options, 'rimraf: invalid options argument provided')
 | 
						|
  assert.equal(typeof options, 'object', 'rimraf: options should be object')
 | 
						|
 | 
						|
  defaults(options)
 | 
						|
 | 
						|
  let busyTries = 0
 | 
						|
  let errState = null
 | 
						|
  let n = 0
 | 
						|
 | 
						|
  const next = (er) => {
 | 
						|
    errState = errState || er
 | 
						|
    if (--n === 0)
 | 
						|
      cb(errState)
 | 
						|
  }
 | 
						|
 | 
						|
  const afterGlob = (er, results) => {
 | 
						|
    if (er)
 | 
						|
      return cb(er)
 | 
						|
 | 
						|
    n = results.length
 | 
						|
    if (n === 0)
 | 
						|
      return cb()
 | 
						|
 | 
						|
    results.forEach(p => {
 | 
						|
      const CB = (er) => {
 | 
						|
        if (er) {
 | 
						|
          if ((er.code === "EBUSY" || er.code === "ENOTEMPTY" || er.code === "EPERM") &&
 | 
						|
              busyTries < options.maxBusyTries) {
 | 
						|
            busyTries ++
 | 
						|
            // try again, with the same exact callback as this one.
 | 
						|
            return setTimeout(() => rimraf_(p, options, CB), busyTries * 100)
 | 
						|
          }
 | 
						|
 | 
						|
          // this one won't happen if graceful-fs is used.
 | 
						|
          if (er.code === "EMFILE" && timeout < options.emfileWait) {
 | 
						|
            return setTimeout(() => rimraf_(p, options, CB), timeout ++)
 | 
						|
          }
 | 
						|
 | 
						|
          // already gone
 | 
						|
          if (er.code === "ENOENT") er = null
 | 
						|
        }
 | 
						|
 | 
						|
        timeout = 0
 | 
						|
        next(er)
 | 
						|
      }
 | 
						|
      rimraf_(p, options, CB)
 | 
						|
    })
 | 
						|
  }
 | 
						|
 | 
						|
  if (options.disableGlob || !glob.hasMagic(p))
 | 
						|
    return afterGlob(null, [p])
 | 
						|
 | 
						|
  options.lstat(p, (er, stat) => {
 | 
						|
    if (!er)
 | 
						|
      return afterGlob(null, [p])
 | 
						|
 | 
						|
    glob(p, options.glob, afterGlob)
 | 
						|
  })
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
// Two possible strategies.
 | 
						|
// 1. Assume it's a file.  unlink it, then do the dir stuff on EPERM or EISDIR
 | 
						|
// 2. Assume it's a directory.  readdir, then do the file stuff on ENOTDIR
 | 
						|
//
 | 
						|
// Both result in an extra syscall when you guess wrong.  However, there
 | 
						|
// are likely far more normal files in the world than directories.  This
 | 
						|
// is based on the assumption that a the average number of files per
 | 
						|
// directory is >= 1.
 | 
						|
//
 | 
						|
// If anyone ever complains about this, then I guess the strategy could
 | 
						|
// be made configurable somehow.  But until then, YAGNI.
 | 
						|
const rimraf_ = (p, options, cb) => {
 | 
						|
  assert(p)
 | 
						|
  assert(options)
 | 
						|
  assert(typeof cb === 'function')
 | 
						|
 | 
						|
  // sunos lets the root user unlink directories, which is... weird.
 | 
						|
  // so we have to lstat here and make sure it's not a dir.
 | 
						|
  options.lstat(p, (er, st) => {
 | 
						|
    if (er && er.code === "ENOENT")
 | 
						|
      return cb(null)
 | 
						|
 | 
						|
    // Windows can EPERM on stat.  Life is suffering.
 | 
						|
    if (er && er.code === "EPERM" && isWindows)
 | 
						|
      fixWinEPERM(p, options, er, cb)
 | 
						|
 | 
						|
    if (st && st.isDirectory())
 | 
						|
      return rmdir(p, options, er, cb)
 | 
						|
 | 
						|
    options.unlink(p, er => {
 | 
						|
      if (er) {
 | 
						|
        if (er.code === "ENOENT")
 | 
						|
          return cb(null)
 | 
						|
        if (er.code === "EPERM")
 | 
						|
          return (isWindows)
 | 
						|
            ? fixWinEPERM(p, options, er, cb)
 | 
						|
            : rmdir(p, options, er, cb)
 | 
						|
        if (er.code === "EISDIR")
 | 
						|
          return rmdir(p, options, er, cb)
 | 
						|
      }
 | 
						|
      return cb(er)
 | 
						|
    })
 | 
						|
  })
 | 
						|
}
 | 
						|
 | 
						|
const fixWinEPERM = (p, options, er, cb) => {
 | 
						|
  assert(p)
 | 
						|
  assert(options)
 | 
						|
  assert(typeof cb === 'function')
 | 
						|
 | 
						|
  options.chmod(p, 0o666, er2 => {
 | 
						|
    if (er2)
 | 
						|
      cb(er2.code === "ENOENT" ? null : er)
 | 
						|
    else
 | 
						|
      options.stat(p, (er3, stats) => {
 | 
						|
        if (er3)
 | 
						|
          cb(er3.code === "ENOENT" ? null : er)
 | 
						|
        else if (stats.isDirectory())
 | 
						|
          rmdir(p, options, er, cb)
 | 
						|
        else
 | 
						|
          options.unlink(p, cb)
 | 
						|
      })
 | 
						|
  })
 | 
						|
}
 | 
						|
 | 
						|
const fixWinEPERMSync = (p, options, er) => {
 | 
						|
  assert(p)
 | 
						|
  assert(options)
 | 
						|
 | 
						|
  try {
 | 
						|
    options.chmodSync(p, 0o666)
 | 
						|
  } catch (er2) {
 | 
						|
    if (er2.code === "ENOENT")
 | 
						|
      return
 | 
						|
    else
 | 
						|
      throw er
 | 
						|
  }
 | 
						|
 | 
						|
  let stats
 | 
						|
  try {
 | 
						|
    stats = options.statSync(p)
 | 
						|
  } catch (er3) {
 | 
						|
    if (er3.code === "ENOENT")
 | 
						|
      return
 | 
						|
    else
 | 
						|
      throw er
 | 
						|
  }
 | 
						|
 | 
						|
  if (stats.isDirectory())
 | 
						|
    rmdirSync(p, options, er)
 | 
						|
  else
 | 
						|
    options.unlinkSync(p)
 | 
						|
}
 | 
						|
 | 
						|
const rmdir = (p, options, originalEr, cb) => {
 | 
						|
  assert(p)
 | 
						|
  assert(options)
 | 
						|
  assert(typeof cb === 'function')
 | 
						|
 | 
						|
  // try to rmdir first, and only readdir on ENOTEMPTY or EEXIST (SunOS)
 | 
						|
  // if we guessed wrong, and it's not a directory, then
 | 
						|
  // raise the original error.
 | 
						|
  options.rmdir(p, er => {
 | 
						|
    if (er && (er.code === "ENOTEMPTY" || er.code === "EEXIST" || er.code === "EPERM"))
 | 
						|
      rmkids(p, options, cb)
 | 
						|
    else if (er && er.code === "ENOTDIR")
 | 
						|
      cb(originalEr)
 | 
						|
    else
 | 
						|
      cb(er)
 | 
						|
  })
 | 
						|
}
 | 
						|
 | 
						|
const rmkids = (p, options, cb) => {
 | 
						|
  assert(p)
 | 
						|
  assert(options)
 | 
						|
  assert(typeof cb === 'function')
 | 
						|
 | 
						|
  options.readdir(p, (er, files) => {
 | 
						|
    if (er)
 | 
						|
      return cb(er)
 | 
						|
    let n = files.length
 | 
						|
    if (n === 0)
 | 
						|
      return options.rmdir(p, cb)
 | 
						|
    let errState
 | 
						|
    files.forEach(f => {
 | 
						|
      rimraf(path.join(p, f), options, er => {
 | 
						|
        if (errState)
 | 
						|
          return
 | 
						|
        if (er)
 | 
						|
          return cb(errState = er)
 | 
						|
        if (--n === 0)
 | 
						|
          options.rmdir(p, cb)
 | 
						|
      })
 | 
						|
    })
 | 
						|
  })
 | 
						|
}
 | 
						|
 | 
						|
// this looks simpler, and is strictly *faster*, but will
 | 
						|
// tie up the JavaScript thread and fail on excessively
 | 
						|
// deep directory trees.
 | 
						|
const rimrafSync = (p, options) => {
 | 
						|
  options = options || {}
 | 
						|
  defaults(options)
 | 
						|
 | 
						|
  assert(p, 'rimraf: missing path')
 | 
						|
  assert.equal(typeof p, 'string', 'rimraf: path should be a string')
 | 
						|
  assert(options, 'rimraf: missing options')
 | 
						|
  assert.equal(typeof options, 'object', 'rimraf: options should be object')
 | 
						|
 | 
						|
  let results
 | 
						|
 | 
						|
  if (options.disableGlob || !glob.hasMagic(p)) {
 | 
						|
    results = [p]
 | 
						|
  } else {
 | 
						|
    try {
 | 
						|
      options.lstatSync(p)
 | 
						|
      results = [p]
 | 
						|
    } catch (er) {
 | 
						|
      results = glob.sync(p, options.glob)
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  if (!results.length)
 | 
						|
    return
 | 
						|
 | 
						|
  for (let i = 0; i < results.length; i++) {
 | 
						|
    const p = results[i]
 | 
						|
 | 
						|
    let st
 | 
						|
    try {
 | 
						|
      st = options.lstatSync(p)
 | 
						|
    } catch (er) {
 | 
						|
      if (er.code === "ENOENT")
 | 
						|
        return
 | 
						|
 | 
						|
      // Windows can EPERM on stat.  Life is suffering.
 | 
						|
      if (er.code === "EPERM" && isWindows)
 | 
						|
        fixWinEPERMSync(p, options, er)
 | 
						|
    }
 | 
						|
 | 
						|
    try {
 | 
						|
      // sunos lets the root user unlink directories, which is... weird.
 | 
						|
      if (st && st.isDirectory())
 | 
						|
        rmdirSync(p, options, null)
 | 
						|
      else
 | 
						|
        options.unlinkSync(p)
 | 
						|
    } catch (er) {
 | 
						|
      if (er.code === "ENOENT")
 | 
						|
        return
 | 
						|
      if (er.code === "EPERM")
 | 
						|
        return isWindows ? fixWinEPERMSync(p, options, er) : rmdirSync(p, options, er)
 | 
						|
      if (er.code !== "EISDIR")
 | 
						|
        throw er
 | 
						|
 | 
						|
      rmdirSync(p, options, er)
 | 
						|
    }
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
const rmdirSync = (p, options, originalEr) => {
 | 
						|
  assert(p)
 | 
						|
  assert(options)
 | 
						|
 | 
						|
  try {
 | 
						|
    options.rmdirSync(p)
 | 
						|
  } catch (er) {
 | 
						|
    if (er.code === "ENOENT")
 | 
						|
      return
 | 
						|
    if (er.code === "ENOTDIR")
 | 
						|
      throw originalEr
 | 
						|
    if (er.code === "ENOTEMPTY" || er.code === "EEXIST" || er.code === "EPERM")
 | 
						|
      rmkidsSync(p, options)
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
const rmkidsSync = (p, options) => {
 | 
						|
  assert(p)
 | 
						|
  assert(options)
 | 
						|
  options.readdirSync(p).forEach(f => rimrafSync(path.join(p, f), options))
 | 
						|
 | 
						|
  // We only end up here once we got ENOTEMPTY at least once, and
 | 
						|
  // at this point, we are guaranteed to have removed all the kids.
 | 
						|
  // So, we know that it won't be ENOENT or ENOTDIR or anything else.
 | 
						|
  // try really hard to delete stuff on windows, because it has a
 | 
						|
  // PROFOUNDLY annoying habit of not closing handles promptly when
 | 
						|
  // files are deleted, resulting in spurious ENOTEMPTY errors.
 | 
						|
  const retries = isWindows ? 100 : 1
 | 
						|
  let i = 0
 | 
						|
  do {
 | 
						|
    let threw = true
 | 
						|
    try {
 | 
						|
      const ret = options.rmdirSync(p, options)
 | 
						|
      threw = false
 | 
						|
      return ret
 | 
						|
    } finally {
 | 
						|
      if (++i < retries && threw)
 | 
						|
        continue
 | 
						|
    }
 | 
						|
  } while (true)
 | 
						|
}
 | 
						|
 | 
						|
module.exports = rimraf
 | 
						|
rimraf.sync = rimrafSync
 |