feat(team): support team

app/router/private/avatar.py

@@ -1,18 +1,17 @@
 from __future__ import annotations
 
 import hashlib
-from io import BytesIO
 
 from app.database.lazer_user import User
 from app.dependencies.database import Database
 from app.dependencies.storage import get_storage_service
 from app.dependencies.user import get_client_user
 from app.storage.base import StorageService
+from app.utils import check_image
 
 from .router import router
 
-from fastapi import Depends, File, HTTPException, Security
-from PIL import Image
+from fastapi import Depends, File, Security
 
 
 @router.post(
@@ -39,20 +38,7 @@ async def upload_avatar(
     """
 
     # check file
-    if len(content) > 5 * 1024 * 1024:  # 5MB limit
-        raise HTTPException(status_code=400, detail="File size exceeds 5MB limit")
-    elif len(content) == 0:
-        raise HTTPException(status_code=400, detail="File cannot be empty")
-    try:
-        with Image.open(BytesIO(content)) as img:
-            if img.format not in ["PNG", "JPEG", "GIF"]:
-                raise HTTPException(status_code=400, detail="Invalid image format")
-            if img.size[0] > 256 or img.size[1] > 256:
-                raise HTTPException(
-                    status_code=400, detail="Image size exceeds 256x256 pixels"
-                )
-    except Exception as e:
-        raise HTTPException(status_code=400, detail=f"Error processing image: {e}")
+    check_image(content, 5 * 1024 * 1024, 256, 256)
 
     filehash = hashlib.sha256(content).hexdigest()
     storage_path = f"avatars/{current_user.id}_{filehash}.png"