1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
package # hide from PAUSE
DBIx::Class::Relationship::Accessor;
use strict;
use warnings;
sub register_relationship {
my ($class, $rel, $info) = @_;
if (my $acc_type = $info->{attrs}{accessor}) {
$class->add_relationship_accessor($rel => $acc_type);
}
$class->next::method($rel => $info);
}
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_exception("No such column $rel to filter")
unless $class->has_column($rel);
my $f_class = $class->relationship_info($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_exception("$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{"${rel}_rs"} = sub { shift->search_related_rs($rel, @_) };
$meth{"add_to_${rel}"} = sub { shift->create_related($rel, @_); };
} else {
$class->throw_exception("No such relationship accessor type $acc_type");
}
{
no strict 'refs';
no warnings 'redefine';
foreach my $meth (keys %meth) {
*{"${class}::${meth}"} = $meth{$meth};
}
}
}
sub new {
my ($class, $attrs, @rest) = @_;
my ($related, $info);
foreach my $key (keys %{$attrs||{}}) {
next unless $info = $class->relationship_info($key);
$related->{$key} = delete $attrs->{$key}
if ref $attrs->{$key}
&& $info->{attrs}{accessor}
&& $info->{attrs}{accessor} eq 'single';
}
my $obj = $class->next::method($attrs, @rest);
if ($related) {
$obj->{_relationship_data} = $related;
foreach my $rel (keys %$related) {
$obj->set_from_related($rel, $related->{$rel});
}
}
return $obj;
}
sub update {
my ($obj, $attrs, @rest) = @_;
my $info;
foreach my $key (keys %{$attrs||{}}) {
next unless $info = $obj->relationship_info($key);
if (ref $attrs->{$key} && $info->{attrs}{accessor}
&& $info->{attrs}{accessor} eq 'single') {
my $rel = delete $attrs->{$key};
$obj->set_from_related($key => $rel);
$obj->{_relationship_data}{$key} = $rel;
}
}
return $obj->next::method($attrs, @rest);
}
1;
|
