From b90b45903b466e05a2326c8b9f5d80d48104328e Mon Sep 17 00:00:00 2001
From: Osama Ragab <osamaragab520@gmail.com>
Date: Tue, 7 Jun 2022 03:21:19 +0200
Subject: [PATCH] feat: add case state update endpoints

---
 api/cases/apis.py     | 16 +++++++++++++++
 api/cases/models.py   |  2 ++
 api/cases/services.py | 46 ++++++++++++++++++++++++++++++++++++++++++-
 api/cases/urls.py     |  4 ++++
 4 files changed, 67 insertions(+), 1 deletion(-)

diff --git a/api/cases/apis.py b/api/cases/apis.py
index 42003fd..7c4493b 100644
--- a/api/cases/apis.py
+++ b/api/cases/apis.py
@@ -12,8 +12,10 @@
     list_user_cases,
 )
 from api.cases.services import (
+    archive_case,
     create_case,
     create_case_contact,
+    finish_case,
     publish_case,
     update_case_contact,
 )
@@ -197,6 +199,20 @@ def get(self, request, case_id):
         return Response(status=status.HTTP_200_OK)
 
 
+class CaseFinishApi(APIView):
+    def get(self, request, case_id):
+        case = get_case(pk=case_id, fetched_by=request.user)
+        finish_case(case=case, performed_by=request.user)
+        return Response(status=status.HTTP_200_OK)
+
+
+class CaseArchiveApi(APIView):
+    def get(self, request, case_id):
+        case = get_case(pk=case_id, fetched_by=request.user)
+        archive_case(case=case, performed_by=request.user)
+        return Response(status=status.HTTP_200_OK)
+
+
 class UserCasesListApi(APIView):
     class Pagination(LimitOffsetPagination):
         default_limit = 10
diff --git a/api/cases/models.py b/api/cases/models.py
index dda4e8c..cd1a5ca 100644
--- a/api/cases/models.py
+++ b/api/cases/models.py
@@ -58,11 +58,13 @@ def activate(self):
     @transition(field=state, source=States.ACTIVE, target=States.FINISHED)
     def finish(self):
         self.is_active = False
+        self.posted_at = None
 
     # Switch to ARCHIVED state from any state except ARCHIVED
     @transition(field=state, source="+", target=States.ARCHIVED)
     def archive(self):
         self.is_active = False
+        self.posted_at = None
 
     # If user selected incorrect match or lost again
     @transition(
diff --git a/api/cases/services.py b/api/cases/services.py
index fda31bd..4241cc1 100644
--- a/api/cases/services.py
+++ b/api/cases/services.py
@@ -76,6 +76,7 @@ def update_case():
     ...
 
 
+@transaction.atomic
 def create_case_details(
     *,
     case: Case,
@@ -166,6 +167,7 @@ def case_matching_binding(*, case: Case, matches_list: List[Dict[int, int]]) ->
     )
 
 
+@transaction.atomic
 def activate_case(case: Case):
     matches = process_case(case)
     case_matching_binding(case=case, matches_list=matches)
@@ -190,6 +192,7 @@ def activate_case(case: Case):
     # case.user.fcmdevice.send_message(msg)
 
 
+# TODO refactor object permission on view level to some mixin or permission class
 def publish_case(*, case: Case, performed_by: User):
     if case.user != performed_by:
         raise PermissionDenied()
@@ -211,7 +214,6 @@ def publish_case(*, case: Case, performed_by: User):
         level=Notification.Level.SUCCESS,
         sent_to=case.user,
     )
-
     # msg = Message(
     #     notification=FirebaseNotification(
     #         title="تم نشر الحاله بنجاح",
@@ -222,6 +224,48 @@ def publish_case(*, case: Case, performed_by: User):
     # device.send_message(msg)
 
 
+def archive_case(*, case: Case, performed_by: User):
+    if case.user != performed_by:
+        raise PermissionDenied()
+
+    if case.state == Case.States.ARCHIVED:
+        raise ValidationError("Case already archived")
+
+    case.archive()
+    case.save()
+    create_notification(
+        case=case,
+        action=Notification.Action.PUBLISH,
+        title="تم ارشفه الحاله بنجاح",
+        body="تم ارشفه الحاله لن يتمكن اى احد للوصول لها غيرك",
+        level=Notification.Level.WARNING,
+        sent_to=case.user,
+    )
+
+
+def finish_case(*, case: Case, performed_by: User):
+    if case.user != performed_by:
+        raise PermissionDenied()
+
+    if not case.is_active:
+        raise ValidationError("Cannot finish inactive case")
+
+    if case.state == Case.States.FINISHED:
+        raise ValidationError("Case already closed")
+
+    case.finish()
+    case.save()
+
+    create_notification(
+        case=case,
+        action=Notification.Action.NONE,
+        title="تم ارشفه الحاله بنجاح",
+        body="تم اغلاق الحاله بنجاح نرجو ان يكون ذويك على ما يرام",
+        level=Notification.Level.SUCCESS,
+        sent_to=case.user,
+    )
+
+
 def create_case_contact(*, user: User, case: Case) -> CaseContact:
     contact = CaseContact(case=case, user=user)
     contact.full_clean()
diff --git a/api/cases/urls.py b/api/cases/urls.py
index 3756b74..1dd08d3 100644
--- a/api/cases/urls.py
+++ b/api/cases/urls.py
@@ -1,8 +1,10 @@
 from django.urls import path
 
 from .apis import (
+    CaseArchiveApi,
     CaseContactCreateApi,
     CaseContactUpdateApi,
+    CaseFinishApi,
     CaseListApi,
     CaseMatchListApi,
     CasePublishApi,
@@ -25,4 +27,6 @@
     # path("<int:case_id>/update/", UpdateCaseApi.as_view(), name="update"),
     path("<int:case_id>/matches/", CaseMatchListApi.as_view(), name="matches"),
     path("<int:case_id>/publish/", CasePublishApi.as_view(), name="publish"),
+    path("<int:case_id>/finish/", CaseFinishApi.as_view(), name="finish"),
+    path("<int:case_id>/archive/", CaseArchiveApi.as_view(), name="archive"),
 ]