Skip to content

Commit

Permalink
clean
Browse files Browse the repository at this point in the history
  • Loading branch information
@TexBlock
TexBlock committed Sep 13, 2023
1 parent 446741c commit d188a95
Show file tree
Hide file tree
Showing 3 changed files with 189 additions and 198 deletions.
192 changes: 188 additions & 4 deletions common/src/main/java/org/localmc/tools/ftbqkeys/FTBQKeysMod.java
View file
Original file line number Diff line number Diff line change
@@ -1,18 +1,36 @@
package org.localmc.tools.ftbqkeys;

import com.google.common.collect.Lists;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.mojang.brigadier.arguments.StringArgumentType;
import com.mojang.brigadier.tree.ArgumentCommandNode;
import com.mojang.brigadier.tree.LiteralCommandNode;
import com.mojang.brigadier.tree.RootCommandNode;
import dev.ftb.mods.ftbquests.FTBQuests;
import dev.ftb.mods.ftbquests.quest.*;
import dev.ftb.mods.ftbquests.quest.loot.RewardTable;
import dev.ftb.mods.ftbquests.quest.reward.Reward;
import dev.ftb.mods.ftbquests.quest.task.Task;
import me.shedaniel.architectury.event.events.CommandRegistrationEvent;
import me.shedaniel.architectury.platform.Platform;
import net.minecraft.Util;
import net.minecraft.client.Minecraft;
import net.minecraft.client.resources.language.LanguageInfo;
import net.minecraft.commands.CommandSourceStack;
import net.minecraft.commands.Commands;
import net.minecraft.commands.SharedSuggestionProvider;
import net.minecraft.network.chat.TranslatableComponent;
import org.apache.commons.io.FileUtils;
import org.localmc.tools.ftbqkeys.command.FTBQKeysCommands;

import java.io.File;
import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;
import java.util.Locale;
import java.util.TreeMap;
import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

