Add a dummy TypeConstraint namespace so you can treat them as objects
[gitmo/Mouse.git] / lib / Mouse / Util / TypeConstraints.pm
index 6410243..4379903 100644 (file)
@@ -30,40 +30,41 @@ sub via (&) {
     $_[0]
 }
 
-my $optimized_constraints;
-my $optimized_constraints_base;
-{
+BEGIN {
     no warnings 'uninitialized';
     %TYPE = (
         Any        => sub { 1 },
         Item       => sub { 1 },
         Bool       => sub {
-            !defined($_) || $_ eq "" || "$_" eq '1' || "$_" eq '0'
+            !defined($_[0]) || $_[0] eq "" || "$_[0]" eq '1' || "$_[0]" eq '0'
         },
-        Undef      => sub { !defined($_) },
-        Defined    => sub { defined($_) },
-        Value      => sub { defined($_) && !ref($_) },
-        Num        => sub { !ref($_) && looks_like_number($_) },
-        Int        => sub { defined($_) && !ref($_) && /^-?[0-9]+$/ },
-        Str        => sub { defined($_) && !ref($_) },
-        ClassName  => sub { Mouse::is_class_loaded($_) },
-        Ref        => sub { ref($_) },
-
-        ScalarRef  => sub { ref($_) eq 'SCALAR' },
-        ArrayRef   => sub { ref($_) eq 'ARRAY'  },
-        HashRef    => sub { ref($_) eq 'HASH'   },
-        CodeRef    => sub { ref($_) eq 'CODE'   },
-        RegexpRef  => sub { ref($_) eq 'Regexp' },
-        GlobRef    => sub { ref($_) eq 'GLOB'   },
+        Undef      => sub { !defined($_[0]) },
+        Defined    => sub { defined($_[0]) },
+        Value      => sub { defined($_[0]) && !ref($_[0]) },
+        Num        => sub { !ref($_[0]) && looks_like_number($_[0]) },
+        Int        => sub { defined($_[0]) && !ref($_[0]) && $_[0] =~ /^-?[0-9]+$/ },
+        Str        => sub { defined($_[0]) && !ref($_[0]) },
+        ClassName  => sub { Mouse::is_class_loaded($_[0]) },
+        Ref        => sub { ref($_[0]) },
+
+        ScalarRef  => sub { ref($_[0]) eq 'SCALAR' },
+        ArrayRef   => sub { ref($_[0]) eq 'ARRAY'  },
+        HashRef    => sub { ref($_[0]) eq 'HASH'   },
+        CodeRef    => sub { ref($_[0]) eq 'CODE'   },
+        RegexpRef  => sub { ref($_[0]) eq 'Regexp' },
+        GlobRef    => sub { ref($_[0]) eq 'GLOB'   },
 
         FileHandle => sub {
-            ref($_) eq 'GLOB' && openhandle($_)
+            ref($_[0]) eq 'GLOB' && openhandle($_[0])
             or
-            blessed($_) && $_->isa("IO::Handle")
+            blessed($_[0]) && $_[0]->isa("IO::Handle")
         },
 
-        Object     => sub { blessed($_) && blessed($_) ne 'Regexp' },
+        Object     => sub { blessed($_[0]) && blessed($_[0]) ne 'Regexp' },
     );
+    foreach my $code (values %TYPE) {
+        bless $code, 'Mouse::Meta::TypeConstraint';
+    }
 
     sub optimized_constraints { \%TYPE }
     my @TYPE_KEYS = keys %TYPE;
@@ -78,10 +79,16 @@ sub type {
     if ($TYPE{$name} && $TYPE_SOURCE{$name} ne $pkg) {
         Carp::croak "The type constraint '$name' has already been created in $TYPE_SOURCE{$name} and cannot be created again in $pkg";
     };
-    my $constraint = $conf{where} || do { $TYPE{delete $conf{as} || 'Any' } };
+    my $constraint = $conf{where} || do {
+        my $as = delete $conf{as} || 'Any';
+        if (! exists $TYPE{$as}) {
+            $TYPE{$as} = _build_type_constraint($as);
+        }
+        $TYPE{$as};
+    };
 
     $TYPE_SOURCE{$name} = $pkg;
-    $TYPE{$name} = $constraint;
+    $TYPE{$name} = sub { local $_ = $_[0]; $constraint->($_[0]) };
 }
 
 sub subtype {
@@ -90,22 +97,22 @@ sub subtype {
     if ($TYPE{$name} && $TYPE_SOURCE{$name} ne $pkg) {
         Carp::croak "The type constraint '$name' has already been created in $TYPE_SOURCE{$name} and cannot be created again in $pkg";
     };
-    my $constraint = $conf{where} || do {
-        my $as = delete $conf{as} || 'Any';
-        if (! exists $TYPE{$as}) { # Perhaps it's a parameterized source?
-            Mouse::Meta::Attribute::_build_type_constraint($as);
-        }
-        $TYPE{$as};
-    };
-    my $as         = $conf{as} || '';
+    my $constraint = $conf{where};
+    my $as_constraint = find_or_create_isa_type_constraint($conf{as} || 'Any');
 
     $TYPE_SOURCE{$name} = $pkg;
+    $TYPE{$name} = $constraint ? 
+        sub {
+            local $_ = $_[0];
+            $as_constraint->($_[0]) && $constraint->($_[0])
+        } :
+        sub {
+            local $_ = $_[0];
+            $as_constraint->($_[0]);
+        }
+    ;
 
-    if ($as = $TYPE{$as}) {
-        $TYPE{$name} = sub { $as->($_) && $constraint->($_) };
-    } else {
-        $TYPE{$name} = $constraint;
-    }
+    return $name;
 }
 
 sub coerce {
@@ -125,7 +132,7 @@ sub coerce {
         if (! $TYPE{$type}) {
             # looks parameterized
             if ($type =~ /^[^\[]+\[.+\]$/) {
-                Mouse::Meta::Attribute::_build_type_constraint($type);
+                _build_type_constraint($type);
             } else {
                 Carp::croak "Could not find the type constraint ($type) to coerce from"
             }
@@ -137,12 +144,16 @@ sub coerce {
 }
 
 sub class_type {
-    my $pkg = caller(0);
     my($name, $conf) = @_;
-    my $class = $conf->{class};
-    subtype(
-        $name => where => sub { $_->isa($class) }
-    );
+    if ($conf && $conf->{class}) {
+        # No, you're using this wrong
+        warn "class_type() should be class_type(ClassName). Perhaps you're looking for subtype $name => as '$conf->{class}'?";
+        subtype($name, as => $conf->{class});
+    } else {
+        subtype(
+            $name => where => sub { $_->isa($name) }
+        );
+    }
 }
 
 sub role_type {
@@ -164,10 +175,10 @@ sub typecast_constraints {
         next unless $COERCE{$type};
         for my $coerce_type (@{ $COERCE_KEYS{$type}}) {
             $_ = $value;
-            next unless $TYPE{$coerce_type}->();
+            next unless $TYPE{$coerce_type}->($value);
             $_ = $value;
-            $_ = $COERCE{$type}->{$coerce_type}->();
-            return $_ if $type_constraint->();
+            $_ = $COERCE{$type}->{$coerce_type}->($value);
+            return $_ if $type_constraint->($_);
         }
     }
     return $value;
@@ -194,6 +205,115 @@ sub enum {
     );
 }
 
+sub _build_type_constraint {
+
+    my $spec = shift;
+    my $code;
+    if ($spec =~ /^([^\[]+)\[(.+)\]$/) {
+        # parameterized
+        my $constraint = $1;
+        my $param      = $2;
+        my $parent;
+        if ($constraint eq 'Maybe') {
+            $parent = _build_type_constraint('Undef');
+        } else {
+            $parent = _build_type_constraint($constraint);
+        }
+        my $child = _build_type_constraint($param);
+        if ($constraint eq 'ArrayRef') {
+            my $code_str = 
+                "#line " . __LINE__ . ' "' . __FILE__ . "\"\n" .
+                "sub {\n" .
+                "    if (\$parent->(\$_[0])) {\n" .
+                "        foreach my \$e (\@{\$_[0]}) {\n" .
+                "            return () unless \$child->(\$e);\n" .
+                "        }\n" .
+                "        return 1;\n" .
+                "    }\n" .
+                "    return ();\n" .
+                "};\n"
+            ;
+            $code = eval $code_str or Carp::confess("Failed to generate inline type constraint: $@");
+        } elsif ($constraint eq 'HashRef') {
+            my $code_str = 
+                "#line " . __LINE__ . ' "' . __FILE__ . "\"\n" .
+                "sub {\n" .
+                "    if (\$parent->(\$_[0])) {\n" .
+                "        foreach my \$e (values \%{\$_[0]}) {\n" .
+                "            return () unless \$child->(\$e);\n" .
+                "        }\n" .
+                "        return 1;\n" .
+                "    }\n" .
+                "    return ();\n" .
+                "};\n"
+            ;
+            $code = eval $code_str or Carp::confess($@);
+        } elsif ($constraint eq 'Maybe') {
+            my $code_str =
+                "#line " . __LINE__ . ' "' . __FILE__ . "\"\n" .
+                "sub {\n" .
+                "    return \$child->(\$_[0]) || \$parent->(\$_[0]);\n" .
+                "};\n"
+            ;
+            $code = eval $code_str or Carp::confess($@);
+        } else {
+            Carp::confess("Support for parameterized types other than ArrayRef or HashRef is not implemented yet");
+        }
+        $TYPE{$spec} = $code;
+    } else {
+        $code = $TYPE{ $spec };
+        if (! $code) {
+            my $code_str = 
+                "#line " . __LINE__ . ' "' . __FILE__ . "\"\n" .
+                "sub {\n" .
+                "    Scalar::Util::blessed(\$_[0]) && \$_[0]->isa('$spec');\n" .
+                "}"
+            ;
+            $code = eval $code_str  or Carp::confess($@);
+            $TYPE{$spec} = bless $code, 'Mouse::Meta::TypeConstraint';
+        }
+    }
+    return bless $code, 'Mouse::Meta::TypeConstraint';
+}
+
+sub find_type_constraint {
+    my $type_constraint = shift;
+    return $TYPE{$type_constraint};
+}
+
+sub find_or_create_isa_type_constraint {
+    my $type_constraint = shift;
+
+    my $code;
+
+    $type_constraint =~ s/\s+//g;
+    my @type_constraints = split /\|/, $type_constraint;
+    if (@type_constraints == 1) {
+        $code = $TYPE{$type_constraints[0]} ||
+            _build_type_constraint($type_constraints[0]);
+    } else {
+        my @code_list = map {
+            $TYPE{$_} || _build_type_constraint($_)
+        } @type_constraints;
+        $code = bless sub {
+            my $i = 0;
+            for my $code (@code_list) {
+                return 1 if $code->($_[0]);
+            }
+            return 0;
+        }, 'Mouse::Meta::TypeConstraint';
+    }
+    return $code;
+}
+
+package # Hide from pause
+    Mouse::Meta::TypeConstraint;
+
+sub check { 
+    $_[0]->($_[1])
+}
+
+
 1;
 
 __END__
@@ -208,6 +328,22 @@ Mouse::Util::TypeConstraints - simple type constraints
 
 Returns the simple type constraints that Mouse understands.
 
+=head1 FUNCTIONS
+
+=over 4
+
+=item B<subtype 'Name' => as 'Parent' => where { } ...>
+
+=item B<subtype as 'Parent' => where { } ...>
+
+=item B<class_type ($class, ?$options)>
+
+=item B<role_type ($role, ?$options)>
+
+=item B<enum (\@values)>
+
+=back
+
 =cut