Added support for temp columns and ->make_read_only

[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / CDBICompat / TempColumns.pm

diff --git a/lib/DBIx/Class/CDBICompat/TempColumns.pm b/lib/DBIx/Class/CDBICompat/TempColumns.pm
new file mode 100644 (file)
index 0000000..64d6d20
--- /dev/null
+++ b/lib/DBIx/Class/CDBICompat/TempColumns.pm
@@ -0,0 +1,64 @@
+package DBIx::Class::CDBICompat::TempColumns;
+
+use strict;
+use warnings;
+use base qw/Class::Data::Inheritable/;
+
+__PACKAGE__->mk_classdata('_temp_columns' => { });
+
+sub _add_column_group {
+  my ($class, $group, @cols) = @_;
+  if ($group eq 'TEMP') {
+    $class->_register_column_group($group => @cols);
+    $class->mk_group_accessors('temp' => @cols);
+    my %tmp = %{$class->_temp_columns};
+    $tmp{$_} = 1 for @cols;
+    $class->_temp_columns(\%tmp);
+  } else {
+    return $class->NEXT::ACTUAL::_add_column_group($group, @cols);
+  }
+}
+
+sub new {
+  my ($class, $attrs, @rest) = @_;
+  my %temp;
+  foreach my $key (keys %$attrs) {
+    $temp{$key} = delete $attrs->{$key} if $class->_temp_columns->{$key};
+  }
+  my $new = $class->NEXT::ACTUAL::new($attrs, @rest);
+  foreach my $key (keys %temp) {
+    $new->set_temp($key, $temp{$key});
+  }
+  return $new;
+}
+
+
+sub find_column {
+  my ($class, $col, @rest) = @_;
+  return $col if $class->_temp_columns->{$col};
+  return $class->NEXT::ACTUAL::find_column($col, @rest);
+}
+
+sub get_temp {
+  my ($self, $column) = @_;
+  $self->throw( "Can't fetch data as class method" ) unless ref $self;
+  $self->throw( "No such TEMP column '${column}'" ) unless $self->_temp_columns->{$column} ;
+  return $self->{_temp_column_data}{$column}
+    if exists $self->{_temp_column_data}{$column};
+  return undef;
+}
+
+sub set_temp {
+  my ($self, $column, $value) = @_;
+  $self->throw( "No such TEMP column '${column}'" )
+    unless $self->_temp_columns->{$column};
+  $self->throw( "set_temp called for ${column} without value" )
+    if @_ < 3;
+  return $self->{_temp_column_data}{$column} = $value;
+}
+
+sub has_real_column {
+  return 1 if shift->_columns->{shift};
+}
+
+1;