# Runs a ruby script and identifies all the libraries that it loads,
# then copies them into a directory

# Run like this
# ruby assemble.rb <script> <output_directory>

require 'fileutils'

unless ARGV.length == 2
  raise "Wrong number of args, should be <script> <output-directory>"
end

RUBY_MAIN_SCRIPT = ARGV.shift
OUTPUT_DIR = ARGV.shift

ASSEMBLING = true

load RUBY_MAIN_SCRIPT

lib_files = {}
$LOADED_FEATURES.each do | lib |
  lib_file = nil
  $LOAD_PATH.each do | path |
    maybe_lib = File.join(path, lib)
    if File.exists?(maybe_lib)
      lib_file = maybe_lib
      break
    end
  end
  if lib_file
    lib_files[lib] = lib_file
  else
    raise "Could not find lib_file #{lib}"
  end
end

FileUtils.mkdir_p(OUTPUT_DIR)

lib_files.each do | lib, lib_file |
  dest = File.join(OUTPUT_DIR, lib)
  dest_path = File.dirname(dest)
  FileUtils.mkdir_p(dest_path)
  FileUtils.install(lib_file, dest)
end