Compare commits

...

4 commits

Author SHA1 Message Date
3ef88bde40
Make polls require manage message perms 2024-11-19 20:27:09 +00:00
cae774a82f
I hate regex 2024-11-19 02:09:13 +00:00
e1d452d5f8 Update README.md 2024-11-19 01:35:31 +00:00
e70ffe2c0e Update README.md 2024-11-19 01:30:16 +00:00
4 changed files with 56 additions and 17 deletions

View file

@ -1,5 +1,10 @@
<img src="https://git.frzn.dev/fwoppydwisk/doki/raw/branch/master/assets/logo.svg" alt="" height="100"/>
<hr>
<a href="https://git.frzn.dev/fwoppydwisk/doki/releases/latest">
<img src="https://git.frzn.dev/fwoppydwisk/doki/badges/release.svg?style=for-the-badge" alt="">
</a>
<img src="https://git.frzn.dev/fwoppydwisk/doki/badges/stars.svg?style=for-the-badge" alt="">
<br>
A multipurpose Discord bot written in Java.
## Tested Environments

View file

@ -27,6 +27,10 @@ public class Timeout extends ApplicationCommand {
@AppOption(name = "member") Member member,
@AppOption(name = "duration", description = "ex: 2h5m, must be between 1m and 7d") String duration) {
Duration timeoutDuration = DurationUtils.parseDuration(duration);
if (!DurationUtils.isDurationBetween(timeoutDuration, Duration.ofMinutes(1), Duration.ofDays(7))) {
event.replyFormat("Invalid duration %s!, must be between 1m and 7d", duration).queue();
return;
}
member.timeoutFor(timeoutDuration).queue();
event.replyFormat("Timed out %s for %s", member.getAsMention(), duration).queue();
}

View file

@ -18,7 +18,7 @@ import java.util.Objects;
@CommandMarker
@BotPermissions(Permission.MESSAGE_SEND_POLLS)
@UserPermissions(Permission.MESSAGE_SEND_POLLS)
@UserPermissions(Permission.MESSAGE_MANAGE)
public class Poll extends ApplicationCommand {
@JDASlashCommand(
name = "poll",
@ -26,11 +26,15 @@ public class Poll extends ApplicationCommand {
)
public void poll(GuildSlashEvent event,
@AppOption(name = "title") String pollTitle,
@AppOption(name = "duration", description = "ex: 2h5m, must be between 1h and 7d") String duration,
@AppOption(name = "duration", description = "ex: 2h, must be between 1h and 7d") String duration,
@AppOption(name = "options", description = "Comma-seperated poll options") String options,
@Optional @AppOption(name = "multiple-choice", description = "Allow multiple choices? (defaults to false)") Boolean multiChoice) {
Boolean pollMultiChoice = Objects.requireNonNullElse(multiChoice, false);
Duration pollDuration = DurationUtils.parseDuration(duration);
if (!DurationUtils.isDurationBetween(pollDuration, Duration.ofHours(1), Duration.ofDays(7))) {
event.replyFormat("Invalid duration %s!, must be between 1h and 7d", duration).queue();
return;
}
String[] pollOptions = options.split(",");
MessagePollBuilder poll = MessagePollData.builder(pollTitle)
.setDuration(pollDuration);

View file

@ -7,26 +7,52 @@ import java.util.regex.Matcher;
import java.util.regex.Pattern;
public class DurationUtils {
private static final Pattern timePattern = Pattern.compile("(\\d+)([dhms])?");
public static Duration parseDuration(String input) throws IllegalArgumentException {
Matcher matcher = timePattern.matcher(input);
private static final Pattern DURATION_PATTERN = Pattern.compile(
"(\\d+d\\s*)?(\\d+h\\s*)?(\\d+m\\s*)?(\\d+s\\s*)?");
if (!matcher.matches()) {
throw new IllegalArgumentException("Invalid duration format");
public static Duration parseDuration(String durationString) {
durationString = durationString.trim();
Matcher matcher = DURATION_PATTERN.matcher(durationString);
long days = 0;
long hours = 0;
long minutes = 0;
long seconds = 0;
if (matcher.matches()) {
// Extract days if present
if (matcher.group(1) != null) {
days = Long.parseLong(matcher.group(1).replace("d", "").trim());
}
// Extract hours if present
if (matcher.group(2) != null) {
hours = Long.parseLong(matcher.group(2).replace("h", "").trim());
}
// Extract minutes if present
if (matcher.group(3) != null) {
minutes = Long.parseLong(matcher.group(3).replace("m", "").trim());
}
// Extract seconds if present
if (matcher.group(4) != null) {
seconds = Long.parseLong(matcher.group(4).replace("s", "").trim());
}
}
int value = Integer.parseInt(matcher.group(1));
String unit = matcher.group(2);
return switch (unit == null || unit.isEmpty() ? "m" : unit.toLowerCase()) {
case "d" -> Duration.ofDays(value);
case "h" -> Duration.ofHours(value);
case "m" -> Duration.ofMinutes(value);
case "s" -> Duration.ofSeconds(value);
default -> throw new IllegalArgumentException("Invalid unit");
};
return Duration.ofDays(days)
.plusHours(hours)
.plusMinutes(minutes)
.plusSeconds(seconds);
}
public static boolean isDurationBetween(Duration target, Duration lowerThreshold, Duration upperThreshold) {
if (target == lowerThreshold || target == upperThreshold) return true;
return !target.isNegative() &&
(target.compareTo(lowerThreshold) >= 0) &&
(target.compareTo(upperThreshold) <= 0);
}
public static String getTimeDifference(OffsetDateTime offsetDateTime) {
if (offsetDateTime == null) return "0 seconds";