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

package noppes.npcs.controllers;

import cpw.mods.fml.common.eventhandler.SubscribeEvent;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.nbt.NBTTagString;
import net.minecraft.server.MinecraftServer;
import net.minecraftforge.event.world.WorldEvent;
import noppes.npcs.CustomNpcs;
import noppes.npcs.LogWriter;
import noppes.npcs.controllers.data.ForgeDataScript;
import noppes.npcs.controllers.data.GlobalNPCDataScript;
import noppes.npcs.controllers.data.PlayerDataScript;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.scripted.ScriptWorld;
import noppes.npcs.util.JsonException;
import noppes.npcs.util.NBTJsonUtil;

import javax.script.ScriptContext;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineFactory;
import javax.script.ScriptEngineManager;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.*;

public class ScriptController {

	public static ScriptController Instance;
	public static boolean HasStart = false;
	private final ScriptEngineManager manager;
	private ScriptEngineFactory nashornFactory;
	public Map<String, String> languages = new HashMap<String, String>();
	public Map<String, String> scripts = new HashMap<String, String>();
	public long lastLoaded = 0;
	public File dir;
	public NBTTagCompound compound = new NBTTagCompound();

	private boolean loaded = false;
	public boolean shouldSave = false;

	public PlayerDataScript playerScripts = new PlayerDataScript((EntityPlayer)null);
	public long lastPlayerUpdate = 0L;

	public ForgeDataScript forgeScripts = new ForgeDataScript();
	public long lastForgeUpdate = 0L;

	public GlobalNPCDataScript globalNpcScripts = new GlobalNPCDataScript((EntityNPCInterface) null);
	public long lastGlobalNpcUpdate = 0L;

	public ScriptController(){
		loaded = false;
		Instance = this;
		manager = new ScriptEngineManager();
		LogWriter.info("Script Engines Available:");

		ScriptEngine engine = manager.getEngineByName("nashorn");
		if (engine != null) {
			this.nashornFactory = engine.getFactory();
		}

		for(ScriptEngineFactory fac : manager.getEngineFactories()){
			if(fac.getExtensions().isEmpty())
				continue;

			ScriptEngine scriptEngine = fac.getScriptEngine();
			try {
				scriptEngine.put("$RunTest",null);
				String ext = "." + fac.getExtensions().get(0).toLowerCase();
				LogWriter.info("Engine " + fac.getEngineName() + " running " + fac.getLanguageName() + " with extension: " + ext);
				languages.put(fac.getLanguageName(), ext);
			} catch (Exception ignored) {}
		}
	}

	private File forgeScriptsFile() {
		return new File(this.dir, "forge_scripts.json");
	}

	public boolean loadForgeScripts() {
		this.forgeScripts.clear();
		File file = this.forgeScriptsFile();

		try {
			if(!file.exists()) {
				return false;
			} else {
				this.forgeScripts.readFromNBT(NBTJsonUtil.LoadFile(file));
				return true;
			}
		} catch (Exception var3) {
			LogWriter.error("Error loading: " + file.getAbsolutePath(), var3);
			return false;
		}
	}

	public void saveForgeScripts() {
		File file = this.forgeScriptsFile();
		try {
			NBTJsonUtil.SaveFile(file, this.forgeScripts.writeToNBT(new NBTTagCompound()));
			this.forgeScripts.lastInited = -1L;
		} catch (IOException | JsonException var4) {
			var4.printStackTrace();
		}
	}

	private File playerScriptsFile() {
		return new File(dir, "player_scripts.json");
	}

	private File npcScriptsFile() {
		return new File(dir, "npc_scripts.json");
	}

	public boolean loadPlayerScripts() {
		this.playerScripts.clear();
		File file = this.playerScriptsFile();

		try {
			if(!file.exists()) {
				return false;
			} else {
				this.playerScripts.readFromNBT(NBTJsonUtil.LoadFile(file));
				shouldSave = false;
				return true;
			}
		} catch (Exception var3) {
			LogWriter.error("Error loading: " + file.getAbsolutePath(), var3);
			return false;
		}
	}

	public void savePlayerScripts() {
		File file = this.playerScriptsFile();
		try {
			NBTJsonUtil.SaveFile(file, this.playerScripts.writeToNBT(new NBTTagCompound()));
			this.playerScripts.lastInited = -1L;
		} catch (IOException | JsonException var4) {
			var4.printStackTrace();
		}
	}

