Commit | Line | Data |
75d07914 |
1 | package # hide from PAUSE |
c0e7b4e5 |
2 | DBIx::Class::Relationship::ProxyMethods; |
b28cc0ba |
3 | |
4 | use strict; |
5 | use warnings; |
8d73fcd4 |
6 | use base 'DBIx::Class'; |
7 | use DBIx::Class::_Util 'quote_sub'; |
8 | use namespace::clean; |
b28cc0ba |
9 | |
8273e845 |
10 | our %_pod_inherit_config = |
044e70c7 |
11 | ( |
12 | class_map => { 'DBIx::Class::Relationship::ProxyMethods' => 'DBIx::Class::Relationship' } |
13 | ); |
14 | |
71e65b39 |
15 | sub register_relationship { |
16 | my ($class, $rel, $info) = @_; |
97c96475 |
17 | if (my $proxy_args = $info->{attrs}{proxy}) { |
18 | $class->proxy_to_related($rel, $proxy_args); |
b28cc0ba |
19 | } |
71e65b39 |
20 | $class->next::method($rel, $info); |
b28cc0ba |
21 | } |
22 | |
223b8fe3 |
23 | sub proxy_to_related { |
97c96475 |
24 | my ($class, $rel, $proxy_args) = @_; |
25 | my %proxy_map = $class->_build_proxy_map_from($proxy_args); |
8d73fcd4 |
26 | |
27 | quote_sub "${class}::$_", sprintf( <<'EOC', $rel, $proxy_map{$_} ) |
28 | my $self = shift; |
29 | my $relobj = $self->%1$s; |
30 | if (@_ && !defined $relobj) { |
31 | $relobj = $self->create_related( %1$s => { %2$s => $_[0] } ); |
32 | @_ = (); |
33 | } |
34 | $relobj ? $relobj->%2$s(@_) : undef; |
35 | EOC |
36 | for keys %proxy_map |
223b8fe3 |
37 | } |
38 | |
97c96475 |
39 | sub _build_proxy_map_from { |
40 | my ( $class, $proxy_arg ) = @_; |
41 | my $ref = ref $proxy_arg; |
42 | |
43 | if ($ref eq 'HASH') { |
44 | return %$proxy_arg; |
45 | } |
46 | elsif ($ref eq 'ARRAY') { |
47 | return map { |
48 | (ref $_ eq 'HASH') |
49 | ? (%$_) |
50 | : ($_ => $_) |
51 | } @$proxy_arg; |
52 | } |
53 | elsif ($ref) { |
54 | $class->throw_exception("Unable to process the 'proxy' argument $proxy_arg"); |
55 | } |
56 | else { |
57 | return ( $proxy_arg => $proxy_arg ); |
58 | } |
59 | } |
60 | |
b28cc0ba |
61 | 1; |