module Embulk
  module Input
    class Elasticsearch < InputPlugin
      class Converter
        def self.get_sources(results, fields)
          hits = results['hits']['hits']
          hits.map { |hit|
            result = hit['_source']
            fields.map { |field|
              value = field['metadata'] ? hit[field['name']] : result[field['name']]
              convert_value(value, field)
            }
          }
        end

        def self.convert_value(value, field)
          return nil if value.nil?
          case field["type"]
          when "string"
            value
          when "long"
            value.to_i
          when "double"
            value.to_f
          when "boolean"
            if value.is_a?(TrueClass) || value.is_a?(FalseClass)
              value
            else
              downcased_val = value.downcase
              case downcased_val
              when 'true' then true
              when 'false' then false
              when '1' then true
              when '0' then false
              else nil
              end
            end
          when "timestamp"
            Time.parse(value)
          when "json"
            value
          else
            raise "Unsupported type #{field['type']}"
          end
        end
      end
    end
  end
end