public class FTBQKeysMod {
public static final String MODID = "ftbqkeys";
Expand All @@ -26,6 +44,172 @@ public static void saveLang(TreeMap<String, String> transKeys, String lang, File
}

public static void init() {
CommandRegistrationEvent.EVENT.register(FTBQKeysCommands::registerCommands);
CommandRegistrationEvent.EVENT.register((dispatcher, selection) -> {
RootCommandNode<CommandSourceStack> rootCommandNode = dispatcher.getRoot();
LiteralCommandNode<CommandSourceStack> commandNode = Commands.literal("ftbqkey").executes(context -> 0).build();

ArgumentCommandNode<CommandSourceStack, String> argumentCommandNode = Commands.argument("lang", StringArgumentType.word()).suggests((commandContext, suggestionsBuilder) -> {
List<String> list = new ArrayList<>();
for (LanguageInfo languageInfo : Minecraft.getInstance().getLanguageManager().getLanguages()) {
String code = languageInfo.getCode();
list.add(code);
}
return SharedSuggestionProvider.suggest(list.toArray(new String[0]), suggestionsBuilder);
}).executes(context -> {
try {
File parent = new File(FTBQKeysMod.gameDir.toFile(), "ftbqkeys");
File transFiles = new File(parent, "export-lang/");
File questsFolder = new File(FTBQKeysMod.configDir.toFile(), "ftbquests");

if (questsFolder.exists()) {
File backup = new File(parent, "backup/ftbquests");
FileUtils.copyDirectory(questsFolder, backup);
}

TreeMap<String, String> transKeys = new TreeMap<>();
QuestFile file = FTBQuests.PROXY.getQuestFile(false);

for (int i = 0; i < file.rewardTables.size(); i++) {
RewardTable table = file.rewardTables.get(i);

transKeys.put("loot_table." + (i + 1), table.title);
table.title = "{" + "loot_table." + (i + 1) + "}";
}

for (int i = 0; i < file.chapterGroups.size(); i++) {
ChapterGroup chapterGroup = file.chapterGroups.get(i);

if (!chapterGroup.title.isEmpty()) {
transKeys.put("category." + (i + 1), chapterGroup.title);
chapterGroup.title = "{" + "category." + (i + 1) + "}";
}
}

for (int i = 0; i < file.getAllChapters().size(); i++) {
Chapter chapter = file.getAllChapters().get(i);

String prefix = "chapter." + (i + 1);

if (!chapter.title.isEmpty()) {
transKeys.put(prefix + ".title", chapter.title);
chapter.title = "{" + prefix + ".title" + "}";
}

if (chapter.subtitle.size() > 0) {
transKeys.put(prefix + ".subtitle", String.join("\n", chapter.subtitle));
chapter.subtitle.clear();
chapter.subtitle.add("{" + prefix + ".subtitle" + "}");
}


for (int i1 = 0; i1 < chapter.images.size(); i1++) {
ChapterImage chapterImage = chapter.images.get(i1);

if (!chapterImage.hover.isEmpty()) {
transKeys.put(prefix + ".image." + (i1 + 1), String.join("\n", chapterImage.hover));
chapterImage.hover.clear();
chapterImage.hover.add("{" + prefix + ".image." + (i1 + 1) + "}");
}
}

for (int i1 = 0; i1 < chapter.quests.size(); i1++) {
Quest quest = chapter.quests.get(i1);

if (!quest.title.isEmpty()) {
transKeys.put(prefix + ".quest." + (i1 + 1) + ".title", quest.title);
quest.title = "{" + prefix + ".quest." + (i1 + 1) + ".title}";
}

if (!quest.subtitle.isEmpty()) {
transKeys.put(prefix + ".quest." + (i1 + 1) + ".subtitle", quest.subtitle);
quest.subtitle = "{" + prefix + ".quest." + (i1 + 1) + ".subtitle" + "}";
}

if (quest.description.size() > 0) {
List<String> descList = Lists.newArrayList();

StringJoiner joiner = new StringJoiner("\n");
int num = 1;

for (int i2 = 0; i2 < quest.description.size(); i2++) {
String desc = quest.description.get(i2);

final String regex = "\\{image:.*?}";

if (desc.contains("{image:")) {
if (!joiner.toString().isEmpty()) {
transKeys.put(prefix + ".quest." + (i1 + 1) + ".description." + num, joiner.toString());
descList.add("{" + prefix + ".quest." + (i1 + 1) + ".description." + num + "}");
joiner = new StringJoiner("\n");
num++;
}

final Pattern pattern = Pattern.compile(regex, Pattern.MULTILINE);
final Matcher matcher = pattern.matcher(desc);

while (matcher.find()) {
desc = desc.replace(matcher.group(0), "");
descList.add(matcher.group(0));
}
} else {
if (desc.isEmpty()) {
joiner.add("\n");
} else {
joiner.add(desc);
}
}
}

if (!joiner.toString().isEmpty()) {
transKeys.put(prefix + ".quest." + (i1 + 1) + ".description." + num, joiner.toString());
descList.add("{" + prefix + ".quest." + (i1 + 1) + ".description." + num + "}");
}

quest.description.clear();
quest.description.addAll(descList);
}

for (int i2 = 0; i2 < quest.tasks.size(); i2++) {
Task task = quest.tasks.get(i2);

if (!task.title.isEmpty()) {
transKeys.put(prefix + ".quest." + (i1 + 1) + ".task." + (i2 + 1) + ".title", task.title);
task.title = "{" + prefix + ".quest." + (i1 + 1) + ".task." + (i2 + 1) + ".title}";
}
}

for (int i2 = 0; i2 < quest.rewards.size(); i2++) {
Reward reward = quest.rewards.get(i2);

if (!reward.title.isEmpty()) {
transKeys.put(prefix + ".quest." + (i1 + 1) + ".reward." + (i2 + 1) + ".title", reward.title);
reward.title = "{" + prefix + ".quest." + (i1 + 1) + ".reward." + (i2 + 1) + ".title}";
}
}
}
}

File output = new File(parent, "config/ftbquests");

file.writeDataFull(output.toPath());

String lang = context.getArgument("lang", String.class);
FTBQKeysMod.saveLang(transKeys, lang, transFiles);

if (!lang.equalsIgnoreCase("en_us")) {
FTBQKeysMod.saveLang(transKeys, "en_us", transFiles);
}

context.getSource().getPlayerOrException().sendMessage(new TranslatableComponent("command.ftbqkeys.message" + parent.getAbsolutePath()), Util.NIL_UUID);

} catch (Exception e) {
e.printStackTrace();
}
return 1;
}).build();

rootCommandNode.addChild(commandNode);
commandNode.addChild(argumentCommandNode);
});
}
}
Loading

0 comments on commit d188a95

Please sign in to comment.