rbs 4.1.0.pre.2-java

data/lib/rbs/environment_loader.rb

@@ -0,0 +1,167 @@
+# frozen_string_literal: true
+
+module RBS
+  class EnvironmentLoader
+    class UnknownLibraryError < StandardError
+      attr_reader :library
+
+      def initialize(lib:)
+        @library = lib
+
+        super("Cannot find type definitions for library: #{lib.name} (#{lib.version || "[nil]"})")
+      end
+    end
+
+    include FileFinder
+
+    class Library < Struct.new(:name, :version, keyword_init: true)
+    end
+
+    attr_reader :core_root
+    attr_reader :repository
+
+    attr_reader :libs
+    attr_reader :dirs
+
+    DEFAULT_CORE_ROOT = Pathname(_ = __dir__) + "../../core"
+
+    def self.gem_sig_path(name, version)
+      requirements = [] #: Array[String]
+      requirements << version if version
+      spec = Gem::Specification.find_by_name(name, *requirements)
+      path = Pathname(spec.gem_dir) + "sig"
+      if path.directory?
+        [spec, path]
+      end
+    rescue Gem::MissingSpecError
+      nil
+    end
+
+    def initialize(core_root: DEFAULT_CORE_ROOT, repository: Repository.new)
+      @core_root = core_root
+      @repository = repository
+
+      @libs = Set.new
+      @dirs = []
+    end
+
+    def add(path: nil, library: nil, version: nil, resolve_dependencies: true)
+      case
+      when path
+        dirs << path
+      when library
+        if libs.add?(Library.new(name: library, version: version)) && resolve_dependencies
+          resolve_dependencies(library: library, version: version)
+        end
+      end
+    end
+
+    def resolve_dependencies(library:, version:)
+      [Collection::Sources::Rubygems.instance, Collection::Sources::Stdlib.instance].each do |source|
+        next unless source.has?(library, version)
+
+        unless version
+          version = source.versions(library).last or raise
+        end
+
+        source.dependencies_of(library, version)&.each do |dep|
+          add(library: dep['name'], version: nil)
+        end
+        return
+      end
+    end
+
+    def add_collection(lockfile)
+      lockfile.check_rbs_availability!
+
+      repository.add(lockfile.fullpath)
+
+      lockfile.gems.each_value do |gem|
+        name = gem[:name]
+        locked_version = gem[:version]
+
+        if (source = gem[:source]).is_a?(Collection::Sources::Rubygems)
+          # allow loading different version of a gem
+
+          unless source.has?(name, locked_version)
+            if (spec, _ = self.class.gem_sig_path(name, nil))
+              RBS.logger.warn { "Loading type definition from gem `#{name}-#{spec.version}` because locked version `#{locked_version}` is unavailable. Try `rbs collection update` to fix the (potential) issue." }
+              locked_version = spec.version.to_s
+            end
+          end
+        end
+
+        add(library: gem[:name], version: locked_version, resolve_dependencies: false)
+      end
+    end
+
+    def has_library?(library:, version:)
+      if self.class.gem_sig_path(library, version) || repository.lookup(library, version)
+        true
+      else
+        false
+      end
+    end
+
+    def load(env:)
+      # @type var loaded: Array[[AST::Declarations::t, Pathname, source]]
+      loaded = []
+
+      # For migrating stringio to stdlib
+      if @core_root && libs.none? { |lib| lib.name == 'stringio' }
+        add(library: 'stringio', version: nil)
+      end
+
+      each_signature do |source, path, buffer, decls, dirs|
+        decls.each do |decl|
+          loaded << [decl, path, source]
+        end
+        env.add_source(Source::RBS.new(buffer, dirs, decls))
+      end
+
+      loaded
+    end
+
+    def each_dir
+      if root = core_root
+        yield :core, root
+      end
+
+      libs.each do |lib|
+        unless has_library?(version: lib.version, library: lib.name)
+          raise UnknownLibraryError.new(lib: lib)
+        end
+
+        case
+        when from_gem = self.class.gem_sig_path(lib.name, lib.version)
+          yield lib, from_gem[1]
+        when from_repo = repository.lookup(lib.name, lib.version)
+          yield lib, from_repo
+        end
+      end
+
+      dirs.each do |dir|
+        yield dir, dir
+      end
+    end
+
+    def each_signature
+      files = Set[]
+
+      each_dir do |source, dir|
+        skip_hidden = !source.is_a?(Pathname)
+
+        FileFinder.each_file(dir, skip_hidden: skip_hidden) do |path|
+          next if files.include?(path)
+
+          files << path
+          buffer = Buffer.new(name: path, content: path.read(encoding: "UTF-8"))
+
+          _, dirs, decls = Parser.parse_signature(buffer)
+
+          yield source, path, buffer, decls, dirs
+        end
+      end
+    end
+  end
+end

