Files
CustomNPC-Plus/src/main/java/noppes/npcs/controllers/data/DataScript.java
T

309 lines
9.4 KiB
Java

package noppes.npcs.controllers.data;
import cpw.mods.fml.common.FMLCommonHandler;
import cpw.mods.fml.common.eventhandler.Event;
import cpw.mods.fml.relauncher.Side;
import net.minecraft.entity.Entity;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagList;
import net.minecraft.world.World;
import net.minecraft.world.WorldServer;
import noppes.npcs.EventHooks;
import noppes.npcs.LogWriter;
import noppes.npcs.NBTTags;
import noppes.npcs.api.IWorld;
import noppes.npcs.api.entity.ICustomNpc;
import noppes.npcs.config.ConfigDebug;
import noppes.npcs.config.ConfigScript;
import noppes.npcs.constants.EnumScriptType;
import noppes.npcs.controllers.ScriptContainer;
import noppes.npcs.controllers.ScriptController;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.scripted.NpcAPI;
import noppes.npcs.scripted.ScriptWorld;
import noppes.npcs.scripted.constants.EntityType;
import noppes.npcs.scripted.constants.JobType;
import noppes.npcs.scripted.constants.RoleType;
import noppes.npcs.scripted.entity.ScriptNpc;
import javax.script.ScriptEngine;
import java.util.*;
public class DataScript implements INpcScriptHandler {
public List<ScriptContainer> eventScripts = new ArrayList<>();
private HashMap<EnumScriptType,ScriptContainer> scripts = new HashMap<>();
private final static EntityType entities = new EntityType();
private final static JobType jobs = new JobType();
private final static RoleType roles = new RoleType();
public String scriptLanguage = "ECMAScript";
private EntityNPCInterface npc;
public boolean enabled = false;
public ICustomNpc dummyNpc;
public IWorld dummyWorld;
public boolean clientNeedsUpdate = false;
public boolean aiNeedsUpdate = false;
public boolean hasInited = false;
public DataScript(EntityNPCInterface npc) {
for (int i = 0; i < 15; i++) {
this.setNPCScript(i, new ScriptContainer(this));
}
this.npc = npc;
if (npc.wrappedNPC == null) {
npc.wrappedNPC = new ScriptNpc(this.npc);
}
dummyNpc = npc.wrappedNPC;
if(npc.worldObj instanceof WorldServer)
dummyWorld = NpcAPI.Instance().getIWorld((WorldServer) npc.worldObj);//new ScriptWorld((WorldServer) npc.worldObj);
}
public void readFromNBT(NBTTagCompound compound) {
scripts = readScript(compound.getTagList("ScriptsContainers", 10));
this.scriptLanguage = compound.getString("ScriptLanguage");
if (!ScriptController.Instance.languages.containsKey(scriptLanguage)) {
if (!ScriptController.Instance.languages.isEmpty()) {
this.scriptLanguage = (String) ScriptController.Instance.languages.keySet().toArray()[0];
} else {
this.scriptLanguage = "ECMAScript";
}
}
enabled = compound.getBoolean("ScriptEnabled");
}
public void readEventsFromNBT(NBTTagCompound compound) {
eventScripts = NBTTags.GetScript(compound,this);
}
public NBTTagCompound writeToNBT(NBTTagCompound compound) {
compound.setTag("ScriptsContainers", writeScript(scripts));
compound.setString("ScriptLanguage", scriptLanguage);
compound.setBoolean("ScriptEnabled", enabled);
return compound;
}
public NBTTagCompound writeEventsToNBT(NBTTagCompound compound) {
compound.setInteger("TotalScripts",this.eventScripts.size());
for (int i = 0; i < this.eventScripts.size(); i++) {
compound.setTag("Tab"+i,this.eventScripts.get(i).writeToNBT(new NBTTagCompound()));
}
return compound;
}
private HashMap<EnumScriptType,ScriptContainer> readScript(NBTTagList list){
HashMap<EnumScriptType,ScriptContainer> scripts = new HashMap<>();
for (int i = 0; i < 15; i++) {
scripts.put(EnumScriptType.values()[i], new ScriptContainer(this));
}
for(int i = 0; i < list.tagCount(); i++){
NBTTagCompound compoundd = list.getCompoundTagAt(i);
ScriptContainer script = new ScriptContainer(this);
script.readFromNBT(compoundd);
if (script.hasCode() || npc.isRemote()) {
scripts.put(EnumScriptType.values()[compoundd.getInteger("Type")], script);
}
}
return scripts;
}
private NBTTagList writeScript(HashMap<EnumScriptType,ScriptContainer> scripts){
NBTTagList list = new NBTTagList();
for (Map.Entry<EnumScriptType,ScriptContainer> entry : scripts.entrySet()) {
NBTTagCompound tagCompound = new NBTTagCompound();
ScriptContainer container = entry.getValue();
if (container.hasCode()) {
tagCompound.setInteger("Type", entry.getKey().ordinal());
container.writeToNBT(tagCompound);
list.appendTag(tagCompound);
}
}
return list;
}
public boolean callScript(EnumScriptType type, Event event, Object... obs){
if(aiNeedsUpdate){
npc.updateAI = true;
aiNeedsUpdate = false;
}
if(clientNeedsUpdate){
npc.updateClient = true;
clientNeedsUpdate = false;
}
if(!isEnabled())
return false;
if(!hasInited && !npc.isRemote() && type != EnumScriptType.INIT){
hasInited = true;
for (ScriptContainer scriptContainer : this.eventScripts) {
scriptContainer.errored = false;
}
EventHooks.onNPCInit(this.npc);
}
for (ScriptContainer script : this.eventScripts) {
script.run(type, event);
}
ScriptContainer script = scripts.get(type);
if(script == null || script.errored || !script.hasCode())
return false;
script.setEngine(this.getLanguage());
if(script.engine == null)
return false;
for(int i = 0; i + 1 < obs.length; i += 2){
Object ob = obs[i + 1];
if(ob instanceof Entity)
ob = NpcAPI.Instance().getIEntity((Entity)ob);
script.engine.put(obs[i].toString(), ob);
}
if(ConfigDebug.ScriptLogging && FMLCommonHandler.instance().getEffectiveSide() == Side.SERVER){
if(obs.length > 1 && obs[1] == null){
LogWriter.postScriptLog(npc.field_110179_h, type, String.format("[%s] NPC %s (%s, %s, %s)", ((String)type.function).toUpperCase(), npc.display.name, (int)npc.posX, (int)npc.posY, (int)npc.posZ));
} else {
LogWriter.postScriptLog(npc.field_110179_h, type, String.format("[%s] NPC %s (%s, %s, %s) | Objects: %s", ((String)type.function).toUpperCase(), npc.display.name, (int)npc.posX, (int)npc.posY, (int)npc.posZ, Arrays.toString(obs)));
}
}
return callScript(script, event);
}
private boolean callScript(ScriptContainer script, Event event){
ScriptEngine engine = script.engine;
engine.put("npc", dummyNpc);
engine.put("world", dummyWorld);
engine.put("event", event);
engine.put("API", NpcAPI.Instance());
engine.put("EntityType", entities);
engine.put("RoleType", roles);
engine.put("JobType", jobs);
for (Map.Entry<String,Object> engineObjects : NpcAPI.engineObjects.entrySet()) {
engine.put(engineObjects.getKey(),engineObjects.getValue());
}
script.run(engine);
if(clientNeedsUpdate){
npc.updateClient = true;
clientNeedsUpdate = false;
}
if(aiNeedsUpdate){
npc.updateAI = true;
aiNeedsUpdate = false;
}
return event.isCanceled();
}
public boolean isEnabled(){
return enabled && ScriptController.HasStart && !npc.worldObj.isRemote && !scripts.isEmpty() && ConfigScript.ScriptingEnabled;
}
public Map<Long, String> getOldConsoleText() {
Map<Long, String> map = new TreeMap<>();
for (Map.Entry<EnumScriptType,ScriptContainer> entry : scripts.entrySet()) {
for (Map.Entry<Long,String> consoleEntry : entry.getValue().console.entrySet()) {
map.put(consoleEntry.getKey(), " tab " + entry.getKey().ordinal() + ":\n" + consoleEntry.getValue());
}
}
return map;
}
public Map<Long, String> getConsoleText() {
TreeMap<Long, String> map = new TreeMap<>();
int tab = 0;
for (ScriptContainer script : this.getScripts()) {
++tab;
for (Map.Entry<Long, String> longStringEntry : script.console.entrySet()) {
map.put(longStringEntry.getKey(), " tab " + tab + ":\n" + longStringEntry.getValue());
}
}
return map;
}
public void clearConsole() {
for (ScriptContainer script : this.getScripts()) {
script.console.clear();
}
}
@Override
public void callScript(EnumScriptType type, Event event) {
callScript(type, event, "$$IGNORED$$", null);
}
@Override
public void callScript(String hookName, Event event) {
try {
EnumScriptType enumScriptType = EnumScriptType.valueOf(hookName);
this.callScript(enumScriptType, event);
} catch (IllegalArgumentException ignored) {}
}
public boolean isClient() {
return this.npc.isRemote();
}
public boolean getEnabled() {
return this.enabled;
}
public void setEnabled(boolean bo) {
this.enabled = bo;
}
public String getLanguage() {
return this.scriptLanguage;
}
public void setLanguage(String lang) {
this.scriptLanguage = lang;
}
public void setScripts(List<ScriptContainer> list) {
this.eventScripts = list;
}
public List<ScriptContainer> getScripts() {
return this.eventScripts;
}
public String noticeString() {
//BlockPos pos = this.npc.func_180425_c();
//return MoreObjects.toStringHelper(this.npc).add("x", pos.func_177958_n()).add("y", pos.func_177956_o()).add("z", pos.func_177952_p()).toString();
return "";
}
public void setWorld(World world) {
if(world instanceof WorldServer)
dummyWorld = new ScriptWorld((WorldServer) world);
}
public ScriptContainer getNPCScript(EnumScriptType scriptType) {
return this.scripts.get(scriptType);
}
public ScriptContainer getNPCScript(int ordinal) {
return this.getNPCScript(EnumScriptType.values()[ordinal]);
}
public void setNPCScript(EnumScriptType scriptType, ScriptContainer container) {
this.scripts.put(scriptType, container);
}
public void setNPCScript(int ordinal, ScriptContainer container) {
this.setNPCScript(EnumScriptType.values()[ordinal], container);
}
public Collection<ScriptContainer> getNPCScripts() {
return this.scripts.values();
}
}