lib/rocktumbler/gemfile.rb in rocktumbler-0.1.2 vs lib/rocktumbler/gemfile.rb in rocktumbler-0.1.3

@@ -1,8 +1,9 @@
 require_relative 'exceptions'
 
 module Rocktumbler
+  # The Gemfile is a class responsible for reading and parsing a Gemfile
   class Gemfile
     attr_accessor :ruby, :source
 
     def initialize(gemfile_location)
       @gemfile_location = gemfile_location
@@ -10,29 +11,27 @@
       @ruby = parse_ruby
       @source = parse_source
     end
 
     def read_contents
-      gemfile = File.open(@gemfile_location,'r')
+      gemfile = File.open(@gemfile_location, 'r')
       str = gemfile.read
       gemfile.close
       str
     end
 
     def print_source_and_ruby
-      str = ""
-      str += "#{@source.gsub(/"/,'\'')}\n\n" if @source
-      str += "#{@ruby.gsub(/"/,'\'')}\n\n" if @ruby
+      str = ''
+      str += "#{@source.tr('\"', '\'')}\n" if @source
+      str += "\n#{@ruby.tr('\"', '\'')}\n" if @ruby
       str
     end
 
     def parse_ruby
-      /^(ruby.*)/.match(@gemfile_contents)
-      $1
+      /^(ruby.*)/.match(@gemfile_contents)[1] if @gemfile_contents =~ /^(ruby.*)/
     end
 
     def parse_source
-      /^(source.*)/.match(@gemfile_contents)
-      $1
+      /^(source.*)/.match(@gemfile_contents)[1]
     end
   end
 end