brew/Library/Homebrew/cmd/unpack.rb

require "stringio"
require "formula"

module Homebrew
  module DATALoader
    # Original code from http://stackoverflow.com/a/2157556/371237
    def self.load(path)
      data = StringIO.new
      path.open("r") do |f|
        begin
          line = f.gets
        end until line.nil? || /^__END__$/ === line
        data << line while line = f.gets
      end
      data.rewind
      data
    end
  end

  module UnpackPatch
    def patch
      return unless ARGV.flag? "--patch"

      begin
        old_verbose, $VERBOSE = $VERBOSE, nil
        Object.const_set "DATA", DATALoader.load(path)
      ensure
        $VERBOSE = old_verbose
      end

      super
    end
  end

  def unpack
    formulae = ARGV.formulae
    raise FormulaUnspecifiedError if formulae.empty?

    if dir = ARGV.value("destdir")
      unpack_dir = Pathname.new(dir)
      unpack_dir.mkpath
    else
      unpack_dir = Pathname.pwd
    end

    raise "Cannot write to #{unpack_dir}" unless unpack_dir.writable_real?

    formulae.each do |f|
      f.extend(UnpackPatch)
      stage_dir = unpack_dir.join("#{f.name}-#{f.version}")

      if stage_dir.exist?
        raise "Destination #{stage_dir} already exists!" unless ARGV.force?
        rm_rf stage_dir
      end

      oh1 "Unpacking #{f.name} to: #{stage_dir}"

      ENV['VERBOSE'] = '1' # show messages about tar
      f.brew { cp_r getwd, stage_dir }
      ENV['VERBOSE'] = nil

      if ARGV.flag? "--git"
        ohai "Setting up git repository"
        cd stage_dir
        system "git", "init", "-q"
        system "git", "add", "-A"
        system "git", "commit", "-q", "-m", "brew-unpack"
      end
    end
  end
end