luislavena/gem-compiler

# frozen_string_literal: true

require "rbconfig"
require "rubygems/command"

class Gem::Commands::CompileCommand < Gem::Command
  ABIs = {
      "ruby"   => :ruby,
      "strict" => :strict,
      "none"   => :none
  }.freeze

  def initialize
    defaults = {
      output: Dir.pwd,
      abi_lock: :ruby
    }

    super "compile", "Create binary pre-compiled gem", defaults

    add_option "-O", "--output DIR", "Directory where binary will be stored" do |value, options|
      options[:output] = File.expand_path(value, Dir.pwd)
    end

    add_option "--include-shared-dir DIR", "Additional directory for shared libraries" do |value, options|
      options[:include_shared_dir] = value
    end

    add_option "--prune", "Clean non-existing files during re-packaging" do |value, options|
      options[:prune] = true
    end

    add_option "--abi-lock MODE",
      "Lock to version of Ruby (ruby, strict, none)" do |value, options|

      mode = ABIs[value]
      unless mode
        valid = ABIs.keys.sort
        raise OptionParser::InvalidArgument, "#{value} (#{valid.join ', '} are valid)"
      end

      options[:abi_lock] = mode
    end

    add_option "-N", "--no-abi-lock", "Do not lock compiled Gem to Ruby's ABI (same as --abi-lock=none)" do |value, options|
      options[:abi_lock] = :none
    end

    add_option "-S", "--strip [CMD]", "Strip symbols from generated binaries" do |value, options|
      if value.nil? || value.empty?
        options[:strip] = RbConfig::CONFIG["STRIP"]
      else
        options[:strip] = value
      end
    end

    add_option "--build-number NUMBER",
      "Append build number to compiled Gem version" do |value, options|

      begin
        options[:build_number] = Integer(value).abs
      rescue ArgumentError
        raise OptionParser::InvalidArgument, "must be a number"
      end
    end
  end

  def arguments
    "GEMFILE       path to the gem file to compile"
  end

  def usage
    "#{program_name} GEMFILE"
  end

  def execute
    gemfile = options[:args].shift

    # no gem, no binary
    unless gemfile
      raise Gem::CommandLineError,
            "Please specify a gem file on the command line (e.g. #{program_name} foo-0.1.0.gem)"
    end

    require "rubygems/compiler"

    compiler = Gem::Compiler.new(gemfile, options)
    compiler.compile
  end
end