035661a3c3f7b681fb7b96ddb50c85ab109ca3f0
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / Relationship / Accessor.pm
1 package # hide from PAUSE
2     DBIx::Class::Relationship::Accessor;
3
4 use strict;
5 use warnings;
6
7 sub register_relationship {
8   my ($class, $rel, $info) = @_;
9   if (my $acc_type = $info->{attrs}{accessor}) {
10     $class->add_relationship_accessor($rel => $acc_type);
11   }
12   $class->next::method($rel => $info);
13 }
14
15 sub add_relationship_accessor {
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 {
27         my $val = $self->find_related($rel, {}, {});
28         return unless $val;
29         return $self->{_relationship_data}{$rel} = $val;
30       }
31     };
32   } elsif ($acc_type eq 'filter') {
33     $class->throw_exception("No such column $rel to filter")
34        unless $class->has_column($rel);
35     my $f_class = $class->relationship_info($rel)->{class};
36     $class->inflate_column($rel,
37       { inflate => sub {
38           my ($val, $self) = @_;
39           return $self->find_or_create_related($rel, {}, {});
40         },
41         deflate => sub {
42           my ($val, $self) = @_;
43           $self->throw_exception("$val isn't a $f_class") unless $val->isa($f_class);
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, @_) };
51     $meth{"add_to_${rel}"} = sub { shift->create_related($rel, @_); };
52   } else {
53     $class->throw_exception("No such relationship accessor type $acc_type");
54   }
55   {
56     no strict 'refs';
57     no warnings 'redefine';
58     foreach my $meth (keys %meth) {
59       *{"${class}::${meth}"} = $meth{$meth};
60     }
61   }
62 }
63
64 1;