Be cleverer about argument parsing.

davorg · Mar 24, 2017 · ccf2744 · ccf2744
1 parent f4cf411
commit ccf2744
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/Lotto.pm b/Lotto.pm
@@ -44,7 +44,11 @@ sub parse_args {
   my @errs;
 
   if (@_ == 2) {
-    $type = shift;
+    ($type, $count) = (shift, shift);
+    ($type, $count) = ($count, $type) if $count =~ /\D/;
+    if ($count =~ /\D/) {
+      push @errs, "$count doesn't look like a positive integer";
+    }
     if (! exists $config->{$type}) {
       push @errs, qq["$type" is not a recognised type of lottery];
     }