path: root/activerecord/lib/active_record/fixtures/file.rb

require 'erb'
require 'yaml'

module ActiveRecord
  class Fixtures
    class File
      include Enumerable

      ##
      # Open a fixture file named +file+.  When called with a block, the block
      # is called with the filehandle and the filehandle is automatically closed
      # when the block finishes.
      def self.open(file)
        x = new file
        block_given? ? yield(x) : x
      end

      def initialize(file)
        @file = file
        @rows = nil
      end

      def each(&block)
        rows.each(&block)
      end

      RESCUE_ERRORS = [ ArgumentError ] # :nodoc:

      private
      if defined?(Psych) && defined?(Psych::SyntaxError)
        RESCUE_ERRORS << Psych::SyntaxError
      end

      def rows
        return @rows if @rows

        begin
          data = YAML.load(render(IO.read(@file)))
        rescue *RESCUE_ERRORS => error
          raise Fixture::FormatError, "a YAML error occurred parsing #{@file}. Please note that YAML must be consistently indented using spaces. Tabs are not allowed. Please have a look at http://www.yaml.org/faq.html\nThe exact error was:\n  #{error.class}: #{error}", error.backtrace
        end
        @rows = data ? validate(data).to_a : []
      end

      def render(content)
        ERB.new(content).result
      end

      # Validate our unmarshalled data.
      def validate(data)
        unless Hash === data || YAML::Omap === data
          raise Fixture::FormatError, 'fixture is not a hash'
        end

        raise Fixture::FormatError unless data.all? { |name, row| Hash === row }
        data
      end
    end
  end
end