Contents

module Avro
  module Builder
    # TODO: eventually this should be refactored into something standalone
    # instead of a module that is included to provide the file handling methods.
    module FileHandler

      module ClassMethods
        # Load paths are used to search for imports and extends.
        def load_paths
          @load_paths ||= Set.new
        end
      end

      def self.included(base)
        base.extend ClassMethods
      end

      def read_file(name)
        File.read(find_file(name))
      end

      FileNotFoundError = Class.new(StandardError)

      def find_file(name)
        # Ensure that the file_name that is searched for begins with a slash (/)
        # and ends with a .rb extension. Additionally, if the name contains
        # a namespace then ensure that periods (.) are replaced by forward
        # slashes. E.g. for 'test.example' search for '/test/example.rb'.
        file_name = "/#{name.to_s.tr('.', '/').sub(/^\//, '').sub(/\.rb$/, '')}.rb"
        matches = real_load_paths.flat_map do |load_path|
          Dir["#{load_path}/**/*.rb"].select do |file_path|
            file_path.end_with?(file_name)
          end
        end.uniq
        raise "Multiple matches: #{matches}" if matches.size > 1
        raise FileNotFoundError.new("File not found #{file_name}") if matches.empty?

        matches.first
      end

      private

      def real_load_paths
        self.class.load_paths.map { |path| File.realpath(path) }.uniq
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
avro-builder-0.16.1	lib/avro/builder/file_handler.rb
avro-builder-0.16.0	lib/avro/builder/file_handler.rb