fe8521466a9ef2c25a66f7149e31f6ab614b0708
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / CDBICompat / HasA.pm
1 package DBIx::Class::CDBICompat::HasA;
2
3 use strict;
4 use warnings;
5
6 sub has_a {
7   my ($self, $col, $f_class) = @_;
8   $self->throw( "No such column ${col}" ) unless $self->_columns->{$col};
9   eval "require $f_class";
10   my ($pri, $too_many) = keys %{ $f_class->_primaries };
11   $self->throw( "has_a only works with a single primary key; ${f_class} has more" )
12     if $too_many;
13   $self->add_relationship($col, $f_class,
14                             { "foreign.${pri}" => "self.${col}" },
15                             { _type => 'has_a' } );
16   $self->inflate_column($col,
17     { inflate => sub { 
18         my ($val, $self) = @_;
19         return ($self->search_related($col, {}, {}))[0]
20           || $f_class->new({ $pri => $val }); },
21       deflate => sub {
22         my ($val, $self) = @_;
23         $self->throw("$val isn't a $f_class") unless $val->isa($f_class);
24         return ($val->_ident_values)[0] } } );
25   return 1;
26 }
27
28 1;