- inject object constructor when call meta->make_immutable.

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

diff --git a/lib/Mouse/TypeRegistry.pm b/lib/Mouse/TypeRegistry.pm
index 757ed85..ecde30d 100644 (file)
--- a/lib/Mouse/TypeRegistry.pm
+++ b/lib/Mouse/TypeRegistry.pm
@@ -3,11 +3,74 @@ package Mouse::TypeRegistry;
 use strict;
 use warnings;
 
-use Scalar::Util qw/looks_like_number blessed openhandle/;
+use Mouse::Util qw/blessed looks_like_number openhandle/;
 
-no warnings 'uninitialized';
-sub optimized_constraints {
-    return {
+my $SUBTYPE = +{};
+my $COERCE = +{};
+
+sub import {
+    my $class  = shift;
+    my %args   = @_;
+    my $caller = caller(0);
+
+    $SUBTYPE->{$caller} ||= +{};
+    $COERCE->{$caller}  ||= +{};
+
+    if (defined $args{'-export'} && ref($args{'-export'}) eq 'ARRAY') {
+        no strict 'refs';
+        *{"$caller\::import"} = sub { _import(@_) };
+    }
+
+    no strict 'refs';
+    *{"$caller\::subtype"}     = \&_subtype;
+    *{"$caller\::coerce"}      = \&_coerce;
+#    *{"$caller\::class_type"}  = \&_class_type;
+#    *{"$caller\::role_type"}   = \&_role_type;
+}
+
+sub _import {
+    my($class, @types) = @_;
+    return unless exists $SUBTYPE->{$class} && exists $COERCE->{$class};
+    my $pkg = caller(1);
+    return unless @types;
+    copy_types($class, $pkg, @types);
+}
+
+sub _subtype {
+    my $pkg = caller(0);
+    my($name, $stuff) = @_;
+    if (ref $stuff eq 'HASH') {
+        my $as = $stuff->{as};
+        $stuff = optimized_constraints()->{$as};
+    }
+    $SUBTYPE->{$pkg}->{$name} = $stuff;
+}
+
+sub _coerce {
+    my $pkg = caller(0);
+    my($name, $conf) = @_;
+    $COERCE->{$pkg}->{$name} = $conf;
+}
+
+sub typecast_constraints {
+    my($class, $pkg, $type, $value) = @_;
+    return $value unless defined $COERCE->{$pkg} && defined $COERCE->{$pkg}->{$type};
+
+    my $optimized_constraints = optimized_constraints();
+    for my $coerce_type (keys %{ $COERCE->{$pkg}->{$type} }) {
+        local $_ = $value;
+        if ($optimized_constraints->{$coerce_type}->()) {
+            local $_ = $value;
+            return $COERCE->{$pkg}->{$type}->{$coerce_type}->();
+        }
+    }
+
+    return $value;
+}
+
+{
+    no warnings 'uninitialized';
+    my $optimized_constraints = {
         Any        => sub { 1 },
         Item       => sub { 1 },
         Bool       => sub {
@@ -39,6 +102,11 @@ sub optimized_constraints {
 
         Object     => sub { blessed($_) && blessed($_) ne 'Regexp' },
     };
+    sub optimized_constraints {
+        my($class, $pkg) = @_;
+        my $subtypes = $SUBTYPE->{$pkg} || {};
+        return { %{ $subtypes }, %{ $optimized_constraints } };
+    }
 }
 
 1;