From 5a3574734c2c28e3422e1c160293d46776df0f2e Mon Sep 17 00:00:00 2001
From: shatteredbeam <shatteredbeam@gmail.com>
Date: Mon, 30 Apr 2012 15:48:59 -0700
Subject: [PATCH] Added automatic backup of datafiles.

---
 src/main/java/com/gmail/nossr50/mcMMO.java    |  12 ++
 .../shatt/backup/ZipLibrary.java              | 122 ++++++++++++++++++
 2 files changed, 134 insertions(+)
 create mode 100644 src/main/java/net/shatteredlands/shatt/backup/ZipLibrary.java

diff --git a/src/main/java/com/gmail/nossr50/mcMMO.java b/src/main/java/com/gmail/nossr50/mcMMO.java
index 46b4a95e4..a8ee833c3 100644
--- a/src/main/java/com/gmail/nossr50/mcMMO.java
+++ b/src/main/java/com/gmail/nossr50/mcMMO.java
@@ -18,6 +18,8 @@ import com.gmail.nossr50.listeners.HardcoreListener;
 import com.gmail.nossr50.listeners.PlayerListener;
 import com.gmail.nossr50.locale.LocaleLoader;
 
+import net.shatteredlands.shatt.backup.ZipLibrary;
+
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
@@ -167,8 +169,18 @@ public class mcMMO extends JavaPlugin {
         for (PlayerProfile x : Users.getProfiles().values()) {
             x.save();
         }
+        
 
         this.getServer().getScheduler().cancelTasks(this); //This removes our tasks
+        
+        //Remove other tasks BEFORE starting the Backup, or we just cancel it straight away.
+        
+        try {  //Try the backup
+			ZipLibrary.mcMMObackup();
+		} catch (IOException e)  {
+			p.getLogger().severe(e.toString());
+		}
+        
         System.out.println("mcMMO was disabled."); //How informative!
     }
 
diff --git a/src/main/java/net/shatteredlands/shatt/backup/ZipLibrary.java b/src/main/java/net/shatteredlands/shatt/backup/ZipLibrary.java
new file mode 100644
index 000000000..7d5b368ac
--- /dev/null
+++ b/src/main/java/net/shatteredlands/shatt/backup/ZipLibrary.java
@@ -0,0 +1,122 @@
+package net.shatteredlands.shatt.backup;
+
+import com.gmail.nossr50.*;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.text.SimpleDateFormat;
+import java.util.ArrayList;
+import java.util.Date;
+import java.util.List;
+import java.util.zip.Deflater;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
+
+public class ZipLibrary {
+	
+	private static String BackupDirectory = mcMMO.mainDirectory + "backup";
+	private static File BackupDir = new File(BackupDirectory);
+	private static File FlatFileDirectory = new File(mcMMO.flatFileDirectory);
+	private static File UsersFile = new File(mcMMO.usersFile);
+	private static File ConfigFile = new File(mcMMO.mainDirectory + "config.yml");
+	private static File Leaderboards = new File(mcMMO.leaderboardDirectory);
+	
+	
+	public static void mcMMObackup() throws IOException {
+		try {
+			if (BackupDir.mkdir()) { mcMMO.p.getLogger().info("Created Backup Directory."); }
+		} catch (Exception h) {
+			mcMMO.p.getLogger().severe(h.toString());
+		}
+		
+		//Generate the proper date for the backup filename
+		Date date = new Date();
+		SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH-mm-ss");
+		File fileZip = new File(BackupDirectory + File.separator + dateFormat.format(date) + ".zip");
+		
+		
+		//Create the Source List, and add directories/etc to the file.
+		List<File> sources = new ArrayList<File>();
+		sources.add(FlatFileDirectory);
+		sources.add(UsersFile);
+		sources.add(ConfigFile);
+		sources.add(Leaderboards);
+		
+		
+		//Actually do something
+		System.out.println("Backing up your mcMMO Configuration... ");
+		
+		packZip(fileZip, sources);
+		
+	}
+	
+	private static void packZip(File output, List<File> sources) throws IOException
+    {
+        ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(output));
+        zipOut.setLevel(Deflater.DEFAULT_COMPRESSION);
+
+        for (File source : sources) {
+        	
+            if (source.isDirectory()) {
+                zipDir(zipOut, "", source);
+            } else {
+                zipFile(zipOut, "", source);
+            }
+        }
+        
+        zipOut.flush();
+        zipOut.close();
+        System.out.println("Backup Completed.");
+    }
+
+    private static String buildPath(String path, String file) {
+        if (path == null || path.isEmpty()) {
+            return file;
+        } else {
+            return path + "/" + file;
+        }
+    }
+
+    private static void zipDir(ZipOutputStream zos, String path, File dir) throws IOException {
+        if (!dir.canRead()) {
+            System.out.println("Cannot read " + dir.getCanonicalPath() + " (maybe because of permissions)");
+            return;
+        }
+
+        File[] files = dir.listFiles();
+        path = buildPath(path, dir.getName());
+
+        for (File source : files) {
+            if (source.isDirectory()) {
+                zipDir(zos, path, source);
+            } else {
+                zipFile(zos, path, source);
+            }
+        }
+
+    }
+
+    private static void zipFile(ZipOutputStream zos, String path, File file) throws IOException {
+        if (!file.canRead()) {
+            System.out.println("Cannot read " + file.getCanonicalPath() + "(File Permissions?)");
+            return;
+        }
+        
+        zos.putNextEntry(new ZipEntry(buildPath(path, file.getName())));
+
+        FileInputStream fis = new FileInputStream(file);
+
+        byte[] buffer = new byte[4092];
+        int byteCount = 0;
+        while ((byteCount = fis.read(buffer)) != -1) {
+            zos.write(buffer, 0, byteCount);
+        }
+
+        fis.close();
+        zos.closeEntry();
+    }
+    
+
+}