require 'active_record'

class Carto::DataImportQueryBuilder

  def initialize
    @order = {}
  end

  def with_user(user)
    @user_id = user.id
    self
  end

  def with_state_not_in(states)
    @excluded_states = states
    self
  end

  def with_created_at_after(date)
    @created_at_after = date
    self
  end

  def with_order(order, asc_desc = :asc)
    @order[order] = asc_desc
    self
  end

  def build
    query = Carto::DataImport.all

    if @user_id
      query = query.where(user_id: @user_id)
    end

    if @excluded_states
      query = query.where('state not in (?)', @excluded_states)
    end

    if @created_at_after
      query = query.where('created_at >= ?', @created_at_after)
    end

    @order.each { |k, v|
      query = query.order(k)
      query = query.reverse_order if v == :desc
    }

    query
  end

end