I hate regex

This commit is contained in:
floppydiskette 2024-11-19 02:07:20 +00:00
parent e1d452d5f8
commit cae774a82f
Signed by: fwoppydwisk
SSH key fingerprint: SHA256:Hqn452XQ1ETzUt/FthJu6+OFkS4NBxCv5VQSEvuk7CE
3 changed files with 50 additions and 16 deletions

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

@ -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());
}
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");
};
// 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());
}
}
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";