Fix an accessor generator to accept role application to instance
[gitmo/Mouse.git] / lib / Mouse / Meta / Method / Accessor.pm
1 package Mouse::Meta::Method::Accessor;
2 use Mouse::Util; # enables strict and warnings
3 use Scalar::Util qw(blessed);
4
5 sub _generate_accessor{
6     my (undef, $attribute, $class, $type) = @_;
7
8     my $name          = $attribute->name;
9     my $default       = $attribute->default;
10     my $constraint    = $attribute->type_constraint;
11     my $builder       = $attribute->builder;
12     my $trigger       = $attribute->trigger;
13     my $is_weak       = $attribute->is_weak_ref;
14     my $should_deref  = $attribute->should_auto_deref;
15     my $should_coerce = (defined($constraint) && $constraint->has_coercion && $attribute->should_coerce);
16
17     my $compiled_type_constraint = defined($constraint) ? $constraint->_compiled_type_constraint : undef;
18
19     my $self  = '$_[0]';
20     my $key   = "q{$name}";
21     my $slot  = "$self\->{$key}";
22
23     $type ||= 'accessor';
24
25     my $accessor = sprintf(qq{#line 1 "%s for %s (%s)"\n}, $type, $name, __FILE__)
26                  . "sub {\n";
27
28     if ($type eq 'accessor' || $type eq 'writer') {
29         if($type eq 'accessor'){
30             $accessor .= 
31                 'if (scalar(@_) >= 2) {' . "\n";
32         }
33         else{ # writer
34             $accessor .= 
35                 'if(@_ < 2){ Carp::confess("Not enough arguments for the writer of '.$name.'") }'.
36                 '{' . "\n";
37         }
38                 
39         my $value = '$_[1]';
40
41         if (defined $constraint) {
42             if ($should_coerce) {
43                 $accessor .=
44                     "\n".
45                     'my $val = $constraint->coerce('.$value.');';
46                 $value = '$val';
47             }
48             $accessor .= 
49                 "\n".
50                 '$compiled_type_constraint->('.$value.') or
51                     $attribute->verify_type_constraint_error($name, '.$value.', $constraint);' . "\n";
52         }
53
54         # if there's nothing left to do for the attribute we can return during
55         # this setter
56         $accessor .= 'return ' if !$is_weak && !$trigger && !$should_deref;
57
58         $accessor .= "$slot = $value;\n";
59
60         if ($is_weak) {
61             $accessor .= "Scalar::Util::weaken($slot) if ref $slot;\n";
62         }
63
64         if ($trigger) {
65             $accessor .= '$trigger->('.$self.', '.$value.');' . "\n";
66         }
67
68         $accessor .= "}\n";
69     }
70     elsif($type eq 'reader') {
71         $accessor .= 'Carp::confess("Cannot assign a value to a read-only accessor") if scalar(@_) >= 2;' . "\n";
72     }
73     else{
74         $class->throw_error("Unknown accessor type '$type'");
75     }
76
77     # XXX: an anon class can be a runtime created class
78     if ($attribute->is_lazy || $class->is_anon_class) {
79         my $value;
80
81         if (defined $builder){
82             $value = "$self->\$builder()";
83         }
84         elsif (ref($default) eq 'CODE'){
85             $value = "$self->\$default()";
86         }
87         else{
88             $value = '$default';
89         }
90
91         if($should_coerce){
92             $value = "\$constraint->coerce($value)";
93         }
94
95         $accessor .= "$slot = $value if !exists $slot;\n";
96     }
97
98     if ($should_deref) {
99         if ($constraint->is_a_type_of('ArrayRef')) {
100             $accessor .= "return \@{ $slot || [] } if wantarray;\n";
101         }
102         elsif($constraint->is_a_type_of('HashRef')){
103             $accessor .= "return \%{ $slot || {} } if wantarray;\n";
104         }
105         else{
106             $class->throw_error("Can not auto de-reference the type constraint " . $constraint->name);
107         }
108     }
109
110     $accessor .= "return $slot;\n}\n";
111
112     #print "# class ", $class->name, "\n", $accessor, "\n";
113     my $code;
114     my $e = do{
115         local $@;
116         $code = eval $accessor;
117         $@;
118     };
119     die $e if $e;
120
121     return $code;
122 }
123
124 sub _generate_reader{
125     my $class = shift;
126     return $class->_generate_accessor(@_, 'reader');
127 }
128
129 sub _generate_writer{
130     my $class = shift;
131     return $class->_generate_accessor(@_, 'writer');
132 }
133
134
135 sub _generate_predicate {
136     my (undef, $attribute, $class) = @_;
137
138     my $slot = $attribute->name;
139     return sub{
140         return exists $_[0]->{$slot};
141     };
142 }
143
144 sub _generate_clearer {
145     my (undef, $attribute, $class) = @_;
146
147     my $slot = $attribute->name;
148
149    return sub{
150         delete $_[0]->{$slot};
151     };
152 }
153
154 sub _generate_delegation{
155     my (undef, $attribute, $class, $reader, $handle_name, $method_to_call) = @_;
156
157     return sub {
158         my $instance = shift;
159         my $proxy    = $instance->$reader();
160
161         my $error = !defined($proxy)                ? ' is not defined'
162                   : ref($proxy) && !blessed($proxy) ? qq{ is not an object (got '$proxy')}
163                                                     : undef;
164         if ($error) {
165             $instance->meta->throw_error(
166                 "Cannot delegate $handle_name to $method_to_call because "
167                     . "the value of "
168                     . $attribute->name
169                     . $error
170              );
171         }
172         $proxy->$method_to_call(@_);
173     };
174 }
175
176
177 1;