Implemented Friends layer

1 files changed, 38 insertions, 0 deletions

diff --git a/src/DevHive.Data/Repositories/FriendsRepository.cs b/src/DevHive.Data/Repositories/FriendsRepository.cs
new file mode 100644
index 0000000..c65ba5e
--- /dev/null
+++ b/src/DevHive.Data/Repositories/FriendsRepository.cs
@@ -0,0 +1,38 @@
+using System;
+using System.Linq;
+using System.Threading.Tasks;
+using DevHive.Common.Models.Data;
+using DevHive.Data.Models;
+using Microsoft.EntityFrameworkCore;
+
+namespace DevHive.Data.Repositories
+{
+	public class FriendsRepository : UserRepository
+	{
+		private readonly DbContext _context;
+
+		public FriendsRepository(DbContext context) 
+			: base(context)
+		{
+			this._context = context;
+		}
+	
+		//Create
+		public async Task<bool> AddFriendAsync(User user, User friend)
+		{
+			this._context.Update(user);
+			user.Friends.Add(friend);
+
+			return await RepositoryMethods.SaveChangesAsync(this._context);
+		}
+
+		//Delete
+		public async Task<bool> RemoveFriendAsync(User user, User friend)
+		{
+			this._context.Update(user);
+			user.Friends.Remove(friend);
+
+			return await RepositoryMethods.SaveChangesAsync(this._context);
+		}
+	}
+}
\ No newline at end of file