Commit | Line | Data |
75d07914 |
1 | package # hide from PAUSE |
c0e7b4e5 |
2 | DBIx::Class::Relationship::Accessor; |
4a07648a |
3 | |
4 | use strict; |
5 | use warnings; |
6 | |
71e65b39 |
7 | sub register_relationship { |
8 | my ($class, $rel, $info) = @_; |
9 | if (my $acc_type = $info->{attrs}{accessor}) { |
223b8fe3 |
10 | $class->add_relationship_accessor($rel => $acc_type); |
4a07648a |
11 | } |
71e65b39 |
12 | $class->next::method($rel => $info); |
4a07648a |
13 | } |
14 | |
223b8fe3 |
15 | sub add_relationship_accessor { |
4a07648a |
16 | my ($class, $rel, $acc_type) = @_; |
17 | my %meth; |
18 | if ($acc_type eq 'single') { |
19 | $meth{$rel} = sub { |
20 | my $self = shift; |
21 | if (@_) { |
22 | $self->set_from_related($rel, @_); |
23 | return $self->{_relationship_data}{$rel} = $_[0]; |
24 | } elsif (exists $self->{_relationship_data}{$rel}) { |
25 | return $self->{_relationship_data}{$rel}; |
26 | } else { |
1a14aa3f |
27 | my $val = $self->find_related($rel, {}, {}); |
b28cc0ba |
28 | return unless $val; |
29 | return $self->{_relationship_data}{$rel} = $val; |
4a07648a |
30 | } |
31 | }; |
32 | } elsif ($acc_type eq 'filter') { |
701da8c4 |
33 | $class->throw_exception("No such column $rel to filter") |
103647d5 |
34 | unless $class->has_column($rel); |
4685e006 |
35 | my $f_class = $class->relationship_info($rel)->{class}; |
4a07648a |
36 | $class->inflate_column($rel, |
37 | { inflate => sub { |
38 | my ($val, $self) = @_; |
e60dc79f |
39 | return $self->find_or_new_related($rel, {}, {}); |
4a07648a |
40 | }, |
41 | deflate => sub { |
42 | my ($val, $self) = @_; |
701da8c4 |
43 | $self->throw_exception("$val isn't a $f_class") unless $val->isa($f_class); |
4a07648a |
44 | return ($val->_ident_values)[0]; |
45 | # WARNING: probably breaks for multi-pri sometimes. FIXME |
46 | } |
47 | } |
48 | ); |
49 | } elsif ($acc_type eq 'multi') { |
50 | $meth{$rel} = sub { shift->search_related($rel, @_) }; |
5b89a768 |
51 | $meth{"${rel}_rs"} = sub { shift->search_related_rs($rel, @_) }; |
4a07648a |
52 | $meth{"add_to_${rel}"} = sub { shift->create_related($rel, @_); }; |
53 | } else { |
701da8c4 |
54 | $class->throw_exception("No such relationship accessor type $acc_type"); |
4a07648a |
55 | } |
56 | { |
57 | no strict 'refs'; |
58 | no warnings 'redefine'; |
59 | foreach my $meth (keys %meth) { |
60 | *{"${class}::${meth}"} = $meth{$meth}; |
61 | } |
62 | } |
63 | } |
64 | |
65 | 1; |