Rakefile

# -----------------------------------------------------------------------------
#
# Generic Gem Rakefile
#
# -----------------------------------------------------------------------------
# Copyright 2010-2012 Daniel Azuma
#
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
#   this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
# * Neither the name of the copyright holder, nor the names of any other
#   contributors to this software, may be used to endorse or promote products
#   derived from this software without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
# -----------------------------------------------------------------------------
;


# Load config if present

config_path_ = ::File.expand_path('rakefile_config.rb', ::File.dirname(__FILE__))
load(config_path_) if ::File.exists?(config_path_)
RAKEFILE_CONFIG = {} unless defined?(::RAKEFILE_CONFIG)


# Gemspec

require 'rubygems'
require 'rubygems/package'
gemspec_ = eval(::File.read(::Dir.glob('*.gemspec').first))
release_gemspec_ = eval(::File.read(::Dir.glob('*.gemspec').first))
release_gemspec_.version = gemspec_.version.to_s.sub(/\.nonrelease$/, '')


# Platform info

dlext_ = ::RbConfig::CONFIG['DLEXT']

platform_ =
  case ::RUBY_DESCRIPTION
  when /^jruby\s/ then :jruby
  when /^ruby\s/ then :mri
  when /^rubinius\s/ then :rubinius
  else :unknown
  end

puts "Platform: #{::RUBY_DESCRIPTION}"

platform_suffix_ =
  case platform_
  when :mri
    if ::RUBY_VERSION =~ /^1\.8\..*$/
      'mri18'
    elsif ::RUBY_VERSION =~ /^1\.9\..*$/
      'mri19'
    elsif ::RUBY_VERSION =~ /^2\..*$/
      'mri2'
    else
      raise "Unknown version of Matz Ruby Interpreter (#{::RUBY_VERSION})"
    end
  when :rubinius then 'rbx'
  when :jruby then 'jruby'
  else 'unknown'
  end


# Directories

doc_directory_ = ::RAKEFILE_CONFIG[:doc_directory] || 'doc'
pkg_directory_ = ::RAKEFILE_CONFIG[:pkg_directory] || 'pkg'
tmp_directory_ = ::RAKEFILE_CONFIG[:tmp_directory] || 'tmp'


# Build tasks

internal_ext_info_ = gemspec_.extensions.map do |extconf_path_|
  source_dir_ = ::File.dirname(extconf_path_)
  name_ = ::File.basename(source_dir_)
  {
    :name => name_,
    :source_dir => source_dir_,
    :extconf_path => extconf_path_,
    :source_glob => "#{source_dir_}/*.{c,h}",
    :obj_glob => "#{source_dir_}/*.{o,dSYM}",
    :suffix_makefile_path => "#{source_dir_}/Makefile_#{platform_suffix_}",
    :built_lib_path => "#{source_dir_}/#{name_}.#{dlext_}",
    :staged_lib_path => "#{source_dir_}/#{name_}_#{platform_suffix_}.#{dlext_}",
  }
end
internal_ext_info_ = [] if platform_ == :jruby

internal_ext_info_.each do |info_|
  file info_[:staged_lib_path] => [info_[:suffix_makefile_path]] + ::Dir.glob(info_[:source_glob]) do
    ::Dir.chdir(info_[:source_dir]) do
      cp "Makefile_#{platform_suffix_}", 'Makefile'
      sh 'make'
      rm 'Makefile'
    end
    mv info_[:built_lib_path], info_[:staged_lib_path]
    rm_r ::Dir.glob(info_[:obj_glob])
  end
  file info_[:suffix_makefile_path] => info_[:extconf_path] do
    ::Dir.chdir(info_[:source_dir]) do
      ruby 'extconf.rb'
      mv 'Makefile', "Makefile_#{platform_suffix_}"
    end
  end
end

