#!/usr/bin/env ruby
#

require "rubygems"
require "optparse"
require "ostruct"
require "erb"

$: << "#{File.dirname(__FILE__)}/../lib"
require "fpm"
require "fpm/flags"

def main(args)
  settings = OpenStruct.new
  settings.exclude = []

  # Source-specific settings/flags go here.
  settings.source = {}

  # Maintainer scripts - https://github.com/jordansissel/fpm/issues/18
  settings.scripts ||= {} 

  opts = OptionParser.new do |opts|
    # TODO(sissel): Maybe this should be '-o OUTPUT' ?
    opts.on("-p PACKAGEFILE", "--package PACKAGEFILE",

    opts.on("--url URL",
            "Add a url for this package.") do |url|
      settings.url = url
    end # --url
  end # OptionParser

  # Add extra flags.
  FPM::Source::Gem.flags(FPM::Flags.new(opts, "gem", "gem source only"), settings)

  opts.parse!(args)

  ok = true
  if settings.package_type.nil?
    $stderr.puts "Missing package target type (no -t flag?)"
    ok = false
  end

      :name => settings.package_name,
      :prefix => settings.prefix,
      :suffix => settings.suffix,
      :exclude => settings.exclude,
      :maintainer => settings.maintainer,
      :provides => [],
      :description => settings.description,
      :url => settings.url,
      :settings => settings.source
    )

    @edit = !!settings.edit

    @paths = paths
    @package = package_class_for(settings.package_type).new(@source)
    # Append dependencies given from settings (-d flag for fpm)

require "fpm/namespace"

class FPM::Flags
  def initialize(opts, flag_prefix, help_prefix)
    @opts = opts
    @flag_prefix = flag_prefix
    @help_prefix = help_prefix
  end # def initialize

  def on(*args, &block)
    fixed_args = args.collect do |arg|
      if arg =~ /^--/
        "--#{@flag_prefix}-#{arg.gsub(/^--/, "")}" 
      else
        "(#{@help_prefix}) #{arg}"
      end
    end
    @opts.on(*fixed_args, &block)
  end # def on
end # class FPM::Flags

require "fpm/namespace"
require "fpm/source"
require "rubygems/package"
require "rubygems"
require "fileutils"

class FPM::Source::Gem < FPM::Source
  def self.flags(opts, settings)
    opts.on("--bin-path DIRECTORY",
            "The directory to install gem executables") do |path|
      settings.source[:bin_path] = path
    end
  end # def flags

  def get_source(params)
    gem = @paths.first
    looks_like_name_re = /^[A-Za-z0-9_-]+$/
    if !File.exists?(gem) 
      if gem =~ looks_like_name_re
        download(gem, params[:version])
      else

  def make_tarball!(tar_path, builddir)
    tmpdir = "#{tar_path}.dir"
    gem = @paths.first
    if self[:prefix]
      installdir = "#{tmpdir}/#{self[:prefix]}"
      # TODO(sissel): Overwriting @paths is bad mojo and confusing...
      # Maybe we shouldn't?
      @paths = [ self[:prefix] ]
    else
      installdir = File.join(tmpdir, ::Gem::dir)
      @paths = [ ::Gem::dir ]
    end

    ::FileUtils.mkdir_p(installdir)
    args = ["gem", "install", "--quiet", "--no-ri", "--no-rdoc",
       "--install-dir", installdir, "--ignore-dependencies"]
    if self[:settings][:bin_path]
      args += ["--bindir", File.join(tmpdir, self[:settings][:bin_path])]
      @paths << self[:settings][:bin_path]
    end

    args << gem
    system(*args)
    
    # make paths relative  (/foo becomes ./foo)
    tar(tar_path, @paths.collect {|p| ".#{p}"}, tmpdir)
    FileUtils.rm_r(tmpdir)

    # TODO(sissel): Make a helper method.
    system(*["gzip", "-f", tar_path])
  end

end # class FPM::Source::Gem