files

Author: itsmeakhil

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

@@ -0,0 +1,27 @@
+from fastapi import APIRouter,Request,Depends,UploadFile
+from app.api.v1.web.drive.files.schema import RenameFile, MoveFile, DeleteFiles, GetPresignedUrl
+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.files.services import rename_file, get_presigned_url, delete_files, move_files
+
+router = APIRouter()
+
+FILE_URL ="/file"
+
+@router.post(FILE_URL+"/get-presigned-url")
+async def get_presigned_url_api(payload: GetPresignedUrl, user: UserDetails = Depends(get_current_user)):
+    return await get_presigned_url(user, payload.model_dump())
+
+@router.post(FILE_URL+"/rename")
+async def rename_file_api(payload: RenameFile, user: UserDetails = Depends(get_current_user)):
+    return await rename_file(user, payload.model_dump())
+
+
+@router.delete(FILE_URL+"/delete")
+async def delete_files_api(payload: DeleteFiles, user: UserDetails = Depends(get_current_user)):
+    return await delete_files(user, payload.model_dump())
+
+
+@router.post(FILE_URL+"/move")
+async def move_files_api(payload: MoveFile, user: UserDetails = Depends(get_current_user)):
+    return await move_files(user, payload.model_dump())

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

@@ -0,0 +1,23 @@
+from pydantic import BaseModel, Field
+from typing import Optional, List
+from datetime import datetime
+
+
+class RenameFile(BaseModel):
+    name: str = Field(..., description="File name")
+    file_id: str = Field(..., description="File ID")
+    parent_id: Optional[str] = Field(None, description="Parent folder ID, None for root")
+
+class MoveFile(BaseModel):
+    file_ids: List[str] = Field(..., description="File IDs")
+    parent_id: str = Field(..., description="Parent folder ID")
+
+class DeleteFiles(BaseModel):
+    file_ids: List[str] = Field(..., description="File IDs")
+
+class GetPresignedUrl(BaseModel):
+    file_type: str = Field(..., description="File type")
+    name: str = Field(..., description="File name")
+    size: int = Field(..., description="File size")
+    file_path: str = Field(..., description="File path")
+    parent_id: Optional[str] = Field(None, description="Parent folder ID")

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

@@ -0,0 +1,117 @@
+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, get_file_extension, get_folders_from_path, create_timestamp
+from app.utils.i8ns import translate
+from app.utils.s3_utils import generate_upload_url
+
+async def get_presigned_url(user, payload):
+    db = user.get("db")
+    user_id = user.get("user_id")
+    file_name = payload.get("name")
+    
+    # Check if file already exists for this user
+    query = {
+        "lower_name": file_name.strip().lower(),
+        "created_by": user_id
+    }
+    if payload.get("parent_id"):
+        query["parent_id"] = payload.get("parent_id")
+    
+    file = db_manager.find_one(db, FILES, query)
+    if file:
+        return response_helper(400, translate("drive.files.already_exists"))
+    
+    # Generate file paths
+    file_id = create_uuid()
+    file_extension = get_file_extension(file_name)
+    final_file_path = f"{user_id}/files/{file_id}.{file_extension}"
+    
+    # Get folders from path
+    folders = get_folders_from_path(payload.get("file_path"))
+    parent_id = payload.get("parent_id")
+    
+    # Create folder hierarchy if folders exist
+    for folder_name in folders:
+        query = {
+            "created_by": user_id, 
+            "lower_name": folder_name.strip().lower()
+        }
+        if parent_id:
+            query["parent_id"] = parent_id
+        
+        folder_details = db_manager.find_one(db, FOLDERS, query)
+        if not folder_details:
+            # Create new folder
+            folder_data = {
+                "doc_id": create_uuid(),
+                "name": folder_name.strip(),
+                "parent_id": parent_id,
+                "created_by": user_id,
+                "lower_name": folder_name.strip().lower(),
+            }
+            db_manager.insert_one(db, FOLDERS, folder_data)
+            parent_id = folder_data.get("doc_id")
+        else:
+            # Use existing folder
+            parent_id = folder_details.get("doc_id")
+    
+    # Insert file record (always, regardless of whether folders exist)
+    file_data = {
+        "doc_id": file_id,
+        "type": payload.get("file_type"),
+        "name": file_name,
+        "lower_name": file_name.strip().lower(),
+        "size": payload.get("size"),
+        "path": payload.get("file_path"),
+        "key": final_file_path,
+        "parent_id": parent_id,
+        "created_by": user_id,
+    }
+    db_manager.insert_one(db, FILES, file_data)
+    
+    # Generate presigned upload URL
+    upload_url = generate_upload_url(final_file_path)
+    
+    return response_helper(200, translate("file.get_presigned_url"), data=upload_url)
+
+
+async def rename_file(user, payload):
+    db = user.get("db")
+    name = payload.get("name").strip()
+    file_id = payload.get("file_id")
+    lower_name = name.lower()
+    query = {"created_by": user.get("user_id"), "lower_name": lower_name, "doc_id": {"$ne": file_id}}
+    
+    if payload.get("parent_id"):
+        query["parent_id"] = payload.get("parent_id")
+    
+    file = db_manager.find_one(db, FILES, query)
+    if file:
+        return response_helper(400, translate("drive.files.already_exists"))
+    
+    db_manager.update_one(db, FILES, {"doc_id": file_id, "created_by": user.get("user_id")}, {"$set": {"name": name, "lower_name": lower_name}})
+    return response_helper(200, translate("drive.files.renamed"))
+
+
+async def delete_files(user, payload):
+    db = user.get("db")
+    file_ids = payload.get("file_ids")
+    user_id = user.get("user_id")
+    db_manager.update_many(db, FILES, {"doc_id": {"$in": file_ids}, "created_by": user_id}, {"$set": {"access": False, "delete_by": user_id, "deleted_at": create_timestamp()}})
+    return response_helper(200, translate("drive.files.deleted"))
+
+
+async def move_files(user, payload):
+    db = user.get("db")
+    file_ids = payload.get("file_ids")
+    parent_id = payload.get("parent_id")
+    query = {"created_by": user.get("user_id"), "doc_id": {"$in": file_ids}}
+    files = db_manager.find(db, FILES, query)
+    files_moved = []
+    for item in files:
+        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, FILES, individual_query):
+            db_manager.update_one(db, FILES, {"doc_id": item.get("doc_id")}, {"$set": {"parent_id": parent_id}})
+            files_moved.append(item.get("name"))
+    
+    return response_helper(200, translate("drive.files.moved"), files_moved=files_moved)

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

