Update file UserController.java

demo/src/main/java/com/example/demo/controller/UserController.java

@@ -1,6 +1,8 @@
 package com.example.demo.controller;
 
+import java.security.Principal;
 import java.util.List;
+import java.util.Optional;
 
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
@@ -16,6 +18,7 @@ import org.springframework.web.bind.annotation.RestController;
 
 import com.example.demo.dto.UserDto;
 import com.example.demo.entity.User;
+import com.example.demo.repository.UserRepo;
 import com.example.demo.service.UserService;
 
 @RestController
@@ -23,10 +26,12 @@ import com.example.demo.service.UserService;
 public class UserController {
 
 	private UserService userService;
+	private UserRepo userRepo;
 
-	public UserController(UserService userService) {
+	public UserController(UserService userService, UserRepo userRepo) {
 		super();
 		this.userService = userService;
+		this.userRepo = userRepo;
 	}
 
 	// build create user REST API
@@ -49,6 +54,15 @@ public class UserController {
 	    return new ResponseEntity<>(users, HttpStatus.OK);
 	}
 
+	// build get current user REST API
+	@GetMapping("/me")
+	public UserDto getLoggedInUser(Principal principal) {
+	    if (principal == null) return null;
+	    String username = principal.getName();
+	    User user = (userRepo.findByUsername(username))
+	    		.orElse(null);
+	    return userService.convertToDto(user);
+	}
 
 	// build get user by id REST API
 	// http://localhost:8080/api/users/(id number goes here)