All TC objects (except unions) now have inlining code, and tests for all the variatio...

[gitmo/Moose.git] / lib / Moose / Meta / TypeConstraint / Enum.pm

diff --git a/lib/Moose/Meta/TypeConstraint/Enum.pm b/lib/Moose/Meta/TypeConstraint/Enum.pm
index 54b5581..2a7beb7 100644 (file)
--- a/lib/Moose/Meta/TypeConstraint/Enum.pm
+++ b/lib/Moose/Meta/TypeConstraint/Enum.pm
@@ -4,6 +4,7 @@ use strict;
 use warnings;
 use metaclass;
 
+use B;
 use Moose::Util::TypeConstraints ();
 
 use base 'Moose::Meta::TypeConstraint';
@@ -12,10 +13,28 @@ __PACKAGE__->meta->add_attribute('values' => (
     accessor => 'values',
 ));
 
+our %ENUMS;
+
+my $inliner = sub {
+    my $self = shift;
+    my $val  = shift;
+
+    my $name = $self->name();
+    $ENUMS{$name} ||= { map { $_ => 1 } @{ $self->values() } };
+
+    return
+          "defined $val && " . '$'
+        . __PACKAGE__
+        . '::ENUMS{'
+        . B::perlstring($name)
+        . "}{ $val }";
+};
+
 sub new {
     my ( $class, %args ) = @_;
 
     $args{parent} = Moose::Util::TypeConstraints::find_type_constraint('Str');
+    $args{inlined} = $inliner;
 
     if ( scalar @{ $args{values} } < 2 ) {
         require Moose;
@@ -33,6 +52,9 @@ sub new {
         }
     }
 
+    my %values = map { $_ => 1 } @{ $args{values} };
+    $args{constraint} = sub { $values{ $_[0] } };
+
     my $self = $class->_new(\%args);
 
     $self->compile_type_constraint()