完成部分帖子相关功能,同时完善了一些用户功能
Change-Id: Ie9d22bcceec197930d48f578d8b9a749ce7a3382
diff --git a/src/main/java/com/pt/controller/UserController.java b/src/main/java/com/pt/controller/UserController.java
index ea4dcab..f72e7db 100644
--- a/src/main/java/com/pt/controller/UserController.java
+++ b/src/main/java/com/pt/controller/UserController.java
@@ -112,4 +112,54 @@
return ResponseEntity.badRequest().body("User not found");
}
}
+
+ @DeleteMapping("/delete")
+ public ResponseEntity<?> deleteUser(@RequestHeader("token") String token,
+ @RequestParam("username") String username,
+ @RequestParam("targetUsername") String targetUsername
+ ) {
+ if(!JWTUtils.checkToken(token, username, Constants.UserRole.ADMIN)) {
+ return ResponseEntity.badRequest().body("Invalid token");
+ }
+
+ User user = userService.findByUsername(targetUsername);
+ if (user != null) {
+ userService.deleteById(user.getUid());
+ return ResponseEntity.ok("User deleted successfully");
+ } else {
+ return ResponseEntity.badRequest().body("User not found");
+ }
+ }
+
+ @GetMapping("/list")
+ public ResponseEntity<?> listUsers(@RequestHeader("token") String token,
+ @RequestParam("username") String username) {
+ if(!JWTUtils.checkToken(token, username, Constants.UserRole.ADMIN)) {
+ return ResponseEntity.badRequest().body("Invalid token");
+ }
+
+ Map<String, Object> ans = new HashMap<>();
+ ans.put("result", "User list retrieved successfully");
+ ans.put("amount", userService.listAll().size());
+ ans.put("users", userService.listAll());
+ return ResponseEntity.ok(ans);
+ }
+
+ @GetMapping("/get/info")
+ public ResponseEntity<?> getUserInfo(@RequestHeader("token") String token,
+ @RequestParam("username") String username) {
+ if(!JWTUtils.checkToken(token, username, Constants.UserRole.USER)) {
+ return ResponseEntity.badRequest().body("Invalid token");
+ }
+
+ User user = userService.findByUsername(username);
+ if (user != null) {
+ Map<String, Object> ans = new HashMap<>();
+ ans.put("result", "User info retrieved successfully");
+ ans.put("user", user);
+ return ResponseEntity.ok(ans);
+ } else {
+ return ResponseEntity.badRequest().body("User not found");
+ }
+ }
}