From faa2e47b6718c59feadecea176020f3326076d5d Mon Sep 17 00:00:00 2001
From: transtrike <transtrike@gmail.com>
Date: Thu, 17 Dec 2020 23:05:20 +0200
Subject: Implemented Friends layer

---
 src/DevHive.Data/Repositories/FriendsRepository.cs | 38 ++++++++++++++++++++++
 src/DevHive.Data/Repositories/UserRepository.cs    |  8 -----
 2 files changed, 38 insertions(+), 8 deletions(-)
 create mode 100644 src/DevHive.Data/Repositories/FriendsRepository.cs

(limited to 'src/DevHive.Data/Repositories')

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
diff --git a/src/DevHive.Data/Repositories/UserRepository.cs b/src/DevHive.Data/Repositories/UserRepository.cs
index 577f02f..6d14a65 100644
--- a/src/DevHive.Data/Repositories/UserRepository.cs
+++ b/src/DevHive.Data/Repositories/UserRepository.cs
@@ -116,13 +116,5 @@ namespace DevHive.Data.Repositories
 				.AsNoTracking()
 				.AnyAsync(u => u.Email == email);
 		}
-	
-		public async Task<bool> AddFriend(User user, User friend)
-		{
-			this._context.Update(user);
-			user.Friends.Add(friend);
-
-			return await RepositoryMethods.SaveChangesAsync(this._context);
-		} 
 	}
 }
-- 
cgit v1.2.3