Merge pull request #91 from zecrypt-io/akhil/drive/apis

folders list

Author: itsmeakhil

packages/backend-server/app/api/v1/web/drive/api.py

@@ -1,9 +1,11 @@
 from fastapi import APIRouter
 from app.api.v1.web.drive.folders import api as folders_router
+from app.api.v1.web.drive.files import api as files_router
 
 
 router = APIRouter()
 
 router.prefix = "/drive"
 
-router.include_router(folders_router.router, tags=["Drive: Folders"])
+router.include_router(folders_router.router, tags=["Drive: Folders"])
+router.include_router(files_router.router, tags=["Drive: Files"])

packages/backend-server/app/api/v1/web/drive/folders/api.py

@@ -1,8 +1,8 @@
-from fastapi import APIRouter,Depends
+from fastapi import APIRouter,Depends,Query
 from app.api.v1.web.drive.folders.schema import CreateFolder, RenameFolder, MoveFolder, DeleteFolders
 from app.api.v1.web.auth.schema import UserDetails
 from app.framework.permission_services.service import get_current_user
-from app.api.v1.web.drive.folders.services import create_folder,delete_folders,rename_folder, move_folders
+from app.api.v1.web.drive.folders.services import create_folder,delete_folders,rename_folder, move_folders, get_folders_list
 
 FOLDER_URL ="/folder"
 router = APIRouter()
@@ -26,3 +26,7 @@ async def rename_folder_api(payload: RenameFolder, user: UserDetails = Depends(g
 async def move_folder_api(payload: MoveFolder, user: UserDetails = Depends(get_current_user)):
     return await move_folders(user, payload.model_dump())
 
+
+@router.get(FOLDER_URL+"/list")
+async def get_all_folders_api(parent_id: str = Query(None, description="Parent folder ID", ge=1), user: UserDetails = Depends(get_current_user)):
+    return await get_folders_list(user, parent_id)

packages/backend-server/app/api/v1/web/drive/folders/services.py

@@ -73,3 +73,22 @@ async def move_folders(user, payload):
     db_manager.update_many(db, FOLDERS, {"doc_id": {"$in":folder_ids}, "created_by": user.get("user_id")}, {"$set": {"parent_id": parent_id}})
     return response_helper(200, translate("drive.folders.moved"), folders_moved=folders_moved)
 
+async def get_folders_list(user, parent_id):
+    db = user.get("db")
+    query = {"created_by": user.get("user_id")}
+    if parent_id:
+        query["parent_id"] = parent_id
+    
+    folders = db_manager.find(db, FOLDERS, query)
+    
+    for folder in folders:
+        parent_query={"parent_id": folder.get("doc_id"),"created_by": user.get("user_id")}
+        files = db_manager.find(db, FILES,parent_query )
+        folder["files"] = files
+        folder["sub_folders"] = db_manager.find(db, FOLDERS, parent_query)
+    
+    data= {
+        "folders": folders,
+        "files": db_manager.find(db, FILES, query)
+    }
+    return response_helper(200, translate("drive.folders.list"), data=data)