mirror of
https://github.com/KAMKEEL/CustomNPC-Plus.git
synced 2026-11-12 09:37:29 +00:00
335 lines
9.7 KiB
Java
335 lines
9.7 KiB
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.LogWriter;
|
|
import noppes.npcs.NoppesStringUtils;
|
|
import noppes.npcs.Server;
|
|
import noppes.npcs.api.handler.IDialogHandler;
|
|
import noppes.npcs.api.handler.data.IDialog;
|
|
import noppes.npcs.api.handler.data.IDialogCategory;
|
|
import noppes.npcs.constants.EnumOptionType;
|
|
import noppes.npcs.constants.EnumPacketClient;
|
|
import noppes.npcs.constants.SyncType;
|
|
import noppes.npcs.controllers.data.Dialog;
|
|
import noppes.npcs.controllers.data.DialogCategory;
|
|
import noppes.npcs.controllers.data.DialogOption;
|
|
import noppes.npcs.util.NBTJsonUtil;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.util.*;
|
|
|
|
public class DialogController implements IDialogHandler {
|
|
public HashMap<Integer,DialogCategory> categoriesSync = new HashMap<Integer, DialogCategory>();
|
|
public HashMap<Integer,DialogCategory> categories = new HashMap<Integer, DialogCategory>();
|
|
public HashMap<Integer,Dialog> dialogs = new HashMap<Integer, Dialog>();
|
|
public static DialogController Instance = new DialogController();;
|
|
|
|
private int lastUsedDialogID = 0;
|
|
private int lastUsedCatID = 0;
|
|
|
|
public DialogController(){
|
|
Instance = this;
|
|
}
|
|
|
|
public void load(){
|
|
LogWriter.info("Loading Dialogs");
|
|
loadCategories();
|
|
LogWriter.info("Done loading Dialogs");
|
|
}
|
|
|
|
private void loadCategories(){
|
|
categories.clear();
|
|
dialogs.clear();
|
|
|
|
lastUsedCatID = 0;
|
|
lastUsedDialogID = 0;
|
|
|
|
try {
|
|
File file = new File(CustomNpcs.getWorldSaveDirectory(), "dialog.dat");
|
|
if(file.exists()){
|
|
loadCategoriesOld(file);
|
|
file.delete();
|
|
file = new File(CustomNpcs.getWorldSaveDirectory(), "dialog.dat_old");
|
|
if(file.exists())
|
|
file.delete();
|
|
return;
|
|
}
|
|
} catch (Exception e) {
|
|
LogWriter.except(e);
|
|
}
|
|
|
|
File dir = getDir();
|
|
if(!dir.exists()){
|
|
dir.mkdir();
|
|
loadDefaultDialogs();
|
|
}
|
|
else{
|
|
for(File file : dir.listFiles()){
|
|
if(!file.isDirectory())
|
|
continue;
|
|
DialogCategory category = loadCategoryDir(file);
|
|
Iterator<Integer> ite = category.dialogs.keySet().iterator();
|
|
while(ite.hasNext()){
|
|
int id = ite.next();
|
|
if(id > lastUsedDialogID)
|
|
lastUsedDialogID = id;
|
|
Dialog dialog = category.dialogs.get(id);
|
|
if(dialogs.containsKey(id)){
|
|
LogWriter.error("Duplicate id " + dialog.id + " from category " + category.title);
|
|
ite.remove();
|
|
}
|
|
else{
|
|
dialogs.put(id, dialog);
|
|
}
|
|
}
|
|
lastUsedCatID++;
|
|
category.id = lastUsedCatID;
|
|
categories.put(category.id, category);
|
|
}
|
|
}
|
|
}
|
|
private DialogCategory loadCategoryDir(File dir) {
|
|
DialogCategory category = new DialogCategory();
|
|
category.title = dir.getName();
|
|
for(File file : dir.listFiles()){
|
|
if(!file.isFile() || !file.getName().endsWith(".json"))
|
|
continue;
|
|
try{
|
|
Dialog dialog = new Dialog();
|
|
dialog.id = Integer.parseInt(file.getName().substring(0, file.getName().length() - 5));
|
|
dialog.readNBTPartial(NBTJsonUtil.LoadFile(file));
|
|
category.dialogs.put(dialog.id, dialog);
|
|
dialog.category = category;
|
|
}
|
|
catch(Exception e){
|
|
LogWriter.error("Error loading: " + file.getAbsolutePath(), e);
|
|
}
|
|
}
|
|
return category;
|
|
}
|
|
|
|
private void loadCategoriesOld(File file) throws Exception{
|
|
File dir = getDir();
|
|
if(!dir.exists()){
|
|
dir.mkdir();
|
|
}
|
|
NBTTagCompound nbttagcompound1 = CompressedStreamTools.readCompressed(new FileInputStream(file));
|
|
NBTTagList list = nbttagcompound1.getTagList("Data", 10);
|
|
if(list == null)
|
|
return;
|
|
|
|
for(int i = 0; i < list.tagCount(); i++){
|
|
DialogCategory category = new DialogCategory();
|
|
category.readNBT(list.getCompoundTagAt(i));
|
|
saveCategory(category);
|
|
Iterator<Map.Entry<Integer, Dialog>> ita = category.dialogs.entrySet().iterator();
|
|
while(ita.hasNext()){
|
|
Map.Entry<Integer, Dialog> entry = ita.next();
|
|
Dialog dialog = entry.getValue();
|
|
dialog.id = entry.getKey();
|
|
dialog.category = category;
|
|
if(dialogs.containsKey(dialog.id))
|
|
ita.remove();
|
|
else{
|
|
saveDialog(category.id, dialog);
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
private void loadDefaultDialogs() {
|
|
DialogCategory cat = new DialogCategory();
|
|
cat.id = lastUsedCatID++;
|
|
cat.title = "Villager";
|
|
|
|
Dialog dia1 = new Dialog();
|
|
dia1.id = 1;
|
|
dia1.category = cat;
|
|
dia1.title = "Start";
|
|
dia1.text = "Hello {player}, "+'\n'+'\n'+"Welcome to our village. I hope you enjoy your stay";
|
|
|
|
Dialog dia2 = new Dialog();
|
|
dia2.id = 2;
|
|
dia2.category = cat;
|
|
dia2.title = "Ask about village";
|
|
dia2.text = "This village has been around for ages. Enjoy your stay here.";
|
|
|
|
Dialog dia3 = new Dialog();
|
|
dia3.id = 3;
|
|
dia3.category = cat;
|
|
dia3.title = "Who are you";
|
|
dia3.text = "I'm a villager here. I have lived in this village my whole life.";
|
|
|
|
cat.dialogs.put(dia1.id, dia1);
|
|
cat.dialogs.put(dia2.id, dia2);
|
|
cat.dialogs.put(dia3.id, dia3);
|
|
|
|
|
|
DialogOption option = new DialogOption();
|
|
option.title = "Tell me something about this village";
|
|
option.dialogId = 2;
|
|
option.optionType = EnumOptionType.DialogOption;
|
|
|
|
DialogOption option2 = new DialogOption();
|
|
option2.title = "Who are you?";
|
|
option2.dialogId = 3;
|
|
option2.optionType = EnumOptionType.DialogOption;
|
|
|
|
DialogOption option3 = new DialogOption();
|
|
option3.title = "Goodbye";
|
|
option3.optionType = EnumOptionType.QuitOption;
|
|
|
|
dia1.options.put(0, option2);
|
|
dia1.options.put(1, option);
|
|
dia1.options.put(2, option3);
|
|
|
|
|
|
DialogOption option4 = new DialogOption();
|
|
option4.title = "Back";
|
|
option4.dialogId = 1;
|
|
|
|
dia2.options.put(1, option4);
|
|
dia3.options.put(1, option4);
|
|
lastUsedDialogID = 3;
|
|
saveCategory(cat);
|
|
saveDialog(cat.id, dia1);
|
|
saveDialog(cat.id, dia2);
|
|
saveDialog(cat.id, dia3);
|
|
}
|
|
|
|
public void saveCategory(DialogCategory category){
|
|
category.title = NoppesStringUtils.cleanFileName(category.title);
|
|
if(categories.containsKey(category.id)){
|
|
DialogCategory currentCategory = categories.get(category.id);
|
|
if(!currentCategory.title.equals(category.title)){
|
|
while(containsCategoryName(category.title))
|
|
category.title += "_";
|
|
File newdir = new File(getDir(), category.title);
|
|
File olddir = new File(getDir(), currentCategory.title);
|
|
if(newdir.exists())
|
|
return;
|
|
if(!olddir.renameTo(newdir))
|
|
return;
|
|
}
|
|
category.dialogs = currentCategory.dialogs;
|
|
}
|
|
else{
|
|
if(category.id < 0){
|
|
lastUsedCatID++;
|
|
category.id = lastUsedCatID;
|
|
}
|
|
while(containsCategoryName(category.title))
|
|
category.title += "_";
|
|
File dir = new File(getDir(), category.title);
|
|
if(!dir.exists())
|
|
dir.mkdirs();
|
|
}
|
|
categories.put(category.id, category);
|
|
Server.sendToAll(EnumPacketClient.SYNC_UPDATE, SyncType.DIALOG_CATEGORY, category.writeNBT(new NBTTagCompound()));
|
|
}
|
|
|
|
public void removeCategory(int category){
|
|
DialogCategory cat = categories.get(category);
|
|
if(cat == null)
|
|
return;
|
|
File dir = new File(getDir(), cat.title);
|
|
if(!dir.delete())
|
|
return;
|
|
for(int dia : cat.dialogs.keySet())
|
|
dialogs.remove(dia);
|
|
categories.remove(category);
|
|
Server.sendToAll(EnumPacketClient.SYNC_REMOVE, SyncType.DIALOG_CATEGORY, category);
|
|
}
|
|
|
|
private boolean containsCategoryName(String name) {
|
|
name = name.toLowerCase();
|
|
for(DialogCategory cat : categories.values()){
|
|
if(cat.title.toLowerCase().equals(name))
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
private boolean containsDialogName(DialogCategory category, Dialog dialog) {
|
|
for(Dialog dia : category.dialogs.values()){
|
|
if(dia.id != dialog.id && dia.title.equalsIgnoreCase(dialog.title))
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
public Dialog saveDialog(int categoryId, Dialog dialog){
|
|
DialogCategory category = categories.get(categoryId);
|
|
if(category == null)
|
|
return dialog;
|
|
dialog.category = category;
|
|
|
|
while(containsDialogName(dialog.category, dialog)){
|
|
dialog.title = dialog.title + "_";
|
|
}
|
|
if(dialog.id < 0){
|
|
lastUsedDialogID++;
|
|
dialog.id = lastUsedDialogID;
|
|
}
|
|
|
|
dialogs.put(dialog.id, dialog);
|
|
category.dialogs.put(dialog.id, dialog);
|
|
|
|
File dir = new File(getDir(), category.title);
|
|
if(!dir.exists())
|
|
dir.mkdirs();
|
|
|
|
File file = new File(dir, dialog.id + ".json_new");
|
|
File file2 = new File(dir, dialog.id + ".json");
|
|
|
|
try {
|
|
NBTTagCompound compound = dialog.writeToNBT(new NBTTagCompound());
|
|
NBTJsonUtil.SaveFile(file, compound);
|
|
if(file2.exists())
|
|
file2.delete();
|
|
file.renameTo(file2);
|
|
Server.sendToAll(EnumPacketClient.SYNC_UPDATE, SyncType.DIALOG, compound, category.id);
|
|
} catch (Exception e) {
|
|
LogWriter.except(e);
|
|
}
|
|
return dialog;
|
|
}
|
|
|
|
public void removeDialog(Dialog dialog) {
|
|
DialogCategory category = dialog.category;
|
|
File file = new File(new File(getDir(), category.title), dialog.id + ".json");
|
|
if(!file.delete())
|
|
return;
|
|
category.dialogs.remove(dialog.id);
|
|
dialogs.remove(dialog.id);
|
|
Server.sendToAll(EnumPacketClient.SYNC_REMOVE, SyncType.DIALOG, dialog.id);
|
|
}
|
|
|
|
private File getDir(){
|
|
return new File(CustomNpcs.getWorldSaveDirectory(), "dialogs");
|
|
}
|
|
|
|
public boolean hasDialog(int dialogId) {
|
|
return dialogs.containsKey(dialogId);
|
|
}
|
|
|
|
public Map<String,Integer> getScroll() {
|
|
Map<String,Integer> map = new HashMap<String,Integer>();
|
|
for(DialogCategory category : categories.values()){
|
|
map.put(category.title, category.id);
|
|
}
|
|
return map;
|
|
}
|
|
|
|
public List<IDialogCategory> categories() {
|
|
return new ArrayList(this.categories.values());
|
|
}
|
|
|
|
public IDialog get(int id) {
|
|
return (IDialog)this.dialogs.get(id);
|
|
}
|
|
}
|