data/lib/rbs/environment_walker.rb

@@ -0,0 +1,155 @@
+# frozen_string_literal: true
+
+module RBS
+  class EnvironmentWalker
+    InstanceNode = _ = Struct.new(:type_name, keyword_init: true)
+    SingletonNode = _ = Struct.new(:type_name, keyword_init: true)
+    TypeNameNode = _ = Struct.new(:type_name, keyword_init: true)
+
+    attr_reader :env
+
+    def initialize(env:)
+      @env = env
+      @only_ancestors = false
+    end
+
+    def builder
+      @builder ||= DefinitionBuilder.new(env: env)
+    end
+
+    def only_ancestors!(only = true)
+      @only_ancestors = only
+      self
+    end
+
+    def only_ancestors?
+      @only_ancestors
+    end
+
+    include TSort
+
+    def tsort_each_node(&block)
+      env.class_decls.each_key do |type_name|
+        yield InstanceNode.new(type_name: type_name)
+        yield SingletonNode.new(type_name: type_name)
+      end
+      env.interface_decls.each_key do |type_name|
+        yield TypeNameNode.new(type_name: type_name)
+      end
+      env.type_alias_decls.each_key do |type_name|
+        yield TypeNameNode.new(type_name: type_name)
+      end
+    end
+
+    def tsort_each_child(node, &block)
+      name = node.type_name
+
+      unless name.namespace.empty?
+        yield SingletonNode.new(type_name: name.namespace.to_type_name)
+      end
+
+      case node
+      when TypeNameNode
+        case
+        when name.interface?
+          definition = builder.build_interface(name)
+          unless only_ancestors?
+            definition.each_type do |type|
+              each_type_node type, &block
+            end
+          end
+        when name.alias?
+          each_type_node builder.expand_alias1(name), &block
+        else
+          raise "Unexpected TypeNameNode with type_name=#{name}"
+        end
+
+      when InstanceNode, SingletonNode
+        definition = if node.is_a?(InstanceNode)
+                       builder.build_instance(name)
+                     else
+                       builder.build_singleton(name)
+                     end
+
+        if ancestors = definition.ancestors
+          ancestors.ancestors.each do |ancestor|
+            case ancestor
+            when Definition::Ancestor::Instance
+              yield InstanceNode.new(type_name: ancestor.name)
+
+              unless only_ancestors?
+                ancestor.args.each do |type|
+                  each_type_node type, &block
+                end
+              end
+            when Definition::Ancestor::Singleton
+              yield SingletonNode.new(type_name: ancestor.name)
+            end
+          end
+        end
+
+        unless only_ancestors?
+          definition.each_type do |type|
+            each_type_node type, &block
+          end
+        end
+      end
+    end
+
+    def each_type_name(type, &block)
+      each_type_node(type) do |node|
+        yield node.type_name
+      end
+    end
+
+    def each_type_node(type, &block)
+      case type
+      when RBS::Types::Bases::Any
+      when RBS::Types::Bases::Class
+      when RBS::Types::Bases::Instance
+      when RBS::Types::Bases::Self
+      when RBS::Types::Bases::Top
+      when RBS::Types::Bases::Bottom
+      when RBS::Types::Bases::Bool
+      when RBS::Types::Bases::Void
+      when RBS::Types::Bases::Nil
+      when RBS::Types::Variable
+      when RBS::Types::ClassSingleton
+        yield SingletonNode.new(type_name: type.name)
+      when RBS::Types::ClassInstance
+        yield InstanceNode.new(type_name: type.name)
+        type.args.each do |ty|
+          each_type_node(ty, &block)
+        end
+      when RBS::Types::Interface
+        yield TypeNameNode.new(type_name: type.name)
+        type.args.each do |ty|
+          each_type_node(ty, &block)
+        end
+      when RBS::Types::Alias
+        yield TypeNameNode.new(type_name: type.name)
+        type.args.each do |ty|
+          each_type_node(ty, &block)
+        end
+      when RBS::Types::Union, RBS::Types::Intersection, RBS::Types::Tuple
+        type.types.each do |ty|
+          each_type_node ty, &block
+        end
+      when RBS::Types::Optional
+        each_type_node type.type, &block
+      when RBS::Types::Literal
+        # nop
+      when RBS::Types::Record
+        type.fields.each_value do |ty|
+          each_type_node ty, &block
+        end
+      when RBS::Types::Proc
+        type.each_type do |ty|
+          each_type_node ty, &block
+        end
+      else
+        raise "Unexpected type given: #{type}"
+      end
+    end
+  end
+end