path: root/src/main/java/io/github/moulberry/notenoughupdates/gamemodes/SBGamemodes.java

package io.github.moulberry.notenoughupdates.gamemodes;

import com.google.gson.Gson;
import com.google.gson.JsonObject;
import io.github.moulberry.notenoughupdates.NotEnoughUpdates;
import io.github.moulberry.notenoughupdates.util.SBInfo;
import net.minecraft.client.Minecraft;
import net.minecraft.client.gui.GuiScreen;
import net.minecraft.client.gui.inventory.GuiChest;
import net.minecraft.inventory.ContainerChest;
import net.minecraft.util.ChatComponentText;
import net.minecraft.util.EnumChatFormatting;
import net.minecraftforge.client.event.ClientChatReceivedEvent;
import net.minecraftforge.event.entity.player.PlayerInteractEvent;
import net.minecraftforge.fml.common.eventhandler.SubscribeEvent;
import net.minecraftforge.fml.common.gameevent.TickEvent;

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import java.io.*;
import java.nio.ByteBuffer;
import java.nio.charset.StandardCharsets;
import java.security.Key;
import java.util.Base64;
import java.util.HashMap;

public class SBGamemodes {

    private static final Gson gson = new Gson();

    public static final int MODIFIER_DEVILISH = 0b1;
    public static final int MODIFIER_NOBANK = 0b10;
    public static final int MODIFIER_SMALLISLAND = 0b100;

    public static final String MODIFIER_DEVILISH_DISPLAY = EnumChatFormatting.DARK_PURPLE+"Devilish";
    public static final String MODIFIER_NOBANK_DISPLAY = EnumChatFormatting.RED+"No"+EnumChatFormatting.GOLD+"Bank";
    public static final String MODIFIER_SMALLISLAND_DISPLAY = EnumChatFormatting.GREEN+"SmallIsland";

    public static final String MODIFIER_DEVILISH_DESC = EnumChatFormatting.DARK_PURPLE+"Devilish\n" +
            "You are NOT allowed to use fairy souls.";
    public static final String MODIFIER_NOBANK_DESC = EnumChatFormatting.RED+"No"+EnumChatFormatting.GOLD+"Bank\n" +
            "You are NOT allowed to use the bank.";
    public static final String MODIFIER_SMALLISLAND_DESC = EnumChatFormatting.GREEN+"SmallIsland\n" +
            "Your private island is 1/4 the normal size.";

    private static HashMap<String, Gamemode> currentGamemode = new HashMap<>();
    private static long lastDeathExemption = 0;

    public static class Gamemode {
        public HardcoreMode hardcoreMode = HardcoreMode.NORMAL;
        public IronmanMode ironmanMode = IronmanMode.NORMAL;
        public int gamemodeModifiers = 0;

        public boolean locked = true;
    }

    public enum HardcoreMode {
        NORMAL("Normal", "Normal"),
        SOFTCORE(EnumChatFormatting.RED+"Soft"+EnumChatFormatting.DARK_RED+"core\n" +
                "You only have 1 life.\nDying will remove your hardcore status.\nDeaths to the void or \'unknown\' are exempted.",
                "You died.", "You fell into the void"),
        HARDCORE(EnumChatFormatting.DARK_RED+"Hardcore\n" +
                "You only have 1 life.\nDying will remove your hardcore status.");

        public final String display;
        public final String desc;
        private String[] exemptions;

        HardcoreMode(String display, String... exemptions) {
            this.display = display.split("\n")[0];
            this.desc = display;
            this.exemptions = exemptions;
        }

        public boolean isExemption(String line) {
            for(String exemption : exemptions) {
                if(line.contains(exemption)) return true;
            }
            return false;
        }
    }

