Handle case where input duration overflows

Incendo · Sep 15, 2024 · d3911ac · d3911ac
1 parent c210a5c
commit d3911ac
Showing 1 changed file with 19 additions and 15 deletions.
diff --git a/cloud-core/src/main/java/org/incendo/cloud/parser/standard/DurationParser.java b/cloud-core/src/main/java/org/incendo/cloud/parser/standard/DurationParser.java
@@ -106,21 +106,25 @@ public final class DurationParser<C> implements ArgumentParser<C, Duration>, Blo
             }
 
             final char timeUnit = input.charAt(cursor);
-            switch (timeUnit) {
-                case 'd':
-                    duration = duration.plusDays(timeValue);
-                    break;
-                case 'h':
-                    duration = duration.plusHours(timeValue);
-                    break;
-                case 'm':
-                    duration = duration.plusMinutes(timeValue);
-                    break;
-                case 's':
-                    duration = duration.plusSeconds(timeValue);
-                    break;
-                default:
-                    return ArgumentParseResult.failure(new DurationParseException(input, commandContext));
+            try {
+                switch (timeUnit) {
+                    case 'd':
+                        duration = duration.plusDays(timeValue);
+                        break;
+                    case 'h':
+                        duration = duration.plusHours(timeValue);
+                        break;
+                    case 'm':
+                        duration = duration.plusMinutes(timeValue);
+                        break;
+                    case 's':
+                        duration = duration.plusSeconds(timeValue);
+                        break;
+                    default:
+                        return ArgumentParseResult.failure(new DurationParseException(input, commandContext));
+                }
+            } catch (final ArithmeticException ex) {
+                return ArgumentParseResult.failure(new DurationParseException(ex, input, commandContext));
             }
 
             // skip unit, reset rangeStart to start of next segment