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

Update api.py

Author: itsmeakhil

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

@@ -6,7 +6,7 @@
 from app.api.v1.web.user import api as user_router
 from app.api.v1.web.secrets import api as secrets_router
 from app.api.v1.web.dashboard import api as dashboard_router
-
+from app.api.v1.web.drive import api as drive_router
 api_router = APIRouter()
 
 api_router.prefix = "/web"
@@ -18,3 +18,5 @@
 api_router.include_router(user_router.router, tags=["User"])
 api_router.include_router(workspace_router.router, tags=["Workspace"])
 api_router.include_router(dashboard_router.router, tags=["Dashboard"])
+api_router.include_router(drive_router.router)
+

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

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

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

@@ -0,0 +1,28 @@
+from fastapi import APIRouter,Depends
+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
+
+FOLDER_URL ="/folder"
+router = APIRouter()
+
+@router.post(FOLDER_URL+"/create")
+async def create_folder_api(payload: CreateFolder, user: UserDetails = Depends(get_current_user)):
+    return  await create_folder(user, payload.model_dump())
+
+
+@router.delete(FOLDER_URL+"/delete")
+async def delete_folder_api(payload: DeleteFolders, user: UserDetails = Depends(get_current_user)):
+    return await delete_folders(user, payload.model_dump())
+
+
+@router.post(FOLDER_URL+"/rename")
+async def rename_folder_api(payload: RenameFolder, user: UserDetails = Depends(get_current_user)):
+    return await rename_folder(user, payload.model_dump())
+
+
+@router.post(FOLDER_URL+"/move")
+async def move_folder_api(payload: MoveFolder, user: UserDetails = Depends(get_current_user)):
+    return await move_folders(user, payload.model_dump())
+

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

@@ -0,0 +1,19 @@
+from pydantic import BaseModel, Field
+from typing import Optional, List
+from datetime import datetime
+
+class CreateFolder(BaseModel):
+    name: str
+    parent_id: str
+    
+class RenameFolder(BaseModel):
+    name: str
+    folder_id: str
+    parent_id: Optional[str] = None
+
+class MoveFolder(BaseModel):
+    folder_ids: List[str]
+    parent_id: str
+
+class DeleteFolders(BaseModel):
+    folder_ids: List[str]

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

@@ -0,0 +1,75 @@
+
+from app.framework.mongo_db import base_manager as db_manager
+from app.managers.collection_names import FILES, FOLDERS
+from app.utils.utils import response_helper, create_uuid, create_timestamp
+from app.utils.i8ns import translate
+
+async def create_folder(user, payload):
+    db = user.get("db")
+    folder_id = create_uuid()
+    query={
+        "created_by": user.get("user_id"),
+        "lower_name": payload.get("name").strip().lower(),
+    }
+    if payload.get("parent_id"):
+        query["parent_id"] = payload.get("parent_id")
+        folder = db_manager.find_one(db, FOLDERS, query)
+        if  folder:
+            return response_helper(400, translate("drive.folders.already_exists"))
+
+    folder = {
+        "doc_id": folder_id,
+        "name": payload.get("name"),
+        "parent_id": payload.get("parent_id"),
+        "created_by": user.get("user_id"),
+        "lower_name": payload.get("name").strip().lower(),
+        "created_at": create_timestamp(),
+
+    }
+    db_manager.insert_one(db, FOLDERS, folder)
+    return response_helper(200, translate("drive.folders.created"))
+
+
+async def delete_folders(user, payload):
+    db = user.get("db")
+    folder_ids = payload.get("folder_ids")
+    user_id = user.get("user_id")
+    db_manager.update_many(db, FOLDERS, {"doc_id": {"$in":folder_ids}, "created_by": user_id}, {"$set": {"access": False,"delete_by":user_id, "deleted_at": create_timestamp()}})
+    return response_helper(200, translate("drive.folders.deleted"))
+
+
+async def rename_folder(user, payload):
+    db = user.get("db")
+    name = payload.get("name").strip()
+    folder_id = payload.get("folder_id")
+    lower_name= name.lower()
+    query={"created_by": user.get("user_id"), "lower_name": lower_name}
+
+    if payload.get("parent_id"):
+        query["parent_id"] = payload.get("parent_id")
+    folder = db_manager.find_one(db, FOLDERS, query)
+    
+    if folder:
+        return response_helper(400, translate("drive.folders.already_exists"))
+    
+    db_manager.update_one(db, FOLDERS, {"doc_id": folder_id, "created_by": user.get("user_id")}, {"$set": {"name": name, "lower_name": lower_name}})
+    return response_helper(200, translate("drive.folders.renamed"))
+
+
+async def move_folders(user, payload):
+    db = user.get("db")
+    folder_ids = payload.get("folder_ids")
+    parent_id = payload.get("parent_id")
+    query={"created_by": user.get("user_id"), "doc_id": {"$in":folder_ids}}
+    folders = db_manager.find(db, FOLDERS, query)
+    folders_moved=[]
+    for item in folders:
+        individual_query={"created_by": user.get("user_id"), "parent_id": parent_id, "lower_name": item.get("lower_name")}
+        if not db_manager.find_one(db, FOLDERS, individual_query):
+            db_manager.update_one(db, FOLDERS, {"doc_id": item.get("doc_id")}, {"$set": {"parent_id": parent_id}})
+            folders_moved.append(item.get("name"))
+
+
+    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)
+