Merge pull request #18 from Team-Kaleidoscope/devv0.1

First stage: Complete. Awaiting further progress...

1 files changed, 105 insertions, 0 deletions

diff --git a/src/DevHive.Data/Repositories/PostRepository.cs b/src/DevHive.Data/Repositories/PostRepository.cs
new file mode 100644
index 0000000..52c5b4e
--- /dev/null
+++ b/src/DevHive.Data/Repositories/PostRepository.cs
@@ -0,0 +1,105 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading.Tasks;
+using DevHive.Data.Interfaces.Repositories;
+using DevHive.Data.Models;
+using DevHive.Data.RelationModels;
+using Microsoft.EntityFrameworkCore;
+
+namespace DevHive.Data.Repositories
+{
+    public class PostRepository : BaseRepository<Post>, IPostRepository
+	{
+		private readonly DevHiveContext _context;
+		private readonly IUserRepository _userRepository;
+
+		public PostRepository(DevHiveContext context, IUserRepository userRepository)
+			: base(context)
+		{
+			this._context = context;
+			this._userRepository = userRepository;
+		}
+
+		public async Task<bool> AddNewPostToCreator(Guid userId, Post post)
+		{
+			User user = await this._userRepository.GetByIdAsync(userId);
+			user.Posts.Add(post);
+
+			return await this.SaveChangesAsync();
+		}
+
+		#region Read
+		public override async Task<Post> GetByIdAsync(Guid id)
+		{
+			return await this._context.Posts
+					.Include(x => x.Comments)
+					.Include(x => x.Creator)
+					.Include(x => x.Attachments)
+					// .Include(x => x.Rating)
+					.FirstOrDefaultAsync(x => x.Id == id);
+		}
+
+		/// <summary>
+        /// This method returns the post that is made at exactly the given time and by the given creator
+        /// </summary>
+		public async Task<Post> GetPostByCreatorAndTimeCreatedAsync(Guid creatorId, DateTime timeCreated)
+		{
+			return await this._context.Posts
+				.FirstOrDefaultAsync(p => p.Creator.Id == creatorId &&
+					p.TimeCreated == timeCreated);
+		}
+
+		public async Task<List<string>> GetFileUrls(Guid postId)
+		{
+			return (await this.GetByIdAsync(postId)).Attachments.Select(x => x.FileUrl).ToList();
+		}
+		#endregion
+
+		#region Update
+		public override async Task<bool> EditAsync(Guid id, Post newEntity)
+		{
+			Post post = await this.GetByIdAsync(id);
+			// var ratingId = post.Rating.Id;
+
+			this._context
+				.Entry(post)
+				.CurrentValues
+				.SetValues(newEntity);
+
+			List<PostAttachments> postAttachments = new();
+			foreach(var attachment in newEntity.Attachments)
+				postAttachments.Add(attachment);
+			post.Attachments = postAttachments;
+
+			post.Comments.Clear();
+			foreach(var comment in newEntity.Comments)
+				post.Comments.Add(comment);
+
+			// post.Rating.Id = ratingId;
+
+			this._context.Entry(post).State = EntityState.Modified;
+
+			return await this.SaveChangesAsync();
+		}
+		#endregion
+
+		#region Validations
+		public async Task<bool> DoesPostExist(Guid postId)
+		{
+			return await this._context.Posts
+				.AsNoTracking()
+				.AnyAsync(r => r.Id == postId);
+		}
+
+		public async Task<bool> DoesPostHaveFiles(Guid postId)
+		{
+			return await this._context.Posts
+				.AsNoTracking()
+				.Where(x => x.Id == postId)
+				.Select(x => x.Attachments)
+				.AnyAsync();
+		}
+		#endregion
+	}
+}