# Copyright 2013-2014 Bazaarvoice, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

require 'detabulator'

class Table
  def self.load(filename)
    self.new File.read filename
  end

  def initialize(table_as_text)
    raw_header, *raw_data = Detabulator.new.detabulate table_as_text
    header = raw_header.map(&:to_sym)
    @records = raw_data.map { |row| Hash[header.zip(row)] }
  end

  # Selects all rows in the table which match the name/value pairs of the predicate object and returns
  # the single distinct value from those rows for the specified key.
  def get(key, predicate)
    distinct_values(filter(predicate), key, false)
  end

  # Selects all rows in the table which match the name/value pairs of the predicate object and returns
  # all distinct values from those rows for the specified key.
  def get_list(key, predicate)
    distinct_values(filter(predicate), key, true)
  end

  # Selects all rows in the table which match the name/value pairs of the predicate object and returns a
  # set of nested maps, where the key for the map at level n is the key at index n in the specified keys,
  # except for the last key in the specified keys which is used to determine the value of the leaf-level map.
  # In the simple case where keys is a list of 2 elements, this returns a map from key[0] to key[1].
  def get_map(predicate, *keys)
    build_nested_map(filter(predicate), keys, false)
  end

  # Selects all rows in the table which match the name/value pairs of the predicate object and returns a
  # set of nested maps, where the key for the map at level n is the key at index n in the specified keys,
  # except for the last key in the specified keys which is used to determine the list of values in the
  # leaf-level map.  In the simple case where keys is a list of 2 elements, this returns a map from key[0]
  # to a list of values for key[1].
  def get_multimap(predicate, *keys)
    build_nested_map(filter(predicate), keys, true)
  end

  private

  # Return the subset of records that match the predicate for all keys in the predicate.
  def filter(predicate)
    @records.select { |record| predicate.all? { |key, value| record[key] == value } }
  end

  def build_nested_map(records, path, multi)
    key, *rest = path
    if rest.empty?
      # Build the leaf level of the data structure
      distinct_values(records, key, multi)
    else
      # Return a hash keyed by the distinct values of the first key and values are the result of a
      # recursive invocation of arrange() with the rest of the keys
      result = {}
      records.group_by do |record|
        record[key]
      end.map do |value, group|
        result[value] = build_nested_map(group, rest, multi)
      end
      result
    end
  end

  def distinct_values(records, key, multi)
    values = records.map { |record| record[key] }.uniq
    if multi
      # In a multimap the leaf level is a list of string values
      values
    else
      # In a non-multimap the leaf level is a single string value
      raise "Multiple distinct values for the same key '#{key}': #{records.inspect}" if values.length > 1
      values[0]
    end
  end
end