lib/DBIx/Class/ResultSet/Role/DQMethods.pm

   1 package DBIx::Class::ResultSet::Role::DQMethods;
   2
   3 use Data::Query::ExprHelpers;
   4 use Safe::Isa;
   5 use Moo::Role;
   6
   7 sub _dq_converter {
   8   shift->result_source->schema->storage->sql_maker->converter;
   9 }
  10
  11 sub where {
  12   my ($self, $where) = @_;
  13   if ($where->$_isa('Data::Query::ExprBuilder')) {
  14     return $self->_apply_dq_where($where->{expr});
  15   } elsif (ref($where) eq 'HASH') {
  16     return $self->_apply_dq_where(
  17              $self->_dq_converter->_where_to_dq($where)
  18            );
  19   }
  20   die "Argument to ->where must be ExprBuilder or SQL::Abstract hashref, got: "
  21       .(defined($where) ? $where : 'undef');
  22 }
  23
  24 sub _apply_dq_where {
  25   my ($self, $expr) = @_;
  26   my ($mapped, $need_join) = $self->_remap_identifiers($expr);
  27   $self->search_rs(\$mapped, (@$need_join ? { join => $need_join } : ()));
  28 }
  29
  30 sub _remap_identifiers {
  31   my ($self, $dq) = @_;
  32   my $map = {
  33     '' => {
  34       -alias => $self->current_source_alias,
  35       -rsrc => $self->result_source,
  36     }
  37   };
  38   my $attrs = $self->_resolved_attrs;
  39   foreach my $j ( @{$attrs->{from}}[1 .. $#{$attrs->{from}} ] ) {
  40     next unless $j->[0]{-alias};
  41     next unless $j->[0]{-join_path};
  42     my $p = $map;
  43     $p = $p->{$_} ||= {} for map { keys %$_ } @{$j->[0]{-join_path}};
  44     $p->{''} = $j->[0];
  45   }
  46
  47   my $seen_join = { %{$attrs->{seen_join}||{}} };
  48   my $storage = $self->result_source->storage;
  49   my @need_join;
  50   my %seen_op;
  51   my $mapped = map_dq_tree {
  52     return $_ unless is_Identifier;
  53     my @el = @{$_->{elements}};
  54     my $last = pop @el;
  55     my $p = $map;
  56     $p = $p->{$_} ||= {} for @el;
  57     unless ($p->{''}) {
  58       my $need = my $j = {};
  59       $j = $j->{$_} = {} for @el;
  60       my $rsrc = $map->{''}{-rsrc};
  61       $rsrc = $rsrc->related_source($_) for @el;
  62       push @need_join, $need;
  63       my $alias = $storage->relname_to_table_alias(
  64         $el[-1], ++$seen_join->{$el[-1]}
  65       );
  66       $p->{''} = { -alias => $alias, -rsrc => $rsrc };
  67     }
  68     my $info = $p->{''};
  69     if ($info->{-rsrc}->has_relationship($last)) {
  70       die "Invalid name on ".(join(',',@el)||'me').": $last is a relationship";
  71     }
  72     my $col_map = $info->{-column_mapping} ||= do {
  73       my $colinfo = $info->{-rsrc}->columns_info;
  74       +{ map +(($colinfo->{$_}{rename_for_dq}||$_) => $_), keys %$colinfo }
  75     };
  76     die "Invalid name on ".(join(',',@el)||'me').": $last"
  77       unless $col_map->{$last};
  78     return Identifier($info->{-alias}, $col_map->{$last});
  79   } $dq;
  80   return ($mapped, \@need_join);
  81 }
  82
  83 1;