From 970b4d5d977ff98779edbab430c6dfd566cdea04 Mon Sep 17 00:00:00 2001
From: Book Pauk <bookpauk@gmail.com>
Date: Mon, 28 Oct 2019 20:52:04 +0700
Subject: [PATCH] =?UTF-8?q?=D0=9C=D0=B5=D0=BB=D0=BA=D0=B8=D0=B9=20=D1=80?=
 =?UTF-8?q?=D0=B5=D1=84=D0=B0=D0=BA=D1=82=D0=BE=D1=80=D0=B8=D0=BD=D0=B3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server/core/LibSharedStorage/MegaStorage.js | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/server/core/LibSharedStorage/MegaStorage.js b/server/core/LibSharedStorage/MegaStorage.js
index a3658b19..7f5037e5 100644
--- a/server/core/LibSharedStorage/MegaStorage.js
+++ b/server/core/LibSharedStorage/MegaStorage.js
@@ -57,23 +57,36 @@ class MegaStorage {
         await fs.writeFile(nameHash.descPath, JSON.stringify(desc, null, 2));
     }
 
-    async readFiles(callback, dir) {
-        if (!callback)
+    async _findFiles(callback, dir) {
+        if (!callback || !this.readingFiles)
             return;
         if (!dir)
             dir = this.megaStorageDir;
 
         const files = await fs.readdir(dir, { withFileTypes: true });
         for (const file of files) {
+            if (!this.readingFiles)
+                return;
             const found = path.resolve(dir, file.name);
             if (file.isDirectory())
-                await this.readFiles(callback, found);
+                await this._findFiles(callback, found);
             else
                 callback(found);
         }
+        return true;
     }
 
-    async stopReadFiles() {
+    async startFindFiles(callback, dir) {
+        this.readingFiles = true;
+        try {
+            return await this._findFiles(callback, dir);
+        } finally {
+            this.readingFiles = false;
+        }
+    }
+
+    async stopFindFiles() {
+        this.readingFiles = false;
     }
 
     async getStats(gather = false) {