Contents

# frozen_string_literal: true

module LazyRecord
  class Relation
    include Enumerable

    attr_reader :model, :all

    def initialize(model:, array: nil)
      raise ArgumentError, 'model must be a class' unless model.is_a?(Class)
      @model = model
      @all   = []
      self_extend_scopes_module
      array&.each do |object|
        @all << object && next if object.is_a?(model)
        message = "Argument must be a collection of #{model.to_s.tableize}"
        raise ArgumentError, message
      end
    end

    def <<(other)
      message = "object must be of type #{model}"
      raise ArgumentError, message unless other.is_a?(model)
      all << other
    end

    def inspect
      "\#<#{model}Relation [#{all.map(&:inspect).join(', ')}]>"
    end

    def where(condition)
      result = all.select do |x|
        if condition.is_a? Hash
          condition.all? { |key, val| x.send(key) == val }
        else
          eval "x.#{condition}"
        end
      end
      self.class.new(model: model, array: result)
    end

    def each(&block)
      all.each(&block)
    end

    def [](index)
      all[index]
    end

    def last
      self[-1]
    end

    def clear
      all.clear
    end

    private :clear, :all

    def self_extend_scopes_module
      return unless model.const_defined?(:ScopeMethods, _search_ancestors = false)
      extend(model::ScopeMethods)
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
lazy_record-0.1.9	lib/lazy_record/relation.rb
lazy_record-0.1.8	lib/lazy_record/relation.rb
lazy_record-0.1.7	lib/lazy_record/relation.rb
lazy_record-0.1.6	lib/lazy_record/relation.rb
lazy_record-0.1.4	lib/lazy_record/relation.rb