126 lines
4.7 KiB
Java
126 lines
4.7 KiB
Java
package io.spring.api;
|
|
|
|
import com.fasterxml.jackson.annotation.JsonRootName;
|
|
import io.spring.api.exception.InvalidRequestException;
|
|
import io.spring.application.UserQueryService;
|
|
import io.spring.application.data.UserWithToken;
|
|
import io.spring.application.data.UserData;
|
|
import io.spring.core.service.JwtService;
|
|
import io.spring.core.user.EncryptService;
|
|
import io.spring.core.user.User;
|
|
import io.spring.core.user.UserRepository;
|
|
import lombok.Getter;
|
|
import lombok.NoArgsConstructor;
|
|
import org.hibernate.validator.constraints.Email;
|
|
import org.hibernate.validator.constraints.NotBlank;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
import org.springframework.http.ResponseEntity;
|
|
import org.springframework.validation.BindingResult;
|
|
import org.springframework.web.bind.annotation.RequestBody;
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|
import org.springframework.web.bind.annotation.RestController;
|
|
|
|
import javax.validation.Valid;
|
|
import java.util.HashMap;
|
|
import java.util.Map;
|
|
import java.util.Optional;
|
|
|
|
import static org.springframework.web.bind.annotation.RequestMethod.POST;
|
|
|
|
@RestController
|
|
public class UsersApi {
|
|
private UserRepository userRepository;
|
|
private UserQueryService userQueryService;
|
|
private String defaultImage;
|
|
private EncryptService encryptService;
|
|
private JwtService jwtService;
|
|
|
|
@Autowired
|
|
public UsersApi(UserRepository userRepository,
|
|
UserQueryService userQueryService,
|
|
EncryptService encryptService,
|
|
@Value("${image.default}") String defaultImage,
|
|
JwtService jwtService) {
|
|
this.userRepository = userRepository;
|
|
this.userQueryService = userQueryService;
|
|
this.encryptService = encryptService;
|
|
this.defaultImage = defaultImage;
|
|
this.jwtService = jwtService;
|
|
}
|
|
|
|
@RequestMapping(path = "/users", method = POST)
|
|
public ResponseEntity createUser(@Valid @RequestBody RegisterParam registerParam, BindingResult bindingResult) {
|
|
checkInput(registerParam, bindingResult);
|
|
|
|
User user = new User(
|
|
registerParam.getEmail(),
|
|
registerParam.getUsername(),
|
|
encryptService.encrypt(registerParam.getPassword()),
|
|
"",
|
|
defaultImage);
|
|
userRepository.save(user);
|
|
UserData userData = userQueryService.findById(user.getId()).get();
|
|
return ResponseEntity.status(201).body(userResponse(new UserWithToken(userData, jwtService.toToken(user))));
|
|
}
|
|
|
|
private void checkInput(@Valid @RequestBody RegisterParam registerParam, BindingResult bindingResult) {
|
|
if (bindingResult.hasErrors()) {
|
|
throw new InvalidRequestException(bindingResult);
|
|
}
|
|
if (userRepository.findByUsername(registerParam.getUsername()).isPresent()) {
|
|
bindingResult.rejectValue("username", "DUPLICATED", "duplicated username");
|
|
}
|
|
|
|
if (userRepository.findByEmail(registerParam.getEmail()).isPresent()) {
|
|
bindingResult.rejectValue("email", "DUPLICATED", "duplicated email");
|
|
}
|
|
|
|
if (bindingResult.hasErrors()) {
|
|
throw new InvalidRequestException(bindingResult);
|
|
}
|
|
}
|
|
|
|
@RequestMapping(path = "/users/login", method = POST)
|
|
public ResponseEntity userLogin(@Valid @RequestBody LoginParam loginParam, BindingResult bindingResult) {
|
|
Optional<User> optional = userRepository.findByEmail(loginParam.getEmail());
|
|
if (optional.isPresent() && encryptService.check(loginParam.getPassword(), optional.get().getPassword())) {
|
|
UserData userData = userQueryService.findById(optional.get().getId()).get();
|
|
return ResponseEntity.ok(userResponse(new UserWithToken(userData, jwtService.toToken(optional.get()))));
|
|
} else {
|
|
bindingResult.rejectValue("password", "INVALID", "invalid email or password");
|
|
throw new InvalidRequestException(bindingResult);
|
|
}
|
|
}
|
|
|
|
private Map<String, Object> userResponse(UserWithToken userWithToken) {
|
|
return new HashMap<String, Object>() {{
|
|
put("user", userWithToken);
|
|
}};
|
|
}
|
|
}
|
|
|
|
@Getter
|
|
@JsonRootName("user")
|
|
@NoArgsConstructor
|
|
class LoginParam {
|
|
@NotBlank(message = "can't be empty")
|
|
@Email(message = "should be an email")
|
|
private String email;
|
|
@NotBlank(message = "can't be empty")
|
|
private String password;
|
|
}
|
|
|
|
@Getter
|
|
@JsonRootName("user")
|
|
@NoArgsConstructor
|
|
class RegisterParam {
|
|
@NotBlank(message = "can't be empty")
|
|
@Email(message = "should be an email")
|
|
private String email;
|
|
@NotBlank(message = "can't be empty")
|
|
private String username;
|
|
@NotBlank(message = "can't be empty")
|
|
private String password;
|
|
}
|