Module: ForestLiana::ActiveRecordOverride::Associations::JoinDependency

Defined in:
lib/forest_liana/active_record_override.rb

Instance Method Summary collapse

Instance Method Details

#apply_column_aliases(relation) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/forest_liana/active_record_override.rb', line 6

def apply_column_aliases(relation)
  if !(@join_root_alias = relation.select_values.empty?) &&
    relation.select_values.first.to_s == '_forest_admin_eager_load'

    relation.select_values.shift
    used_cols = {}
    # Find and expand out all column names being used in select(...)
    new_select_values = relation.select_values.map(&:to_s).each_with_object([]) do |col, select|
      unless col.include?(' ') # Pass it through if it's some expression (No chance for a simple column reference)
        col = if (col_parts = col.split('.')).length == 1
          [col]
        else
          [col_parts[0..-2].join('.'), col_parts.last]
        end
        used_cols[col] = nil
      end
      select << col
    end

    if new_select_values.present?
      relation.select_values = new_select_values
    else
      relation.select_values.clear
    end

    @aliases ||= ActiveRecord::Associations::JoinDependency::Aliases.new(join_root.each_with_index.map do |join_part, i|
      join_alias = join_part.table&.table_alias || join_part.table_name
      keys = [join_part.base_klass.primary_key] # Always include the primary key

      # # %%% Optional to include all foreign keys:
      # keys.concat(join_part.base_klass.reflect_on_all_associations.select { |a| a.belongs_to? }.map(&:foreign_key))
      # Add foreign keys out to referenced tables that we belongs_to
      join_part.children.each { |child| keys << child.reflection.foreign_key if child.reflection.belongs_to? }

      # Add the foreign key that got us here -- "the train we rode in on" -- if we arrived from
      # a has_many or has_one:
      if join_part.is_a?(ActiveRecord::Associations::JoinDependency::JoinAssociation) &&
        !join_part.reflection.belongs_to?
        keys << join_part.reflection.foreign_key
      end
      keys = keys.compact # In case we're using composite_primary_keys
      j = 0
      columns = join_part.column_names.each_with_object([]) do |column_name, s|
        # Include columns chosen in select(...) as well as the PK and any relevant FKs
        if used_cols.keys.find { |c| (c.length == 1 || c.first == join_alias) && c.last == column_name } ||
          keys.find { |c| c == column_name }
          s << ActiveRecord::Associations::JoinDependency::Aliases::Column.new(column_name, "t#{i}_r#{j}")
        end
        j += 1
      end
      ActiveRecord::Associations::JoinDependency::Aliases::Table.new(join_part, columns)
    end)
    relation.select_values.clear
  end
  relation._select!(-> { aliases.columns })
end