Merge remote-tracking branch 'origin/master'

# Conflicts:
#	src/main/java/com/keuin/kbackupfabric/operation/backup/PrimitiveBackupMethod.java

3 files changed, 217 insertions, 0 deletions

diff --git a/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredBackupMethod.java b/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredBackupMethod.java
new file mode 100644
index 0000000..bb80c80
--- /dev/null
+++ b/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredBackupMethod.java
@@ -0,0 +1,37 @@
+package com.keuin.kbackupfabric.operation.backup.method;
+
+import com.keuin.kbackupfabric.operation.backup.feedback.BackupFeedback;
+
+import java.io.IOException;
+
+/**
+ * Provide specific backup method, which has been configured with proper settings,
+ * such as saving directory and level path.
+ */
+public interface ConfiguredBackupMethod {
+
+    /**
+     * Perform a backup with given method. The backup will be saved as the given name.
+     * Note: real file name depends on the backup type.
+     *
+     * @return backup result.
+     */
+    BackupFeedback backup() throws IOException;
+
+    boolean restore() throws IOException;
+
+    /**
+     * Create backup save directory and do some essential initialization before the backup process.
+     *
+     * @return false if failed, then the backup process won't proceed.
+     */
+    boolean touch();
+
+    /**
+     * Get the used backup file name.
+     *
+     * @return the file name.
+     */
+    String getBackupFileName();
+
+}
diff --git a/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredIncrementalBackupMethod.java b/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredIncrementalBackupMethod.java
new file mode 100644
index 0000000..b5d2463
--- /dev/null
+++ b/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredIncrementalBackupMethod.java
@@ -0,0 +1,92 @@
+package com.keuin.kbackupfabric.operation.backup.method;
+
+import com.keuin.kbackupfabric.operation.backup.feedback.IncrementalBackupFeedback;
+import com.keuin.kbackupfabric.util.FilesystemUtil;
+import com.keuin.kbackupfabric.util.PrintUtil;
+import com.keuin.kbackupfabric.util.backup.incremental.ObjectCollection;
+import com.keuin.kbackupfabric.util.backup.incremental.ObjectCollectionFactory;
+import com.keuin.kbackupfabric.util.backup.incremental.ObjectCollectionSerializer;
+import com.keuin.kbackupfabric.util.backup.incremental.identifier.Sha256Identifier;
+import com.keuin.kbackupfabric.util.backup.incremental.manager.IncrementalBackupStorageManager;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Paths;
+import java.util.Arrays;
+import java.util.HashSet;
+
+public class ConfiguredIncrementalBackupMethod implements ConfiguredBackupMethod {
+
+    private final String backupIndexFileName;
+    private final String levelPath;
+    private final String backupIndexFileSaveDirectory;
+    private final String backupBaseDirectory;
+
+    public ConfiguredIncrementalBackupMethod(String backupIndexFileName, String levelPath, String backupIndexFileSaveDirectory, String backupBaseDirectory) {
+        this.backupIndexFileName = backupIndexFileName;
+        this.levelPath = levelPath;
+        this.backupIndexFileSaveDirectory = backupIndexFileSaveDirectory;
+        this.backupBaseDirectory = backupBaseDirectory;
+    }
+
+    @Override
+    public IncrementalBackupFeedback backup() throws IOException {
+        File levelPathFile = new File(levelPath);
+
+        // construct incremental backup index
+        PrintUtil.info("Hashing files...");
+        ObjectCollection collection = new ObjectCollectionFactory<>(Sha256Identifier.getFactory())
+                .fromDirectory(levelPathFile, new HashSet<>(Arrays.asList("session.lock", "kbackup_metadata")));
+
+        // update storage
+        PrintUtil.info("Copying files...");
+        IncrementalBackupStorageManager storageManager = new IncrementalBackupStorageManager(Paths.get(backupBaseDirectory));
+        int filesAdded = storageManager.addObjectCollection(collection, levelPathFile);
+
+        // save index file
+        PrintUtil.info("Saving index file...");
+        ObjectCollectionSerializer.toFile(collection, new File(backupIndexFileSaveDirectory, backupIndexFileName));
+
+        // return result
+        PrintUtil.info("Incremental backup finished.");
+        return new IncrementalBackupFeedback(filesAdded >= 0, filesAdded);
+    }
+
+    @Override
+    public boolean restore() throws IOException {
+        // load collection
+        PrintUtil.info("Loading file list...");
+        ObjectCollection collection = ObjectCollectionSerializer.fromFile(
+                new File(backupIndexFileSaveDirectory, backupIndexFileName)
+        );
+
+        // delete old level
+        File levelPathFile = new File(levelPath);
+        PrintUtil.info("Deleting old level...");
+        if (!FilesystemUtil.forceDeleteDirectory(levelPathFile)) {
+            PrintUtil.info("Failed to delete old level!");
+            return false;
+        }
+
+        // restore file
+        PrintUtil.info("Copying files...");
+        IncrementalBackupStorageManager storageManager = new IncrementalBackupStorageManager(Paths.get(backupBaseDirectory));
+        int restoreObjectCount = storageManager.restoreObjectCollection(collection, levelPathFile);
+
+        PrintUtil.info(String.format("%d file(s) restored.", restoreObjectCount));
+        return true;
+    }
+
+    @Override
+    public boolean touch() {
+        File baseDirectoryFile = new File(backupBaseDirectory);
+        return baseDirectoryFile.isDirectory() || baseDirectoryFile.mkdir();
+    }
+
+    @Override
+    public String getBackupFileName() {
+        return backupIndexFileName;
+    }
+
+
+}
diff --git a/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredPrimitiveBackupMethod.java b/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredPrimitiveBackupMethod.java
new file mode 100644
index 0000000..1c3c9f6
--- /dev/null
+++ b/src/main/java/com/keuin/kbackupfabric/operation/backup/method/ConfiguredPrimitiveBackupMethod.java
@@ -0,0 +1,88 @@
+package com.keuin.kbackupfabric.operation.backup.method;
+
+import com.keuin.kbackupfabric.exception.ZipUtilException;
+import com.keuin.kbackupfabric.metadata.BackupMetadata;
+import com.keuin.kbackupfabric.operation.backup.feedback.PrimitiveBackupFeedback;
+import com.keuin.kbackupfabric.util.FilesystemUtil;
+import com.keuin.kbackupfabric.util.PrintUtil;
+import com.keuin.kbackupfabric.util.ZipUtil;
+import com.keuin.kbackupfabric.util.backup.BackupFilesystemUtil;
+import com.keuin.kbackupfabric.util.backup.BackupNameTimeFormatter;
+import com.keuin.kbackupfabric.util.backup.name.PrimitiveBackupFileNameEncoder;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Paths;
+import java.time.LocalDateTime;
+
+public class ConfiguredPrimitiveBackupMethod implements ConfiguredBackupMethod {
+
+    private final String backupFileName;
+    private final String levelPath;
+    private final String backupSavePath;
+
+    public ConfiguredPrimitiveBackupMethod(String backupFileName, String levelPath, String backupSavePath) {
+        this.backupFileName = backupFileName;
+        this.levelPath = levelPath;
+        this.backupSavePath = backupSavePath;
+    }
+
+    @Deprecated
+    private String getBackupFileName(LocalDateTime time, String backupName) {
+        String timeString = BackupNameTimeFormatter.localDateTimeToString(time);
+        return String.format("%s%s_%s%s", BackupFilesystemUtil.getBackupFileNamePrefix(), timeString, backupName, ".zip");
+    }
+
+    @Override
+    public PrimitiveBackupFeedback backup() throws IOException {
+        try {
+            String customBackupName = new PrimitiveBackupFileNameEncoder().decode(backupFileName).customName;
+            BackupMetadata backupMetadata = new BackupMetadata(System.currentTimeMillis(), customBackupName);
+            PrintUtil.info(String.format("zip(srcPath=%s, destPath=%s)", levelPath, backupSavePath));
+            PrintUtil.info("Compressing level ...");
+            ZipUtil.makeBackupZip(levelPath, backupSavePath, backupFileName, backupMetadata);
+        } catch (ZipUtilException exception) {
+            PrintUtil.info("Infinite recursive of directory tree detected, backup was aborted.");
+            return new PrimitiveBackupFeedback(false, 0);
+        }
+
+        // Get backup file size and return
+        return new PrimitiveBackupFeedback(true, FilesystemUtil.getFileSizeBytes(backupSavePath, backupFileName));
+    }
+
+    @Override
+    public boolean restore() throws IOException {
+        // Delete old level
+        PrintUtil.info("Server stopped. Deleting old level ...");
+        if (!FilesystemUtil.forceDeleteDirectory(new File(levelPath))) {
+            PrintUtil.info("Failed to delete old level!");
+            return false;
+        }
+
+
+        // TODO: Refactor this to the concrete BackupMethod.
+        // Decompress archive
+        PrintUtil.info("Decompressing archived level ...");
+        ZipUtil.unzip(Paths.get(backupSavePath, backupFileName).toString(), levelPath, false);
+
+
+//        try {
+//            Thread.sleep(1000);
+//        } catch (InterruptedException ignored) {
+//        }
+
+        return true;
+    }
+
+    @Override
+    public boolean touch() {
+        File backupSaveDirectoryFile = new File(backupSavePath);
+        return backupSaveDirectoryFile.isDirectory() || backupSaveDirectoryFile.mkdir();
+    }
+
+    @Override
+    public String getBackupFileName() {
+        return backupFileName;
+    }
+
+}