113 lines
3.7 KiB
Java
Executable File
113 lines
3.7 KiB
Java
Executable File
package org.codiki.images;
|
|
|
|
import org.codiki.core.entities.dto.ImageDTO;
|
|
import org.codiki.core.entities.persistence.Image;
|
|
import org.codiki.core.entities.persistence.User;
|
|
import org.codiki.core.repositories.ImageRepository;
|
|
import org.codiki.core.repositories.UserRepository;
|
|
import org.codiki.core.services.FileUploadService;
|
|
import org.codiki.core.services.UserService;
|
|
import org.springframework.core.io.Resource;
|
|
import org.springframework.stereotype.Service;
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
import javax.servlet.http.HttpServletResponse;
|
|
import java.io.IOException;
|
|
import java.security.Principal;
|
|
import java.util.Date;
|
|
import java.util.List;
|
|
import java.util.NoSuchElementException;
|
|
import java.util.Optional;
|
|
import java.util.stream.Collectors;
|
|
|
|
@Service
|
|
public class ImageService {
|
|
/** User service. */
|
|
private UserService userService;
|
|
/** User repository. */
|
|
private UserRepository userRepository;
|
|
/** File upload service. */
|
|
private FileUploadService fileUploadService;
|
|
/** Image repository. */
|
|
private ImageRepository imageRepository;
|
|
|
|
/**
|
|
* Constructor.
|
|
* @param userService User service.
|
|
* @param userRepository User repository.
|
|
* @param fileUploadService File upload service.
|
|
* @param imageRepository Image repository.
|
|
*/
|
|
public ImageService(UserService userService,
|
|
UserRepository userRepository,
|
|
FileUploadService fileUploadService,
|
|
ImageRepository imageRepository) {
|
|
this.userService = userService;
|
|
this.userRepository = userRepository;
|
|
this.fileUploadService = fileUploadService;
|
|
this.imageRepository = imageRepository;
|
|
}
|
|
|
|
public String uploadAvatar(final MultipartFile pFile,
|
|
final HttpServletResponse pResponse,
|
|
final Principal pPrincipal) throws IOException {
|
|
final String avatarFileName = fileUploadService.uploadProfileImage(pFile);
|
|
|
|
final Optional<User> connectedUser = userService.getUserByPrincipal(pPrincipal);
|
|
if(connectedUser.isPresent()) {
|
|
final Optional<User> userFromDb = userRepository.findById(connectedUser.get().getId());
|
|
if(userFromDb.isPresent()) {
|
|
userFromDb.get().setImage(avatarFileName);
|
|
userRepository.save(userFromDb.get());
|
|
} else {
|
|
pResponse.sendError(HttpServletResponse.SC_UNAUTHORIZED);
|
|
}
|
|
} else {
|
|
pResponse.sendError(HttpServletResponse.SC_UNAUTHORIZED);
|
|
}
|
|
|
|
return avatarFileName;
|
|
}
|
|
|
|
public String uploadImage(final MultipartFile pFile, final Principal pPrincipal) {
|
|
final String imageFileName = fileUploadService.uploadImage(pFile);
|
|
|
|
userService.getUserByPrincipal(pPrincipal)
|
|
.map(User::getId)
|
|
.map(userRepository::findById)
|
|
.orElseThrow(NoSuchElementException::new)
|
|
.ifPresent(user -> {
|
|
final Image image = new Image();
|
|
image.setLink(imageFileName);
|
|
image.setDate(new Date());
|
|
image.setUser(user);
|
|
imageRepository.save(image);
|
|
});
|
|
|
|
return imageFileName;
|
|
}
|
|
|
|
public Optional<Resource> loadAvatar(final String pAvatarFileName) {
|
|
return fileUploadService.loadAvatar(pAvatarFileName);
|
|
}
|
|
|
|
public Optional<Resource> loadImage(final String pImageLink) {
|
|
return fileUploadService.loadImage(pImageLink);
|
|
}
|
|
|
|
public List<ImageDTO> getUserImages(final Principal pPrincipal) {
|
|
return userService.getUserByPrincipal(pPrincipal)
|
|
.map(User::getId)
|
|
.map(imageRepository::getByUserId)
|
|
.orElseThrow(NoSuchElementException::new)
|
|
.stream()
|
|
.map(ImageDTO::new)
|
|
.collect(Collectors.toList());
|
|
}
|
|
|
|
public Optional<ImageDTO> getImageDetails(final String pImageLink) {
|
|
return imageRepository.findByLink(pImageLink)
|
|
.map(ImageDTO::new);
|
|
}
|
|
}
|