lib/MooseX/Getopt/OptionTypeMap.pm

   1 package MooseX::Getopt::OptionTypeMap;
   2 # ABSTRACT: Storage for the option to type mappings
   3
   4 use Moose 'confess', 'blessed';
   5 use Moose::Util::TypeConstraints 'find_type_constraint';
   6
   7 my %option_type_map = (
   8     'Bool'     => '!',
   9     'Str'      => '=s',
  10     'Int'      => '=i',
  11     'Num'      => '=f',
  12     'ArrayRef' => '=s@',
  13     'HashRef'  => '=s%',
  14 );
  15
  16 sub has_option_type {
  17     my (undef, $type_or_name) = @_;
  18
  19     return 1 if exists $option_type_map{blessed($type_or_name) ? $type_or_name->name : $type_or_name};
  20
  21     my $current = blessed($type_or_name) ? $type_or_name : find_type_constraint($type_or_name);
  22
  23     (defined $current)
  24         || confess "Could not find the type constraint for '$type_or_name'";
  25
  26     while (my $parent = $current->parent) {
  27         return 1 if exists $option_type_map{$parent->name};
  28         $current = $parent;
  29     }
  30
  31     return 0;
  32 }
  33
  34 sub get_option_type {
  35     my (undef, $type_or_name) = @_;
  36
  37     my $name = blessed($type_or_name) ? $type_or_name->name : $type_or_name;
  38
  39     return $option_type_map{$name} if exists $option_type_map{$name};
  40
  41     my $current = ref $type_or_name ? $type_or_name : find_type_constraint($type_or_name);
  42
  43     (defined $current)
  44         || confess "Could not find the type constraint for '$type_or_name'";
  45
  46     while ( $current = $current->parent ) {
  47         return $option_type_map{$current->name}
  48             if exists $option_type_map{$current->name};
  49     }
  50
  51     return;
  52 }
  53
  54 sub add_option_type_to_map {
  55     my (undef, $type_name, $option_string) = @_;
  56     (defined $type_name && defined $option_string)
  57         || confess "You must supply both a type name and an option string";
  58
  59     if ( blessed($type_name) ) {
  60         $type_name = $type_name->name;
  61     } else {
  62         (find_type_constraint($type_name))
  63             || confess "The type constraint '$type_name' does not exist";
  64     }
  65
  66     $option_type_map{$type_name} = $option_string;
  67 }
  68
  69 no Moose::Util::TypeConstraints;
  70 no Moose;
  71
  72 1;
  73
  74 =head1 DESCRIPTION
  75
  76 See the I<Custom Type Constraints> section in the L<MooseX::Getopt> docs
  77 for more info about how to use this module.
  78
  79 =method B<has_option_type ($type_or_name)>
  80
  81 =method B<get_option_type ($type_or_name)>
  82
  83 =method B<add_option_type_to_map ($type_name, $option_spec)>
  84
  85 =cut