sane-ify DSL a bit

rubygems · Apr 12, 2011 · bbaaec8 · bbaaec8
1 parent 2016e47
commit bbaaec8
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/lib/bundler/dsl.rb b/lib/bundler/dsl.rb
@@ -76,18 +76,18 @@ def gem(name, *args)
     def source(source, options = {})
       case source
       when :gemcutter, :rubygems, :rubyforge then
-        rubygems_source "http://rubygems.org"
+        @rubygems_source.add_remote "http://rubygems.org"
         return
       when String
-        rubygems_source source
+        @rubygems_source.add_remote source
         return
-      end
-
-      @source = source
-      options[:prepend] ? @sources.unshift(@source) : @sources << @source
+      else
+        @source = source
+        options[:prepend] ? @sources.unshift(@source) : @sources << @source
 
-      yield if block_given?
-      @source
+        yield if block_given?
+        return @source
+      end
     ensure
       @source = nil
     end