Jaki/SteamWar
1
0
Fork 0
forked from SteamWar/SteamWar
Code Pull Requests Activity

Add VacationCommand

Browse Source
This commit is contained in:
YoyoNow
2025-08-01 10:43:37 +02:00
parent 9279d9cd8e
commit 6623e9d808
2 changed files with 197 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
VelocityCore/src/de/steamwar/velocitycore/discord/DiscordBot.java
+8 -2
View File
@@ -54,8 +54,8 @@ import net.dv8tion.jda.api.utils.MemberCachePolicy;
import net.dv8tion.jda.api.utils.messages.MessageCreateBuilder;
import java.awt.*;
import java.util.List;
import java.util.*;
import java.util.List;
import java.util.concurrent.TimeUnit;
import java.util.function.Consumer;
@@ -193,14 +193,20 @@ public class DiscordBot {
VelocityCore.schedule(CouncilChannel::updateAll).repeat(1, TimeUnit.HOURS).schedule();
VacationCommand vacationCommand = new VacationCommand();
jda.addEventListener(
new DiscordTicketHandler(),
new DiscordTeamEvent(),
new ChannelListener(),
new DiscordSchemUpload()
new DiscordSchemUpload(),
vacationCommand
);
commandSetup(jda.retrieveCommands().complete(), jda.updateCommands());
jda.getGuildById(1241489896909180998L)
.upsertCommand(vacationCommand.COMMAND)
.queue();
}
private final OptionData commandArgument = new OptionData(OptionType.STRING, ARGUMENT_NAME, "Command arguments", false);
VelocityCore/src/de/steamwar/velocitycore/discord/VacationCommand.java
+189
View File
@@ -0,0 +1,189 @@
/*
* This file is a part of the SteamWar software.
*
* Copyright (C) 2020 SteamWar.de-Serverteam
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package de.steamwar.velocitycore.discord;
import it.unimi.dsi.fastutil.Pair;
import net.dv8tion.jda.api.entities.Guild;
import net.dv8tion.jda.api.entities.ScheduledEvent;
import net.dv8tion.jda.api.events.interaction.command.CommandAutoCompleteInteractionEvent;
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
import net.dv8tion.jda.api.hooks.ListenerAdapter;
import net.dv8tion.jda.api.interactions.InteractionHook;
import net.dv8tion.jda.api.interactions.commands.Command;
import net.dv8tion.jda.api.interactions.commands.OptionType;
import net.dv8tion.jda.api.interactions.commands.build.OptionData;
import net.dv8tion.jda.api.interactions.commands.build.SubcommandData;
import net.dv8tion.jda.internal.interactions.CommandDataImpl;
import org.jetbrains.annotations.NotNull;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.OffsetDateTime;
import java.time.ZoneId;
import java.time.format.DateTimeFormatter;
import java.time.format.DateTimeParseException;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
public class VacationCommand extends ListenerAdapter {
private final Guild guild = Objects.requireNonNull(DiscordBot.getInstance().getJda().getGuildById(1241489896909180998L));
public final CommandDataImpl COMMAND = new CommandDataImpl("vacation", "Verwalte deinen Urlaub");
public VacationCommand() {
COMMAND.addSubcommands(new SubcommandData("create", "Erstelle deinen Urlaub")
.addOptions(new OptionData(OptionType.STRING, "from", "Datum (TT.MM.JJJJ)", true),
new OptionData(OptionType.STRING, "to", "Datum (TT.MM.JJJJ)", true)));
COMMAND.addSubcommands(new SubcommandData("delete", "Lösche deinen Urlaub")
.addOptions(new OptionData(OptionType.INTEGER, "vacation", "Dein Urlaub", true, true)));
}
@Override
public void onSlashCommandInteraction(@NotNull SlashCommandInteractionEvent event) {
if (!event.getName().equals("vacation")) return;
switch (event.getSubcommandName()) {
case "create":
createVacation(event);
break;
case "delete":
deleteVacation(event);
break;
default:
break;
}
}
public static final DateTimeFormatter PARSER = DateTimeFormatter.ofPattern("dd.MM.uuuu");
private void createVacation(SlashCommandInteractionEvent event) {
InteractionHook interactionHook = event.deferReply(true).complete();
String from = event.getOption("from").getAsString();
String to = event.getOption("to").getAsString();
LocalDateTime fromDate;
try {
fromDate = LocalDate.parse(from, PARSER).atStartOfDay();
} catch (DateTimeParseException e) {
interactionHook.editOriginal("Das Datumsformat ist falsch! Bitte verwenden Sie TT.MM.JJJJ für den ersten Tag deines Urlaubs").queue();
return;
}
if (fromDate == null) {
interactionHook.editOriginal("Das Datumsformat ist falsch! Bitte verwenden Sie TT.MM.JJJJ für den ersten Tag deines Urlaubs").queue();
return;
}
if (!fromDate.isAfter(LocalDateTime.now())) {
interactionHook.editOriginal("Bitte gib ein Datum in der Zukunft an!").queue();
return;
}
LocalDateTime toDate;
try {
toDate = LocalDate.parse(to, PARSER).atTime(23, 59, 59);
} catch (DateTimeParseException e) {
interactionHook.editOriginal("Das Datumsformat ist falsch! Bitte verwenden Sie TT.MM.JJJJ für den letzten Tag deines Urlaubs").queue();
return;
}
if (toDate == null) {
interactionHook.editOriginal("Das Datumsformat ist falsch! Bitte verwenden Sie TT.MM.JJJJ für den letzten Tag deines Urlaubs").queue();
return;
}
if (!toDate.isAfter(LocalDateTime.now())) {
interactionHook.editOriginal("Bitte gib ein Datum in der Zukunft an!").queue();
return;
}
if (!toDate.isAfter(fromDate)) {
interactionHook.editOriginal("Bitte gib ein Datum nach dem ersten Urlaubstag an!").queue();
return;
}
guild.createScheduledEvent(
"Urlaub " + event.getMember().getEffectiveName(),
event.getMember().getId(),
OffsetDateTime.of(fromDate, ZoneId.of("Europe/Berlin").getRules().getOffset(fromDate)),
OffsetDateTime.of(toDate, ZoneId.of("Europe/Berlin").getRules().getOffset(toDate))
).onSuccess(scheduledEvent -> {
interactionHook.editOriginal("Urlaub erstellt!").queue();
})
.onErrorMap(throwable -> {
interactionHook.editOriginal("Urlaub konnte nicht erstellt werden!").queue();
return null;
})
.queue();
}
private void deleteVacation(SlashCommandInteractionEvent event) {
InteractionHook interactionHook = event.deferReply(true).complete();
long eventId = event.getOption("vacation").getAsLong();
ScheduledEvent scheduledEvent = guild.getScheduledEventById(eventId);
if (scheduledEvent == null) {
interactionHook.editOriginal("Konnte den Urlaub nicht finden!").queue();
return;
}
scheduledEvent.delete()
.onSuccess(unused -> {
interactionHook.editOriginal("Urlaub gelöscht!").queue();
})
.onErrorMap(throwable -> {
interactionHook.editOriginal("Urlaub konnte nicht gelöscht werden!").queue();
return null;
})
.queue();
}
@Override
public void onCommandAutoCompleteInteraction(@NotNull CommandAutoCompleteInteractionEvent event) {
if (!event.getName().equals("vacation")) return;
switch (event.getFocusedOption().getName()) {
case "vacation":
listVacations(event);
break;
default:
break;
}
}
private void listVacations(CommandAutoCompleteInteractionEvent event) {
String vacation = event.getOption("vacation").getAsString();
List<Command.Choice> choices = guild.getScheduledEvents()
.stream()
.filter(scheduledEvent -> scheduledEvent.getLocation().equals(event.getMember().getId()))
.map(scheduledEvent -> {
StringBuilder st = new StringBuilder();
st.append(String.format("%02d", scheduledEvent.getStartTime().getDayOfMonth())).append(".");
st.append(String.format("%02d", scheduledEvent.getStartTime().getMonth())).append(".");
st.append(scheduledEvent.getStartTime().getYear());
st.append(" - ");
st.append(String.format("%02d", scheduledEvent.getEndTime().getDayOfMonth())).append(".");
st.append(String.format("%02d", scheduledEvent.getEndTime().getDayOfMonth())).append(".");
st.append(scheduledEvent.getEndTime().getYear());
return Pair.of(scheduledEvent, st.toString());
})
.filter(pair -> pair.right().startsWith(vacation))
.limit(25)
.map(pair -> {
return new Command.Choice(pair.right(), pair.left().getLocation());
})
.collect(Collectors.toList());
event.replyChoices(choices)
.queue();
}
}