Commit | Line | Data |
51a3b106 |
1 | package Method::Generate::Accessor; |
2 | |
3 | use strictures 1; |
b1eebd55 |
4 | use Moo::_Utils; |
5 | use base qw(Moo::Object); |
51a3b106 |
6 | use Sub::Quote; |
7 | use B 'perlstring'; |
daa05b62 |
8 | BEGIN { |
6e294299 |
9 | our $CAN_HAZ_XS = |
10 | _maybe_load_module('Class::XSAccessor') |
11 | && |
12 | (Class::XSAccessor->VERSION > 1.06) |
13 | ; |
daa05b62 |
14 | } |
51a3b106 |
15 | |
316917c9 |
16 | sub generate_method { |
6f68f022 |
17 | my ($self, $into, $name, $spec, $quote_opts) = @_; |
51a3b106 |
18 | die "Must have an is" unless my $is = $spec->{is}; |
dcae37d3 |
19 | if ($is eq 'ro') { |
20 | $spec->{reader} = $name unless exists $spec->{reader}; |
21 | } elsif ($is eq 'rw') { |
22 | $spec->{accessor} = $name unless exists $spec->{accessor}; |
23 | } elsif ($is eq 'lazy') { |
24 | $spec->{init_arg} = undef unless exists $spec->{init_arg}; |
25 | $spec->{reader} = $name unless exists $spec->{reader}; |
26 | $spec->{lazy} = 1; |
27 | $spec->{builder} ||= '_build_'.$name unless $spec->{default}; |
28 | } elsif ($is ne 'bare') { |
29 | die "Unknown is ${is}"; |
30 | } |
31 | my %methods; |
32 | if (my $reader = $spec->{reader}) { |
33 | if (our $CAN_HAZ_XS && $self->is_simple_get($name, $spec)) { |
34 | $methods{$reader} = $self->_generate_xs( |
35 | getters => $into, $reader, $name |
36 | ); |
37 | } else { |
e57f338d |
38 | $self->{captures} = {}; |
dcae37d3 |
39 | $methods{$reader} = |
40 | quote_sub "${into}::${reader}" |
41 | => $self->_generate_get($name, $spec) |
e57f338d |
42 | => delete $self->{captures} |
dcae37d3 |
43 | ; |
44 | } |
45 | } |
46 | if (my $accessor = $spec->{accessor}) { |
47 | if ( |
48 | our $CAN_HAZ_XS |
49 | && $self->is_simple_get($name, $spec) |
50 | && $self->is_simple_set($name, $spec) |
51 | ) { |
52 | $methods{$accessor} = $self->_generate_xs( |
53 | accessors => $into, $accessor, $name |
54 | ); |
55 | } else { |
e57f338d |
56 | $self->{captures} = {}; |
dcae37d3 |
57 | $methods{$accessor} = |
58 | quote_sub "${into}::${accessor}" |
59 | => $self->_generate_getset($name, $spec) |
e57f338d |
60 | => delete $self->{captures} |
dcae37d3 |
61 | ; |
62 | } |
63 | } |
64 | if (my $writer = $spec->{writer}) { |
65 | if ( |
66 | our $CAN_HAZ_XS |
67 | && $self->is_simple_set($name, $spec) |
68 | ) { |
69 | $methods{$writer} = $self->_generate_xs( |
70 | setters => $into, $writer, $name |
71 | ); |
51a3b106 |
72 | } else { |
e57f338d |
73 | $self->{captures} = {}; |
dcae37d3 |
74 | $methods{$writer} = |
75 | quote_sub "${into}::${writer}" |
76 | => $self->_generate_set($name, $spec) |
e57f338d |
77 | => delete $self->{captures} |
dcae37d3 |
78 | ; |
51a3b106 |
79 | } |
dcae37d3 |
80 | } |
8f333548 |
81 | if (my $pred = $spec->{predicate}) { |
dcae37d3 |
82 | $methods{$pred} = |
83 | quote_sub "${into}::${pred}" => |
84 | ' '.$self->_generate_simple_has('$_[0]', $name)."\n" |
85 | ; |
8f333548 |
86 | } |
87 | if (my $cl = $spec->{clearer}) { |
dcae37d3 |
88 | $methods{$cl} = |
89 | quote_sub "${into}::${cl}" => |
90 | " delete \$_[0]->{${\perlstring $name}}\n" |
91 | ; |
901efe1a |
92 | } |
4db3a740 |
93 | if (my $hspec = $spec->{handles}) { |
94 | my $asserter = $spec->{asserter} ||= '_assert_'.$name; |
95 | my @specs = do { |
96 | if (ref($hspec) eq 'ARRAY') { |
97 | map [ $_ => $_ ], @$hspec; |
98 | } elsif (ref($hspec) eq 'HASH') { |
99 | map [ $_ => ref($hspec->{$_}) ? @{$hspec->{$_}} : $hspec->{$_} ], |
100 | keys %$hspec; |
101 | } elsif (!ref($hspec)) { |
102 | map [ $_ => $_ ], Role::Tiny->methods_provided_by($hspec); |
103 | } else { |
104 | die "You gave me a handles of ${hspec} and I have no idea why"; |
105 | } |
106 | }; |
107 | foreach my $spec (@specs) { |
108 | my ($proxy, $target, @args) = @$spec; |
e57f338d |
109 | $self->{captures} = {}; |
4db3a740 |
110 | $methods{$proxy} = |
111 | quote_sub "${into}::${proxy}" => |
112 | $self->_generate_delegation($asserter, $target, \@args), |
e57f338d |
113 | delete $self->{captures} |
4db3a740 |
114 | ; |
115 | } |
116 | } |
117 | if (my $asserter = $spec->{asserter}) { |
e57f338d |
118 | $self->{captures} = {}; |
4db3a740 |
119 | $methods{$asserter} = |
120 | quote_sub "${into}::${asserter}" => |
121 | 'do { '.$self->_generate_get($name, $spec).qq! }||die "Attempted to access '${name}' but it is not set"!, |
e57f338d |
122 | delete $self->{captures} |
4db3a740 |
123 | ; |
124 | } |
dcae37d3 |
125 | \%methods; |
51a3b106 |
126 | } |
127 | |
3a9a65a4 |
128 | sub is_simple_attribute { |
129 | my ($self, $name, $spec) = @_; |
8f333548 |
130 | # clearer doesn't have to be listed because it doesn't |
131 | # affect whether defined/exists makes a difference |
901efe1a |
132 | !grep $spec->{$_}, |
32381de9 |
133 | qw(lazy default builder isa trigger predicate weak_ref); |
3a9a65a4 |
134 | } |
135 | |
daa05b62 |
136 | sub is_simple_get { |
137 | my ($self, $name, $spec) = @_; |
901efe1a |
138 | !($spec->{lazy} and ($spec->{default} or $spec->{builder})); |
daa05b62 |
139 | } |
140 | |
141 | sub is_simple_set { |
142 | my ($self, $name, $spec) = @_; |
32381de9 |
143 | !grep $spec->{$_}, qw(isa trigger weak_ref); |
daa05b62 |
144 | } |
145 | |
4ced3a94 |
146 | sub has_eager_default { |
147 | my ($self, $name, $spec) = @_; |
148 | (!$spec->{lazy} and ($spec->{default} or $spec->{builder})); |
149 | } |
150 | |
51a3b106 |
151 | sub _generate_get { |
46389f86 |
152 | my ($self, $name, $spec) = @_; |
153 | my $simple = $self->_generate_simple_get('$_[0]', $name); |
901efe1a |
154 | if ($self->is_simple_get($name, $spec)) { |
155 | $simple; |
156 | } else { |
157 | 'do { '.$self->_generate_use_default( |
158 | '$_[0]', $name, $spec, |
159 | $self->_generate_simple_has('$_[0]', $name), |
160 | ).'; '.$simple.' }'; |
161 | } |
46389f86 |
162 | } |
163 | |
164 | sub _generate_simple_has { |
165 | my ($self, $me, $name) = @_; |
166 | "exists ${me}->{${\perlstring $name}}"; |
167 | } |
168 | |
649ac264 |
169 | sub generate_get_default { |
170 | my $self = shift; |
e57f338d |
171 | $self->{captures} = {}; |
649ac264 |
172 | my $code = $self->_generate_get_default(@_); |
e57f338d |
173 | ($code, delete $self->{captures}); |
649ac264 |
174 | } |
175 | |
176 | sub _generate_use_default { |
177 | my ($self, $me, $name, $spec, $test) = @_; |
46389f86 |
178 | $self->_generate_simple_set( |
32381de9 |
179 | $me, $name, $spec, $self->_generate_get_default($me, $name, $spec) |
46389f86 |
180 | ).' unless '.$test; |
a16d301e |
181 | } |
182 | |
649ac264 |
183 | sub _generate_get_default { |
184 | my ($self, $me, $name, $spec) = @_; |
185 | $spec->{default} |
186 | ? $self->_generate_call_code($name, 'default', $me, $spec->{default}) |
187 | : "${me}->${\$spec->{builder}}" |
188 | } |
189 | |
a16d301e |
190 | sub generate_simple_get { |
901efe1a |
191 | my ($self, @args) = @_; |
192 | $self->_generate_simple_get(@args); |
a16d301e |
193 | } |
194 | |
195 | sub _generate_simple_get { |
196 | my ($self, $me, $name) = @_; |
197 | my $name_str = perlstring $name; |
198 | "${me}->{${name_str}}"; |
51a3b106 |
199 | } |
200 | |
201 | sub _generate_set { |
dcae37d3 |
202 | my ($self, $name, $spec) = @_; |
901efe1a |
203 | if ($self->is_simple_set($name, $spec)) { |
32381de9 |
204 | $self->_generate_simple_set('$_[0]', $name, $spec, '$_[1]'); |
6d377074 |
205 | } else { |
901efe1a |
206 | my ($trigger, $isa_check) = @{$spec}{qw(trigger isa)}; |
32381de9 |
207 | my $simple = $self->_generate_simple_set('$self', $name, $spec, '$value'); |
e57f338d |
208 | my $code = "do { my (\$self, \$value) = \@_;\n"; |
901efe1a |
209 | if ($isa_check) { |
210 | $code .= |
e57f338d |
211 | " ".$self->_generate_isa_check($name, '$value', $isa_check).";\n"; |
901efe1a |
212 | } |
213 | if ($trigger) { |
e57f338d |
214 | my $fire = $self->_generate_trigger($name, '$self', '$value', $trigger); |
901efe1a |
215 | $code .= |
e57f338d |
216 | " ".$simple.";\n ".$fire.";\n" |
901efe1a |
217 | ." \$value;\n"; |
218 | } else { |
219 | $code .= " ".$simple.";\n"; |
220 | } |
221 | $code .= " }"; |
222 | $code; |
a16d301e |
223 | } |
a16d301e |
224 | } |
901efe1a |
225 | |
a16d301e |
226 | sub generate_trigger { |
227 | my $self = shift; |
e57f338d |
228 | $self->{captures} = {}; |
a16d301e |
229 | my $code = $self->_generate_trigger(@_); |
e57f338d |
230 | ($code, delete $self->{captures}); |
a16d301e |
231 | } |
232 | |
233 | sub _generate_trigger { |
234 | my ($self, $name, $obj, $value, $trigger) = @_; |
6d377074 |
235 | $self->_generate_call_code($name, 'trigger', "${obj}, ${value}", $trigger); |
236 | } |
237 | |
238 | sub generate_isa_check { |
901efe1a |
239 | my ($self, @args) = @_; |
e57f338d |
240 | $self->{captures} = {}; |
901efe1a |
241 | my $code = $self->_generate_isa_check(@args); |
e57f338d |
242 | ($code, delete $self->{captures}); |
6d377074 |
243 | } |
244 | |
245 | sub _generate_isa_check { |
246 | my ($self, $name, $value, $check) = @_; |
247 | $self->_generate_call_code($name, 'isa_check', $value, $check); |
248 | } |
249 | |
250 | sub _generate_call_code { |
251 | my ($self, $name, $type, $values, $sub) = @_; |
252 | if (my $quoted = quoted_from_sub($sub)) { |
625d6219 |
253 | my $code = $quoted->[1]; |
e57f338d |
254 | my $at_ = '@_ = ('.$values.');'; |
8c6626cf |
255 | if (my $captures = $quoted->[2]) { |
6d377074 |
256 | my $cap_name = qq{\$${type}_captures_for_${name}}; |
8c6626cf |
257 | $self->{captures}->{$cap_name} = \$captures; |
e57f338d |
258 | Sub::Quote::inlinify( |
259 | $code, $values, Sub::Quote::capture_unroll($cap_name, $captures, 6) |
260 | ); |
261 | } else { |
262 | Sub::Quote::inlinify($code, $values); |
8c6626cf |
263 | } |
901efe1a |
264 | } else { |
265 | my $cap_name = qq{\$${type}_for_${name}}; |
266 | $self->{captures}->{$cap_name} = \$sub; |
267 | "${cap_name}->(${values})"; |
625d6219 |
268 | } |
a16d301e |
269 | } |
270 | |
3a9a65a4 |
271 | sub generate_populate_set { |
272 | my $self = shift; |
e57f338d |
273 | $self->{captures} = {}; |
3a9a65a4 |
274 | my $code = $self->_generate_populate_set(@_); |
e57f338d |
275 | ($code, delete $self->{captures}); |
3a9a65a4 |
276 | } |
277 | |
278 | sub _generate_populate_set { |
279 | my ($self, $me, $name, $spec, $source, $test) = @_; |
4ced3a94 |
280 | if ($self->has_eager_default($name, $spec)) { |
3a9a65a4 |
281 | my $get_indent = ' ' x ($spec->{isa} ? 6 : 4); |
d02da2bc |
282 | my $get_default = $self->_generate_get_default( |
283 | '$new', $_, $spec |
284 | ); |
3a9a65a4 |
285 | my $get_value = |
d02da2bc |
286 | defined($spec->{init_arg}) |
287 | ? "(\n${get_indent} ${test}\n${get_indent} ? ${source}\n${get_indent} : " |
288 | .$get_default |
289 | ."\n${get_indent})" |
290 | : $get_default; |
3a9a65a4 |
291 | ($spec->{isa} |
292 | ? " {\n my \$value = ".$get_value.";\n " |
5d349892 |
293 | .$self->_generate_isa_check( |
294 | $name, '$value', $spec->{isa} |
295 | ).";\n" |
32381de9 |
296 | .' '.$self->_generate_simple_set($me, $name, $spec, '$value').";\n" |
5d349892 |
297 | ." }\n" |
32381de9 |
298 | : ' '.$self->_generate_simple_set($me, $name, $spec, $get_value).";\n" |
3a9a65a4 |
299 | ) |
300 | .($spec->{trigger} |
301 | ? ' ' |
5d349892 |
302 | .$self->_generate_trigger( |
303 | $name, $me, $self->_generate_simple_get($me, $name), |
304 | $spec->{trigger} |
305 | )." if ${test};\n" |
3a9a65a4 |
306 | : '' |
307 | ); |
308 | } else { |
309 | " if (${test}) {\n" |
310 | .($spec->{isa} |
311 | ? " " |
5d349892 |
312 | .$self->_generate_isa_check( |
313 | $name, $source, $spec->{isa} |
314 | ).";\n" |
3a9a65a4 |
315 | : "" |
316 | ) |
32381de9 |
317 | ." ".$self->_generate_simple_set($me, $name, $spec, $source).";\n" |
3a9a65a4 |
318 | .($spec->{trigger} |
5d349892 |
319 | ? " " |
320 | .$self->_generate_trigger( |
321 | $name, $me, $self->_generate_simple_get($me, $name), |
322 | $spec->{trigger} |
323 | ).";\n" |
324 | : "" |
3a9a65a4 |
325 | ) |
326 | ." }\n"; |
327 | } |
328 | } |
329 | |
330 | sub generate_multi_set { |
331 | my ($self, $me, $to_set, $from) = @_; |
332 | "\@{${me}}{qw(${\join ' ', @$to_set})} = $from"; |
333 | } |
334 | |
a16d301e |
335 | sub _generate_simple_set { |
32381de9 |
336 | my ($self, $me, $name, $spec, $value) = @_; |
a16d301e |
337 | my $name_str = perlstring $name; |
32381de9 |
338 | my $simple = "${me}->{${name_str}} = ${value}"; |
339 | if ($spec->{weak_ref}) { |
340 | require Scalar::Util; |
341 | "Scalar::Util::weaken(${simple})"; |
342 | } else { |
343 | $simple; |
344 | } |
51a3b106 |
345 | } |
346 | |
347 | sub _generate_getset { |
a16d301e |
348 | my ($self, $name, $spec) = @_; |
dcae37d3 |
349 | q{(@_ > 1}."\n ? ".$self->_generate_set($name, $spec) |
f537c364 |
350 | ."\n : ".$self->_generate_get($name, $spec)."\n )"; |
51a3b106 |
351 | } |
352 | |
4db3a740 |
353 | sub _generate_delegation { |
354 | my ($self, $asserter, $target, $args) = @_; |
355 | my $arg_string = do { |
356 | if (@$args) { |
357 | # I could, I reckon, linearise out non-refs here using perlstring |
358 | # plus something to check for numbers but I'm unsure if it's worth it |
359 | $self->{captures}{'@curries'} = $args; |
360 | '@curries, @_'; |
361 | } else { |
362 | '@_'; |
363 | } |
364 | }; |
365 | "shift->${asserter}->${target}(${arg_string});"; |
366 | } |
367 | |
daa05b62 |
368 | sub _generate_xs { |
dcae37d3 |
369 | my ($self, $type, $into, $name, $slot) = @_; |
daa05b62 |
370 | Class::XSAccessor->import( |
e6f2e914 |
371 | class => $into, |
dcae37d3 |
372 | $type => { $name => $slot } |
daa05b62 |
373 | ); |
901efe1a |
374 | $into->can($name); |
daa05b62 |
375 | } |
376 | |
51a3b106 |
377 | 1; |