lib/DBIx/Class/InflateColumn.pm

   1 package DBIx::Class::InflateColumn;
   2
   3 use strict;
   4 use warnings;
   5
   6 sub inflate_column {
   7   my ($self, $col, $attrs) = @_;
   8   die "No such column $col to inflate" unless exists $self->_columns->{$col};
   9   die "inflate_column needs attr hashref" unless ref $attrs eq 'HASH';
  10   $self->_columns->{$col}{_inflate_info} = $attrs;
  11   $self->mk_group_accessors('inflated_column' => $col);
  12   return 1;
  13 }
  14
  15 sub _inflated_column {
  16   my ($self, $col, $value) = @_;
  17   return $value unless defined $value; # NULL is NULL is NULL
  18   return $value unless exists $self->_columns->{$col}{_inflate_info};
  19   return $value unless exists $self->_columns->{$col}{_inflate_info}{inflate};
  20   my $inflate = $self->_columns->{$col}{_inflate_info}{inflate};
  21   return $inflate->($value, $self);
  22 }
  23
  24 sub _deflated_column {
  25   my ($self, $col, $value) = @_;
  26   return $value unless ref $value; # If it's not an object, don't touch it
  27   return $value unless exists $self->_columns->{$col}{_inflate_info};
  28   return $value unless exists $self->_columns->{$col}{_inflate_info}{deflate};
  29   my $deflate = $self->_columns->{$col}{_inflate_info}{deflate};
  30   return $deflate->($value, $self);
  31 }
  32
  33 sub get_inflated_column {
  34   my ($self, $col) = @_;
  35   $self->throw("$col is not an inflated column") unless
  36     exists $self->_columns->{$col}{_inflate_info};
  37
  38   return $self->{_inflated_column}{$col}
  39     if exists $self->{_inflated_column}{$col};
  40   return $self->{_inflated_column}{$col} =
  41            $self->_inflated_column($col, $self->get_column($col));
  42 }
  43
  44 sub set_inflated_column {
  45   my ($self, $col, @rest) = @_;
  46   my $ret = $self->store_inflated_column($col, @rest);
  47   $self->{_dirty_columns}{$col} = 1;
  48   return $ret;
  49 }
  50
  51 sub store_inflated_column {
  52   my ($self, $col, $obj) = @_;
  53   unless (ref $obj) {
  54     delete $self->{_inflated_column}{$col};
  55     return $self->store_column($col, $obj);
  56   }
  57
  58   my $deflated = $self->_deflated_column($col, $obj);
  59            # Do this now so we don't store if it's invalid
  60
  61   $self->{_inflated_column}{$col} = $obj;
  62   #warn "Storing $obj: ".($obj->_ident_values)[0];
  63   $self->store_column($col, $deflated);
  64   return $obj;
  65 }
  66
  67 sub new {
  68   my ($class, $attrs, @rest) = @_;
  69   $attrs ||= {};
  70   foreach my $key (keys %$attrs) {
  71     if (ref $attrs->{$key} && exists $class->_columns->{$key}{_inflate_info}) {
  72       $attrs->{$key} = $class->_deflated_column($key, $attrs->{$key});
  73     }
  74   }
  75   return $class->NEXT::ACTUAL::new($attrs, @rest);
  76 }
  77
  78 1;