$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
$LOAD_PATH.unshift(File.dirname(__FILE__))

require 'rspec'
require 'zip_tricks'
require 'digest'
require 'fileutils'
require 'shellwords'

module Keepalive
  # Travis-CI kills the build if it does not receive output on standard out or standard error
  # for longer than a few minutes. We have a few tests that take a _very_ long time, and during
  # those tests this method has to be called every now and then to revive the output and let the
  # build proceed.
  def still_alive!
    $keepalive_last_out_ping_at ||= Time.now
    if (Time.now - $keepalive_last_out_ping_at) > 3
      $keepalive_last_out_ping_at = Time.now
      $stdout << '_'
    end
  end
  extend self
end


class ManagedTempfile < Tempfile
  @@managed_tempfiles = []
  
  def initialize(*)
    super
    @@managed_tempfiles << self
  end
  
  def self.prune!
    @@managed_tempfiles.each do |tf|
      (tf.close; tf.unlink) rescue nil
    end
    @@managed_tempfiles.clear
  end
end

# A Tempfile filled with N bytes of random data, that also knows the CRC32 of that data
class RandomFile < ManagedTempfile
  attr_reader :crc32
  RANDOM_MEG = Random.new.bytes(1024 * 1024) # Allocate it once to prevent heap churn
  def initialize(size)
    super('random-bin')
    binmode
    crc = ZipTricks::StreamCRC32.new
    bytes = size % (1024 * 1024)
    megs = size / (1024 * 1024)
    megs.times do
      Keepalive.still_alive!
      self << RANDOM_MEG
      crc << RANDOM_MEG
    end
    random_blob = Random.new.bytes(bytes)
    self << random_blob
    crc << random_blob
    @crc32 = crc.to_i
    rewind
  end

  def copy_to(io)
    rewind
    while data = read(10*1024*1024)
      io << data
      Keepalive.still_alive!
    end
    rewind
  end
end

module ZipInspection
  def inspect_zip_with_external_tool(path_to_zip)
    zipinfo_path = 'zipinfo'
    $zip_inspection_buf ||= StringIO.new
    $zip_inspection_buf.puts "\n"
    $zip_inspection_buf.puts "Inspecting ZIP output of #{inspect}." # The only way to get at the RSpec example without using the block argument
    $zip_inspection_buf.puts "Be aware that the zipinfo version on OSX is too old to deal with Zip6."
    escaped_cmd = Shellwords.join([zipinfo_path, '-tlhvz', path_to_zip])
    $zip_inspection_buf.puts `#{escaped_cmd}`
  end
  
  def open_with_external_app(app_path, path_to_zip, skip_if_missing)
    bin_exists = File.exist?(app_path)
    skip "This system does not have #{File.basename(app_path)}" if skip_if_missing && !bin_exists
    return unless bin_exists
    `#{Shellwords.join([app_path, path_to_zip])}`
  end
  
  def open_zip_with_archive_utility(path_to_zip, skip_if_missing: false)
    # ArchiveUtility sometimes puts the stuff it unarchives in ~/Downloads etc. so do
    # not perform any checks on the files since we do not really know where they are on disk.
    # Visual inspection should show whether the unarchiving is handled correctly.
    au_path = '/System/Library/CoreServices/Applications/Archive Utility.app/Contents/MacOS/Archive Utility'
    open_with_external_app(au_path, path_to_zip, skip_if_missing)
  end
  
  def open_zip_with_unarchiver(path_to_zip, skip_if_missing: false)
    ua_path = '/Applications/The Unarchiver.app/Contents/MacOS/The Unarchiver'
    open_with_external_app(ua_path, path_to_zip, skip_if_missing)
  end
end

RSpec.configure do |config|
  config.include Keepalive
  config.include ZipInspection
  
  config.after :each do
    ManagedTempfile.prune!
  end
  
  config.after :suite do
    $stderr << $zip_inspection_buf.string if $zip_inspection_buf
  end
end