Fixed comment and post getting and updating

1 files changed, 23 insertions, 0 deletions

diff --git a/src/DevHive.Data/Repositories/CommentRepository.cs b/src/DevHive.Data/Repositories/CommentRepository.cs
index d33b7bf..8ddc628 100644
--- a/src/DevHive.Data/Repositories/CommentRepository.cs
+++ b/src/DevHive.Data/Repositories/CommentRepository.cs
@@ -17,6 +17,14 @@ namespace DevHive.Data.Repositories
 		}
 
 		#region Read
+		public override async Task<Comment> GetByIdAsync(Guid id)
+		{
+			return await this._context.Comments
+				.Include(x => x.Creator)
+				.Include(x => x.Post)
+				.FirstOrDefaultAsync(x => x.Id == id);
+		}
+
 		public async Task<Comment> GetCommentByIssuerAndTimeCreatedAsync(Guid issuerId, DateTime timeCreated)
 		{
 			return await this._context.Comments
@@ -25,6 +33,21 @@ namespace DevHive.Data.Repositories
 		}
 		#endregion
 
+		#region Update
+		public override async Task<bool> EditAsync(Guid id, Comment newEntity)
+		{
+			Comment comment = await this.GetByIdAsync(id);
+
+			this._context
+				.Entry(comment)
+				.CurrentValues
+				.SetValues(newEntity);
+
+			return await this.SaveChangesAsync(this._context);
+		}
+		#endregion
+
+
 		#region Validations
 		public async Task<bool> DoesCommentExist(Guid id)
 		{