	public boolean loadGlobalNPCScripts() {
		this.globalNpcScripts.clear();
		File file = this.npcScriptsFile();

		try {
			if(!file.exists()) {
				return false;
			} else {
				this.globalNpcScripts.readFromNBT(NBTJsonUtil.LoadFile(file));
				shouldSave = false;
				return true;
			}
		} catch (Exception var3) {
			LogWriter.error("Error loading: " + file.getAbsolutePath(), var3);
			return false;
		}
	}

	public void saveGlobalNpcScripts() {
		File file = this.npcScriptsFile();
		try {
			NBTJsonUtil.SaveFile(file, this.globalNpcScripts.writeToNBT(new NBTTagCompound()));
			this.globalNpcScripts.lastInited = -1L;
		} catch (IOException | JsonException var4) {
			var4.printStackTrace();
		}
	}

	public void loadCategories(){
		dir = new File(CustomNpcs.getWorldSaveDirectory(), "scripts");
		if(!dir.exists())
			dir.mkdir();
		if(!getSavedFile().exists())
			shouldSave = true;
		new ScriptWorld(null).clearTempData();
		scripts.clear();
		for(String language : languages.keySet()){
			String ext = languages.get(language);
			File scriptDir = new File(dir,language.toLowerCase());
			if(!scriptDir.exists())
				scriptDir.mkdir();
			else
				loadDir(scriptDir, "", ext);
		}
		lastLoaded = System.currentTimeMillis();
	}
	private void loadDir(File dir, String name, String ext){
		for(File file : dir.listFiles()){
			String filename = name + file.getName().toLowerCase();
			if(file.isDirectory()){
				loadDir(file, filename + "/", ext);
				continue;
			}
			if(!filename.endsWith(ext))
				continue;
			try {
				scripts.put(filename, readFile(file));
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	public boolean loadStoredData(){
		loadCategories();
		File file = getSavedFile();
		try {
			if(!file.exists())
				return false;
			this.compound = NBTJsonUtil.LoadFile(file);
			shouldSave = false;
		}
		catch (Exception e) {
			LogWriter.error("Error loading: " + file.getAbsolutePath(), e);
			return false;
		}
		return true;
	}

	private File getSavedFile(){
		return new File(dir, "world_data.json");
	}

	private String readFile(File file) throws IOException {
		BufferedReader br = new BufferedReader(new FileReader(file));
		try {
			StringBuilder sb = new StringBuilder();
			String line = br.readLine();

			while (line != null) {
				sb.append(line);
				sb.append("\n");
				line = br.readLine();
			}
			return sb.toString();
		} finally {
			br.close();
		}
	}

	private static final List<String> nashornNames = immutableList("nashorn", "Nashorn", "js", "JS", "JavaScript", "javascript", "ECMAScript", "ecmascript");
	public ScriptEngine getEngineByName(String language) {
		if (nashornNames.contains(language) && this.nashornFactory != null) {
			ScriptEngine scriptEngine = this.nashornFactory.getScriptEngine();
			scriptEngine.setBindings(this.manager.getBindings(), ScriptContext.GLOBAL_SCOPE);
			return scriptEngine;
		}
		return manager.getEngineByName(language);
	}
	private static List<String> immutableList(String... elements) {
		return Collections.unmodifiableList(Arrays.asList(elements));
	}

	public NBTTagList nbtLanguages() {
		NBTTagList list = new NBTTagList();
		for(String language : languages.keySet()){
			NBTTagCompound compound = new NBTTagCompound();
			NBTTagList scripts = new NBTTagList();
			for(String script : getScripts(language)){
				scripts.appendTag(new NBTTagString(script));
			}
			compound.setTag("Scripts", scripts);
			compound.setString("Language", language);
			list.appendTag(compound);
		}
		return list;
	}

	private List<String> getScripts(String language){
		List<String> list = new ArrayList<String>();
		String ext = languages.get(language);
		if(ext == null)
			return list;
		for(String script : scripts.keySet()){
			if(script.endsWith(ext)){
				list.add(script);
			}
		}
		return list;
	}

	@SubscribeEvent
	public void invoke(WorldEvent.Save event){
		if(!shouldSave || event.world.isRemote || event.world != MinecraftServer.getServer().worldServers[0])
			return;

		try {
			NBTJsonUtil.SaveFile(getSavedFile(), compound);
		}
		catch (Exception e) {
			LogWriter.except(e);
		}

		shouldSave = false;
	}
}