lib/DBIx/Class/CDBICompat/ImaDBI.pm

   1 package DBIx::Class::CDBICompat::ImaDBI;
   2
   3 use strict;
   4 use warnings;
   5
   6 use NEXT;
   7 use base qw/Class::Data::Inheritable/;
   8
   9 __PACKAGE__->mk_classdata('_transform_sql_handler_order'
  10                             => [ qw/TABLE ESSENTIAL JOIN/ ] );
  11
  12 __PACKAGE__->mk_classdata('_transform_sql_handlers' =>
  13   {
  14     'TABLE' =>
  15       sub {
  16         my ($self, $class, $data) = @_;
  17         return $class->_table_name unless $data;
  18         my ($f_class, $alias) = split(/=/, $data);
  19         $f_class ||= $class;
  20         $self->{_classes}{$alias} = $f_class;
  21         return $f_class->_table_name." ${alias}";
  22       },
  23     'ESSENTIAL' =>
  24       sub {
  25         my ($self, $class, $data) = @_;
  26         return join(' ', $class->columns('Essential')) unless $data;
  27         return join(' ', $self->{_classes}{$data}->columns('Essential'));
  28       },
  29     'JOIN' =>
  30       sub {
  31         my ($self, $class, $data) = @_;
  32         my ($from, $to) = split(/ /, $data);
  33         my ($from_class, $to_class) = @{$self->{_classes}}{$from, $to};
  34         my ($rel_obj) = grep { $_->{class} && $_->{class} eq $to_class }
  35                           values %{ $from_class->_relationships };
  36         unless ($rel_obj) {
  37           ($from, $to) = ($to, $from);
  38           ($from_class, $to_class) = ($to_class, $from_class);
  39           ($rel_obj) = grep { $_->{class} && $_->{class} eq $to_class }
  40                          values %{ $from_class->_relationships };
  41         }
  42         $self->throw( "No relationship to JOIN from ${from_class} to ${to_class}" )
  43           unless $rel_obj;
  44         my $attrs = {
  45           %$self,
  46           _aliases => { self => $from, foreign => $to },
  47           _action => 'join',
  48         };
  49         my $join = $from_class->_cond_resolve($rel_obj->{cond}, $attrs);
  50         return $join;
  51       }
  52
  53   } );
  54
  55 sub db_Main {
  56   return $_[0]->_get_dbh;
  57 }
  58
  59 sub _dbi_connect {
  60   my ($class, @info) = @_;
  61   $info[3] = { %{ $info[3] || {}} };
  62   $info[3]->{RootClass} = 'DBIx::ContextualFetch';
  63   return $class->NEXT::_dbi_connect(@info);
  64 }
  65
  66 sub __driver {
  67   return $_[0]->_get_dbh->{Driver}->{Name};
  68 }
  69
  70 sub set_sql {
  71   my ($class, $name, $sql) = @_;
  72   my $table = $class->_table_name;
  73   #$sql =~ s/__TABLE__/$table/;
  74   no strict 'refs';
  75   *{"${class}::sql_${name}"} =
  76     sub {
  77       my $sql = $sql;
  78       my $class = shift;
  79       return $class->_sql_to_sth($class->transform_sql($sql, @_));
  80     };
  81   if ($sql =~ /select/i) {
  82     my $meth = "sql_${name}";
  83     *{"${class}::search_${name}"} =
  84       sub {
  85         my ($class, @args) = @_;
  86         $class->sth_to_objects($class->$meth, \@args);
  87       };
  88   }
  89 }
  90
  91 sub transform_sql {
  92   my ($class, $sql, @args) = @_;
  93   my $table = $class->_table_name;
  94   my $attrs = { };
  95   foreach my $key (@{$class->_transform_sql_handler_order}) {
  96     my $h = $class->_transform_sql_handlers->{$key};
  97     $sql =~ s/__$key(?:\(([^\)]+)\))?__/$h->($attrs, $class, $1)/eg;
  98   }
  99   return sprintf($sql, @args);
 100 }
 101
 102 1;