CustomNPC-Plus/src/main/java/noppes/npcs/controllers/BankController.java

package noppes.npcs.controllers;

import net.minecraft.nbt.CompressedStreamTools;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import noppes.npcs.CustomNpcs;
import noppes.npcs.controllers.data.Bank;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap;

public class BankController {
    public HashMap<Integer, Bank> banks;
    private String filePath = "";

    private static BankController instance;

    public BankController() {
        instance = this;
        banks = new HashMap<Integer, Bank>();
        loadBanks();

        if (banks.isEmpty()) {
            Bank bank = new Bank();
            bank.id = 0;
            bank.name = "Default Bank";
            for (int i = 0; i < 6; i++)
                bank.slotTypes.put(i, 0);

            banks.put(bank.id, bank);
        }
    }

    public static BankController getInstance() {
        if (newInstance()) {
            instance = new BankController();
        }
        return instance;
    }

    private static boolean newInstance() {
        if (instance == null)
            return true;
        File file = CustomNpcs.getWorldSaveDirectory();
        if (file == null)
            return false;
        if (!instance.filePath.equals(file.getAbsolutePath())) {
            return true;
        }

        return false;
    }

    private void loadBanks() {
        File saveDir = CustomNpcs.getWorldSaveDirectory();
        if (saveDir == null)
            return;

        filePath = saveDir.getAbsolutePath();

        try {
            File file = new File(saveDir, "bank.dat");
            if (file.exists()) {
                loadBanks(file);
            }
        } catch (Exception e) {
            try {
                File file = new File(saveDir, "bank.dat_old");
                if (file.exists()) {
                    loadBanks(file);
                }

            } catch (Exception ee) {
            }
        }
    }

    private void loadBanks(File file) throws IOException {
        try (FileInputStream fis = new FileInputStream(file)) {
            loadBanks(CompressedStreamTools.readCompressed(fis));
        }
    }

    public void loadBanks(NBTTagCompound nbttagcompound1) throws IOException {
        HashMap<Integer, Bank> banks = new HashMap<Integer, Bank>();
        NBTTagList list = nbttagcompound1.getTagList("Data", 10);
        if (list != null) {
            for (int i = 0; i < list.tagCount(); i++) {
                NBTTagCompound nbttagcompound = list.getCompoundTagAt(i);
                Bank bank = new Bank();
                bank.readEntityFromNBT(nbttagcompound);
                banks.put(bank.id, bank);
            }
        }
        this.banks = banks;
    }

    public NBTTagCompound getNBT() {

        NBTTagList list = new NBTTagList();
        for (Bank bank : banks.values()) {
            NBTTagCompound nbtfactions = new NBTTagCompound();
            bank.writeEntityToNBT(nbtfactions);
            list.appendTag(nbtfactions);
        }
        NBTTagCompound nbttagcompound = new NBTTagCompound();
        nbttagcompound.setTag("Data", list);
        return nbttagcompound;
    }

    public Bank getBank(int bankId) {
        Bank bank = banks.get(bankId);
        if (bank != null)
            return bank;
        return banks.values().iterator().next();
    }

    public void saveBanks() {
        try {
            File saveDir = CustomNpcs.getWorldSaveDirectory();
            File file = new File(saveDir, "bank.dat_new");
            File file1 = new File(saveDir, "bank.dat_old");
            File file2 = new File(saveDir, "bank.dat");
            CompressedStreamTools.writeCompressed(getNBT(), new FileOutputStream(file));
            if (file1.exists()) {
                file1.delete();
            }
            file2.renameTo(file1);
            if (file2.exists()) {
                file2.delete();
            }
            file.renameTo(file2);
            if (file.exists()) {
                file.delete();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void saveBank(Bank bank) {
        if (bank.id < 0) {
            bank.id = getUnusedId();
        }
        banks.put(bank.id, bank);
        saveBanks();
    }

    public int getUnusedId() {
        int id = 0;
        while (true) {
            if (!banks.containsKey(id))
                return id;
            id++;
        }
    }

    public void removeBank(int bank) {
        if (bank < 0 || banks.size() <= 1)
            return;
        banks.remove(bank);
        saveBanks();
    }
}