Add support for Type in native traits
[gitmo/Moose.git] / lib / Moose / Meta / Method / Accessor / Native / Writer.pm
index 1d124ec..2b55aae 100644 (file)
@@ -3,135 +3,170 @@ package Moose::Meta::Method::Accessor::Native::Writer;
 use strict;
 use warnings;
 
-our $VERSION = '1.13';
-$VERSION = eval $VERSION;
-our $AUTHORITY = 'cpan:STEVAN';
+use List::MoreUtils qw( any );
+use Moose::Util;
 
-use base 'Moose::Meta::Method::Accessor::Native';
+use Moose::Role;
 
-sub _generate_method {
-    my $self = shift;
-
-    my $inv = '$self';
-
-    my $slot_access = $self->_inline_get($inv);
-
-    my $code = 'sub {';
+with 'Moose::Meta::Method::Accessor::Native';
 
-    $code .= "\n" . $self->_inline_pre_body(@_);
+requires '_potential_value';
 
-    $code .= "\n" . 'my $self = shift;';
-
-    $code .= "\n" . $self->_inline_curried_arguments;
-
-    $code .= $self->_writer_core( $inv, $slot_access );
+sub _generate_method {
+    my $self = shift;
 
-    $code .= "\n" . $self->_inline_post_body(@_);
+    my $inv         = '$self';
+    my $slot_access = $self->_get_value($inv);
 
-    $code .= "\n}";
+    return (
+        'sub {',
+            'my ' . $inv . ' = shift;',
+            $self->_inline_curried_arguments,
+            $self->_inline_writer_core($inv, $slot_access),
+        '}',
+    );
+}
 
-    return $code;
+sub _inline_writer_core {
+    my $self = shift;
+    my ($inv, $slot_access) = @_;
+
+    my $potential = $self->_potential_value($slot_access);
+    my $old       = '@old';
+
+    my @code;
+    push @code, (
+        $self->_inline_check_argument_count,
+        $self->_inline_process_arguments($inv, $slot_access),
+        $self->_inline_check_arguments('for writer'),
+        $self->_inline_check_lazy($inv, '$type_constraint', '$type_coercion', '$type_message'),
+    );
+
+    if ($self->_return_value($slot_access)) {
+        # some writers will save the return value in this variable when they
+        # generate the potential value.
+        push @code, 'my @return;'
+    }
+
+    push @code, (
+        $self->_inline_coerce_new_values,
+        $self->_inline_copy_native_value(\$potential),
+        $self->_inline_tc_code($potential, '$type_constraint', '$type_coercion', '$type_message'),
+        $self->_inline_get_old_value_for_trigger($inv, $old),
+        $self->_inline_capture_return_value($slot_access),
+        $self->_inline_set_new_value($inv, $potential, $slot_access),
+        $self->_inline_trigger($inv, $slot_access, $old),
+        $self->_inline_return_value($slot_access, 'for writer'),
+    );
+
+    return @code;
 }
 
-sub _writer_core {
-    my ( $self, $inv, $slot_access ) = @_;
+sub _inline_process_arguments { return }
 
-    my $code = q{};
+sub _inline_check_arguments { return }
 
-    $code .= "\n" . $self->_inline_check_argument_count;
-    $code .= "\n" . $self->_inline_process_arguments( $inv, $slot_access );
-    $code .= "\n" . $self->_inline_check_arguments('for writer');
+sub _inline_coerce_new_values { return }
 
-    $code .= "\n" . $self->_inline_check_lazy($inv);
+sub _writer_value_needs_copy {
+    my $self = shift;
 
-    my $new_value       = $self->_new_value($slot_access);
-    my $potential_value = $self->_potential_value($slot_access);
+    return $self->_constraint_must_be_checked;
+}
 
-    $code .= "\n" . $self->_inline_copy_value( \$potential_value );
-    $code .= "\n"
-        . $self->_inline_tc_code(
-        $new_value,
-        $potential_value
-        );
+sub _constraint_must_be_checked {
+    my $self = shift;
 
-    $code .= "\n" . $self->_inline_get_old_value_for_trigger($inv);
-    $code .= "\n" . $self->_inline_capture_return_value($slot_access);
-    $code .= "\n"
-        . $self->_inline_set_new_value(
-        $inv,
-        $potential_value,
-        $slot_access,
-        );
-    $code .= "\n" . $self->_inline_trigger( $inv, $slot_access, '@old' );
-    $code .= "\n" . $self->_return_value( $slot_access, 'for writer' );
+    my $attr = $self->associated_attribute;
 
-    return $code;
+    return $attr->has_type_constraint
+        && ( !$self->_is_root_type( $attr->type_constraint )
+        || ( $attr->should_coerce && $attr->type_constraint->has_coercion ) );
 }
 
-sub _inline_process_arguments {q{}}
-
-sub _inline_check_arguments {q{}}
-
-sub _value_needs_copy {
+sub _is_root_type {
     my $self = shift;
-
-    return $self->_constraint_must_be_checked;
+    my $type = shift;
+
+    if (
+        Moose::Util::does_role( $type, 'Type::Constraint::Role::Interface' ) )
+    {
+        require Type::Library::Builtins;
+        return
+            any { $type->is_same_type_as( Type::Library::Builtins::t($_) ) }
+        @{ $self->root_types };
+    }
+    else {
+        my $name = $type->name;
+        return any { $name eq $_ } @{ $self->root_types };
+    }
 }
 
-sub _inline_copy_value {
-    my ( $self, $potential_ref ) = @_;
+sub _inline_copy_native_value {
+    my $self = shift;
+    my ($potential_ref) = @_;
 
-    return q{} unless $self->_value_needs_copy;
+    return unless $self->_writer_value_needs_copy;
 
-    my $code = "my \$potential = ${$potential_ref};";
+    my $code = 'my $potential = ' . ${$potential_ref} . ';';
 
     ${$potential_ref} = '$potential';
 
     return $code;
 }
 
-sub _inline_tc_code {
-    my ( $self, $new_value, $potential_value ) = @_;
-
-    return q{} unless $self->_constraint_must_be_checked;
+around _inline_tc_code => sub {
+    my $orig = shift;
+    my $self = shift;
+    my ($value, $tc, $coercion, $message, $for_lazy) = @_;
 
-    return $self->_inline_check_coercion($potential_value) . "\n"
-        . $self->_inline_check_constraint($potential_value);
-}
+    return unless $for_lazy || $self->_constraint_must_be_checked;
 
-sub _inline_check_coercion {
-    die '_inline_check_coercion must be overridden by ' . ref $_[0];
-}
+    return $self->$orig(@_);
+};
 
-sub _inline_check_constraint {
+around _inline_check_constraint => sub {
+    my $orig = shift;
     my $self = shift;
+    my ($value, $tc, $message, $for_lazy) = @_;
 
-    return q{} unless $self->_constraint_must_be_checked;
+    return unless $for_lazy || $self->_constraint_must_be_checked;
 
-    return $self->SUPER::_inline_check_constraint( $_[0] );
-}
+    return $self->$orig(@_);
+};
 
-sub _constraint_must_be_checked {
-    die '_constraint_must_be_checked must be overridden by ' . ref $_[0];
-}
-
-sub _inline_capture_return_value { return q{} }
+sub _inline_capture_return_value { return }
 
 sub _inline_set_new_value {
     my $self = shift;
 
-    return $self->SUPER::_inline_store(@_)
-        if $self->_value_needs_copy;
+    return $self->_inline_store_value(@_)
+        if $self->_writer_value_needs_copy
+        || !$self->_slot_access_can_be_inlined
+        || !$self->_get_is_lvalue;
 
     return $self->_inline_optimized_set_new_value(@_);
 }
 
+sub _get_is_lvalue {
+    my $self = shift;
+
+    return $self->associated_attribute->associated_class->instance_metaclass->inline_get_is_lvalue;
+}
+
 sub _inline_optimized_set_new_value {
     my $self = shift;
 
-    return $self->SUPER::_inline_store(@_)
+    return $self->_inline_store_value(@_);
+}
+
+sub _return_value {
+    my $self = shift;
+    my ($slot_access) = @_;
+
+    return $slot_access;
 }
 
-sub _return_value { return q{} }
+no Moose::Role;
 
 1;