package DBIx::Class::Relationship::Accessor;
use strict;
use warnings;
sub add_relationship {
my ($class, $rel, @rest) = @_;
my $ret = $class->NEXT::ACTUAL::add_relationship($rel => @rest);
my $rel_obj = $class->_relationships->{$rel};
if (my $acc_type = $rel_obj->{attrs}{accessor}) {
$class->add_relationship_accessor($rel => $acc_type);
}
return $ret;
}
sub add_relationship_accessor {
my ($class, $rel, $acc_type) = @_;
my %meth;
if ($acc_type eq 'single') {
$meth{$rel} = sub {
my $self = shift;
if (@_) {
$self->set_from_related($rel, @_);
return $self->{_relationship_data}{$rel} = $_[0];
} elsif (exists $self->{_relationship_data}{$rel}) {
return $self->{_relationship_data}{$rel};
} else {
my $val = $self->find_related($rel, {}, {});
return unless $val;
return $self->{_relationship_data}{$rel} = $val;
}
};
} elsif ($acc_type eq 'filter') {
$class->throw("No such column $rel to filter")
unless exists $class->_columns->{$rel};
my $f_class = $class->_relationships->{$rel}{class};
$class->inflate_column($rel,
{ inflate => sub {
my ($val, $self) = @_;
return $self->find_or_create_related($rel, {}, {});
},
deflate => sub {
my ($val, $self) = @_;
$self->throw("$val isn't a $f_class") unless $val->isa($f_class);
return ($val->_ident_values)[0];
# WARNING: probably breaks for multi-pri sometimes. FIXME
}
}
);
} elsif ($acc_type eq 'multi') {
$meth{$rel} = sub { shift->search_related($rel, @_) };
$meth{"add_to_${rel}"} = sub { shift->create_related($rel, @_); };
} else {
$class->throw("No such relationship accessor type $acc_type");
}
{
no strict 'refs';
no warnings 'redefine';
foreach my $meth (keys %meth) {
*{"${class}::${meth}"} = $meth{$meth};
}
}
}
1;