optimized

[gitmo/Mouse.git] / lib / Mouse / TypeRegistry.pm

diff --git a/lib/Mouse/TypeRegistry.pm b/lib/Mouse/TypeRegistry.pm
index 6568990..b5d069c 100644 (file)
--- a/lib/Mouse/TypeRegistry.pm
+++ b/lib/Mouse/TypeRegistry.pm
@@ -2,11 +2,52 @@
 package Mouse::TypeRegistry;
 use strict;
 use warnings;
-use Scalar::Util qw/looks_like_number blessed openhandle/;
 
-no warnings 'uninitialized';
-sub optimized_constraints {
-    return {
+use Carp ();
+use Mouse::Util qw/blessed looks_like_number openhandle/;
+
+my $SUBTYPE = +{};
+my $COERCE = +{};
+
+#find_type_constraint register_type_constraint
+sub import {
+    my $class  = shift;
+    my %args   = @_;
+    my $caller = $args{callee} || caller(0);
+
+    no strict 'refs';
+    *{"$caller\::as"}          = \&_as;
+    *{"$caller\::where"}       = \&_where;
+    *{"$caller\::message"}     = \&_message;
+    *{"$caller\::from"}        = \&_from;
+    *{"$caller\::via"}         = \&_via;
+    *{"$caller\::subtype"}     = \&_subtype;
+    *{"$caller\::coerce"}      = \&_coerce;
+    *{"$caller\::class_type"}  = \&_class_type;
+    *{"$caller\::role_type"}   = \&_role_type;
+}
+
+
+sub _as ($) {
+    as => $_[0]
+}
+sub _where (&) {
+    where => $_[0]
+}
+sub _message ($) {
+    message => $_[0]
+}
+
+sub _from { @_ }
+sub _via (&) {
+    $_[0]
+}
+
+my $optimized_constraints;
+my $optimized_constraints_base;
+{
+    no warnings 'uninitialized';
+    $optimized_constraints = $optimized_constraints_base = {
         Any        => sub { 1 },
         Item       => sub { 1 },
         Bool       => sub {
@@ -18,6 +59,7 @@ sub optimized_constraints {
         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' },
@@ -33,40 +75,86 @@ sub optimized_constraints {
             or
                 blessed($_)
                 && $_->isa("IO::Handle")
-        },
+            },
 
         Object     => sub { blessed($_) && blessed($_) ne 'Regexp' },
-
-        ClassName  => sub {
-            return if ref($_);
-            return unless defined($_) && length($_);
-
-            # walk the symbol table tree to avoid autovififying
-            # \*{${main::}{"Foo::"}} == \*main::Foo::
-
-            my $pack = \*::;
-            foreach my $part (split('::', $_)) {
-                return unless exists ${$$pack}{"${part}::"};
-                $pack = \*{${$$pack}{"${part}::"}};
-            }
-
-            # check for $VERSION or @ISA
-            return 1 if exists ${$$pack}{VERSION}
-                    && defined *{${$$pack}{VERSION}}{SCALAR};
-            return 1 if exists ${$$pack}{ISA}
-                    && defined *{${$$pack}{ISA}}{ARRAY};
-
-            # check for any method
-            foreach ( keys %{$$pack} ) {
-                next if substr($_, -2, 2) eq '::';
-                return 1 if defined *{${$$pack}{$_}}{CODE};
-            }
-
-            # fail
-            return;
-        },
     };
 }
+sub _subtype {
+    my $pkg = caller(0);
+    my($name, %conf) = @_;
+    if (my $type = $SUBTYPE->{$name}) {
+        Carp::croak "The type constraint '$name' has already been created, cannot be created again in $pkg";
+    };
+    my $as = $conf{as};
+    my $stuff = $conf{where} || optimized_constraints()->{$as};
+
+    $SUBTYPE->{$name} = $stuff;
+    $optimized_constraints = +{ %{ $SUBTYPE }, %{ $optimized_constraints_base } };
+}
+
+sub _coerce {
+    my($name, %conf) = @_;
+
+    Carp::croak "Cannot find type '$name', perhaps you forgot to load it."
+        unless optimized_constraints()->{$name};
+
+    my $subtypes = optimized_constraints();
+    $COERCE->{$name} ||= {};
+    while (my($type, $code) = each %conf) {
+        Carp::croak "A coercion action already exists for '$type'"
+            if $COERCE->{$name}->{$type};
+
+        Carp::croak "Could not find the type constraint ($type) to coerce from"
+            unless $subtypes->{$type};
+
+        $COERCE->{$name}->{$type} = $code;
+    }
+}
+
+sub _class_type {
+    my $pkg = caller(0);
+    my($name, $conf) = @_;
+    my $class = $conf->{class};
+    _subtype(
+        $name => where => sub {
+            defined $_ && ref($_) eq $class;
+        }
+    );
+}
+
+sub _role_type {
+    my($name, $conf) = @_;
+    my $role = $conf->{role};
+    _subtype(
+        $name => where => sub {
+            return unless defined $_ && ref($_) && $_->isa('Mouse::Object');
+            $_->meta->does_role($role);
+        }
+    );
+}
+
+sub typecast_constraints {
+    my($class, $pkg, $type, $value) = @_;
+    return $value unless $COERCE->{$type};
+
+    my $optimized_constraints = optimized_constraints();
+    for my $coerce_type (keys %{ $COERCE->{$type} }) {
+        local $_ = $value;
+        if ($optimized_constraints->{$coerce_type}->()) {
+            local $_ = $value;
+            return $COERCE->{$type}->{$coerce_type}->();
+        }
+    }
+
+    return $value;
+}
+
+sub optimized_constraints { $optimized_constraints }
+{
+    my @optimized_constraints_keys = keys %{ $optimized_constraints };
+    sub list_all_builtin_type_constraints { @optimized_constraints_keys }
+}
 
 1;