From 5b5aef5c210b5458767400ea7998f998f0f97757 Mon Sep 17 00:00:00 2001
From: Daniel Gyulai <gydani4@hotmail.hu>
Date: Thu, 19 Nov 2020 15:57:03 +0100
Subject: [PATCH] Updated "Update" class, added method to query the updates of
 a case

---
 CoviDok/BLL/User/Handlers/ICaseHandler.cs |  2 ++
 CoviDok/BLL/User/Managers/CaseManager.cs  | 16 +++++++++++++++-
 CoviDok/Controllers/CaseController.cs     | 18 ++++++++++++++++++
 CoviDok/Data/Model/Update.cs              |  7 +++++--
 CoviDok/Data/MySQL/MySqlCaseHandler.cs    |  5 +++++
 5 files changed, 45 insertions(+), 3 deletions(-)

diff --git a/CoviDok/BLL/User/Handlers/ICaseHandler.cs b/CoviDok/BLL/User/Handlers/ICaseHandler.cs
index 88a407a..aa4e6b3 100644
--- a/CoviDok/BLL/User/Handlers/ICaseHandler.cs
+++ b/CoviDok/BLL/User/Handlers/ICaseHandler.cs
@@ -19,5 +19,7 @@ namespace CoviDok.BLL.User.Handlers
 
         public Task SetCase(int id, CaseStatus status);
         public bool IsAuthorized(int ID, Case c);
+
+        public List<Update> GetUpdatesForCase(int id);
     }
 }
diff --git a/CoviDok/BLL/User/Managers/CaseManager.cs b/CoviDok/BLL/User/Managers/CaseManager.cs
index 0d6dd4c..754bcf2 100644
--- a/CoviDok/BLL/User/Managers/CaseManager.cs
+++ b/CoviDok/BLL/User/Managers/CaseManager.cs
@@ -21,6 +21,18 @@ namespace CoviDok.BLL.User.Managers
             return await handler.Filter(filter);
         }
 
+        public async Task<List<Update>> GetUpdatesForCase(Session s, int id)
+        {
+            Case c = await handler.GetCase(id);
+            if (c == null) throw new KeyNotFoundException();
+            if (handler.IsAuthorized(s.ID, c))
+            {
+                return handler.GetUpdatesForCase(id);
+            }
+            else throw new UnauthorizedAccessException();
+            
+        }
+
         public async Task SetCertified(Session s, int id)
         {
             Case c = await handler.GetCase(id);
@@ -96,7 +108,9 @@ namespace CoviDok.BLL.User.Managers
 
                 c.LastModificationDate = DateTime.Now;
                 Update update = new Update {
-                    Sender = s.ID,
+                    CaseID = c.Id,
+                    SenderID = s.ID,
+                    SenderRole = s.Type,
                     Content = updateMsg,
                     CreatedDate = DateTime.Now,
                     Images = Images
diff --git a/CoviDok/Controllers/CaseController.cs b/CoviDok/Controllers/CaseController.cs
index c54d71f..8a5f0fd 100644
--- a/CoviDok/Controllers/CaseController.cs
+++ b/CoviDok/Controllers/CaseController.cs
@@ -82,6 +82,24 @@ namespace CoviDok.Controllers
             }
         }
 
+        [HttpPost("{id}/updates")]
+        public async Task<ActionResult<List<Update>>> GetUpdatesForCase(int id, AuthGet get)
+        {
+            Session s = await Handler.GetSession(get.SessionID);
+            if (s == null) return Unauthorized();
+            try {
+                return await mgr.GetUpdatesForCase(s, id);
+            }
+            catch (UnauthorizedAccessException)
+            {
+                return Unauthorized();
+            }
+            catch (KeyNotFoundException)
+            {
+                return NotFound();
+            }
+        }
+
         // POST /api/Case/{id}/close
         [HttpPost("{id}/close")]
         public async Task<IActionResult> PostClose(int id, CaseUpdate data)
diff --git a/CoviDok/Data/Model/Update.cs b/CoviDok/Data/Model/Update.cs
index 93f744a..6213085 100644
--- a/CoviDok/Data/Model/Update.cs
+++ b/CoviDok/Data/Model/Update.cs
@@ -1,4 +1,5 @@
-using System;
+using CoviDok.Api;
+using System;
 using System.Collections.Generic;
 using System.Linq;
 using System.Threading.Tasks;
@@ -8,7 +9,9 @@ namespace CoviDok.Data.Model
     public class Update
     {
         public int Id { get; set; }
-        public int Sender { get; set; }
+        public int CaseID { get; set; }
+        public int SenderID { get; set; }
+        public Role SenderRole { get; set; }
         public string Content { get; set; }
         public DateTime CreatedDate { get; set; }
 
diff --git a/CoviDok/Data/MySQL/MySqlCaseHandler.cs b/CoviDok/Data/MySQL/MySqlCaseHandler.cs
index 02f37b1..e48b8a4 100644
--- a/CoviDok/Data/MySQL/MySqlCaseHandler.cs
+++ b/CoviDok/Data/MySQL/MySqlCaseHandler.cs
@@ -28,6 +28,11 @@ namespace CoviDok.Data.MySQL
             await context.SaveChangesAsync();
         }
 
+        public List<Update> GetUpdatesForCase(int id)
+        {
+            return (from u in context.Updates where u.CaseID == id select u).ToList();
+        }
+
         public async Task<List<Case>> Filter(CaseFilter filter)
         {
             List<Case> ret = new List<Case>();