-
Notifications
You must be signed in to change notification settings - Fork 30.2k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
254 additions
and
19 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,192 @@ | ||
'use strict'; | ||
|
||
const { EventEmitter } = require('events'); | ||
const path = require('path'); | ||
const { SafeMap, Symbol, StringPrototypeStartsWith } = primordials; | ||
const { validateObject } = require('internal/validators'); | ||
const { kEmptyObject } = require('internal/util'); | ||
const { ERR_FEATURE_UNAVAILABLE_ON_PLATFORM } = require('internal/errors'); | ||
|
||
const kFSWatchStart = Symbol('kFSWatchStart'); | ||
|
||
let internalSync; | ||
let internalPromises; | ||
|
||
function lazyLoadFsPromises() { | ||
internalPromises ??= require('fs/promises'); | ||
return internalPromises; | ||
} | ||
|
||
function lazyLoadFsSync() { | ||
internalSync ??= require('fs'); | ||
return internalSync; | ||
} | ||
|
||
async function traverse(dir, files = new SafeMap()) { | ||
const { stat, opendir } = lazyLoadFsPromises(); | ||
|
||
files.set(dir, await stat(dir)); | ||
|
||
try { | ||
const directory = await opendir(dir); | ||
|
||
for await (const file of directory) { | ||
const f = path.join(dir, file.name); | ||
|
||
try { | ||
const stats = await stat(f); | ||
|
||
files.set(f, stats); | ||
|
||
if (stats.isDirectory()) { | ||
await traverse(f, files); | ||
} | ||
} catch (error) { | ||
if (error.code !== 'ENOENT' || error.code !== 'EPERM') { | ||
this.emit('error', error); | ||
} | ||
} | ||
|
||
} | ||
} catch (error) { | ||
if (error.code !== 'EACCES') { | ||
this.emit('error', error); | ||
} | ||
} | ||
|
||
return files; | ||
} | ||
|
||
class FSWatcher extends EventEmitter { | ||
#options = null; | ||
#closed = false; | ||
#files = new SafeMap(); | ||
|
||
/** | ||
* @param {{ | ||
* persistent?: boolean; | ||
* recursive?: boolean; | ||
* encoding?: string; | ||
* signal?: AbortSignal; | ||
* }} [options] | ||
*/ | ||
constructor(options = kEmptyObject) { | ||
super(); | ||
|
||
validateObject(options, 'options'); | ||
this.#options = options; | ||
} | ||
|
||
close() { | ||
const { unwatchFile } = lazyLoadFsSync(); | ||
this.#closed = true; | ||
|
||
for (const file of this.#files.keys()) { | ||
unwatchFile(file); | ||
} | ||
|
||
this.emit('close'); | ||
} | ||
|
||
#unwatchFolder(file) { | ||
const { unwatchFile } = lazyLoadFsSync(); | ||
|
||
for (const filename in this.#files) { | ||
if (StringPrototypeStartsWith(filename, file)) { | ||
unwatchFile(filename); | ||
} | ||
} | ||
} | ||
|
||
async #watchFolder(folder) { | ||
const { opendir, stat } = lazyLoadFsPromises(); | ||
|
||
try { | ||
const files = await opendir(folder); | ||
|
||
for await (const file of files) { | ||
const f = path.join(folder, file.name); | ||
|
||
if (this.#closed) { | ||
return; | ||
} | ||
|
||
if (!this.#files.has(f)) { | ||
const fileStats = await stat(f); | ||
this.#files.set(f, fileStats); | ||
this.emit('change', 'rename', f); | ||
this.#watchFile(f); | ||
} | ||
} | ||
} catch (error) { | ||
this.emit('error', error); | ||
} | ||
} | ||
|
||
/** | ||
* @param {string} file | ||
*/ | ||
#watchFile(file) { | ||
const { watchFile } = lazyLoadFsSync(); | ||
|
||
if (this.#closed) { | ||
return; | ||
} | ||
|
||
const existingStat = this.#files.get(file); | ||
|
||
watchFile(file, { | ||
persistent: this.#options.persistent, | ||
}, (statWatcher, previousStatWatcher) => { | ||
if (existingStat && !existingStat.isDirectory() && | ||
statWatcher.nlink !== 0 && existingStat.mtime.getTime() === statWatcher.mtime.getTime()) { | ||
return; | ||
} | ||
|
||
this.#files.set(file, statWatcher); | ||
|
||
if (statWatcher.isDirectory()) { | ||
this.#watchFolder(file); | ||
} else if (statWatcher.birthtimeMs === 0 && previousStatWatcher.birthtimeMs !== 0) { | ||
// The file is now deleted | ||
this.#files.delete(file); | ||
this.emit('change', 'rename', file); | ||
|
||
if (statWatcher.isDirectory()) { | ||
this.#unwatchFolder(file); | ||
} | ||
} else { | ||
this.emit('change', 'change', file); | ||
} | ||
}); | ||
} | ||
|
||
/** | ||
* @param {string | Buffer | URL} filename | ||
*/ | ||
async [kFSWatchStart](filename) { | ||
this.#closed = false; | ||
this.#files = await traverse(filename); | ||
|
||
this.#watchFile(filename); | ||
|
||
for (const f in this.#files) { | ||
this.#watchFile(f); | ||
} | ||
} | ||
|
||
ref() { | ||
// This is kept to have the same API with FSWatcher | ||
throw new ERR_FEATURE_UNAVAILABLE_ON_PLATFORM('ref'); | ||
} | ||
|
||
unref() { | ||
// This is kept to have the same API with FSWatcher | ||
throw new ERR_FEATURE_UNAVAILABLE_ON_PLATFORM('unref'); | ||
} | ||
} | ||
|
||
module.exports = { | ||
FSWatcher, | ||
kFSWatchStart, | ||
}; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,44 @@ | ||
'use strict'; | ||
|
||
const common = require('../common'); | ||
|
||
if (!common.hasCrypto) | ||
common.skip('missing crypto'); | ||
|
||
// Only run these tests on Linux. | ||
if (!common.isLinux) { | ||
return; | ||
} | ||
|
||
const { randomUUID } = require('crypto'); | ||
const assert = require('assert'); | ||
const path = require('path'); | ||
const fs = require('fs'); | ||
|
||
const tmpdir = require('../common/tmpdir'); | ||
const testDir = tmpdir.path; | ||
tmpdir.refresh(); | ||
|
||
{ | ||
const file = `${randomUUID()}.txt`; | ||
const testsubdir = fs.mkdtempSync(testDir + path.sep); | ||
const filePath = path.join(testsubdir, file); | ||
|
||
const watcher = fs.watch(testsubdir, { recursive: true }); | ||
|
||
let watcherClosed = false; | ||
watcher.on('change', function(event, filename) { | ||
assert.ok(event === 'change' || event === 'rename'); | ||
|
||
watcher.close(); | ||
watcherClosed = true; | ||
}); | ||
|
||
setTimeout(() => { | ||
fs.writeFileSync(filePath, 'world'); | ||
}, 100); | ||
|
||
process.on('exit', function() { | ||
assert(watcherClosed, 'watcher Object was not closed'); | ||
}); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters