Merge pull request #2 from evgeni/getopt_colon_spec

GetOpt::Long optional arguments using a colon instead of an equal sign

2 files changed, 43 insertions, 5 deletions

diff --git a/lib/Monitoring/Plugin/Getopt.pm b/lib/Monitoring/Plugin/Getopt.pm
index 354f713..db98567 100644
--- a/lib/Monitoring/Plugin/Getopt.pm
+++ b/lib/Monitoring/Plugin/Getopt.pm
@@ -81,7 +81,8 @@ sub _spec_to_help
 {
   my ($self, $spec, $label) = @_;
 
-  my ($opts, $type) = split /=/, $spec, 2;
+  my ($opts, $type) = split /=|:/, $spec, 2;
+  my $optional = ($spec =~ m/:/);
   my (@short, @long);
   for (split /\|/, $opts) {
     if (length $_ == 1) {
@@ -93,11 +94,20 @@ sub _spec_to_help
 
   my $help = join(', ', @short, @long);
   if ($type) {
-    if ($label) {
-      $help .= '=' . $label;
+    if (!$label) {
+      if ($type eq 'i' || $type eq '+' || $type =~ /\d+/) {
+        $label = 'INTEGER';
+      }
+      else {
+        $label = 'STRING';
+      }
+    }
+
+    if ($optional) {
+      $help .= '[=' . $label . ']';
     }
     else {
-      $help .= $type eq 'i' ? '=INTEGER' : '=STRING';
+      $help .= '=' . $label;
     }
   }
   elsif ($label) {
diff --git a/t/Monitoring-Plugin-Getopt-04.t b/t/Monitoring-Plugin-Getopt-04.t
index b6345d0..2c91e23 100644
--- a/t/Monitoring-Plugin-Getopt-04.t
+++ b/t/Monitoring-Plugin-Getopt-04.t
@@ -2,7 +2,7 @@
 
 use strict;
 
-use Test::More tests => 11;
+use Test::More tests => 15;
 BEGIN { use_ok('Monitoring::Plugin::Getopt') };
 
 # Needed to get evals to work in testing
@@ -78,6 +78,30 @@ sub setup
     [ undef, 'PERCENT%' ],
   );
 
+  # Named args with *optional* but pre-set value
+  $ng->arg(
+    spec => 'dirport|d:9030',
+    help => 'dirport',
+  );
+
+  # Named args with *optional* string value
+  $ng->arg(
+    spec => 'enablesomething|s:s',
+    help => 'something',
+  );
+
+  # Named args with *optional* integer value (same as ":0")
+  $ng->arg(
+    spec => 'testtimeout|T:i',
+    help => 'testtimeout',
+  );
+
+  # Named args with *optional* but increasing integer value
+  $ng->arg(
+    spec => 'verbosity|v:+',
+    help => 'verbosity',
+  );
+
   return $ng;
 }
 
@@ -94,4 +118,8 @@ like($@, qr/\n -H, --hostname=ADDRESS\n   Hostname\n/, 'hostname ok');
 like($@, qr/\n --avatar=AVATAR\n   Avatar\n/, 'avatar ok');
 like($@, qr/\n --disk=BYTES\n   Disk limit in BYTES\n --disk=PERCENT%\n   Disk limit in PERCENT\n --disk=STRING\n   Disk limit in FOOBARS \(Default: 1024\)\n/, 'disk multiline ok');
 like($@, qr/\n --limit=STRING\n   Limit in BYTES\n --limit=PERCENT%\n   Limit in PERCENT\n/, 'limit multiline ok');
+like($@, qr/\n -d, --dirport\[=INTEGER\]/, 'dirport ok');
+like($@, qr/\n -s, --enablesomething\[=STRING\]/, 'enablesomething ok');
+like($@, qr/\n -T, --testtimeout\[=INTEGER\]/, 'testtimeout ok');
+like($@, qr/\n -v, --verbosity\[=INTEGER\]/, 'verbosity ok');
 #print $@;