1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
package de.hysky.skyblocker.skyblock;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParser;
import com.mojang.brigadier.arguments.StringArgumentType;
import com.mojang.brigadier.tree.LiteralCommandNode;
import de.hysky.skyblocker.SkyblockerMod;
import de.hysky.skyblocker.annotations.Init;
import com.mojang.serialization.Codec;
import com.mojang.serialization.JsonOps;
import de.hysky.skyblocker.utils.CodecUtils;
import de.hysky.skyblocker.utils.Http;
import de.hysky.skyblocker.utils.Utils;
import it.unimi.dsi.fastutil.objects.Object2BooleanMap;
import it.unimi.dsi.fastutil.objects.Object2BooleanMaps;
import net.azureaaron.hmapi.data.rank.PackageRank;
import net.azureaaron.hmapi.data.rank.RankType;
import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource;
import net.minecraft.command.CommandSource;
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.NoSuchFileException;
import java.nio.file.Path;
import java.util.Optional;
import java.util.concurrent.CompletableFuture;
import java.util.stream.Stream;
import static net.fabricmc.fabric.api.client.command.v2.ClientCommandManager.argument;
import static net.fabricmc.fabric.api.client.command.v2.ClientCommandManager.literal;
/**
* the mixin {@link de.hysky.skyblocker.mixins.CommandTreeS2CPacketMixin}
*/
public class WarpAutocomplete {
private static final Path FILE = SkyblockerMod.CONFIG_DIR.resolve("warp_autocomplete.json");
private static final Logger LOGGER = LoggerFactory.getLogger(WarpAutocomplete.class);
private static final Codec<Object2BooleanMap<String>> MAP_CODEC = CodecUtils.object2BooleanMapCodec(Codec.STRING);
@Nullable
public static LiteralCommandNode<FabricClientCommandSource> commandNode;
@Init
public static void init() {
CompletableFuture.supplyAsync(() -> {
try {
String warps = Http.sendGetRequest("https://hysky.de/api/locations");
return MAP_CODEC.parse(JsonOps.INSTANCE, JsonParser.parseString(warps)).getOrThrow();
} catch (Exception e) {
LOGGER.error("[Skyblocker] Failed to download warps list", e);
}
return Object2BooleanMaps.<String>emptyMap();
}).thenAccept(warps -> {
if (warps.isEmpty()) {
getWarpsFromFile();
} else {
CompletableFuture.runAsync(() -> {
Optional<JsonElement> result = MAP_CODEC.encodeStart(JsonOps.INSTANCE, warps).result();
if (result.isEmpty()) return;
JsonElement warpsJson = result.get();
try (BufferedWriter writer = Files.newBufferedWriter(FILE, StandardCharsets.UTF_8)) {
SkyblockerMod.GSON.toJson(warpsJson, writer);
} catch (Exception e) {
LOGGER.error("[Skyblocker] Failed to save warps auto complete", e);
}
});
createCommandNode(warps);
}
}
);
}
private static void getWarpsFromFile() {
CompletableFuture.supplyAsync(() -> {
JsonObject object;
try (BufferedReader reader = Files.newBufferedReader(FILE)) {
object = SkyblockerMod.GSON.fromJson(reader, JsonObject.class);
} catch (NoSuchFileException e) {
return Object2BooleanMaps.<String>emptyMap();
} catch (Exception e) {
LOGGER.error("[Skyblocker] Failed to read warp autocomplete file", e);
return Object2BooleanMaps.<String>emptyMap();
}
return MAP_CODEC.parse(JsonOps.INSTANCE, object).result().orElse(Object2BooleanMaps.emptyMap());
}).thenAccept(WarpAutocomplete::createCommandNode);
}
private static void createCommandNode(Object2BooleanMap<String> warps) {
commandNode = literal("warp")
.requires(fabricClientCommandSource -> Utils.isOnSkyblock())
.then(argument("destination", StringArgumentType.string())
.suggests((context, builder) -> CommandSource.suggestMatching(getEligibleWarps(warps), builder))
).build();
}
private static Stream<String> getEligibleWarps(Object2BooleanMap<String> warps) {
return warps.object2BooleanEntrySet().stream()
.filter(WarpAutocomplete::shouldShowWarp)
.map(Object2BooleanMap.Entry::getKey);
}
private static boolean shouldShowWarp(Object2BooleanMap.Entry<String> entry) {
return !entry.getBooleanValue() || Utils.isOnBingo() || RankType.compare(Utils.getRank(), PackageRank.MVP_PLUS) >= 0;
}
}
|
