archivesspace/archivesspace

require 'jsonmodel'

class AdvancedQueryBuilder

  attr_reader :query

  RangeValue = Struct.new(:from, :to)

  def initialize
    @query = nil
  end

  def and(field_or_subquery, value = nil, type = 'text', literal = false, negated = false)
    if field_or_subquery.is_a?(AdvancedQueryBuilder)
      push_subquery('AND', field_or_subquery)
    elsif value.is_a? RangeValue
      push_range('AND', field_or_subquery, value, 'range', literal, negated)
    else
      raise "Missing value" if value.nil?
      push_term('AND', field_or_subquery, value, type, literal, negated)
    end

    self
  end

  def or(field_or_subquery, value = nil, type = 'text', literal = false, negated = false)
    if field_or_subquery.is_a?(AdvancedQueryBuilder)
      push_subquery('OR', field_or_subquery)
    elsif value.is_a? RangeValue
      push_range('AND', field_or_subquery, value, 'range', literal, negated)
    else
      raise "Missing value" unless value
      push_term('OR', field_or_subquery, value, type, literal, negated)
    end

    self
  end

  def empty?
    @query.nil?
  end
  alias_method :empty, :empty?

  def build
    JSONModel::JSONModel(:advanced_query).from_hash({"query" => build_query(@query)})
  end

  def self.from_json_filter_terms(array_of_json)
    builder = new

    array_of_json.each do |json_str|
      json = ASUtils.json_parse(json_str)
      builder.and(json.keys[0], json.values[0])
    end

    builder.build
  end

  def self.build_query_from_form(queries)
    query = if queries.length > 1
              stack = queries.reverse.clone

              while stack.length > 1
                a = stack.pop
                b = stack.pop

                stack.push(JSONModel::JSONModel(:boolean_query).from_hash({
                                                                 :op => b["op"],
                                                                 :subqueries => [as_field_query(a), as_field_query(b)]
                                                               }))
              end

              stack.pop
            else
              as_field_query(queries[0])
            end

    JSONModel::JSONModel(:advanced_query).from_hash({"query" => query})
  end


  private

  def push_subquery(operator, subquery)
    new_query = {
      'operator' => operator,
      'type' => 'boolean_query',
      'arg1' => subquery.query,
      'arg2' => @query,
    }

    @query = new_query
  end

  def push_term(operator, field, value, type = 'text', literal = false, negated = false)
    new_query = {
      'operator' => operator,
      'type' => 'boolean_query',
      'arg1' => {
        'field' => field,
        'value' => value,
        'type' => type,
        'negated' => negated,
        'literal' => literal,
      },
      'arg2' => @query,
    }

    @query = new_query
  end


  def push_range(operator, field, range, type = 'range', literal = false, negated = false)
    new_query = {
      'operator' => operator,
      'type' => 'boolean_query',
      'arg1' => {
        'field' => field,
        'from' => range.from,
        'to' => range.to,
        'type' => type,
        'negated' => negated,
        'literal' => literal,
      },
      'arg2' => @query,
    }

    @query = new_query
  end

  def build_query(query)
    if query['type'] == 'boolean_query'
      subqueries = [query['arg1'], query['arg2']].compact.map {|subquery|
        build_query(subquery)
      }

      JSONModel::JSONModel(:boolean_query).from_hash({
                                            'op' => query['operator'],
                                            'subqueries' => subqueries
                                          })
    else
      self.class.as_field_query(query)
    end
  end

  def self.as_field_query(query_data)
    raise "keys should be strings only" if query_data.is_a?(Hash) && query_data.any? { |k, _| k.is_a? Symbol }
    if query_data.is_a?(JSONModelType)
      query_data
    elsif query_data['type'] == "date"
      query_precision = query_data['value'].split('-').count
      case query_precision
      when 3 then query_data['precision'] = "day"
      when 2 then query_data['precision'] = "month"
      when 1 then query_data['precision'] = "year"
      end

      query_data['value'] = JSONModel::Validations.normalise_date(query_data['value'])
      JSONModel::JSONModel(:date_field_query).from_hash(query_data)
    elsif query_data['type'] == "boolean"
      JSONModel::JSONModel(:boolean_field_query).from_hash(query_data)
    elsif query_data['type'] == "range"
      JSONModel::JSONModel(:range_query).from_hash(query_data)
    else
      if query_data["type"] == "enum" && query_data["value"].blank?
        query_data["comparator"] = "empty"
      end

      # Looks like sometimes the value is set to a Boolean, but :field_query
      # schema insists this should be a String.
      query_data["value"] = query_data["value"].to_s
      query = JSONModel::JSONModel(:field_query).from_hash(query_data)

      if query_data['type'] == "enum"
        query.literal = true
      end

      query
    end
  end

end