feat: created FileStorageService to store files in localhost

joaov1ct0r · joaov1ct0r · commit 49bcb8f2e55c · 2025-04-08T22:54:39.000-03:00
diff --git a/src/main/java/com/joaov1ct0r/restful_api_users_java/modules/domain/services/FileStorageService.java b/src/main/java/com/joaov1ct0r/restful_api_users_java/modules/domain/services/FileStorageService.java
@@ -0,0 +1,35 @@
+package com.joaov1ct0r.restful_api_users_java.modules.domain.services;
+
+import org.springframework.stereotype.Service;
+import org.springframework.util.StringUtils;
+import org.springframework.web.multipart.MultipartFile;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.nio.file.StandardCopyOption;
+import java.util.UUID;
+
+@Service
+public class FileStorageService {
+    private final Path fileStorageLocation = Paths.get("uploads").toAbsolutePath().normalize();
+
+    public FileStorageService() throws IOException {
+        Files.createDirectories(this.fileStorageLocation);
+    }
+
+    public String storeFile(MultipartFile file) throws IOException {
+        String fileName = StringUtils.cleanPath(file.getOriginalFilename());
+
+        if (fileName.contains("..")) {
+            throw new IOException("Sorry! Filename contains invalid path sequence " + fileName);
+        }
+
+        String newFileName = UUID.randomUUID().toString() + "_" + fileName;
+
+        Path targetLocation = this.fileStorageLocation.resolve(newFileName);
+        Files.copy(file.getInputStream(), targetLocation, StandardCopyOption.REPLACE_EXISTING);
+
+        return "http://localhost:8080/uploads/" + newFileName;
+    }
+}
diff --git a/src/main/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/CreateUserService.java b/src/main/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/CreateUserService.java
@@ -1,6 +1,7 @@
 package com.joaov1ct0r.restful_api_users_java.modules.users.services;
 
 import com.joaov1ct0r.restful_api_users_java.modules.domain.services.BaseService;
+import com.joaov1ct0r.restful_api_users_java.modules.domain.services.FileStorageService;
 import com.joaov1ct0r.restful_api_users_java.modules.domain.services.S3Service;
 import com.joaov1ct0r.restful_api_users_java.modules.users.dtos.CreateUserDTO;
 import com.joaov1ct0r.restful_api_users_java.modules.users.dtos.UserDTO;
