feat: create_notification

ara/domain/notification/notification_domain.py

@@ -1,4 +1,4 @@
-from apps.core.models import Notification, NotificationReadLog
+from apps.core.models import Comment
 from ara.domain.notification.type import NotificationInfo
 from ara.infra.notification.notification_infra import NotificationInfra
 
@@ -11,20 +11,13 @@ 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]:
-        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()
-        ]
+        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, notification_info: NotificationInfo):
-        return self.notification_infra.create_notification(notification_info)
+    def create_notification(self, comment: Comment):
+        return self.notification_infra.create_notification(comment)

ara/domain/notification/type.py

@@ -1,16 +1,13 @@
 from typing import Optional
 
-from django.contrib.auth import get_user_model
 from pydantic import BaseModel
 
 from apps.core.models import Article, Comment, Notification
 
-User = get_user_model()
 
-
-class NotificationReadLogInfo(BaseModel):  # 사용 안하는데 ?? 그래도 적어두는게 낫겠죠 ??
+class NotificationReadLogInfo(BaseModel):
     is_read: bool
-    read_by: int  # ??? 모르겠다 int ?? user ?? Foreign Key 인데 ??
+    read_by: int
     notification: Notification
 
     class Config:

ara/infra/notification/notification_infra.py

@@ -1,12 +1,13 @@
 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, user_id: int) -> None:
+    def __init__(self) -> None:
         super().__init__(Notification)
-        self.user_id = user_id
 
     def get_all_notifications(self, user_id: int) -> list[NotificationInfo]:
         queryset = Notification.objects.select_related(
@@ -18,9 +19,19 @@ def get_all_notifications(self, user_id: int) -> list[NotificationInfo]:
         )
         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,  # 이렇게 써도 되나요?
+            id=notification.id,
             type=notification.type,
             title=notification.title,
             content=notification.content,
@@ -43,26 +54,76 @@ def read_notification(self, user_id: int, notification_id: int) -> None:
         notification_read_log.is_read = True
         notification_read_log.save()
 
-    """
-    ##수정해야함##
-
-    def create_notification(self, article: Article, comment: Comment) -> None:
-        if comment.parent_comment:
-            parent_comment = comment.parent_comment
-            related_comment = parent_comment
+    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:
-            related_comment = None
+            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,
+            )
 
-        related_article = comment.parent_article if comment.parent_article else parent_comment.parent_article
+            fcm_notify_comment(
+                _parent_article.created_by,
+                title,
+                _comment.content[:32],
+                f"post/{_parent_article.id}",
+            )
 
-        title = f"{article.title}에 새로운 {'대댓글' if parent_comment else '댓글'}이 작성되었습니다."
-        content = comment.content[:32]
+        def notify_comment_commented(_parent_article: Article, _comment: Comment):
+            name = self.get_display_name(_parent_article, _comment.created_by_id)
+            title = f"{name} 님이 새로운 대댓글을 작성했습니다."
 
-        Notification.objects.create(
-            type="comment_commented" if parent_comment else "article_commented",
-            title=title,
-            content=content,
-            related_article=related_article,
-            related_comment=related_comment,
+            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)