From e365537b7031c71fef821236b395fc7bb776f9f3 Mon Sep 17 00:00:00 2001
From: johnniang <johnniang@foxmail.com>
Date: Wed, 5 Feb 2020 18:08:29 +0800
Subject: [PATCH] Create temporary file after halo is started

---
 .../app/config/properties/HaloProperties.java |  6 ------
 .../halo/app/listener/StartedListener.java    | 21 +++++++++++++++++++
 2 files changed, 21 insertions(+), 6 deletions(-)

diff --git a/src/main/java/run/halo/app/config/properties/HaloProperties.java b/src/main/java/run/halo/app/config/properties/HaloProperties.java
index f683761a4..59f68d322 100644
--- a/src/main/java/run/halo/app/config/properties/HaloProperties.java
+++ b/src/main/java/run/halo/app/config/properties/HaloProperties.java
@@ -68,10 +68,4 @@ public class HaloProperties {
      */
     private String cache = "memory";
 
-
-    public HaloProperties() throws IOException {
-        // Create work directory if not exist
-        Files.createDirectories(Paths.get(workDir));
-        Files.createDirectories(Paths.get(backupDir));
-    }
 }
diff --git a/src/main/java/run/halo/app/listener/StartedListener.java b/src/main/java/run/halo/app/listener/StartedListener.java
index 882463485..0ba175e7f 100644
--- a/src/main/java/run/halo/app/listener/StartedListener.java
+++ b/src/main/java/run/halo/app/listener/StartedListener.java
@@ -57,6 +57,7 @@ public class StartedListener implements ApplicationListener<ApplicationStartedEv
     public void onApplicationEvent(ApplicationStartedEvent event) {
         this.migrate();
         this.initThemes();
+        this.initDirectory();
         this.printStartInfo();
     }
 
@@ -140,4 +141,24 @@ public class StartedListener implements ApplicationListener<ApplicationStartedEv
 
         return fileSystem;
     }
+
+    private void initDirectory() {
+        Path workPath = Paths.get(haloProperties.getWorkDir());
+        Path backupPath = Paths.get(haloProperties.getBackupDir());
+
+        try {
+            if (Files.notExists(workPath)) {
+                Files.createDirectories(workPath);
+                log.info("Created work directory: [{}]", workPath);
+            }
+
+            if (Files.notExists(backupPath)) {
+                Files.createDirectories(backupPath);
+                log.info("Created backup directory: [{}]", backupPath);
+            }
+
+        } catch (IOException ie) {
+            throw new RuntimeException("Failed to initialize directories", ie);
+        }
+    }
 }