class PgSearch::Features::Feature

def columns
  if options[:only]
    all_columns.select do |column|
      Array.wrap(options[:only]).map(&:to_s).include? column.name
    end
  else
    all_columns
  end
end

def document
  columns.map { |column| column.to_sql }.join(" || ' ' || ")
end

def initialize(query, options, all_columns, model, normalizer)
  @query = query
  @options = options || {}
  @all_columns = all_columns
  @model = model
  @normalizer = normalizer
end

def normalize(expression)
  normalizer.add_normalization(expression)
end