From 83c0689b825dd4c0f7b560255e843129fcbb9d40 Mon Sep 17 00:00:00 2001
From: shri <shrirang@bhate.de>
Date: Wed, 4 Sep 2024 17:00:07 +0200
Subject: [PATCH] Add endpoint to serve user profile pics

---
 src/app/domain/accounts/controllers/users.py | 54 +++++++++++++++++++-
 src/app/domain/accounts/schemas.py           |  5 ++
 src/app/domain/accounts/urls.py              |  1 +
 3 files changed, 58 insertions(+), 2 deletions(-)

diff --git a/src/app/domain/accounts/controllers/users.py b/src/app/domain/accounts/controllers/users.py
index 55442657..6edb8359 100644
--- a/src/app/domain/accounts/controllers/users.py
+++ b/src/app/domain/accounts/controllers/users.py
@@ -2,15 +2,20 @@
 
 from __future__ import annotations
 
+import os
 from typing import TYPE_CHECKING, Annotated
 
+import boto3
 from litestar import Controller, delete, get, patch, post
 from litestar.di import Provide
 from litestar.params import Dependency, Parameter
+from litestar.response import Response
+from litestar.exceptions import NotFoundException
 
+from app.db.models import User as UserModel
 from app.domain.accounts import urls
 from app.domain.accounts.dependencies import provide_users_service
-from app.domain.accounts.guards import requires_superuser
+from app.domain.accounts.guards import requires_superuser, requires_active_user
 from app.domain.accounts.schemas import User, UserCreate, UserUpdate
 from app.domain.accounts.services import UserService
 
@@ -20,6 +25,8 @@
     from advanced_alchemy.filters import FilterTypes
     from advanced_alchemy.service import OffsetPagination
 
+app_s3_bucket_name = os.environ["APP_S3_BUCKET_NAME"]
+
 
 class UserController(Controller):
     """User Account Controller."""
@@ -27,7 +34,10 @@ class UserController(Controller):
     tags = ["User Accounts"]
     guards = [requires_superuser]
     dependencies = {"users_service": Provide(provide_users_service)}
-    signature_namespace = {"UserService": UserService}
+    signature_namespace = {
+        "UserService": UserService,
+        "UserModel": UserModel,
+    }
     dto = None
     return_dto = None
 
@@ -69,6 +79,46 @@ async def get_user(
         db_obj = await users_service.get(user_id)
         return users_service.to_schema(db_obj, schema_type=User)
 
+    @get(
+        operation_id="GetUserProfilePic",
+        guards=[requires_active_user],
+        name="users:get-profile-pic",
+        path=urls.ACCOUNT_PROFILE_PIC,
+        summary="Retrieve the profile pic of a user.",
+    )
+    async def get_user_profile_pic(
+        self,
+        users_service: UserService,
+        current_user: UserModel,
+        user_id: Annotated[
+            UUID,
+            Parameter(
+                title="User ID",
+                description="The user to retrieve.",
+            ),
+        ],
+    ) -> User:
+        """Get a user."""
+        db_obj = await users_service.get(user_id)
+        if current_user.tenant_id != db_obj.tenant_id:
+            raise Exception("User not found")
+
+        try:
+            # Retrieve the file from S3
+            s3_client = boto3.client("s3")
+            file_object = s3_client.get_object(Bucket=app_s3_bucket_name, Key="users/avatars/{user_id}.webp")
+
+            # Extract the file content
+            file_content = file_object["Body"].read()
+
+            return Response(
+                content=file_content,
+                media_type="image/webp",  # Ensure correct media type
+                headers={"Content-Disposition": "inline"},  # Optionally, force the browser to display the image inline
+            )
+        except s3_client.exceptions.NoSuchKey:
+            raise NotFoundException(detail=f"Profile pic for user {user_id} not found.")
+
     @post(
         operation_id="CreateUser",
         name="users:create",
diff --git a/src/app/domain/accounts/schemas.py b/src/app/domain/accounts/schemas.py
index 07769ef7..61d3b7b2 100644
--- a/src/app/domain/accounts/schemas.py
+++ b/src/app/domain/accounts/schemas.py
@@ -70,6 +70,11 @@ class User(CamelizedBaseStruct):
     teams: list[UserTeam] = []
     roles: list[UserRole] = []
     oauth_accounts: list[OauthAccount] = []
+    profile_pic_url: str | None = None
+
+    def __post_init__(self):
+        """Build a profile pic url from company url."""
+        self.profile_pic_url = f"https://api.chapter.show/users/{id}/profile_pic"
 
 
 class UserCreate(CamelizedBaseStruct):
diff --git a/src/app/domain/accounts/urls.py b/src/app/domain/accounts/urls.py
index 8fd881b1..bf558a57 100644
--- a/src/app/domain/accounts/urls.py
+++ b/src/app/domain/accounts/urls.py
@@ -5,6 +5,7 @@
 ACCOUNT_LIST = "/api/users"
 ACCOUNT_DELETE = "/api/users/{user_id:uuid}"
 ACCOUNT_DETAIL = "/api/users/{user_id:uuid}"
+ACCOUNT_PROFILE_PIC = "/api/users/{user_id:uuid}/profile-pic"
 ACCOUNT_UPDATE = "/api/users/{user_id:uuid}"
 ACCOUNT_CREATE = "/api/users"
 ACCOUNT_ASSIGN_ROLE = "/api/roles/{role_slug:str}/assign"