feat: Create notification FastAPI

Makefile

@@ -14,7 +14,7 @@ flush_test:
 reset: flush init
 
 run:
-	python manage.py runserver 0.0.0.0:9000
+	python3 manage.py runserver 0.0.0.0:9000
 
 shell:
 	python manage.py shell -i ipython

ara/controller/api.py

@@ -0,0 +1,42 @@
+from fastapi import APIRouter, Depends, HTTPException
+from ara.controller.authentication import get_current_user
+from ara.service.notification_service import NotificationService
+from ara.domain.user import User
+from ara.domain.exceptions import EntityDoesNotExist
+from pydantic import BaseModel
+
+router = APIRouter()
+notification_service = NotificationService()
+
+class NotificationRead(BaseModel):
+    notification_id: int
+
+@router.get("/notifications")
+async def list_notifications(current_user: User = Depends(get_current_user)):
+    notifications = notification_service.get_notifications_for_user(current_user)
+    return notifications
+
+@router.post("/notifications/{notification_id}/read")
+async def mark_notification_as_read(notification_id: int, current_user: User = Depends(get_current_user)):
+    try:
+        notification_service.mark_notification_as_read(notification_id, current_user)
+    except EntityDoesNotExist:
+        raise HTTPException(status_code=404, detail="Notification not found")
+    except PermissionError:
+        raise HTTPException(status_code=403, detail="You are not allowed to mark this notification as read")
+    return {"message": "Notification marked as read successfully"}
+
+@router.post("/notifications/read-all")
+async def mark_all_notifications_as_read(current_user: User = Depends(get_current_user)):
+    notification_service.mark_all_notifications_as_read(current_user)
+    return {"message": "All notifications marked as read successfully"}
+
+@router.post("/notifications/send-push-notification")
+async def send_push_notification(notification: NotificationRead, current_user: User = Depends(get_current_user)):
+    try:
+        notification_service.send_push_notification(notification.notification_id, current_user)
+    except EntityDoesNotExist:
+        raise HTTPException(status_code=404, detail="Notification not found")
+    except PermissionError:
+        raise HTTPException(status_code=403, detail="You are not allowed to send push notification for this notification")
+    return {"message": "Push notification sent successfully"}

ara/domain/exceptions.py

@@ -0,0 +1,2 @@
+class EntityDoesNotExist(Exception):
+    pass

ara/domain/notification.py

@@ -0,0 +1,14 @@
+from pydantic import BaseModel
+from typing import Optional
+
+class Notification(BaseModel):
+    id: int
+    type: str
+    title: str
+    content: str
+    related_article_id: Optional[int]
+    related_comment_id: Optional[int]
+    is_read: bool
+
+    class Config:
+        orm_mode = True

ara/domain/notification/notification_domain.py

@@ -0,0 +1,23 @@
+from apps.core.models import Comment
+from ara.domain.notification.type import NotificationInfo
+from ara.infra.notification.notification_infra import NotificationInfra
+
+
+class NotificationDomain:
+    def __init__(self) -> None:
+        self.notification_infra = NotificationInfra()
+
+    def get_all_notifications(self, user_id: int) -> list[NotificationInfo]:
+        return self.notification_infra.get_all_notifications(user_id)
+
+    def get_unread_notifications(self, user_id: int) -> list[NotificationInfo]:
+        return self.notification_infra.get_unread_notifications(user_id)
+
+    def read_all_notifications(self, user_id: int) -> None:
+        return self.notification_infra.read_all_notifications(user_id)
+
+    def read_notification(self, user_id: int, notification_id: int) -> None:
+        return self.notification_infra.read_notification(user_id, notification_id)
+
+    def create_notification(self, comment: Comment):
+        return self.notification_infra.create_notification(comment)

ara/domain/notification/type.py

@@ -0,0 +1,26 @@
+from typing import Optional
+
+from pydantic import BaseModel
+
+from apps.core.models import Article, Comment, Notification
+
+
+class NotificationReadLogInfo(BaseModel):
+    is_read: bool
+    read_by: int
+    notification: Notification
+
+    class Config:
+        arbitrary_types_allowed = True
+
+
+class NotificationInfo(BaseModel):
+    id: int
+    type: str
+    title: str
+    content: str
+    related_article: Article | None
+    related_comment: Optional[Comment]
+
+    class Config:
+        arbitrary_types_allowed = True

ara/infra/notification/notification_infra.py

