Merge pull request #431 from bounswe/feature/425-selfFollowing-2

[BE] Feature/425 self following
bounswe · Oct 22, 2024 · 95937cb · 95937cb
2 parents 978adcb + fc14cae
commit 95937cb
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/backend/src/main/java/com/group1/programminglanguagesforum/Controllers/UserController.java b/backend/src/main/java/com/group1/programminglanguagesforum/Controllers/UserController.java
@@ -67,6 +67,8 @@ public ResponseEntity<GenericApiResponse<UserProfileResponseDto>> getUserById(@P
             Optional<User> user = userService.getUserById(id);
             if (user.isPresent()) {
                 UserProfileResponseDto userProfileResponseDto = modelMapper.map(user.get(), UserProfileResponseDto.class);
+                userProfileResponseDto.setSelfFollowing(userService.selfFollowing(user.get()));
+
                 GenericApiResponse<UserProfileResponseDto> response = ApiResponseBuilder.buildSuccessResponse(
                         userProfileResponseDto.getClass(),
                         "User retrieved successfully",
@@ -152,6 +154,8 @@ public ResponseEntity<GenericApiResponse<UserProfileResponseDto>> followUser(@Pa
             User user = userContextService.getCurrentUser();
             User followedUser = userService.followUser(user, id);
             UserProfileResponseDto updatedUserProfileResponseDto = modelMapper.map(followedUser, UserProfileResponseDto.class);
+            updatedUserProfileResponseDto.setSelfFollowing(userService.selfFollowing(followedUser));
+
             GenericApiResponse<UserProfileResponseDto> response = ApiResponseBuilder.buildSuccessResponse(
                     updatedUserProfileResponseDto.getClass(),
                     "User followed successfully",
@@ -195,6 +199,8 @@ public ResponseEntity<GenericApiResponse<UserProfileResponseDto>> unfollowUser(@
             User user = userContextService.getCurrentUser();
             User unfollowedUser = userService.unfollowUser(user, id);
             UserProfileResponseDto updatedUserProfileResponseDto = modelMapper.map(unfollowedUser, UserProfileResponseDto.class);
+            updatedUserProfileResponseDto.setSelfFollowing(userService.selfFollowing(unfollowedUser));
+
             GenericApiResponse<UserProfileResponseDto> response = ApiResponseBuilder.buildSuccessResponse(
                     updatedUserProfileResponseDto.getClass(),
                     "User unfollowed successfully",

diff --git a/...main/java/com/group1/programminglanguagesforum/DTOs/Responses/UserProfileResponseDto.java b/...main/java/com/group1/programminglanguagesforum/DTOs/Responses/UserProfileResponseDto.java
@@ -16,5 +16,6 @@ public class UserProfileResponseDto {
     private Long answerCount;
     private int followersCount;
     private int followingCount;
+    private boolean selfFollowing;
     private int reputationPoints;
 }
diff --git a/backend/src/main/java/com/group1/programminglanguagesforum/Services/UserService.java b/backend/src/main/java/com/group1/programminglanguagesforum/Services/UserService.java
@@ -2,6 +2,7 @@
 
 import com.group1.programminglanguagesforum.DTOs.Requests.UserProfileUpdateRequestDto;
 import com.group1.programminglanguagesforum.Entities.User;
+import com.group1.programminglanguagesforum.Exceptions.UnauthorizedAccessException;
 import com.group1.programminglanguagesforum.Exceptions.UserNotFoundException;
 import com.group1.programminglanguagesforum.Repositories.UserRepository;
 import lombok.RequiredArgsConstructor;
@@ -15,6 +16,8 @@
 @RequiredArgsConstructor
 public class UserService {
     private final UserRepository userRepository;
+    private final UserContextService userContextService;
+
     public Optional<User> getUserById(Long id) {
         return userRepository.findById(id);
     }
@@ -59,6 +62,14 @@ public User unfollowUser(User user, Long id) throws UserNotFoundException {
         return userRepository.save(userToUnfollow);
     }
 
+    public boolean selfFollowing(User userToCheck) {
+        try {
+            return userContextService.getCurrentUser().getFollowing().contains(userToCheck);
+        } catch (UnauthorizedAccessException e) {
+            return false;
+        }
+    }
+
     public List<User> getFollowers(User user) {
         return user.getFollowers().stream().toList();
     }