Fixed for polling servers

It should now be able to watch symlinks that lead to outside of the watched folder when polling

Author: ArjhanToteck

lib/DirectoryWatcher.js

@@ -26,7 +26,7 @@ function withoutCase(str) {
 }
 
 function needCalls(times, callback) {
-	return function() {
+	return function () {
 		if (--times === 0) {
 			return callback();
 		}
@@ -61,6 +61,7 @@ class DirectoryWatcher extends EventEmitter {
 		this.watcherManager = watcherManager;
 		this.options = options;
 		this.path = directoryPath;
+		this.watchingSymlink = false;
 		// safeTime is the point in time after which reading is safe to be unchanged
 		// timestamp is a value that should be compared with another timestamp (mtime)
 		/** @type {Map<string, { safeTime: number, timestamp: number }} */
@@ -76,8 +77,8 @@ class DirectoryWatcher extends EventEmitter {
 			typeof options.poll === "number"
 				? options.poll
 				: options.poll
-				? 5007
-				: false;
+					? 5007
+					: false;
 		this.timeout = undefined;
 		this.initialScanRemoved = new Set();
 		this.initialScanFinished = undefined;
@@ -111,8 +112,8 @@ class DirectoryWatcher extends EventEmitter {
 					this.watchInParentDirectory();
 				}
 				this.watcher = watchEventSource.watch(this.path);
-				this.watcher.on("change", this.onWatchEvent.bind(this));
 				this.watcher.on("error", this.onWatcherError.bind(this));
+				this.watcher.on("change", this.onWatchEvent.bind(this));
 			}
 		} catch (err) {
 			this.onWatcherError(err);
@@ -550,11 +551,6 @@ class DirectoryWatcher extends EventEmitter {
 			fs.readdir(this.path, (err, items) => {
 				if (this.closed) return;
 				if (err) {
-					if (err.code === "ENOENT" || err.code === "EPERM") {
-						this.onDirectoryRemoved("scan readdir failed");
-					} else {
-						this.onScanError(err);
-					}
 					this.initialScan = false;
 					this.initialScanFinished = Date.now();
 					if (initial) {
@@ -627,22 +623,7 @@ class DirectoryWatcher extends EventEmitter {
 					}
 				});
 				for (const itemPath of itemPaths) {
-					const handleStats = (err2, stats) => {
-						if (this.closed) return;
-						if (err2) {
-							if (
-								err2.code === "ENOENT" ||
-								err2.code === "EPERM" ||
-								err2.code === "EACCES" ||
-								err2.code === "EBUSY"
-							) {
-								this.setMissing(itemPath, initial, "scan (" + err2.code + ")");
-							} else {
-								this.onScanError(err2);
-							}
-							itemFinished();
-							return;
-						}
+					const handleStats = (stats, symlinkStats) => {
 						if (stats.isFile() || stats.isSymbolicLink()) {
 							if (stats.mtime) {
 								ensureFsAccuracy(stats.mtime);
@@ -654,7 +635,11 @@ class DirectoryWatcher extends EventEmitter {
 								true,
 								"scan (file)"
 							);
-						} else if (stats.isDirectory()) {
+						}
+						if (
+							stats.isDirectory() ||
+							(symlinkStats && symlinkStats.isDirectory())
+						) {
 							if (!initial || !this.directories.has(itemPath))
 								this.setDirectory(
 									itemPath,
@@ -665,11 +650,42 @@ class DirectoryWatcher extends EventEmitter {
 						}
 						itemFinished();
 					};
-					if (this.watcherManager.options.followSymlinks) {
-						fs.stat(itemPath, handleStats);
-					} else {
-						fs.lstat(itemPath, handleStats);
-					}
+					fs.lstat(itemPath, (err2, stats) => {
+						if (this.closed) return;
+						if (err2) {
+							if (
+								err2.code === "ENOENT" ||
+								err2.code === "EPERM" ||
+								err2.code === "EACCES" ||
+								err2.code === "EBUSY"
+							) {
+								this.setMissing(itemPath, initial, "scan (" + err2.code + ")");
+							} else {
+								this.onScanError(err2);
+							}
+							itemFinished();
+							return;
+						}
+						if (
+							stats.isSymbolicLink() &&
+							this.watcherManager.options.followSymlinks
+						) {
+							fs.stat(itemPath, (err3, symlinkStats) => {
+								if (this.closed) return;
+								// something is wrong with the symlink, but not with the file itself
+								if (err3) {
+									handleStats(stats);
+									this.watchingSymlink = false;
+									return;
+								}
+								this.watchingSymlink = true;
+								handleStats(stats, symlinkStats);
+							});
+						} else {
+							this.watchingSymlink = false;
+							handleStats(stats);
+						}
+					});
 				}
 				itemFinished();
 			});