@muhgholy/next-drive 4.23.1 → 4.23.2

package/dist/{chunk-BGCV4Y2V.cjs → chunk-RBLDH7CP.cjs}

@@ -3,7 +3,7 @@
 var formidable = require('formidable');
 var path = require('path');
 var fs = require('fs');
-var os3 = require('os');
+var os2 = require('os');
 var crypto2 = require('crypto');
 var mongoose = require('mongoose');
 var sharp2 = require('sharp');
@@ -16,7 +16,7 @@ function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
 var formidable__default = /*#__PURE__*/_interopDefault(formidable);
 var path__default = /*#__PURE__*/_interopDefault(path);
 var fs__default = /*#__PURE__*/_interopDefault(fs);
-var os3__default = /*#__PURE__*/_interopDefault(os3);
+var os2__default = /*#__PURE__*/_interopDefault(os2);
 var crypto2__default = /*#__PURE__*/_interopDefault(crypto2);
 var mongoose__default = /*#__PURE__*/_interopDefault(mongoose);
 var sharp2__default = /*#__PURE__*/_interopDefault(sharp2);
@@ -273,7 +273,7 @@ var driveConfiguration = async (config) => {
     if (g.migrationPromise) await g.migrationPromise;
     return g.config;
   }
-  const resolvedPath = config.storage?.path || path__default.default.join(os3__default.default.tmpdir(), "next-drive-data");
+  const resolvedPath = config.storage?.path || path__default.default.join(os2__default.default.tmpdir(), "next-drive-data");
   const mode = config.mode || "NORMAL";
   if (mode === "ROOT") {
     g.config = {
@@ -1607,7 +1607,7 @@ var driveUpload = async (source, key, options) => {
     const stats = fs__default.default.statSync(source);
     fileSize = stats.size;
   } else if (Buffer.isBuffer(source)) {
-    const tempDir = path__default.default.join(os3__default.default.tmpdir(), "next-drive-uploads");
+    const tempDir = path__default.default.join(os2__default.default.tmpdir(), "next-drive-uploads");
     if (!fs__default.default.existsSync(tempDir)) {
       fs__default.default.mkdirSync(tempDir, { recursive: true });
     }
@@ -1616,7 +1616,7 @@ var driveUpload = async (source, key, options) => {
     sourceFilePath = tempFilePath;
     fileSize = source.length;
   } else {
-    const tempDir = path__default.default.join(os3__default.default.tmpdir(), "next-drive-uploads");
+    const tempDir = path__default.default.join(os2__default.default.tmpdir(), "next-drive-uploads");
     if (!fs__default.default.existsSync(tempDir)) {
       fs__default.default.mkdirSync(tempDir, { recursive: true });
     }
@@ -1716,6 +1716,81 @@ var driveUpload = async (source, key, options) => {
     }
   }
 };
+var driveCleanup = async () => {
+  const config = getDriveConfig();
+  const fileDir = path__default.default.join(config.storage.path, "file");
+  if (!fs__default.default.existsSync(fileDir)) {
+    return { removed: [], totalFreedInBytes: 0 };
+  }
+  const folderNames = fs__default.default.readdirSync(fileDir).filter((name) => {
+    const fullPath = path__default.default.join(fileDir, name);
+    return fs__default.default.statSync(fullPath).isDirectory();
+  });
+  if (folderNames.length === 0) {
+    return { removed: [], totalFreedInBytes: 0 };
+  }
+  const getDirSize = (dirPath) => {
+    let size = 0;
+    try {
+      const entries = fs__default.default.readdirSync(dirPath, { withFileTypes: true });
+      for (const entry of entries) {
+        const entryPath = path__default.default.join(dirPath, entry.name);
+        if (entry.isFile()) {
+          size += fs__default.default.statSync(entryPath).size;
+        } else if (entry.isDirectory()) {
+          size += getDirSize(entryPath);
+        }
+      }
+    } catch {
+    }
+    return size;
+  };
+  const BATCH_SIZE = 500;
+  const existingIds = /* @__PURE__ */ new Set();
+  for (let i = 0; i < folderNames.length; i += BATCH_SIZE) {
+    const batch = folderNames.slice(i, i + BATCH_SIZE);
+    const docs = await drive_default.find(
+      { _id: { $in: batch } },
+      { _id: 1 }
+    ).lean();
+    for (const doc of docs) {
+      existingIds.add(doc._id.toString());
+    }
+  }
+  const removed = [];
+  let totalFreedInBytes = 0;
+  for (const name of folderNames) {
+    if (!existingIds.has(name)) {
+      const dirPath = path__default.default.join(fileDir, name);
+      try {
+        totalFreedInBytes += getDirSize(dirPath);
+        fs__default.default.rmSync(dirPath, { recursive: true, force: true });
+        removed.push(name);
+      } catch (e) {
+        console.error(`[next-drive] Failed to remove orphaned folder ${name}:`, e);
+      }
+    }
+  }
+  const tempDir = path__default.default.join(config.storage.path, "temp");
+  if (fs__default.default.existsSync(tempDir)) {
+    try {
+      totalFreedInBytes += getDirSize(tempDir);
+      fs__default.default.rmSync(tempDir, { recursive: true, force: true });
+    } catch (e) {
+      console.error("[next-drive] Failed to remove temp directory:", e);
+    }
+  }
+  const systemTmpDir = path__default.default.join(os2__default.default.tmpdir(), "next-drive-uploads");
+  if (fs__default.default.existsSync(systemTmpDir)) {
+    try {
+      totalFreedInBytes += getDirSize(systemTmpDir);
+      fs__default.default.rmSync(systemTmpDir, { recursive: true, force: true });
+    } catch (e) {
+      console.error("[next-drive] Failed to remove system temp directory:", e);
+    }
+  }
+  return { removed, totalFreedInBytes };
+};
 
 // src/server/index.ts
 var getProvider = async (req, owner) => {
@@ -2133,7 +2208,7 @@ var driveAPIHandler = async (req, res) => {
       // ** 3. UPLOAD **
       case "upload": {
         if (req.method !== "POST") return res.status(405).json({ status: 405, message: "Only POST allowed" });
-        const systemTmpDir = path__default.default.join(os3__default.default.tmpdir(), "next-drive-uploads");
+        const systemTmpDir = path__default.default.join(os2__default.default.tmpdir(), "next-drive-uploads");
         if (!fs__default.default.existsSync(systemTmpDir)) fs__default.default.mkdirSync(systemTmpDir, { recursive: true });
         const form = formidable__default.default({
           multiples: false,
@@ -2169,7 +2244,7 @@ var driveAPIHandler = async (req, res) => {
         }
         const { chunkIndex, totalChunks, driveId, fileName, fileSize: fileSizeInBytes, fileType, folderId } = uploadData.data;
         let currentUploadId = driveId;
-        const tempBaseDir = path__default.default.join(os3__default.default.tmpdir(), "next-drive-uploads");
+        const tempBaseDir = path__default.default.join(os2__default.default.tmpdir(), "next-drive-uploads");
         if (!currentUploadId) {
           if (chunkIndex !== 0) return res.status(400).json({ message: "Missing upload ID for non-zero chunk" });
           if (fileType && config.security) {
@@ -2317,7 +2392,7 @@ var driveAPIHandler = async (req, res) => {
         const cancelData = cancelQuerySchema.safeParse(req.query);
         if (!cancelData.success) return res.status(400).json({ status: 400, message: "Invalid ID" });
         const { id } = cancelData.data;
-        const tempUploadDir = path__default.default.join(os3__default.default.tmpdir(), "next-drive-uploads", id);
+        const tempUploadDir = path__default.default.join(os2__default.default.tmpdir(), "next-drive-uploads", id);
         if (fs__default.default.existsSync(tempUploadDir)) {
           try {
             fs__default.default.rmSync(tempUploadDir, { recursive: true, force: true });
@@ -2463,6 +2538,7 @@ var driveAPIHandler = async (req, res) => {
 };
 
 exports.driveAPIHandler = driveAPIHandler;
+exports.driveCleanup = driveCleanup;
 exports.driveConfiguration = driveConfiguration;
 exports.driveDelete = driveDelete;
 exports.driveFilePath = driveFilePath;
@@ -2476,5 +2552,5 @@ exports.driveUpload = driveUpload;
 exports.drive_default = drive_default;
 exports.getDriveConfig = getDriveConfig;
 exports.getDriveInformation = getDriveInformation;
-//# sourceMappingURL=chunk-BGCV4Y2V.cjs.map
-//# sourceMappingURL=chunk-BGCV4Y2V.cjs.map
+//# sourceMappingURL=chunk-RBLDH7CP.cjs.map
+//# sourceMappingURL=chunk-RBLDH7CP.cjs.map