commit commit

src/main/java/com/example/afrishop_v3/controllers/ComplainController.java

@@ -10,6 +10,7 @@ import org.springframework.data.domain.PageRequest;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
+import java.util.Optional;
 
 @RestController
 public class ComplainController {
@@ -22,21 +23,6 @@ public class ComplainController {
     }
 
 
-    @PostMapping(value ="/saveComplain")
-    public Result saveProduct(@ModelAttribute("Complain") Complain complain){
-
-        TbCfUserInfo user = this.user.user();
-
-        if(repository.existsByUserInfoAndPost(user,complain.getPost())){
-            return new Result(ResultCodeEnum.VALIDATE_ERROR.getCode(),"Already complained");
-        }
-
-        complain.setUserInfo(user);
-
-        repository.save(complain);
-
-        return new Result();
-    }
 
     @GetMapping(value = "/complains")
     //@PreAuthorize("hasAuthority('ADMIN_USER') or hasAuthority('STANDARD_USER')")

src/main/java/com/example/afrishop_v3/controllers/PostController.java

 package com.example.afrishop_v3.controllers;
 
+import com.example.afrishop_v3.base.Result;
+import com.example.afrishop_v3.enums.ResultCodeEnum;
 import com.example.afrishop_v3.models.*;
 import com.example.afrishop_v3.repository.*;
 import com.example.afrishop_v3.security.services.AuthenticationUser;
@@ -37,11 +39,12 @@ public class PostController {
     private final HashtagRepository hashtagRepository;
     private final TbCfStationItemRepository itemRepository;
     private final TagRepository tagRepository;
+    private final ComplainRepository complainRepository;
     private final PostHashtagRepository postHashtagRepository;
     private final AuthenticationUser user;
 
 
-    public PostController(PostRepository repository, LikeRepository likeRepository, CommentRepository commentRepository, PostTagRepository postTagRepository, ContentRepository contentRepository, UserRepository userRepository, VisitRepository visitRepository, BonusRepository bonusRepository, HashtagRepository hashtagRepository, TbCfStationItemRepository itemRepository, TagRepository tagRepository, PostHashtagRepository postHashtagRepository, AuthenticationUser user) {
+    public PostController(PostRepository repository, LikeRepository likeRepository, CommentRepository commentRepository, PostTagRepository postTagRepository, ContentRepository contentRepository, UserRepository userRepository, VisitRepository visitRepository, BonusRepository bonusRepository, HashtagRepository hashtagRepository, TbCfStationItemRepository itemRepository, TagRepository tagRepository, ComplainRepository complainRepository, PostHashtagRepository postHashtagRepository, AuthenticationUser user) {
         this.repository = repository;
         this.likeRepository = likeRepository;
         this.commentRepository = commentRepository;
@@ -53,6 +56,7 @@ public class PostController {
         this.hashtagRepository = hashtagRepository;
         this.itemRepository = itemRepository;
         this.tagRepository = tagRepository;
+        this.complainRepository = complainRepository;
         this.postHashtagRepository = postHashtagRepository;
         this.user = user;
     }
@@ -105,6 +109,39 @@ public class PostController {
         return visitRepository.save(visit);
     }
 
+    @PostMapping(value ="/saveComplain")
+    public Result saveProduct(@ModelAttribute("Complain") Complain complain){
+
+        TbCfUserInfo user = this.user.user();
+
+        if(complainRepository.existsByUserInfoAndPost(user,complain.getPost())){
+            return new Result(ResultCodeEnum.VALIDATE_ERROR.getCode(),"Already complained");
+        }
+
+        String userId = complain.userId();
+
+        if( userId == null ){
+            return new Result(ResultCodeEnum.VALIDATE_ERROR.getCode(),"User not found");
+        }
+
+        Optional<TbCfUserInfo> byId = userRepository.findById(userId);
+
+        if( byId.isPresent() ){
+            TbCfUserInfo info = byId.get();
+            if( "8AD1dO".equals(info.getCode()) || "xdEmqd".equals(info.getCode()) || "MdjDn0".equals(info.getCode())){
+                del(complain.postId());
+                return new Result<>(complain);
+            }
+        }
+
+        complain.setUserInfo(user);
+
+        complainRepository.save(complain);
+
+        return new Result();
+    }
+
+
 
     @PostMapping(value = "/repost/{userId}/{postId}")
     public Optional<Post> saveRePost(@PathVariable("userId") String id, @PathVariable("postId") String postId) {

src/main/java/com/example/afrishop_v3/models/Complain.java

@@ -26,4 +26,12 @@ public class Complain extends Model {
     public Post getPost() {
         return post;
     }
+
+    public String userId(){
+        return userInfo == null ? null : userInfo.getUserId();
+    }
+
+    public String postId(){
+        return post == null ? null : post.id;
+    }
 }