Fix: NoSuchFileException if configs\db\backup is not present on f…

…irst start
Stirling-Tools · Jan 11, 2025 · fbcdcb3 · fbcdcb3
1 parent 2241e2c
commit fbcdcb3
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/main/java/stirling/software/SPDF/config/security/database/DatabaseService.java b/src/main/java/stirling/software/SPDF/config/security/database/DatabaseService.java
@@ -55,6 +55,7 @@ public DatabaseService(ApplicationProperties applicationProperties, DataSource d
      */
     @Override
     public boolean hasBackup() {
+        createBackupDirectory();
         Path filePath = Paths.get(BACKUP_DIR);
 
         if (Files.exists(filePath)) {
@@ -74,6 +75,8 @@ public List<FileInfo> getBackupList() {
         List<FileInfo> backupFiles = new ArrayList<>();
 
         if (isH2Database()) {
+            createBackupDirectory();
+
             Path backupPath = Paths.get(BACKUP_DIR);
 
             try (DirectoryStream<Path> stream =
@@ -110,6 +113,18 @@ public List<FileInfo> getBackupList() {
         return backupFiles;
     }
 
+    private void createBackupDirectory() {
+        Path backupPath = Paths.get(BACKUP_DIR);
+        if (!Files.exists(backupPath)) {
+            try {
+                Files.createDirectories(backupPath);
+                log.debug("create backup directory: {}", BACKUP_DIR);
+            } catch (IOException e) {
+                log.error("Error create backup directory: {}", e.getMessage(), e);
+            }
+        }
+    }
+
     @Override
     public void importDatabase() {
         if (!hasBackup()) throw new BackupNotFoundException("No backup scripts were found.");
@@ -255,6 +270,7 @@ public boolean deleteBackupFile(String fileName) throws IOException {
      * @return the <code>Path</code> object for the given file name
      */
     public Path getBackupFilePath(String fileName) {
+        createBackupDirectory();
         Path filePath = Paths.get(BACKUP_DIR, fileName).normalize();
         if (!filePath.startsWith(BACKUP_DIR)) {
             throw new SecurityException("Path traversal detected");