Add support for source block

This allow you to use source block in both Appraisals and Gemfile.

Close #99

lib/appraisal/appraisal.rb

@@ -18,8 +18,8 @@ def gem(*args)
  gemfile.gem(*args)
  end
 
- def source(*args)
- gemfile.source(*args)
+ def source(*args, &block)
+ gemfile.source(*args, &block)
  end
 
  def ruby(*args)

lib/appraisal/bundler_dsl.rb

@@ -6,7 +6,7 @@ class BundlerDSL
  attr_reader :dependencies
 
  PARTS = %w(source ruby_version git_sources path_sources dependencies groups
- platforms gemspec)
+ platforms source_blocks gemspec)
 
  def initialize
  @sources = []
@@ -17,6 +17,7 @@ def initialize
  @platforms = OrderedHash.new
  @git_sources = OrderedHash.new
  @path_sources = OrderedHash.new
+ @source_blocks = OrderedHash.new
  end
 
  def run(&block)
@@ -39,8 +40,13 @@ def platforms(*names, &block)
 
  alias_method :platform, :platforms
 
- def source(source)
- @sources << source
+ def source(source, &block)
+ if block_given?
+ @source_blocks[source] ||= Source.new(source)
+ @source_blocks[source].run(&block)
+ else
+ @sources << source
+ end
  end
 
  def ruby(ruby_version)
@@ -103,7 +109,8 @@ def #{method_name}_entry_for_dup
  METHODS
  end
 
- [:git_sources, :path_sources, :platforms, :groups].each do |method_name|
+ [:git_sources, :path_sources, :platforms, :groups, :source_blocks].
+ each do |method_name|
  class_eval <<-METHODS, __FILE__, __LINE__
  private
 

lib/appraisal/gemfile.rb

@@ -3,9 +3,10 @@
 module Appraisal
  autoload :Gemspec, "appraisal/gemspec"
  autoload :GitSource, "appraisal/git_source"
- autoload :PathSource, "appraisal/path_source"
  autoload :Group, "appraisal/group"
+ autoload :PathSource, "appraisal/path_source"
  autoload :Platform, "appraisal/platform"
+ autoload :Source, "appraisal/source"
 
  # Load bundler Gemfiles and merge dependencies
  class Gemfile < BundlerDSL

lib/appraisal/source.rb

@@ -0,0 +1,30 @@
+require "appraisal/bundler_dsl"
+require "appraisal/utils"
+
+module Appraisal
+ class Source < BundlerDSL
+ def initialize(source)
+ super()
+ @source = source
+ end
+
+ def to_s
+ formatted_output indent(super)
+ end
+
+ # :nodoc:
+ def for_dup
+ formatted_output indent(super)
+ end
+
+ private
+
+ def formatted_output(output_dependencies)
+ <<-OUTPUT.strip
+source #{@source.inspect} do
+#{output_dependencies}
+end
+ OUTPUT
+ end
+ end
+end

spec/acceptance/appraisals_file_bundler_dsl_compatibility_spec.rb

@@ -2,7 +2,7 @@
 
 describe 'Appraisals file Bundler DSL compatibility' do
  it 'supports all Bundler DSL in Appraisals file' do
- build_gems %w(bagel orange_juice milk waffle coffee ham)
+ build_gems %w(bagel orange_juice milk waffle coffee ham sausage pancake)
  build_git_gem 'egg'
 
  build_gemfile <<-Gemfile
@@ -31,6 +31,10 @@
  end
  end
 
+ source "https://other-rubygems.org" do
+ gem "sausage"
+ end
+
  gem 'appraisal', :path => #{PROJECT_ROOT.inspect}
  Gemfile
 
@@ -61,6 +65,10 @@
  gem 'yoghurt'
  end
 
+ source "https://other-rubygems.org" do
+ gem "pancake"
+ end
+
  gemspec
  end
  Appraisals
@@ -108,6 +116,11 @@
  end
  end
 
+ source "https://other-rubygems.org" do
+ gem "sausage"
+ gem "pancake"
+ end
+
  gemspec :path => "../"
  Gemfile
  end