@muhgholy/next-drive 4.23.1 → 4.23.2

package/README.md

@@ -521,6 +521,20 @@ const stats = await DatabaseMongoDrive.aggregate([
 ]);
 ```
 
+### Cleanup Orphaned Files
+
+Remove orphaned file folders from storage that no longer exist in the database, and clean up leftover temp directories:
+
+```typescript
+import { driveCleanup } from "@muhgholy/next-drive/server";
+
+const result = await driveCleanup();
+console.log(`Removed ${result.removed.length} orphaned folders`);
+console.log(`Freed ${result.totalFreedInBytes} bytes`);
+```
+
+**Returns:** `{ removed: string[], totalFreedInBytes: number }`
+
 ### Delete File or Folder
 
 Permanently delete a file or folder from the drive system:

package/dist/{chunk-MZFX6M44.js → chunk-R43JCXQB.js}

@@ -1,7 +1,7 @@
 import formidable from 'formidable';
 import path from 'path';
 import fs from 'fs';
-import os3 from 'os';
+import os2 from 'os';
 import crypto2 from 'crypto';
 import mongoose, { Schema, isValidObjectId } from 'mongoose';
 import sharp2 from 'sharp';
@@ -260,7 +260,7 @@ var driveConfiguration = async (config) => {
     if (g.migrationPromise) await g.migrationPromise;
     return g.config;
   }
-  const resolvedPath = config.storage?.path || path.join(os3.tmpdir(), "next-drive-data");
+  const resolvedPath = config.storage?.path || path.join(os2.tmpdir(), "next-drive-data");
   const mode = config.mode || "NORMAL";
   if (mode === "ROOT") {
     g.config = {
@@ -1594,7 +1594,7 @@ var driveUpload = async (source, key, options) => {
     const stats = fs.statSync(source);
     fileSize = stats.size;
   } else if (Buffer.isBuffer(source)) {
-    const tempDir = path.join(os3.tmpdir(), "next-drive-uploads");
+    const tempDir = path.join(os2.tmpdir(), "next-drive-uploads");
     if (!fs.existsSync(tempDir)) {
       fs.mkdirSync(tempDir, { recursive: true });
     }
@@ -1603,7 +1603,7 @@ var driveUpload = async (source, key, options) => {
     sourceFilePath = tempFilePath;
     fileSize = source.length;
   } else {
-    const tempDir = path.join(os3.tmpdir(), "next-drive-uploads");
+    const tempDir = path.join(os2.tmpdir(), "next-drive-uploads");
     if (!fs.existsSync(tempDir)) {
       fs.mkdirSync(tempDir, { recursive: true });
     }
@@ -1703,6 +1703,81 @@ var driveUpload = async (source, key, options) => {
     }
   }
 };
+var driveCleanup = async () => {
+  const config = getDriveConfig();
+  const fileDir = path.join(config.storage.path, "file");
+  if (!fs.existsSync(fileDir)) {
+    return { removed: [], totalFreedInBytes: 0 };
+  }
+  const folderNames = fs.readdirSync(fileDir).filter((name) => {
+    const fullPath = path.join(fileDir, name);
+    return fs.statSync(fullPath).isDirectory();
+  });
+  if (folderNames.length === 0) {
+    return { removed: [], totalFreedInBytes: 0 };
+  }
+  const getDirSize = (dirPath) => {
+    let size = 0;
+    try {
+      const entries = fs.readdirSync(dirPath, { withFileTypes: true });
+      for (const entry of entries) {
+        const entryPath = path.join(dirPath, entry.name);
+        if (entry.isFile()) {
+          size += fs.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.join(fileDir, name);
+      try {
+        totalFreedInBytes += getDirSize(dirPath);
+        fs.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.join(config.storage.path, "temp");
+  if (fs.existsSync(tempDir)) {
+    try {
+      totalFreedInBytes += getDirSize(tempDir);
+      fs.rmSync(tempDir, { recursive: true, force: true });
+    } catch (e) {
+      console.error("[next-drive] Failed to remove temp directory:", e);
+    }
+  }
+  const systemTmpDir = path.join(os2.tmpdir(), "next-drive-uploads");
+  if (fs.existsSync(systemTmpDir)) {
+    try {
+      totalFreedInBytes += getDirSize(systemTmpDir);
+      fs.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) => {
@@ -2120,7 +2195,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.join(os3.tmpdir(), "next-drive-uploads");
+        const systemTmpDir = path.join(os2.tmpdir(), "next-drive-uploads");
         if (!fs.existsSync(systemTmpDir)) fs.mkdirSync(systemTmpDir, { recursive: true });
         const form = formidable({
           multiples: false,
@@ -2156,7 +2231,7 @@ var driveAPIHandler = async (req, res) => {
         }
         const { chunkIndex, totalChunks, driveId, fileName, fileSize: fileSizeInBytes, fileType, folderId } = uploadData.data;
         let currentUploadId = driveId;
-        const tempBaseDir = path.join(os3.tmpdir(), "next-drive-uploads");
+        const tempBaseDir = path.join(os2.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) {
@@ -2304,7 +2379,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.join(os3.tmpdir(), "next-drive-uploads", id);
+        const tempUploadDir = path.join(os2.tmpdir(), "next-drive-uploads", id);
         if (fs.existsSync(tempUploadDir)) {
           try {
             fs.rmSync(tempUploadDir, { recursive: true, force: true });
@@ -2449,6 +2524,6 @@ var driveAPIHandler = async (req, res) => {
   }
 };
 
-export { driveAPIHandler, driveConfiguration, driveDelete, driveFilePath, driveFileSchemaZod, driveGetUrl, driveInfo, driveList, driveListFiles, driveReadFile, driveUpload, drive_default, getDriveConfig, getDriveInformation };
-//# sourceMappingURL=chunk-MZFX6M44.js.map
-//# sourceMappingURL=chunk-MZFX6M44.js.map
+export { driveAPIHandler, driveCleanup, driveConfiguration, driveDelete, driveFilePath, driveFileSchemaZod, driveGetUrl, driveInfo, driveList, driveListFiles, driveReadFile, driveUpload, drive_default, getDriveConfig, getDriveInformation };
+//# sourceMappingURL=chunk-R43JCXQB.js.map
+//# sourceMappingURL=chunk-R43JCXQB.js.map