Contents

# frozen_string_literal: true
module Praxis
  module Extensions
    module FieldSelection
      class ActiveRecordQuerySelector
        attr_reader :selector, :ds, :top_model, :resolved, :root
        # Gets a dataset, a selector...and should return a dataset with the selector definition applied.
        def initialize(ds:, model:, selectors:, resolved:)
          @selector = selectors
          @ds = ds
          @top_model = model
          @resolved = resolved
          @seen = Set.new
          @root = model.table_name
        end

        def add_select(ds:, model:, table_name:)
          if (fields = fields_for(model))
            # Note, let's always add the pk fields so that associations can load properly
            fields = fields | [model.primary_key.to_sym]
            ds.select(*fields)
          else
            ds
          end
        end

        def generate
          # TODO: unfortunately, I think we can only control the select clauses for the top model 
          # (as I'm not sure ActiveRecord supports expressing it in the join...)
          @ds = add_select(ds: ds, model: top_model, table_name: root)

          @ds.includes(_eager(top_model, resolved) )
        end

        def _eager(model, resolved)
            # Cannot select fields in included rels...boooo :()
            # d = add_select(ds: dset, model: model, table_name: model.table_name)
            tracks = only_assoc_for(model, resolved)
            tracks.inject([]) do |dataset, track|
              next dataset if @seen.include?([model, track])
              @seen << [model, track]
              assoc_model = model.associations[track][:model]
              dataset << { track => _eager(assoc_model, resolved[track]) }
            end
        end

        def only_assoc_for(model, hash)
          hash.keys.reject { |assoc| model.associations[assoc].nil? }
        end

        def fields_for(model)
          selector[model][:select].to_a
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
praxis-0.22.pre.2	lib/praxis/extensions/field_selection/active_record_query_selector.rb
praxis-0.22.pre.1	lib/praxis/extensions/field_selection/active_record_query_selector.rb