Ownercz/MGPT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Small code improvements

Browse Source
This commit is contained in:
ohAlee 2023-03-08 16:58:57 +01:00
parent 27ef6d5b2d
commit 19432ca0fe
3 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/it/ohalee/minecraftgpt/Main.java
View File

@ -13,28 +13,30 @@ import java.util.concurrent.TimeUnit;
public class Main extends JavaPlugin { public class Main extends JavaPlugin {
public static Cache<Player, StringBuilder> CACHE; public static Cache<Player, StringBuilder> CACHE;
public static Cache<Player, Type> USER_TYPE; public static Cache<Player, Type> USER_TYPE = CacheBuilder.newBuilder().build();
@Override @Override
public void onEnable() { public void onEnable() {
saveDefaultConfig(); saveDefaultConfig();
OpenAI.init(getConfig().getString("API_KEY")); OpenAI.init(getConfig().getString("API_KEY")).exceptionallyAsync(throwable -> {
getLogger().severe("Error while initializing OpenAI service! Is your API key valid?");
throwable.printStackTrace();
return null;
});
CACHE = CacheBuilder.newBuilder() CACHE = CacheBuilder.newBuilder()
.expireAfterWrite(30, TimeUnit.MINUTES) .expireAfterWrite(30, TimeUnit.MINUTES)
.removalListener((RemovalListener<Player, StringBuilder>) notification -> { .removalListener((RemovalListener<Player, StringBuilder>) notification -> {
if (notification.getKey() == null) { if (notification.getKey() == null) return;
return;
}
USER_TYPE.invalidate(notification.getKey()); USER_TYPE.invalidate(notification.getKey());
if (notification.getCause() == RemovalCause.EXPIRED) { if (notification.getCause() == RemovalCause.EXPIRED) {
notification.getKey().sendMessage(Messages.format(getConfig().getString("command.toggle.disabled"))); notification.getKey().sendMessage(Messages.format(getConfig().getString("command.toggle.disabled")));
} }
}).build(); }).build();
USER_TYPE = CacheBuilder.newBuilder().build();
getServer().getPluginManager().registerEvents(new PlayerHandlers(this), this); getServer().getPluginManager().registerEvents(new PlayerHandlers(this), this);
ChatCommand command = new ChatCommand(this); ChatCommand command = new ChatCommand(this);
PluginCommand chatgpt = getCommand("chatgpt"); PluginCommand chatgpt = getCommand("chatgpt");
chatgpt.setExecutor(command); chatgpt.setExecutor(command);

4
src/main/java/it/ohalee/minecraftgpt/OpenAI.java
View File

@ -11,8 +11,8 @@ public class OpenAI {
private static OpenAiService service; private static OpenAiService service;
public static void init(String key) { public static CompletableFuture<Void> init(String key) {
service = new OpenAiService(key, 0); return CompletableFuture.runAsync(() -> service = new OpenAiService(key, 5));
} }
public static CompletableFuture<String> getResponse(ConfigurationSection section, StringBuilder cached, String message) { public static CompletableFuture<String> getResponse(ConfigurationSection section, StringBuilder cached, String message) {

2
src/main/java/it/ohalee/minecraftgpt/handler/PlayerHandlers.java
View File

@ -37,7 +37,7 @@ public class PlayerHandlers implements Listener {
return; return;
} }
Collection<Player> recipients = switch (Main.USER_TYPE.asMap().getOrDefault(player, hasFull ? Type.FULL : Type.SINGLE) ) { Collection<Player> recipients = switch (Main.USER_TYPE.asMap().getOrDefault(player, hasFull ? Type.FULL : Type.SINGLE)) {
case SINGLE -> Collections.singletonList(player); case SINGLE -> Collections.singletonList(player);
case FULL, BROADCAST -> e.getRecipients(); case FULL, BROADCAST -> e.getRecipients();
}; };