#!/usr/bin/env ruby
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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_relative "../lib/arrow-format"
require_relative "../lib/arrow-format/integration/options"

options = ArrowFormat::Integration::Options.singleton
case options.command
when "validate"
  require_relative "../lib/arrow-format/integration/validate"
when "json-to-file"
  require_relative "../lib/arrow-format/integration/json-reader"
  File.open(options.json, "r") do |input|
    reader = ArrowFormat::Integration::JSONReader.new(input)
    File.open(options.arrow, "wb") do |output|
      writer = ArrowFormat::FileWriter.new(output)
      writer.start(reader.schema)
      reader.each do |record_batch|
        writer.write_record_batch(record_batch)
      end
      writer.finish
    end
  end
when "stream-to-file"
  File.open(options.arrows, "rb") do |input|
    reader = ArrowFormat::StreamingReader.new(input)
    File.open(options.arrow, "wb") do |output|
      writer = ArrowFormat::FileWriter.new(output)
      writer.start(reader.schema)
      reader.each do |record_batch|
        writer.write_record_batch(record_batch)
      end
      writer.finish
    end
  end
when "file-to-stream"
  File.open(options.arrow, "rb") do |input|
    reader = ArrowFormat::FileReader.new(input)
    File.open(options.arrows, "wb") do |output|
      writer = ArrowFormat::StreamingWriter.new(output)
      writer.start(reader.schema)
      reader.each do |record_batch|
        writer.write_record_batch(record_batch)
      end
      writer.finish
    end
  end
end