    public enum IronmanMode {
        NORMAL("Normal"),
        IRONMAN(EnumChatFormatting.WHITE+"Ironman\n" +
                "You are NOT allowed to trade or use the auction house.",
                "You   ", "Auction House", "Auctions Browser", "Auction View"),
        IRONMANPLUS(EnumChatFormatting.WHITE+"Ironman"+EnumChatFormatting.GOLD+"+\n" +
                "You are NOT allowed to trade, use the auction house or bazaar.",
                "You   ", "Auction House", "Auctions Browser", "Auction View", "Bazaar"),
        ULTIMATE_IRONMAN(EnumChatFormatting.DARK_AQUA+"Ultimate "+EnumChatFormatting.WHITE+"Ironman\n" +
                 "You are NOT allowed to trade or use the auction house.\n" +
                      "You are restricted to 1 inventory. (No containers, no echest, no wardrobe).",
                "You   ", "Auction House", "Auctions Browser", "Auction View", "Chest",
                "Wardrobe", "Weapon Rack", "Shelves"),
        ULTIMATE_IRONMANPLUS(EnumChatFormatting.DARK_AQUA+"Ultimate "+EnumChatFormatting.WHITE+"Ironman"+EnumChatFormatting.GOLD+"+\n" +
                "You are NOT allowed to trade, use the auction house or bazaar.\n" +
                     "You are restricted to 1 inventory. (No containers, no echest, no wardrobe).",
                "You   ", "Auction House", "Auctions Browser", "Auction View", "Bazaar",
                "Chest", "Wardrobe", "Weapon Rack", "Shelves");

        public final String display;
        public final String desc;
        private final String[] bannedInventories;

        IronmanMode(String display, String... bannedInventories) {
            this.display = display.split("\n")[0];
            this.desc = display;
            this.bannedInventories = bannedInventories;
        }

        public boolean isBanned(String inventoryName) {
            for(String banned : bannedInventories) {
                if(inventoryName.contains(banned + " ") || inventoryName.endsWith(banned)) return true;
            }
            return false;
        }
    }

    public static Gamemode getGamemode() {
        String currentProfile = SBInfo.getInstance().currentProfile;

        if(currentProfile == null || currentProfile.isEmpty()) return null;

        return currentGamemode.computeIfAbsent(currentProfile, k -> new Gamemode());
    }

    public static void loadFromFile() {
        File configDir = NotEnoughUpdates.INSTANCE.manager.configLocation;
        File gamemodeFile = new File(configDir,
                "gamemodes/gamemodes-"+Minecraft.getMinecraft().thePlayer.getUniqueID().toString()+".json");
        gamemodeFile.getParentFile().mkdirs();

        if(!gamemodeFile.exists()) {
            return;
        }

        try(BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(gamemodeFile), StandardCharsets.UTF_8))) {
            String line = reader.readLine();
            String decoded = decrypt(line);
            currentGamemode = gson.fromJson(decoded, GamemodeWrapper.class).currentGamemode;
        } catch(Exception e) {
            e.printStackTrace();
        }
    }

    public static class GamemodeWrapper {
        private HashMap<String, Gamemode> currentGamemode;

        public GamemodeWrapper(HashMap<String, Gamemode> currentGamemode) {
            this.currentGamemode = currentGamemode;
        }
    }

    public static void saveToFile() {
        File configDir = NotEnoughUpdates.INSTANCE.manager.configLocation;
        File gamemodeFile = new File(configDir,
                "gamemodes/gamemodes-"+Minecraft.getMinecraft().thePlayer.getUniqueID().toString()+".json");
        gamemodeFile.getParentFile().mkdirs();

        try {
            gamemodeFile.createNewFile();

            try(BufferedWriter writer = new BufferedWriter(
                    new OutputStreamWriter(new FileOutputStream(gamemodeFile), StandardCharsets.UTF_8))) {
                JsonObject obj = new JsonObject();
                writer.write(encrypt(gson.toJson(new GamemodeWrapper(currentGamemode), GamemodeWrapper.class)));
            }
        } catch(Exception e) {
            e.printStackTrace();
        }
    }

    public static Key getKeyFromPlayerUUID() {
        byte[] bytes = ByteBuffer.allocate(2 * Long.SIZE / Byte.SIZE)
                .putLong(Minecraft.getMinecraft().thePlayer.getUniqueID().getLeastSignificantBits())
                .putLong(Minecraft.getMinecraft().thePlayer.getUniqueID().getMostSignificantBits())
                .array();
        SecretKeySpec key = new SecretKeySpec(bytes, "AES");

        return key;
    }


    public static String encrypt(String value) {
        try {
            Cipher cipher = Cipher.getInstance("AES");
            cipher.init(Cipher.ENCRYPT_MODE, getKeyFromPlayerUUID());
            String encrypt = Base64.getEncoder().encodeToString(cipher.doFinal(value.getBytes()));
            return encrypt;
        } catch (Exception e) {
            e