@muhgholy/next-drive 4.23.0 → 4.23.2

package/dist/{chunk-EUAHSB2J.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 = {
@@ -1464,6 +1464,51 @@ var driveList = async (options) => {
   const items = await drive_default.find(query, {}, { sort: { order: 1, _id: -1 }, limit });
   return await Promise.all(items.map((item) => item.toClient()));
 };
+var driveListFiles = async (options) => {
+  const { key, folderId, accountId } = options;
+  const page = Math.max(1, options.page ?? 1);
+  const limit = Math.min(Math.max(1, options.limit ?? 50), 100);
+  let providerName = "LOCAL";
+  if (accountId && accountId !== "LOCAL") {
+    const account = await drive_default.db.model("StorageAccount").findOne({ _id: accountId, owner: key });
+    if (!account) {
+      throw new Error("Invalid Storage Account");
+    }
+    if (account.metadata.provider === "GOOGLE") {
+      providerName = "GOOGLE";
+    }
+  }
+  const query = {
+    "provider.type": providerName,
+    "information.type": "FILE",
+    storageAccountId: accountId || null,
+    trashedAt: null
+  };
+  if (key !== void 0) {
+    query.owner = key;
+  }
+  if (folderId && folderId !== "root") {
+    query.parentId = folderId;
+  } else if (folderId === "root" || folderId === null) {
+    query.parentId = null;
+  }
+  const skip = (page - 1) * limit;
+  const [totalCount, items] = await Promise.all([
+    drive_default.countDocuments(query),
+    drive_default.find(query, {}, { sort: { createdAt: -1 }, skip, limit })
+  ]);
+  const totalPages = Math.ceil(totalCount / limit);
+  return {
+    items: await Promise.all(items.map((item) => item.toClient())),
+    pagination: {
+      page,
+      limit,
+      totalCount,
+      totalPages,
+      hasMore: page < totalPages
+    }
+  };
+};
 var driveDelete = async (source, options) => {
   const { recurse = true } = options || {};
   let drive;
@@ -1562,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 });
     }
@@ -1571,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 });
     }
@@ -1671,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) => {
@@ -2088,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,
@@ -2124,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) {
@@ -2272,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 });
@@ -2418,6 +2538,7 @@ var driveAPIHandler = async (req, res) => {
 };
 
 exports.driveAPIHandler = driveAPIHandler;
+exports.driveCleanup = driveCleanup;
 exports.driveConfiguration = driveConfiguration;
 exports.driveDelete = driveDelete;
 exports.driveFilePath = driveFilePath;
@@ -2425,9 +2546,11 @@ exports.driveFileSchemaZod = driveFileSchemaZod;
 exports.driveGetUrl = driveGetUrl;
 exports.driveInfo = driveInfo;
 exports.driveList = driveList;
+exports.driveListFiles = driveListFiles;
 exports.driveReadFile = driveReadFile;
 exports.driveUpload = driveUpload;
+exports.drive_default = drive_default;
 exports.getDriveConfig = getDriveConfig;
 exports.getDriveInformation = getDriveInformation;
-//# sourceMappingURL=chunk-EUAHSB2J.cjs.map
-//# sourceMappingURL=chunk-EUAHSB2J.cjs.map
+//# sourceMappingURL=chunk-RBLDH7CP.cjs.map
+//# sourceMappingURL=chunk-RBLDH7CP.cjs.map