clean up the constructor generation code
[gitmo/Class-MOP.git] / lib / Class / MOP / Method / Constructor.pm
index aeaa607..56f38c5 100644 (file)
@@ -5,13 +5,14 @@ use strict;
 use warnings;
 
 use Carp         'confess';
-use Scalar::Util 'blessed', 'weaken', 'looks_like_number';
+use Scalar::Util 'blessed', 'weaken';
+use Try::Tiny;
 
-our $VERSION   = '0.71_01';
+our $VERSION   = '1.11';
 $VERSION = eval $VERSION;
 our $AUTHORITY = 'cpan:STEVAN';
 
-use base 'Class::MOP::Method::Generated';
+use base 'Class::MOP::Method::Inlined';
 
 sub new {
     my $class   = shift;
@@ -31,29 +32,40 @@ sub new {
     # needed
     weaken($self->{'associated_metaclass'});
 
-    $self->initialize_body;
+    $self->_initialize_body;
 
     return $self;
 }
 
 sub _new {
     my $class = shift;
-    my $options = @_ == 1 ? $_[0] : {@_};
-
-    bless {
-        # from our superclass
-        'body'                 => undef,
-        'package_name'         => $options->{package_name},
-        'name'                 => $options->{name},        
-        # specific to this subclass
-        'options'              => $options->{options} || {},
-        'associated_metaclass' => $options->{metaclass},
-        'is_inline'            => ($options->{is_inline} || 0),
+
+    return Class::MOP::Class->initialize($class)->new_object(@_)
+        if $class ne __PACKAGE__;
+
+    my $params = @_ == 1 ? $_[0] : {@_};
+
+    return bless {
+        # inherited from Class::MOP::Method
+        body                 => $params->{body},
+        # associated_metaclass => $params->{associated_metaclass}, # overriden
+        package_name         => $params->{package_name},
+        name                 => $params->{name},
+        original_method      => $params->{original_method},
+
+        # inherited from Class::MOP::Generated
+        is_inline            => $params->{is_inline} || 0,
+        definition_context   => $params->{definition_context},
+
+        # inherited from Class::MOP::Inlined
+        _expected_method_class => $params->{_expected_method_class},
+
+        # defined in this subclass
+        options              => $params->{options} || {},
+        associated_metaclass => $params->{metaclass},
     }, $class;
 }
 
-sub can_be_inlined { 1 }
-
 ## accessors
 
 sub options              { (shift)->{'options'}              }
@@ -61,74 +73,117 @@ sub associated_metaclass { (shift)->{'associated_metaclass'} }
 
 ## cached values ...
 
-sub meta_instance {
+sub _attributes {
     my $self = shift;
-    $self->{'meta_instance'} ||= $self->associated_metaclass->get_meta_instance;
-}
-
-sub attributes {
-    my $self = shift;
-    $self->{'attributes'} ||= [ $self->associated_metaclass->compute_all_applicable_attributes ]
+    $self->{'attributes'} ||= [ $self->associated_metaclass->get_all_attributes ]
 }
 
 ## method
 
-sub initialize_body {
+sub _initialize_body {
     my $self        = shift;
-    my $method_name = 'generate_constructor_method';
+    my $method_name = '_generate_constructor_method';
 
     $method_name .= '_inline' if $self->is_inline;
 
     $self->{'body'} = $self->$method_name;
 }
 
-sub generate_constructor_method {
+sub _generate_constructor_method {
     return sub { Class::MOP::Class->initialize(shift)->new_object(@_) }
 }
 
-sub generate_constructor_method_inline {
+sub _generate_constructor_method_inline {
     my $self = shift;
 
-    my $close_over = {};
-
-    my $source = 'sub {';
-    $source .= "\n" . 'my $class = shift;';
-
-    $source .= "\n" . 'return Class::MOP::Class->initialize($class)->new_object(@_)';
-    $source .= "\n" . '    if $class ne \'' . $self->associated_metaclass->name . '\';';
+    my $defaults = [map { $_->default } @{ $self->_attributes }];
+    my $close_over = {
+        '$defaults' => \$defaults,
+    };
+
+    my $meta = $self->associated_metaclass;
+
+    my $idx = 0;
+    my @source = (
+        'sub {',
+            'my $class = shift;',
+            'return Class::MOP::Class->initialize($class)->new_object(@_)',
+                'if $class ne \'' . $meta->name . '\';',
+            'my $params = @_ == 1 ? $_[0] : {@_};',
+            'my $instance = ' . $meta->inline_create_instance('$class') . ';',
+            (map { $self->_generate_slot_initializer($_, $idx++) }
+                 @{ $self->_attributes }),
+            $self->_preserve_weak_metaclasses,
+            'return $instance',
+        '}',
+    );
+
+    warn join("\n", @source) if $self->options->{debug};
+
+    my $code = try {
+        $self->_compile_code(
+            source      => \@source,
+            environment => $close_over,
+        );
+    }
+    catch {
+        my $source = join("\n", @source);
+        confess "Could not eval the constructor :\n\n$source\n\nbecause :\n\n$_";
+    };
 
-    $source .= "\n" . 'my $params = @_ == 1 ? $_[0] : {@_};';
+    return $code;
+}
 
-    $source .= "\n" . 'my $instance = ' . $self->meta_instance->inline_create_instance('$class');
-    $source .= ";\n" . (join ";\n" => map {
-        $self->_generate_slot_initializer($_)
-    } 0 .. (@{$self->attributes} - 1));
-    $source .= ";\n" . 'return $instance';
-    $source .= ";\n" . '}';
-    warn $source if $self->options->{debug};
+sub _generate_slot_initializer {
+    my $self  = shift;
+    my ($attr, $idx) = @_;
 
-    my $code;
-    {
-        # NOTE:
-        # create the nessecary lexicals
-        # to be picked up in the eval
+    my $default = $self->_generate_default_value($attr, $idx);
 
-        $code = $self->_eval_closure(
-            { '$attrs' => \$self->attributes },
-            $source
+    if (defined(my $init_arg = $attr->init_arg)) {
+        my @source = (
+            'if (exists $params->{\'' . $init_arg . '\'}) {',
+                $attr->inline_set(
+                    '$instance', '$params->{\'' . $init_arg . '\'}'
+                ) . ';',
+            '}',
         );
-        confess "Could not eval the constructor :\n\n$source\n\nbecause :\n\n$@" if $@;
+        if (defined $default) {
+            push @source, (
+                'else {',
+                    $attr->inline_set('$instance', $default) . ';',
+                '}',
+            );
+        }
+        return @source;
+    }
+    elsif (defined $default) {
+        return ($attr->inline_set('$instance', $default) . ';');
+    }
+    else {
+        return ();
     }
-    return $code;
 }
 
-sub _generate_slot_initializer {
-    my $self  = shift;
-    my $index = shift;
+sub _preserve_weak_metaclasses {
+    my $self = shift;
+    my $meta = $self->associated_metaclass;
+    if (Class::MOP::metaclass_is_weak($meta->name)) {
+        return (
+            $meta->_inline_set_mop_slot(
+                '$instance', 'Class::MOP::class_of($class)'
+            ) . ';'
+        );
+    }
+    else {
+        return ();
+    }
+}
 
-    my $attr = $self->attributes->[$index];
+sub _generate_default_value {
+    my $self = shift;
+    my ($attr, $index) = @_;
 
-    my $default;
     if ($attr->has_default) {
         # NOTE:
         # default values can either be CODE refs
@@ -137,41 +192,18 @@ sub _generate_slot_initializer {
         # in which case we can just deal with them
         # in the code we eval.
         if ($attr->is_default_a_coderef) {
-            $default = '$attrs->[' . $index . ']->default($instance)';
+            return '$defaults->[' . $index . ']->($instance)';
         }
         else {
-            $default = $attr->default;
-            # make sure to quote strings ...
-            unless (looks_like_number($default)) {
-                $default = "'$default'";
-            }
+            return '$defaults->[' . $index . ']';
         }
-    } elsif( $attr->has_builder ) {
-        $default = '$instance->'.$attr->builder;
     }
-
-    if ( defined $attr->init_arg ) {
-      return (
-          'if(exists $params->{\'' . $attr->init_arg . '\'}){' . "\n" .
-                $self->meta_instance->inline_set_slot_value(
-                    '$instance',
-                    ("'" . $attr->name . "'"),
-                    '$params->{\'' . $attr->init_arg . '\'}' ) . "\n" .
-           '} ' . (!defined $default ? '' : 'else {' . "\n" .
-                $self->meta_instance->inline_set_slot_value(
-                    '$instance',
-                    ("'" . $attr->name . "'"),
-                     $default ) . "\n" .
-           '}')
-        );
-    } elsif ( defined $default ) {
-        return (
-            $self->meta_instance->inline_set_slot_value(
-                '$instance',
-                ("'" . $attr->name . "'"),
-                 $default ) . "\n"
-        );
-    } else { return '' }
+    elsif ($attr->has_builder) {
+        return '$instance->' . $attr->builder;
+    }
+    else {
+        return;
+    }
 }
 
 1;
@@ -200,58 +232,47 @@ Class::MOP::Method::Constructor - Method Meta Object for constructors
 
 =head1 DESCRIPTION
 
-This is a subclass of C<Class::MOP::Method> which deals with
-class constructors. This is used when making a class immutable
-to generate an optimized constructor.
+This is a subclass of C<Class::MOP::Method> which generates
+constructor methods.
 
 =head1 METHODS
 
 =over 4
 
-=item B<new (metaclass => $meta, options => \%options)>
+=item B<< Class::MOP::Method::Constructor->new(%options) >>
 
-=item B<options>
+This creates a new constructor object. It accepts a hash reference of
+options.
 
-This returns the options HASH which is passed into C<new>.
+=over 8
 
-=item B<associated_metaclass>
+=item * metaclass
 
-This returns the metaclass which is passed into C<new>.
+This should be a L<Class::MOP::Class> object. It is required.
 
-=item B<attributes>
+=item * name
 
-This returns the list of attributes which are associated with the
-metaclass which is passed into C<new>.
+The method name (without a package name). This is required.
 
-=item B<meta_instance>
+=item * package_name
 
-This returns the meta instance which is associated with the
-metaclass which is passed into C<new>.
+The package name for the method. This is required.
 
-=item B<is_inline>
+=item * is_inline
 
-This returns a boolean, but since constructors are very rarely
-not inlined, this always returns true for now.
-
-=item B<can_be_inlined>
-
-This method always returns true in this class. It exists so that
-subclasses (like in Moose) can override and do some sort of checking
-to determine whether or not inlining the constructor is safe.
-
-=item B<initialize_body>
-
-This creates the code reference for the constructor itself.
+This indicates whether or not the constructor should be inlined. This
+defaults to false.
 
 =back
 
-=head2 Method Generators 
+=item B<< $metamethod->is_inline >>
 
-=over 4
+Returns a boolean indicating whether or not the constructor is
+inlined.
 
-=item B<generate_constructor_method>
+=item B<< $metamethod->associated_metaclass >>
 
-=item B<generate_constructor_method_inline>
+This returns the L<Class::MOP::Class> object for the method.
 
 =back
 
@@ -261,7 +282,7 @@ Stevan Little E<lt>stevan@iinteractive.comE<gt>
 
 =head1 COPYRIGHT AND LICENSE
 
-Copyright 2006-2008 by Infinity Interactive, Inc.
+Copyright 2006-2010 by Infinity Interactive, Inc.
 
 L<http://www.iinteractive.com>