task :build_ext => internal_ext_info_.map{ |info_| info_[:staged_lib_path] } do
  internal_ext_info_.each do |info_|
    target_prefix_ = target_name_ = nil
    ::Dir.chdir(info_[:source_dir]) do
      ruby 'extconf.rb'
      ::File.open('Makefile') do |file_|
        file_.each do |line_|
          if line_ =~ /^target_prefix\s*=\s*(\S+)\s/
            target_prefix_ = $1
          elsif line_ =~ /^TARGET\s*=\s*(\S+)\s/
            target_name_ = $1
          end
        end
      end
      rm 'Makefile'
    end
    raise "Could not find target_prefix in makefile for #{info_[:name]}" unless target_prefix_
    raise "Could not find TARGET in makefile for #{info_[:name]}" unless target_name_
    cp info_[:staged_lib_path], "lib#{target_prefix_}/#{target_name_}.#{dlext_}"
  end
end


# Clean task

clean_files_ = [doc_directory_, pkg_directory_, tmp_directory_] +
  ::Dir.glob('ext/**/Makefile*') +
  ::Dir.glob('ext/**/*.{o,class,log,dSYM}') +
  ::Dir.glob("**/*.{bundle,so,dll,rbc,jar}") +
  ::Dir.glob('**/.rbx') +
  (::RAKEFILE_CONFIG[:extra_clean_files] || [])
task :clean do
  clean_files_.each{ |path_| rm_rf path_ }
end


# RDoc tasks

task :build_rdoc => "#{doc_directory_}/index.html"
all_rdoc_files_ = ::Dir.glob("lib/**/*.rb") + gemspec_.extra_rdoc_files
main_rdoc_file_ = ::RAKEFILE_CONFIG[:main_rdoc_file]
main_rdoc_file_ = 'README.rdoc' if !main_rdoc_file_ && ::File.readable?('README.rdoc')
main_rdoc_file_ = ::Dir.glob("*.rdoc").first unless main_rdoc_file_
file "#{doc_directory_}/index.html" => all_rdoc_files_ do
  rm_r doc_directory_ rescue nil
  args_ = []
  args_ << '-o' << doc_directory_
  args_ << '--main' << main_rdoc_file_ if main_rdoc_file_
  args_ << '--title' << "#{::RAKEFILE_CONFIG[:product_visible_name] || gemspec_.name.capitalize} #{release_gemspec_.version} Documentation"
  args_ << '-f' << 'darkfish'
  args_ << '--verbose' if ::ENV['VERBOSE']
  gem 'rdoc'
  require 'rdoc/rdoc'
  ::RDoc::RDoc.new.document(args_ + all_rdoc_files_)
end


# Gem release tasks

task :build_other

task :build_gem => :build_other do
  if defined?(::Gem::Builder)
    ::Gem::Builder.new(gemspec_).build
  else
    ::Gem::Package.build(gemspec_)
  end
  mkdir_p(pkg_directory_)
  mv "#{gemspec_.name}-#{gemspec_.version}.gem", "#{pkg_directory_}/"
end

task :build_release => :build_other do
  if defined?(::Gem::Builder)
    ::Gem::Builder.new(release_gemspec_).build
  else
    ::Gem::Package.build(release_gemspec_)
  end
  mkdir_p(pkg_directory_)
  mv "#{release_gemspec_.name}-#{release_gemspec_.version}.gem", "#{pkg_directory_}/"
end

task :release_gem => :build_release do
  ::Dir.chdir(pkg_directory_) do
    sh "#{::RbConfig::TOPDIR}/bin/gem push #{release_gemspec_.name}-#{release_gemspec_.version}.gem"
  end
end


# Unit test task

task :test => [:build_ext, :build_other] do
  $:.unshift(::File.expand_path('lib', ::File.dirname(__FILE__)))
  if (cases_ = ::ENV['TESTCASE'])
    test_files_ = cases_.split(',').map{ |c_| ::Dir.glob("test/#{c_}.rb") }.flatten
  else
    test_files_ = ::Dir.glob("test/**/tc_*.rb")
  end
  $VERBOSE = true
  test_files_.each do |path_|
    load path_
    puts "Loaded testcase #{path_}"
  end
end


# Default task

task :default => [:clean, :test]