@@ -9,12 +10,16 @@
 import com.joaov1ct0r.restful_api_users_java.modules.users.repositories.UserRepository;
 import jakarta.annotation.Nullable;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.web.multipart.MultipartFile;
 
 @Service
 public class CreateUserService extends BaseService {
+    @Value("${SPRING_PROFILES_ACTIVE:prod}")
+    private String env;
+
     @Autowired
     private UserRepository userRepository;
 
@@ -24,6 +29,9 @@ public class CreateUserService extends BaseService {
     @Autowired
     private S3Service s3Service;
 
+    @Autowired
+    private FileStorageService fileStorageService;
+
     public UserDTO execute(CreateUserDTO user, @Nullable MultipartFile file) throws Exception {
         boolean nameIsInUseByOtherUser = this.userRepository.findByUsername(user.getUsername()).isPresent();
 
@@ -54,11 +62,16 @@ public UserDTO execute(CreateUserDTO user, @Nullable MultipartFile file) throws
 
         boolean fileIsPresent = file != null;
 
-        if (fileIsPresent) {
+        if (fileIsPresent && this.env.equals("prod")) {
             String fileName = this.s3Service.uploadFile(file);
             user.setPhotoUrl(fileName);
         }
 
+        if (fileIsPresent && this.env.equals("dev")) {
+            String fileName = this.fileStorageService.storeFile(file);
+            user.setPhotoUrl(fileName);
+        }
+
         UserEntity userDTO = UserMapper.toPersistence(user);
 
         UserEntity createdUser = this.userRepository.save(userDTO);
diff --git a/src/main/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/UpdateUserService.java b/src/main/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/UpdateUserService.java
@@ -1,6 +1,7 @@
 package com.joaov1ct0r.restful_api_users_java.modules.users.services;
 
 import com.joaov1ct0r.restful_api_users_java.modules.domain.services.BaseService;
+import com.joaov1ct0r.restful_api_users_java.modules.domain.services.FileStorageService;
 import com.joaov1ct0r.restful_api_users_java.modules.domain.services.S3Service;
 import com.joaov1ct0r.restful_api_users_java.modules.users.dtos.UpdateUserDTO;
 import com.joaov1ct0r.restful_api_users_java.modules.users.dtos.UserDTO;
@@ -9,6 +10,7 @@
 import com.joaov1ct0r.restful_api_users_java.modules.users.repositories.UserRepository;
 import jakarta.annotation.Nullable;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
@@ -18,6 +20,9 @@
 
 @Service
 public class UpdateUserService extends BaseService {
+    @Value("${SPRING_PROFILES_ACTIVE:prod}")
+    private String env;
+
     @Autowired
     private UserRepository userRepository;
 
@@ -27,7 +32,10 @@ public class UpdateUserService extends BaseService {
     @Autowired
     private S3Service s3Service;
 
-    public UserDTO execute(UpdateUserDTO userDTO, String tokenUserId, @Nullable MultipartFile file) {
+    @Autowired
+    private FileStorageService fileStorageService;
+
+    public UserDTO execute(UpdateUserDTO userDTO, String tokenUserId, @Nullable MultipartFile file) throws Exception {
         var isUserRegistered = this.userRepository.findById(UUID.fromString(tokenUserId));
 
         var userWasntFound = isUserRegistered.isEmpty();
@@ -82,11 +90,16 @@ public UserDTO execute(UpdateUserDTO userDTO, String tokenUserId, @Nullable Mult
 
         boolean fileIsPresent = file != null;
 
-        if (fileIsPresent) {
+        if (fileIsPresent && this.env.equals("prod")) {
             String fileName = this.s3Service.uploadFile(file);
             userToUpdate.setPhotoUrl(fileName);
         }
 
+        if (fileIsPresent && this.env.equals("dev")) {
+            String fileName = this.fileStorageService.storeFile(file);
+            userToUpdate.setPhotoUrl(fileName);
+        }
+
         userToUpdate.setName(userDTO.getName());
         userToUpdate.setEmail(userDTO.getEmail());
         userToUpdate.setUsername(userDTO.getUsername());
diff --git a/src/test/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/CreateUserServiceTest.java b/src/test/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/CreateUserServiceTest.java
@@ -2,6 +2,7 @@
 
 import com.joaov1ct0r.restful_api_users_java.modules.domain.exceptions.BadRequestException;
 import com.joaov1ct0r.restful_api_users_java.modules.domain.entities.ErrorLogEntity;
+import com.joaov1ct0r.restful_api_users_java.modules.domain.services.FileStorageService;
 import com.joaov1ct0r.restful_api_users_java.modules.users.dtos.CreateUserDTO;
 import com.joaov1ct0r.restful_api_users_java.modules.users.entities.UserEntity;
 import com.joaov1ct0r.restful_api_users_java.modules.domain.repositories.ErrorLogsRepository;
@@ -42,11 +43,15 @@ public class CreateUserServiceTest {
     @Mock
     private PasswordEncoder passwordEncoder;
 
+    @Mock
+    private FileStorageService fileStorageService;
+
    @BeforeEach
     public void beforeEachSetUp() {
         Mockito.reset(this.userRepository);
         Mockito.reset(this.passwordEncoder);
         Mockito.reset(this.errorLogsRepository);
+        Mockito.reset(this.fileStorageService);
 
         when(this.errorLogsRepository.save(any())).thenReturn(
                 new ErrorLogEntity(
diff --git a/src/test/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/UpdateUserServiceTest.java b/src/test/java/com/joaov1ct0r/restful_api_users_java/modules/users/services/UpdateUserServiceTest.java
@@ -4,6 +4,7 @@
 import com.joaov1ct0r.restful_api_users_java.modules.domain.exceptions.BadRequestException;
 import com.joaov1ct0r.restful_api_users_java.modules.domain.exceptions.ForbiddenException;
 import com.joaov1ct0r.restful_api_users_java.modules.domain.repositories.ErrorLogsRepository;
+import com.joaov1ct0r.restful_api_users_java.modules.domain.services.FileStorageService;
 import com.joaov1ct0r.restful_api_users_java.modules.users.dtos.UpdateUserDTO;
 import com.joaov1ct0r.restful_api_users_java.modules.users.entities.UserEntity;
 import com.joaov1ct0r.restful_api_users_java.modules.users.repositories.UserRepository;
@@ -42,11 +43,15 @@ public class UpdateUserServiceTest {
     @Mock
     private PasswordEncoder passwordEncoder;
 
+    @Mock
+    private FileStorageService fileStorageService;
+
     @BeforeEach
     public void beforeEachSetUp() {
         Mockito.reset(this.userRepository);
         Mockito.reset(this.errorLogsRepository);
         Mockito.reset(this.passwordEncoder);
+        Mockito.reset(this.fileStorageService);
 
         when(
                 this.errorLogsRepository.save(
@@ -172,7 +177,7 @@ public void shouldNotBeAbleToUpdateUserWithUnavailableEmail() {
 
     @Test
     @DisplayName("Should be able to update user")
-    public void shouldBeAbleToUpdateUser() {
+    public void shouldBeAbleToUpdateUser() throws Exception {
         var userDTO = new UpdateUserDTO(
                 UUID.randomUUID().toString(),
                 "any_username",
