aboutsummaryrefslogtreecommitdiff
path: root/src/DevHive.Data
diff options
context:
space:
mode:
Diffstat (limited to 'src/DevHive.Data')
-rw-r--r--src/DevHive.Data/Repositories/FriendsRepository.cs38
-rw-r--r--src/DevHive.Data/Repositories/UserRepository.cs8
2 files changed, 38 insertions, 8 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
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);
- }
}
}