Add an UpgradeManager

src/main/java/com/gmail/nossr50/util/upgrade/UpgradeManager.java

@@ -0,0 +1,65 @@
+package com.gmail.nossr50.util.upgrade;
+
+import java.util.Arrays;
+import java.util.EnumSet;
+import java.util.Set;
+
+import com.gmail.nossr50.config.ConfigLoader;
+import com.gmail.nossr50.datatypes.database.UpgradeType;
+
+public class UpgradeManager extends ConfigLoader {
+    private final Set<UpgradeType> setNeededUpgrades;
+
+    public UpgradeManager() {
+        super("upgrades.yml");
+
+        setNeededUpgrades = EnumSet.allOf(UpgradeType.class);
+
+        loadKeys();
+    }
+
+    /**
+     * Check if the given {@link UpgradeType} is necessary.
+     *
+     * @param type Upgrade type to check
+     *
+     * @return true if plugin data needs to have the given upgrade
+     */
+    public boolean shouldUpgrade(final UpgradeType type) {
+        return setNeededUpgrades.contains(type);
+    }
+
+    /**
+     * Set the given {@link UpgradeType} as completed. Does nothing if
+     * the upgrade was applied previously.
+     *
+     * @param type Upgrade type to set as complete
+     */
+    public void setUpgradeCompleted(final UpgradeType type) {
+        if (!setNeededUpgrades.remove(type)) {
+            return;
+        }
+
+        plugin.debug("Saving upgrade status for type " + type.toString() + "...");
+
+        config.set("Upgrades_Finished." + type.toString(), true);
+
+        try {
+            config.save(getFile());
+        }
+        catch (Exception e) {
+            e.printStackTrace();
+        }
+    }
+
+    @Override
+    protected void loadKeys() {
+        for (UpgradeType type : UpgradeType.values()) {
+            if (config.getBoolean("Upgrades_Finished." + type.toString())) {
+                setNeededUpgrades.remove(type);
+            }
+        }
+
+        plugin.debug("Needed upgrades: " + Arrays.toString(setNeededUpgrades.toArray(new UpgradeType[setNeededUpgrades.size()])));
+    }
+}