myDrive/backend/db/utils/folderUtils/index.ts

import Folder, { FolderInterface } from "../../../models/folder";
import { ObjectId } from "mongodb";

class DbUtil {
  constructor() {}

  getFolderSearchList = async (userID: string, searchQuery: RegExp) => {
    let query: any = { owner: userID, name: searchQuery };

    const folderList = (await Folder.find(query).limit(
      10
    )) as FolderInterface[];

    return folderList;
  };

  getFolderInfo = async (folderID: string, userID: string) => {
    const folder = (await Folder.findOne({
      owner: userID,
      _id: new ObjectId(folderID),
    })) as FolderInterface;

    return folder;
  };

  getFolderListByParent = async (
    userID: string,
    parent: string,
    sortBy: string,
    s3Enabled: boolean,
    type: string,
    storageType: string,
    itemType: string
  ) => {
    let query: any = { owner: userID, parent: parent };

    if (!s3Enabled) {
      query = { ...query, personalFolder: null };
    }

    if (type) {
      if (type === "mongo") {
        query = { ...query, personalFolder: null };
      } else if (type === "s3") {
        query = { ...query, personalFolder: true };
      }
    }

    if (itemType) {
      if (itemType === "personal") query = { ...query, personalFolder: true };
      if (itemType === "nonpersonal")
        query = { ...query, personalFolder: null };
    }

    const folderList = (await Folder.find(query).sort(
      sortBy
    )) as FolderInterface[];

    return folderList;
  };

  getFolderListBySearch = async (
    userID: string,
    searchQuery: string,
    sortBy: string,
    type: string,
    parent: string,
    storageType: string,
    folderSearch: boolean,
    itemType: string,
    s3Enabled: boolean
  ) => {
    let query: any = { name: searchQuery, owner: userID };

    if (type) {
      if (type === "mongo") {
        query = { ...query, personalFolder: null };
      } else {
        query = { ...query, personalFolder: true };
      }
    }

    if (storageType === "s3") {
      query = { ...query, personalFolder: true };
    }

    if (parent && (parent !== "/" || folderSearch)) {
      query = { ...query, parent };
    }

    if (!s3Enabled) {
      query = { ...query, personalFolder: null };
    }

    if (itemType) {
      if (itemType === "personal") query = { ...query, personalFolder: true };
      if (itemType === "nonpersonal")
        query = { ...query, personalFolder: null };
    }

    const folderList = (await Folder.find(query).sort(
      sortBy
    )) as FolderInterface[];

    return folderList;
  };

  moveFolder = async (
    folderID: string,
    userID: string,
    parent: string,
    parentList: string[]
  ) => {
    const folder = (await Folder.findOneAndUpdate(
      { _id: new ObjectId(folderID), owner: userID },
      { $set: { parent: parent, parentList: parentList } }
    )) as FolderInterface;

    return folder;
  };

  renameFolder = async (folderID: string, userID: string, title: string) => {
    const folder = (await Folder.findOneAndUpdate(
      { _id: new ObjectId(folderID), owner: userID },
      { $set: { name: title } }
    )) as FolderInterface;

    return folder;
  };

  findAllFoldersByParent = async (parentID: string, userID: string) => {
    const folderList = (await Folder.find({
      parentList: parentID,
      owner: userID,
    })) as FolderInterface[];

    return folderList;
  };
}

export default DbUtil;