# example
#!/usr/bin/env ruby
require 'google/cloud/vision'
require 'pathname'

def dbg(msg)
  puts "#{Time.now.inspect} : #{msg}"
end

# Demonstrate the Google OCR system with a few image files.
class Demo
  def ocr_text_count
    1000
  end

  def ocr_timeout
    60
  end

  def image_list(image_path)
    my_images = Dir.glob(image_path + '/*')
    my_images.reject { |img| File.directory?(img) }
  end

  def do_multiple_ocr(image_path)
    annotator = Google::Cloud::Vision::ImageAnnotator.new(timeout: ocr_timeout)
    annotator.text_detection(images: image_list(image_path),
                             async: false,
                             max_results: ocr_text_count)
             .responses.map do |res|
               { 'image_text' => res.text_annotations.first.description }
             end
  end

  def print_texts(texts)
    texts.each_with_index do |text, index|
      dbg "#{index}: TEXT:\n#{text['image_text']}\n"
    end
  end

  def run
    @home_dir = ENV['HOME']
    @image_path = @home_dir + '/demo/image'
    texts = do_multiple_ocr(@image_path)
    print_texts(texts)
  end
end

def write_pid_file
  pid_filename = "#{Pathname.pwd}/demo.pid"
  dbg "Write PID (#{Process.pid}) to file: #{pid_filename}"
  pid_file = File.open(pid_filename, 'w')
  pid_file.write(Process.pid.to_s)
  pid_file.close
end

if $PROGRAM_NAME == __FILE__
  dbg 'DEMO.LAUNCH STARTED.'
  Demo.new.run
  puts "\n\n"
  dbg  'DEMO.LAUNCH COMPLETED.'

  write_pid_file

  # Windows process kill attempt:
  dbg "taskkill /im #{Process.pid} /f /t >nul 2>&1"

  # The following suppresses all the program output to the console.
  #   system("taskkill /im #{Process.pid} /f /t")

  # Native ruby form:
  #   Process.kill('QUIT', Process.pid)

  # Try something stupid to make it exit.
  exit 

  # It doesn't exit. Do we need a way to clean up the annotation client when done?
end