Contents

class ActiveRepository::ResultSet
	def initialize(klass, query={}, attributes={})
    @klass = klass
    convert_query(query)
    @attributes = attributes.merge(SqlQueryExecutor::Query::Normalizers::QueryNormalizer.attributes_from_query(query))
  end

  def all
    @query ? get_result(@query) : @klass.all
  end

  def build(attributes)
    @klass.new(@attributes.merge(attributes))
  end

  def create(attributes)
    @klass.create(@attributes.merge(attributes))
  end

  def count
    all.size
  end

  def first
    @query ? all.first : @klass.all.first
  end

  def first_or_initialize
    object = all.first

    object ? object : @klass.new(@attributes)
  end

  def first_or_create
    object = first_or_initialize

    object.new_record? ? object.save : object

    object.reload
  end

  def last
    @query ? all.last : @klass.all.last
  end

  def where(query)
    @attributes = @attributes.merge(query) if query.is_a?(Hash)
    query = join_query(query, 'and')

    ActiveRepository::ResultSet.new(@klass, query, @attributes)
  end
  alias_method :and, :where

  def or(query)
    query = join_query(query, 'or')

    ActiveRepository::ResultSet.new(@klass, query, @attributes)
  end

private
  def convert_query(query)
    @query = SqlQueryExecutor::Query::Normalizers::QueryNormalizer.clean_query(query)
  end

  def get_result(args)
    if @klass.repository?
      args = args.first if args.is_a?(Array) && args.size == 1
      query_executor = SqlQueryExecutor::Base.new(@klass.all, args)
      query_executor.execute!
    else
      query = SqlQueryExecutor::Base.new([], args)
      objects = PersistenceAdapter.where(@klass, query).map do |object|
        @klass.serialize!(object.attributes)
      end

      objects
    end
  end

  def join_query(query, separator)
    query = SqlQueryExecutor::Query::Normalizers::QueryNormalizer.clean_query(query)
    query.blank? ? @query : (@query.blank? ? query : "(#{@query}) #{separator} (#{query})")
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
active_repository-0.3.1	lib/active_repository/result_set.rb