mirror of
https://github.com/KAMKEEL/CustomNPC-Plus.git
synced 2026-11-14 00:14:09 +00:00
KAMKEEL
179fd4f834
The free-ID search only consulted the global map, so it could pick an id the same category still held further down the iteration - and renameTo silently overwrites on POSIX, so the rename then destroyed that entry's file. It now skips ids held by the category and any file already on disk. Quests had the same discard-on-collision behaviour and now get the same reassignment, including re-registering shared quests under the new id.
317 lines
12 KiB
Java
317 lines
12 KiB
Java
package noppes.npcs.controllers;
|
|
|
|
import kamkeel.npcs.controllers.SyncController;
|
|
import kamkeel.npcs.controllers.sync.handlers.QuestCategorySyncHandler;
|
|
import kamkeel.npcs.network.enums.SyncType;
|
|
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.api.handler.IQuestHandler;
|
|
import noppes.npcs.api.handler.data.IQuest;
|
|
import noppes.npcs.api.handler.data.IQuestCategory;
|
|
import noppes.npcs.constants.EnumProfileSync;
|
|
import noppes.npcs.controllers.data.Quest;
|
|
import noppes.npcs.controllers.data.QuestCategory;
|
|
import noppes.npcs.util.NBTJsonUtil;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
public class QuestController implements IQuestHandler {
|
|
public HashMap<Integer, QuestCategory> categoriesSync = new HashMap<Integer, QuestCategory>();
|
|
public HashMap<Integer, QuestCategory> categories = new HashMap<Integer, QuestCategory>();
|
|
public HashMap<Integer, Quest> quests = new HashMap<Integer, Quest>();
|
|
public HashMap<Integer, Quest> sharedQuests = new HashMap<>();
|
|
|
|
public static QuestController Instance = new QuestController();
|
|
;
|
|
|
|
private int lastUsedCatID = 0;
|
|
private int lastUsedQuestID = 0;
|
|
|
|
public QuestController() {
|
|
Instance = this;
|
|
}
|
|
|
|
public void load() {
|
|
categories.clear();
|
|
quests.clear();
|
|
|
|
lastUsedCatID = 0;
|
|
lastUsedQuestID = 0;
|
|
|
|
try {
|
|
File file = new File(CustomNpcs.getWorldSaveDirectory(), "quests.dat");
|
|
if (file.exists()) {
|
|
loadCategoriesOld(file);
|
|
file.delete();
|
|
file = new File(CustomNpcs.getWorldSaveDirectory(), "quests.dat_old");
|
|
if (file.exists())
|
|
file.delete();
|
|
return;
|
|
}
|
|
} catch (Exception e) {
|
|
|
|
}
|
|
|
|
File dir = getDir();
|
|
if (!dir.exists()) {
|
|
dir.mkdir();
|
|
} else {
|
|
for (File file : dir.listFiles()) {
|
|
if (!file.isDirectory())
|
|
continue;
|
|
QuestCategory category = loadCategoryDir(file);
|
|
List<Quest> reassigned = new ArrayList<Quest>();
|
|
Iterator<Integer> ite = category.quests.keySet().iterator();
|
|
while (ite.hasNext()) {
|
|
int id = ite.next();
|
|
if (id > lastUsedQuestID)
|
|
lastUsedQuestID = id;
|
|
Quest quest = category.quests.get(id);
|
|
if (quests.containsKey(id)) {
|
|
// Quest IDs live in the filename, not inside the JSON body,
|
|
// so a simple rename permanently resolves the conflict.
|
|
File catDir = new File(getDir(), category.title);
|
|
|
|
// Find the next free ID. It has to clear three things: the ids already
|
|
// registered globally, the ids this category still holds but has not
|
|
// reached yet, and any file already on disk - renameTo overwrites
|
|
// silently on POSIX, so landing on an existing file would destroy it.
|
|
int newId = lastUsedQuestID;
|
|
do {
|
|
newId++;
|
|
} while (quests.containsKey(newId)
|
|
|| category.quests.containsKey(newId)
|
|
|| new File(catDir, newId + ".json").exists());
|
|
lastUsedQuestID = newId;
|
|
|
|
File oldFile = new File(catDir, id + ".json");
|
|
File newFile = new File(catDir, newId + ".json");
|
|
boolean renamed = oldFile.renameTo(newFile);
|
|
|
|
quest.id = newId;
|
|
ite.remove();
|
|
reassigned.add(quest);
|
|
|
|
if (renamed) {
|
|
LogWriter.info("Quest ID conflict: reassigned \"" + quest.title
|
|
+ "\" from id " + id + " to " + newId
|
|
+ " in category \"" + category.title + "\"");
|
|
} else {
|
|
LogWriter.error("Quest ID conflict: could not rename file for \""
|
|
+ quest.title + "\" (id " + id
|
|
+ " in category \"" + category.title + "\") to " + newId);
|
|
}
|
|
} else {
|
|
quests.put(id, quest);
|
|
if (quest.profileOptions.enableOptions
|
|
&& (quest.profileOptions.completeControl == EnumProfileSync.Shared
|
|
|| quest.profileOptions.cooldownControl == EnumProfileSync.Shared)) {
|
|
sharedQuests.put(id, quest);
|
|
}
|
|
}
|
|
}
|
|
// Re-register reassigned quests under their new IDs.
|
|
for (Quest quest : reassigned) {
|
|
quests.put(quest.id, quest);
|
|
category.quests.put(quest.id, quest);
|
|
if (quest.profileOptions.enableOptions
|
|
&& (quest.profileOptions.completeControl == EnumProfileSync.Shared
|
|
|| quest.profileOptions.cooldownControl == EnumProfileSync.Shared)) {
|
|
sharedQuests.put(quest.id, quest);
|
|
}
|
|
}
|
|
lastUsedCatID++;
|
|
category.id = lastUsedCatID;
|
|
categories.put(category.id, category);
|
|
}
|
|
}
|
|
}
|
|
|
|
private QuestCategory loadCategoryDir(File dir) {
|
|
QuestCategory category = new QuestCategory();
|
|
category.title = dir.getName();
|
|
for (File file : dir.listFiles()) {
|
|
if (!file.isFile() || !file.getName().endsWith(".json"))
|
|
continue;
|
|
try {
|
|
Quest quest = new Quest();
|
|
quest.id = Integer.parseInt(file.getName().substring(0, file.getName().length() - 5));
|
|
quest.readNBTPartial(NBTJsonUtil.LoadFile(file));
|
|
category.quests.put(quest.id, quest);
|
|
quest.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;
|
|
try (FileInputStream fis = new FileInputStream(file)) {
|
|
nbttagcompound1 = CompressedStreamTools.readCompressed(fis);
|
|
}
|
|
lastUsedCatID = nbttagcompound1.getInteger("lastID");
|
|
lastUsedQuestID = nbttagcompound1.getInteger("lastQuestID");
|
|
NBTTagList list = nbttagcompound1.getTagList("Data", 10);
|
|
if (list != null) {
|
|
for (int i = 0; i < list.tagCount(); i++) {
|
|
QuestCategory category = new QuestCategory();
|
|
category.readNBT(list.getCompoundTagAt(i));
|
|
categories.put(category.id, category);
|
|
saveCategory(category);
|
|
Iterator<Map.Entry<Integer, Quest>> ita = category.quests.entrySet().iterator();
|
|
while (ita.hasNext()) {
|
|
Map.Entry<Integer, Quest> entry = ita.next();
|
|
Quest quest = entry.getValue();
|
|
quest.id = entry.getKey();
|
|
quest.category = category;
|
|
if (quests.containsKey(quest.id))
|
|
ita.remove();
|
|
else {
|
|
saveQuest(category.id, quest);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
public void removeCategory(int category) {
|
|
QuestCategory cat = categories.get(category);
|
|
if (cat == null)
|
|
return;
|
|
File dir = new File(getDir(), cat.title);
|
|
if (!dir.delete())
|
|
return;
|
|
for (int dia : cat.quests.keySet())
|
|
quests.remove(dia);
|
|
categories.remove(category);
|
|
SyncController.syncRemove(SyncType.QUEST_CATEGORY, category);
|
|
}
|
|
|
|
public void saveCategory(QuestCategory category) {
|
|
category.title = NoppesStringUtils.cleanFileName(category.title);
|
|
if (categories.containsKey(category.id)) {
|
|
QuestCategory 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.quests = currentCategory.quests;
|
|
} 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);
|
|
SyncController.syncUpdate(SyncType.QUEST_CATEGORY, -1, QuestCategorySyncHandler.serializeCategory(category));
|
|
}
|
|
|
|
private boolean containsCategoryName(String name) {
|
|
name = name.toLowerCase();
|
|
for (QuestCategory cat : categories.values()) {
|
|
if (cat.title.toLowerCase().equals(name))
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
private boolean containsQuestName(QuestCategory category, Quest quest) {
|
|
for (Quest q : category.quests.values()) {
|
|
if (q.id != quest.id && q.title.equalsIgnoreCase(quest.title))
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public void saveQuest(int categoryID, Quest quest) {
|
|
QuestCategory category = categories.get(categoryID);
|
|
if (category == null)
|
|
return;
|
|
quest.category = category;
|
|
|
|
while (containsQuestName(quest.category, quest)) {
|
|
quest.title = quest.title + "_";
|
|
}
|
|
|
|
if (quest.id < 0) {
|
|
lastUsedQuestID++;
|
|
quest.id = lastUsedQuestID;
|
|
}
|
|
|
|
quests.put(quest.id, quest);
|
|
category.quests.put(quest.id, quest);
|
|
|
|
sharedQuests.remove(quest.id);
|
|
if (quest.profileOptions.enableOptions
|
|
&& (quest.profileOptions.completeControl == EnumProfileSync.Shared
|
|
|| quest.profileOptions.cooldownControl == EnumProfileSync.Shared)) {
|
|
sharedQuests.put(quest.id, quest);
|
|
}
|
|
|
|
File dir = new File(getDir(), category.title);
|
|
if (!dir.exists())
|
|
dir.mkdirs();
|
|
|
|
File file = new File(dir, quest.id + ".json_new");
|
|
File file2 = new File(dir, quest.id + ".json");
|
|
|
|
try {
|
|
NBTJsonUtil.SaveFile(file, quest.writeToNBTPartial(new NBTTagCompound()));
|
|
if (file2.exists())
|
|
file2.delete();
|
|
file.renameTo(file2);
|
|
SyncController.syncUpdate(SyncType.QUEST, category.id, quest.writeToNBT(new NBTTagCompound()));
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public void removeQuest(Quest quest) {
|
|
File file = new File(new File(getDir(), quest.category.title), quest.id + ".json");
|
|
if (!file.delete())
|
|
return;
|
|
quests.remove(quest.id);
|
|
quest.category.quests.remove(quest.id);
|
|
}
|
|
|
|
private File getDir() {
|
|
return new File(CustomNpcs.getWorldSaveDirectory(), "quests");
|
|
}
|
|
|
|
public List<IQuestCategory> categories() {
|
|
return new ArrayList(this.categories.values());
|
|
}
|
|
|
|
public IQuest get(int id) {
|
|
return (IQuest) this.quests.get(id);
|
|
}
|
|
}
|