@@ -0,0 +1,129 @@
+from apps.core.models import Article, Comment, Notification, NotificationReadLog
+from apps.core.models.board import NameType
+from ara.domain.notification.type import NotificationInfo
+from ara.firebase import fcm_notify_comment
+from ara.infra.django_infra import AraDjangoInfra
+
+
+class NotificationInfra(AraDjangoInfra[Notification]):
+    def __init__(self) -> None:
+        super().__init__(Notification)
+
+    def get_all_notifications(self, user_id: int) -> list[NotificationInfo]:
+        queryset = Notification.objects.select_related(
+            "related_article",
+            "related_comment",
+        ).prefetch_related(
+            "related_article__attachments",
+            NotificationReadLog.prefetch_my_notification_read_log(user_id),
+        )
+        return [self._to_notification_info(notification) for notification in queryset]
+
+    def get_unread_notifications(self, user_id: int) -> list[NotificationInfo]:
+        notifications = self.notification_infra.get_all_notifications(user_id)
+        return [
+            notification
+            for notification in notifications
+            if NotificationReadLog.objects.filter(
+                notification=notification, read_by=user_id, is_read=False
+            ).exists()
+        ]
+
+    def _to_notification_info(self, notification: Notification) -> NotificationInfo:
+        return NotificationInfo(
+            id=notification.id,
+            type=notification.type,
+            title=notification.title,
+            content=notification.content,
+            related_article=notification.related_article,
+            related_comment=notification.related_comment,
+        )
+
+    def read_all_notifications(self, user_id: int) -> None:
+        notifications = self.get_all_notifications(user_id)
+        notification_ids = [notification.id for notification in notifications]
+        NotificationReadLog.objects.filter(
+            notification__in=notification_ids, read_by=user_id
+        ).update(is_read=True)
+
+    def read_notification(self, user_id: int, notification_id: int) -> None:
+        notification = self.get_by_id(notification_id)
+        notification_read_log = NotificationReadLog.objects.get(
+            notification=notification, read_by=user_id
+        )
+        notification_read_log.is_read = True
+        notification_read_log.save()
+
+    def get_display_name(self, article: Article, profile: int):
+        if article.name_type == NameType.REALNAME:
+            return "실명"
+        elif article.name_type == NameType.REGULAR:
+            return "nickname"
+        else:
+            return "익명"
+
+    def create_notification(self, comment: Comment) -> None:
+        def notify_article_commented(_parent_article: Article, _comment: Comment):
+            name = self.get_display_name(_parent_article, _comment.created_by_id)
+            title = f"{name} 님이 새로운 댓글을 작성했습니다."
+
+            notification = Notification(
+                type="article_commented",
+                title=title,
+                content=_comment.content[:32],
+                related_article=_parent_article,
+                related_comment=None,
+            )
+            notification.save()
+
+            NotificationReadLog.objects.create(
+                read_by=_parent_article.created_by,
+                notification=notification,
+            )
+
+            fcm_notify_comment(
+                _parent_article.created_by,
+                title,
+                _comment.content[:32],
+                f"post/{_parent_article.id}",
+            )
+
+        def notify_comment_commented(_parent_article: Article, _comment: Comment):
+            name = self.get_display_name(_parent_article, _comment.created_by_id)
+            title = f"{name} 님이 새로운 대댓글을 작성했습니다."
+
+            notification = Notification(
+                type="comment_commented",
+                title=title,
+                content=_comment.content[:32],
+                related_article=_parent_article,
+                related_comment=_comment.parent_comment,
+            )
+            notification.save()
+
+            NotificationReadLog.objects.create(
+                read_by=_comment.parent_comment.created_by,
+                notification=notification,
+            )
+
+            fcm_notify_comment(
+                _comment.parent_comment.created_by,
+                title,
+                _comment.content[:32],
+                f"post/{_parent_article.id}",
+            )
+
+        article = (
+            comment.parent_article
+            if comment.parent_article
+            else comment.parent_comment.parent_article
+        )
+
+        if comment.created_by != article.created_by:
+            notify_article_commented(article, comment)
+
+        if (
+            comment.parent_comment
+            and comment.created_by != comment.parent_comment.created_by
+        ):
+            notify_comment_commented(article, comment)

ara/settings/dev/__init__.py

@@ -11,6 +11,7 @@
 CSRF_TRUSTED_ORIGINS = [
     "https://*.sparcs.org",
     "http://localhost",
+    "http://127.0.0.1:8000",
 ]
 
 CORS_ALLOW_ALL_ORIGINS = True
@@ -32,11 +33,12 @@
 
 # https://django-debug-toolbar.readthedocs.io/
 
-_, _, ips = socket.gethostbyname_ex(socket.gethostname())
-INTERNAL_IPS = [ip[: ip.rfind(".")] + ".1" for ip in ips] + [
-    "127.0.0.1",
-    "10.0.2.2",
-]
+# _, _, ips = socket.gethostbyname_ex(socket.gethostname())
+# INTERNAL_IPS = [ip[: ip.rfind(".")] + ".1" for ip in ips] + [
+#     "127.0.0.1",
+#     "10.0.2.2",
+# ]
+INTERNAL_IPS = ["127.0.0.1"]
 
 DEBUG_TOOLBAR_CONFIG = {
     "SHOW_TOOLBAR_CALLBACK": lambda _: True,

ara/settings/prod/__init__.py

@@ -14,6 +14,8 @@
     "https://newara.sparcs.org",
     "https://newara-api.sparcs.org",
     "https://ara.sparcs.org",
+    "http://localhost",
+    "http://127.0.0.1:8000",
 ]
 
 SSO_IS_BETA = False