diff --git a/VelocityCore/src/de/steamwar/velocitycore/discord/DiscordBot.java b/VelocityCore/src/de/steamwar/velocitycore/discord/DiscordBot.java
index 28cd0c22..53714c52 100644
--- a/VelocityCore/src/de/steamwar/velocitycore/discord/DiscordBot.java
+++ b/VelocityCore/src/de/steamwar/velocitycore/discord/DiscordBot.java
@@ -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);
diff --git a/VelocityCore/src/de/steamwar/velocitycore/discord/VacationCommand.java b/VelocityCore/src/de/steamwar/velocitycore/discord/VacationCommand.java
new file mode 100644
index 00000000..ace8dcf6
--- /dev/null
+++ b/VelocityCore/src/de/steamwar/velocitycore/discord/VacationCommand.java
@@ -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 .
+ */
+
+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 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();
+ }
+}