Merge pull request #59 merge from branch 26 - 4

Author: noobdevsam

compose.yml

@@ -36,6 +36,8 @@ services:
             # Mount the local './storage' directory into the container.
             # This makes it easy to see generated files on your host machine.
             - ./storage:/app/storage
+        command: >
+            sh -c "mkdir -p /app/storage && chown -R cnb:cnb /app/storage && java -jar /workspace/BOOT-INF/lib/spring-project-steganography-tool-v1.jar"
         networks:
             - stego-network
         depends_on:

src/main/java/com/example/springprojectsteganographytool/services/StorageService.java

@@ -4,12 +4,11 @@
 import com.example.springprojectsteganographytool.exceptions.data.StorageSecurityException;
 import org.springframework.core.io.Resource;
 
-import java.io.IOException;
 import java.nio.file.Path;
 
 public interface StorageService {
     Path save(String relativeFileName, byte[] content) throws
-            StorageSecurityException, StorageException, IOException;
+            StorageSecurityException, StorageException;
 
     // Returns true if the file was deleted, false if the file did not exist
     boolean delete(String relativeFileName) throws StorageException;

src/main/java/com/example/springprojectsteganographytool/services/impl/StorageServiceImpl.java

@@ -47,19 +47,22 @@ public StorageServiceImpl(@Value("${app.storage.base-path}") Path basePath) thro
      * @param relativeFileName the relative name of the file to save.
      * @param content          the content of the file as a byte array.
      * @return the path to the saved file.
-     * @throws StorageSecurityException if the file name is invalid or attempts to escape the storage directory.
+     * @throws StorageSecurityException if an attempt is made to escape the storage directory.
      * @throws StorageException         if an error occurs during file operations.
-     * @throws IOException              if an I/O error occurs.
      */
     @Override
-    public Path save(String relativeFileName, byte[] content) throws StorageSecurityException, StorageException, IOException {
+    public Path save(String relativeFileName, byte[] content) throws StorageSecurityException, StorageException {
         log.debug("Saving file: {} ({} bytes)", relativeFileName, content.length);
 
-        var targetPath = safeResolve(relativeFileName);
-        Files.createDirectories(targetPath.getParent());
-        Files.write(targetPath, content, StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
-        log.debug("Saved file: {}  ({} bytes)", targetPath, content.length);
-        return targetPath;
+        try {
+            var targetPath = safeResolve(relativeFileName);
+            Files.createDirectories(targetPath.getParent());
+            Files.write(targetPath, content, StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING);
+            log.debug("Saved file: {}  ({} bytes)", targetPath, content.length);
+            return targetPath;
+        } catch (IOException e) {
+            throw new StorageException("Failed to store file " + relativeFileName, e);
+        }
     }
 
     /**