Sha256: 3bb108c01fc4957f467f780b2c46f1525b64f81022f6c665c7fdcef1dd25b34e

Contents?: true

Size: 1.47 KB

Versions: 9

Compression:

Stored size: 1.47 KB

Contents

# frozen_string_literal: true

module RuboCop
  module Rails
    # It loads db/schema.rb and return Schema object.
    # Cops refers database schema information with this module.
    module SchemaLoader
      extend self

      # It parses `db/schema.rb` and return it.
      # It returns `nil` if it can't find `db/schema.rb`.
      # So a cop that uses the loader should handle `nil` properly.
      #
      # @return [Schema, nil]
      def load(target_ruby_version)
        return @load if defined?(@load)

        @load = load!(target_ruby_version)
      end

      def reset!
        return unless instance_variable_defined?(:@load)

        remove_instance_variable(:@load)
      end

      def db_schema_path
        path = Pathname.pwd
        until path.root?
          schema_path = path.join('db/schema.rb')
          return schema_path if schema_path.exist?

          path = path.join('../').cleanpath
        end

        nil
      end

      private

      def load!(target_ruby_version)
        path = db_schema_path
        return unless path

        ast = parse(path, target_ruby_version)
        Schema.new(ast) if ast
      end

      def parse(path, target_ruby_version)
        klass_name = :"Ruby#{target_ruby_version.to_s.sub('.', '')}"
        klass = ::Parser.const_get(klass_name)
        parser = klass.new(RuboCop::AST::Builder.new)

        buffer = Parser::Source::Buffer.new(path, 1)
        buffer.source = path.read

        parser.parse(buffer)
      end
    end
  end
end

Version data entries

9 entries across 9 versions & 2 rubygems

Version Path
mlh-rubocop-config-1.0.3 vendor/bundle/ruby/3.2.0/gems/rubocop-rails-2.23.1/lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.23.1 lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.23.0 lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.22.2 lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.22.1 lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.22.0 lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.21.2 lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.21.1 lib/rubocop/rails/schema_loader.rb
rubocop-rails-2.21.0 lib/rubocop/rails/schema_loader.rb