@@ -15,6 +15,11 @@ class Settings(BaseSettings):
 
     VALKEY_URL: str
 
+    DO_SPACES_KEY:str
+    DO_SPACES_SECRET:str
+    DO_SPACES_REGION:str
+    DO_SPACES_BUCKET:str
+    DO_SPACES_ENDPOINT:str
     class Config:
         case_sensitive = True
         env_file = ".env"

packages/backend-server/app/core/config.py

@@ -0,0 +1,38 @@
+import boto3
+from app.core.config import settings
+
+
+# DigitalOcean Spaces config
+session = boto3.session.Session()
+client = session.client(
+    "s3",
+    region_name=settings.DO_SPACES_REGION,
+    endpoint_url=f"https://{settings.DO_SPACES_REGION}.digitaloceanspaces.com",
+    aws_access_key_id=settings.DO_SPACES_KEY,
+    aws_secret_access_key=settings.DO_SPACES_SECRET,
+)
+
+BUCKET_NAME = settings.DO_SPACES_BUCKET
+
+
+def generate_upload_url(file_name,expires_in=3600):
+    try:
+        url = client.generate_presigned_url(
+            "put_object",
+            Params={"Bucket": BUCKET_NAME, "Key": file_name},
+            ExpiresIn=expires_in,
+        )
+        return {"upload_url": url}
+    except Exception as e:
+        return {"error": str(e)}
+
+def generate_download_url(file_name,expires_in=3600):
+    try:
+        url = client.generate_presigned_url(
+            "get_object",
+            Params={"Bucket": BUCKET_NAME, "Key": file_name},
+            ExpiresIn=expires_in,
+        )
+        return {"download_url": url}
+    except Exception as e:
+        return {"error": str(e)}

packages/backend-server/app/utils/s3_utils.py

@@ -6,7 +6,9 @@
 import json
 from datetime import datetime
 import pytz
+import os
 
+from pathlib import Path
 
 
 def response_helper(
@@ -63,3 +65,12 @@ def generate_query_hash(query, projection=None):
 
 def create_timestamp():
     return datetime.now(pytz.utc)
+
+def get_file_extension(filename):
+    _, ext = os.path.splitext(filename)
+    return ext.lstrip(".")
+
+
+
+def get_folders_from_path(path: str):
+    return [part for part in Path(path).parent.parts]