orthoses 0.3.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2d5b96650f0b18fdb84d67bc6821887ac974e529bf13e05c079ad703645a0583
-  data.tar.gz: 50a0101c732e7a66d166e0ff4a3f6fd439d7c50ab2985ffb49c618e698f5873f
+  metadata.gz: 3fe5907a7edff87d73372b0e6f556d5f3b08a3ae6d7407447f249a25953d069e
+  data.tar.gz: 7564345ffdc6363a5ba03bb9c70299b17cb15cdb3cc501bcbd3b221c7b0a823a
 SHA512:
-  metadata.gz: 2a9443b508c4dbb2518c656d34441fcedf4d571254834b375a99d97fd3c1443dd8bc4ef165c48334a66e64267efa7e9da7db43f9fcfea7c96d90a023f83e25b9
-  data.tar.gz: aa13ee94cb12d5729204fa8ecd926142bf56c58f30e9a6e13ee1c73914748fda0d752d3aeb7d31292de677c8ad4c19c5dd8636d919b05664b154cd4d62e0188a
+  metadata.gz: 35acabbfbd4ab4c549d8cefdecc829074ade2bc041093432b3a1ae52c095f1d2e8a4420d30754dfd0a1dfb9d09f999fb2885b4f87c7e14c2c3386afff886082f
+  data.tar.gz: 01f8881d042a106e720574902abc2227161c65c78b35277618aea2343f8be43dfdcb4d94735ce2f5a7fd62ffb5137e5c928e0bd1e9dd50cd583ecd6819c80d38

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    orthoses (0.3.0)
+    orthoses (0.6.0)
       rbs (~> 2.0)
 
 GEM
@@ -30,9 +30,9 @@ GEM
     rb-fsevent (0.11.1)
     rb-inotify (0.10.1)
       ffi (~> 1.0)
-    rbs (2.4.0)
+    rbs (2.5.0)
     rgot (1.1.0)
-    steep (0.52.2)
+    steep (1.0.0)
       activesupport (>= 5.1)
       language_server-protocol (>= 3.15, < 4.0)
       listen (~> 3.0)

data/README.md

@@ -89,18 +89,6 @@ This is very useful for middleware development and debugging.
 If a class is defined using the `DelegateClass` method
 RBS is automatically generated as inherited.
 
-### Orthoses::Writer
-
-Write output RBS to `io` option object.
-`io` should be able to call `#write` method.
-This is useful when you want to combine outputs into a single output.
-
-### Orthoses::RBSPrototypeRB
-
-Run `rbs prototype rb` command process to `paths` option files.
-`content_filter` option could filter with content decl.
-`mixin_filter` option could filter with mixin(include, extend, prepend) decl.
-
 .rb
 
 ```rb
@@ -115,6 +103,22 @@ class Tempfile < File
 end
 ```
 
+### Orthoses::Writer
+
+Write output RBS to `io` option object.
+`io` should be able to call `#write` method.
+This is useful when you want to combine outputs into a single output.
+
+### Orthoses::RBSPrototypeRB
+
+Run `rbs prototype rb` command process to `paths` option files.
+`content_filter` option could filter with content decl.
+`mixin_filter` option could filter with mixin(include, extend, prepend) decl.
+
+### Orthoses::AvoidRecursiveAncestorError
+
+Mixin a module into an Object class raises `RBS::RecursiveAncestorError` when validation.
+Please add this middleware then.
 
 ## Development
 

data/examples/minitest/Rakefile

@@ -0,0 +1,56 @@
+file :src do
+  sh "git clone https://github.com/minitest/minitest.git src"
+end
+
+desc "generate minitest"
+task :minitest => :src do
+  $:.unshift(File.expand_path("src/lib"))
+
+  out = Pathname('out')
+  out.rmtree rescue nil
+  require 'orthoses'
+
+  Orthoses::Builder.new do
+    use Orthoses::CreateFileByName,
+      base_dir: out.to_s
+    use Orthoses::Filter, if: ->(name, content) {
+      name.start_with?("Minitest") || name == "Kernel"
+    }
+    use Orthoses::Constant,
+      strict: false,
+      if: ->(current, const, val, rbs){
+        !Orthoses::Utils.rbs_defined_const?("#{current}::#{const}")
+      }
+    use Orthoses::Mixin
+    use Orthoses::Attribute
+    use Orthoses::RBSPrototypeRB,
+      paths: Dir.glob('src/lib/minitest{.rb,/*.rb}'),
+      constant_filter: ->(_) { false },
+      mixin_filter: ->(_) { false },
+      attribute_filter: ->(_) { false }
+    run ->(){
+      require_relative 'src/lib/minitest'
+      require_relative 'src/lib/minitest/test'
+      require_relative 'src/lib/minitest/spec'
+      require_relative 'src/lib/minitest/parallel'
+      require_relative 'src/lib/minitest/mock'
+      require_relative 'src/lib/minitest/benchmark'
+
+      require_relative 'src/lib/minitest/pride_plugin'
+    }
+  end.call
+
+  sh "rbs -r mutex_m -I #{out} validate --silent"
+  out.join("manifest.yaml").write <<~YAML
+    dependencies:
+      - name: mutex_m
+  YAML
+  tmp = out.join("kernel.rbs").read
+  out.join("kernel.rbs").open("w") do |f|
+    f.puts "%a{annotate:rdoc:skip}"
+    f.write tmp
+  end
+  sh "rbs annotate --gems out"
+end
+
+task default: :minitest

data/examples/simple_middleware.rb

@@ -0,0 +1,11 @@
+module Orthoses
+  class Simple
+    def initialize(loader)
+      @loader = loader
+    end
+
+    def call
+      @loader.call
+    end
+  end
+end

data/examples/simple_middleware.rbs

@@ -0,0 +1,7 @@
+module Orthoses
+  class Simple
+    @loader: Orthoses::_Call
+    include Orthoses::_MiddleWare
+    def initialize: (Orthoses::_Call loader) -> void
+  end
+end

data/lib/orthoses/attribute.rb

@@ -0,0 +1,98 @@
+module Orthoses
+  class Attribute
+    module Hook
+      def attr(*names)
+        super
+      end
+
+      def attr_accessor(*names)
+        super
+      end
+
+      def attr_reader(*names)
+        super
+      end
+
+      def attr_writer(*names)
+        super
+      end
+    end
+
+    def initialize(loader)
+      @loader = loader
+    end
+
+    def call
+      Module.prepend(Hook)
+
+      attr = CallTracer.new
+      attr_accessor = CallTracer.new
+      attr_reader = CallTracer.new
+      attr_writer = CallTracer.new
+
+      store = attr.trace(Hook.method(:attr)) do
+        attr_accessor.trace(Hook.method(:attr_accessor)) do
+          attr_reader.trace(Hook.method(:attr_reader)) do
+            attr_writer.trace(Hook.method(:attr_writer)) do
+              @loader.call
+            end
+          end
+        end
+      end
+
+      store["Module"].body.delete("prepend Orthoses::Attribute::Hook")
+
+      attr.result.each do |method, argument|
+        m = method.receiver.to_s.match(/#<Class:([\w:]+)>/)
+        if m && m[1]
+          receiver_name = m[1]
+          prefix = "self."
+        else
+          receiver_name = Utils.module_name(method.receiver) or next
+          prefix = nil
+        end
+        content = store[receiver_name]
+        if argument[:names][1].equal?(true)
+          content << "attr_accessor #{prefix}#{argument[:names][0]}: untyped"
+        elsif argument[:names][1].equal?(false)
+          content << "attr_reader #{prefix}#{argument[:names][0]}: untyped"
+        else
+          argument[:names].each do |name|
+            content << "attr_reader #{prefix}#{name}: untyped"
+          end
+        end
+      end
+
+      each_definition(attr_accessor) do |receiver_name, name|
+        store[receiver_name] << "attr_accessor #{name}: untyped"
+      end
+      each_definition(attr_reader) do |receiver_name, name|
+        store[receiver_name] << "attr_reader #{name}: untyped"
+      end
+      each_definition(attr_writer) do |receiver_name, name|
+        store[receiver_name] << "attr_writer #{name}: untyped"
+      end
+
+      store
+    end
+
+    private
+
+    def each_definition(call_tracer)
+      call_tracer.result.each do |method, argument|
+        m = method.receiver.to_s.match(/#<Class:([\w:]+)>/)
+        if m && m[1]
+          receiver_name = m[1]
+          argument[:names].each do |name|
+            yield [receiver_name, "self.#{name}"]
+          end
+        else
+          receiver_name = Utils.module_name(method.receiver) or next
+          argument[:names].each do |name|
+            yield [receiver_name, name]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/orthoses/avoid_recursive_ancestor_error.rb

@@ -0,0 +1,22 @@
+module Orthoses
+  class AvoidRecursiveAncestorError
+    def initialize(loader)
+      @loader = loader
+    end
+
+    def call
+      @loader.call.tap do |store|
+        object_mixins = {}
+        store["Object"].to_decl.members.each do |member|
+          case member
+          when RBS::AST::Members::Mixin
+            object_mixins[member.name.relative!.to_s] = true
+          end
+        end
+        object_mixins.each_key do |object_mixin|
+          store[object_mixin].header = "module #{object_mixin} : BasicObject"
+        end
+      end
+    end
+  end
+end

data/lib/orthoses/builder.rb

@@ -27,6 +27,7 @@ module Orthoses
     end
 
     def use(middleware, *args, **key, &block)
+      Orthoses.logger.debug("use #{middleware}")
       @use << proc do |loader|
         middleware.new(loader, *args, **key, &block).tap do |m|
           m.extend CallLogable

data/lib/orthoses/content/duplication_checker.rb

@@ -4,24 +4,35 @@ module Orthoses
     class DuplicationChecker
       def initialize(decl)
         @decl = decl
-        @uniq_map = {}
       end
 
-      def <<(member)
-        key = member_key(member)
-        if @uniq_map.key?(key)
-          Orthoses.logger.warn("#{@decl.name}::#{member_to_s(member)} was droped since duplication")
-        else
-          @uniq_map[key] = member
+      def update_decl
+        uniq_map = {}
+        @decl.members.each do |member|
+          key = member_key(member)
+          drop_member = uniq_map[key]
+          uniq_map[key] = member
+          if drop_member
+            Orthoses.logger.warn("#{@decl.name}::#{drop_member} was droped since duplication")
+          end
         end
-      end
-
-      def uniq_members
-        @uniq_map.values
+        drop_known_method_definition(uniq_map)
+        @decl.members.replace(uniq_map.values)
       end
 
       private
 
+      def drop_known_method_definition(uniq_map)
+        env = Utils.rbs_environment(collection: true)
+        if m_entry = env.class_decls[@decl.name.absolute!]
+          m_entry.decls.each do |d|
+            d.decl.members.grep_v(RBS::AST::Members::LocationOnly).each do |member|
+              uniq_map.delete(member_key(member))
+            end
+          end
+        end
+      end
+
       def member_to_s(member)
         out = StringIO.new
         RBS::Writer.new(out: out).write_member(member)
@@ -36,7 +47,14 @@ module Orthoses
         when RBS::AST::Members::LocationOnly
           [member.class]
         when RBS::AST::Members::Alias
-          [member.class, member.new_name, member.old_name, member.kind]
+          [RBS::AST::Members::MethodDefinition, member.new_name, member.kind]
+        when RBS::AST::Members::AttrAccessor
+          # FIXME: how to check "#{member.name}=" ?
+          [RBS::AST::Members::MethodDefinition, member.name, member.kind]
+        when RBS::AST::Members::AttrReader
+          [RBS::AST::Members::MethodDefinition, member.name, member.kind]
+        when RBS::AST::Members::AttrWriter
+          [RBS::AST::Members::MethodDefinition, "#{member.name}=", member.kind]
         else
           [member.class, member.name]
         end

data/lib/orthoses/content/environment.rb

@@ -1,87 +1,82 @@
 module Orthoses
   class Content
     class Environment
-      def initialize(constant_filter: nil, mixin_filter: nil)
-        @env = RBS::Environment.new
+      class << self
+        def load_from_paths(paths)
+          new.tap do |env|
+            paths.each do |path|
+              Orthoses.logger.debug("Load #{path}")
+              buffer = RBS::Buffer.new(name: path.to_s, content: File.read(path.to_s, encoding: "UTF-8"))
+              RBS::Parser.parse_signature(buffer).each do |decl|
+                env << decl
+              end
+            end
+          end
+        end
+      end
+
+      def initialize(constant_filter: nil, mixin_filter: nil, attribute_filter: nil)
+        @load_env = RBS::Environment.new
+        @known_env = Utils.rbs_environment(collection: true, cache: false)
         @constant_filter = constant_filter
         @mixin_filter = mixin_filter
+        @attribute_filter = attribute_filter
       end
 
       def <<(decl)
-        @env << decl
+        @load_env << decl
+        @known_env << decl
+      rescue RBS::DuplicatedDeclarationError => err
+        Orthoses.logger.warn(err.inspect)
       end
 
       def write_to(store:)
         each do |add_content|
           content = store[add_content.name]
-          # content.header = add_content.header
+          content.header = add_content.header
           content.concat(add_content.body)
         end
       end
 
       def each
-        each_class do |content|
-          yield content
-        end
-        each_interface do |content|
-          yield content
-        end
-      end
+        avoid_generic_parameter_mismatch_error
 
-      private
+        header_builder = HeaderBuilder.new(env: @known_env)
 
-      def each_class
-        @env.class_decls.each do |type_name, m_entry|
+        @load_env.class_decls.each do |type_name, m_entry|
           name = type_name.relative!.to_s
           content = Content.new(name: name)
-          case decl = m_entry.decls.first.decl
-          when RBS::AST::Declarations::Module
-            self_types = decl.self_types.empty? ? nil : " : #{decl.self_types.join(', ')}"
-            content.header = "module #{name_and_params(name, decl.type_params)}#{self_types}"
-          when RBS::AST::Declarations::Class
-            super_class = decl.super_class.nil? ? nil : " < #{name_and_args(decl.super_class.name, decl.super_class.args)}"
-            content.header = "class #{name_and_params(name, decl.type_params)}#{super_class}"
-          else
-            raise
-          end
+          content.header = header_builder.build(entry: m_entry, name_hint: name)
           decls_to_lines(m_entry.decls.map(&:decl)).each do |line|
             content << line
           end
           yield content
         end
-      end
 
-      def each_interface
-        @env.interface_decls.each do |type_name, s_entry|
+        @load_env.interface_decls.each do |type_name, s_entry|
           name = type_name.relative!.to_s
           content = Content.new(name: name)
-          decl = s_entry.decl
-          content.header = "interface #{name_and_params(name, decl.type_params)}"
-          decls_to_lines([decl]).each do |line|
+          content.header = header_builder.build(entry: s_entry)
+          decls_to_lines([s_entry.decl]).each do |line|
             content << line
           end
           yield content
         end
       end
 
-      def name_and_params(name, params)
-        if params.empty?
-          "#{name}"
-        else
-          ps = params.each.map do |param|
-            param.to_s
-          end
-
-          "#{name}[#{ps.join(", ")}]"
-        end
-      end
+      private
 
-      def name_and_args(name, args)
-        if name && args
-          if args.empty?
-            "#{name}"
-          else
-            "#{name}[#{args.join(", ")}]"
+      # Avoid `RBS::GenericParameterMismatchError` from like rbs_prototype_rb
+      #     class Array # <= RBS::GenericParameterMismatchError
+      #     end
+      def avoid_generic_parameter_mismatch_error
+        @known_env.class_decls.each do |type_name, m_entry|
+          tmp_primary_d = m_entry.decls.find { |d| !d.decl.type_params.empty? }
+          next unless tmp_primary_d
+          m_entry.decls.each do |d|
+            if d.decl.type_params.empty?
+              d.decl.type_params.replace(tmp_primary_d.decl.type_params)
+            end
           end
         end
       end
@@ -98,6 +93,8 @@ module Orthoses
               next unless @constant_filter.nil? || @constant_filter.call(member)
             when RBS::AST::Members::Mixin
               next unless @mixin_filter.nil? || @mixin_filter.call(member)
+            when RBS::AST::Members::Attribute
+              next unless @attribute_filter.nil? || @attribute_filter.call(member)
             end
             writer.write_member(member)
           end

data/lib/orthoses/content/header_builder.rb

@@ -0,0 +1,94 @@
+module Orthoses
+  class Content
+    class HeaderBuilder
+      def initialize(env:)
+        @env = env
+        @resolver = RBS::TypeNameResolver.from_env(env)
+      end
+
+      def build(entry:, name_hint: nil)
+        case entry
+        when RBS::Environment::ModuleEntry
+          build_module(entry: entry, name_hint: name_hint)
+        when RBS::Environment::ClassEntry
+          build_class(entry: entry, name_hint: name_hint)
+        when RBS::Environment::SingleEntry
+          build_interface(entry: entry, name_hint: name_hint)
+        else
+          raise
+        end
+      end
+
+      private
+
+      def build_module(entry:, name_hint: nil)
+        primary = entry.primary
+        full_name = name_hint || primary.decl.name.relative!
+
+        self_types =
+          if primary.decl.self_types.empty?
+            nil
+          else
+            " : #{primary.decl.self_types.join(', ')}"
+          end
+
+        "module #{name_and_params(full_name, primary.decl.type_params)}#{self_types}"
+      end
+
+      def build_class(entry:, name_hint: nil)
+        primary = entry.primary
+        full_name = name_hint || primary.decl.name.relative!
+
+        "class #{name_and_params(full_name, primary.decl.type_params)}#{build_super_class(primary)}"
+      end
+
+      def build_super_class(primary)
+        return nil if primary.decl.super_class.then { |s| s.nil? || s.name.relative!.to_s.then { |n| n == "Object" || n == "Random::Base" } }
+
+        context = primary.outer.length.times.map do |i|
+          primary.outer[0, i + 1].map(&:name).inject(:+).to_namespace.absolute!
+        end
+        context.push(RBS::Namespace.root)
+
+        super_class_name = @resolver.resolve(primary.decl.super_class.name, context: context) || primary.decl.super_class.name
+        if primary.decl.super_class.args.empty?
+          if super_class_entry = @env.class_decls[super_class_name]
+            super_primary = super_class_entry.primary
+            " < #{name_and_args(super_class_name, super_primary.decl.type_params.map { :untyped })}"
+          else
+            " < #{name_and_args(super_class_name, [])}"
+          end
+        else
+          " < #{name_and_args(super_class_name, primary.decl.super_class.args)}"
+        end
+      end
+
+      def build_interface(entry:, name_hint: nil)
+        full_name = name_hint || entry.decl.name.relative!
+        "interface #{name_and_params(full_name, entry.decl.type_params)}"
+      end
+
+      def name_and_params(name, params)
+        if params.empty?
+          "#{name}"
+        else
+          ps = params.each.map do |param|
+            param.to_s
+          end
+
+          "#{name}[#{ps.join(", ")}]"
+        end
+      end
+
+      def name_and_args(name, args)
+        if name && args
+          if args.empty?
+            "#{name}"
+          else
+            "#{name}[#{args.join(", ")}]"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/orthoses/content.rb

@@ -2,6 +2,7 @@
 
 require 'orthoses/content/duplication_checker'
 require 'orthoses/content/environment'
+require 'orthoses/content/header_builder'
 
 module Orthoses
   # Common interface for output.
@@ -20,36 +21,79 @@ module Orthoses
     attr_accessor :header
 
     def initialize(name:)
+      Orthoses.logger.debug("Create Orthoses::Content for #{name}")
       @name = name
       @body = []
       @header = nil
     end
 
     def <<(line)
-      Orthoses.logger.debug("[#{name}] << #{line} on #{__FILE__}:#{__LINE__}")
       @body << line
     end
 
     def concat(other)
-      Orthoses.logger.debug("[#{name}] concat #{other} on #{__FILE__}:#{__LINE__}")
       @body.concat(other)
     end
 
     def to_rbs
-      if @header.nil?
-        auto_header
-      end
-      body_uniq("#{@header}\n#{@body.map { |b| "  #{b}\n" }.join}end\n")
+      auto_header
+      uniqed_body_string
+    end
+
+    def to_decl
+      auto_header
+      uniqed_body_decl
     end
 
     private
 
+    def original_rbs
+      <<~RBS
+        #{@header}
+          #{@body.join("\n")}
+        end
+      RBS
+    end
+
+    def uniqed_body_string
+      out = StringIO.new
+      writer = RBS::Writer.new(out: out)
+      writer.write_decl(uniqed_body_decl)
+      out.string
+    end
+
+    def uniqed_body_decl
+      buffer = RBS::Buffer.new(
+        name: "orthoses/content.rb",
+        content: original_rbs
+      )
+      parsed_decls = RBS::Parser.parse_signature(buffer)
+      unless parsed_decls.length == 1
+        raise "expect decls.length == 1, but got #{parsed_decls.length}"
+      end
+      parsed_decl = parsed_decls.first or raise
+      parsed_decl.tap do |decl|
+        DuplicationChecker.new(decl).update_decl
+      end
+    rescue RBS::ParsingError
+      Orthoses.logger.error "```rbs\n#{original_rbs}```"
+      raise
+    end
+
     def auto_header
       if name.split('::').last.start_with?('_')
         self.header = "interface #{name}"
         return
       end
 
+      env = Utils.rbs_environment(collection: true)
+      if entry = env.class_decls[TypeName(name).absolute!]
+        @header = Content::HeaderBuilder.new(env: env).build(entry: entry)
+        return
+      end
+
+      return unless @header.nil?
+
       val = Object.const_get(name)
 
       case val
@@ -93,28 +137,5 @@ module Orthoses
         end
       end
     end
-
-    def body_uniq(rbs)
-      buffer = RBS::Buffer.new(
-        name: "orthoses/content.rb",
-        content: rbs
-      )
-      out = StringIO.new
-      writer = RBS::Writer.new(out: out)
-      decls = RBS::Parser.parse_signature(buffer).map do |parsed_decl|
-        parsed_decl.tap do |decl|
-          duplicate_checker = DuplicationChecker.new(decl)
-          decl.members.each do |member|
-            duplicate_checker << member
-          end
-          decl.members.replace(duplicate_checker.uniq_members)
-        end
-      end
-      writer.write(decls)
-      out.string
-    rescue RBS::ParsingError
-      Orthoses.logger.error "```rbs\n#{rbs}```"
-      raise
-    end
   end
 end

data/lib/orthoses/load_rbs.rb

@@ -11,14 +11,7 @@ module Orthoses
 
     def call
       @loader.call.tap do |store|
-        env = Content::Environment.new
-        @paths.each do |path|
-          Orthoses.logger.debug("Load #{path}")
-          buffer = RBS::Buffer.new(name: path.to_s, content: File.read(path.to_s, encoding: "UTF-8"))
-          RBS::Parser.parse_signature(buffer).each do |decl|
-            env << decl
-          end
-        end
+        env = Content::Environment.load_from_paths(@paths)
         env.write_to(store: store)
       end
     end

data/lib/orthoses/mixin.rb

@@ -37,15 +37,7 @@ module Orthoses
           next unless known_type_params.nil? || known_type_params.empty?
           next unless @if.nil? || @if.call(base_mod, how, mod)
 
-          if base_mod_name == "Object"
-            # avoid RecursiveAncestorError
-            content = store[mod_name]
-            if content.header.nil?
-              content.header = "module #{Utils.module_to_type_name(mod)} : ::BasicObject"
-            end
-          else
-            store[mod_name]
-          end
+          store[mod_name]
 
           "#{how} #{mod_name}"
         end

data/lib/orthoses/rbs_prototype_rb.rb

@@ -3,11 +3,12 @@ module Orthoses
   #     use Orthoses::RBSPrototypeRB,
   #       paths: Dir.glob("lib/**/*.rb")
   class RBSPrototypeRB
-    def initialize(loader, paths:, constant_filter: nil, mixin_filter: nil)
+    def initialize(loader, paths:, constant_filter: nil, mixin_filter: nil, attribute_filter: nil)
       @loader = loader
       @paths = paths
       @constant_filter = constant_filter
       @mixin_filter = mixin_filter
+      @attribute_filter = attribute_filter
     end
 
     def call
@@ -15,6 +16,7 @@ module Orthoses
         parser = RBS::Prototype::RB.new
         @paths.each do |path|
           begin
+            Orthoses.logger.debug("parse #{path} by RBS::Prototype::RB")
             parser.parse File.read(path.to_s)
           rescue => err
             Orthoses.logger.error("Parse error #{err.inspect} by RBS::Prototype::RB #{path}")
@@ -23,6 +25,7 @@ module Orthoses
         env = Orthoses::Content::Environment.new(
           constant_filter: @constant_filter,
           mixin_filter: @mixin_filter,
+          attribute_filter: @attribute_filter,
         )
         parser.decls.each do |decl|
           env << decl

data/lib/orthoses/utils.rb

@@ -46,17 +46,39 @@ module Orthoses
       RBS::Namespace.parse(name).to_type_name
     end
 
-    def self.rbs_environment(library: nil, collection: false)
+    class << self
+      # Generated scripts are not always located in the root.
+      # By default, it follows the upper level directory from the current directory
+      # to find the rbs_collection.yaml file and set it to `rbs_collection_pathname`.
+      # It can be reconfigured if necessary.
+      attr_accessor :rbs_collection_pathname
+    end
+    # set default
+    self.rbs_collection_pathname = ->() {
+      begin
+        start = here = Dir.pwd
+        until ok = RBS::Collection::Config::PATH.exist?
+          Dir.chdir("..")
+          return nil if Dir.pwd == here
+          here = Dir.pwd
+        end
+        Pathname(here) + RBS::Collection::Config::PATH
+      ensure
+        Dir.chdir(start)
+      end
+    }.call
+
+    def self.rbs_environment(library: nil, collection: false, cache: true)
       @env_cache ||= {}
-      if hit = @env_cache[[library, collection]]
+      if cache && hit = @env_cache[[library, collection]]
         return hit
       end
 
       loader = RBS::EnvironmentLoader.new
 
-      if collection
-        lock = RBS::Collection::Config::PATH&.then { |p| RBS::Collection::Config.lockfile_of(p) }
-        loader.add_collection(lock) if lock
+      if collection && rbs_collection_pathname
+        config = RBS::Collection::Config.lockfile_of(rbs_collection_pathname) or raise
+        loader.add_collection(config)
       end
 
       case library
@@ -71,8 +93,9 @@ module Orthoses
         end
       end
 
-      environment = RBS::Environment.from_loader(loader).resolve_type_names
-      @env_cache[[library, collection]] = environment
+      RBS::Environment.from_loader(loader).resolve_type_names.tap do |env|
+        @env_cache[[library, collection]] = env if cache
+      end
     end
 
     def self.object_to_rbs(object, strict:)
@@ -121,6 +144,9 @@ module Orthoses
             "Hash[#{keys.join(' | ')}, #{values.join(' | ')}]"
           end
         end
+      when Range
+        type = object_to_rbs(object.begin || object.end, strict: false)
+        "Range[#{type}]"
       when ARGF
         # see also https://github.com/ruby/rbs/pull/975
         'untyped'

data/lib/orthoses/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Orthoses
-  VERSION = "0.3.0"
+  VERSION = "0.6.0"
 end

data/lib/orthoses.rb

@@ -3,6 +3,8 @@
 require 'rbs'
 require 'pathname'
 
+require_relative 'orthoses/attribute'
+require_relative 'orthoses/avoid_recursive_ancestor_error'
 require_relative 'orthoses/builder'
 require_relative 'orthoses/call_tracer'
 require_relative 'orthoses/constant'

data/sig/orthoses/_call.rbs

@@ -1,5 +1,5 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 interface Orthoses::_Call
-  def call: () -> Hash[String, Orthoses::Content]
+  def call: () -> Orthoses::store
 end

data/sig/orthoses/attribute/hook.rbs

@@ -0,0 +1,11 @@
+# THIS IS GENERATED CODE from `$ rake generate_self_sig`
+
+module Orthoses::Attribute::Hook
+  def attr: (*untyped names) -> untyped
+
+  def attr_accessor: (*untyped names) -> untyped
+
+  def attr_reader: (*untyped names) -> untyped
+
+  def attr_writer: (*untyped names) -> untyped
+end

data/sig/orthoses/attribute.rbs

@@ -0,0 +1,9 @@
+# THIS IS GENERATED CODE from `$ rake generate_self_sig`
+
+class Orthoses::Attribute
+  def initialize: (Orthoses::_Call loader) -> void
+  def call: () -> Orthoses::store
+  private
+  def each_definition: (untyped call_tracer) { (untyped) -> untyped } -> untyped
+  @loader: Orthoses::_Call
+end

data/sig/orthoses/avoid_recursive_ancestor_error.rbs

@@ -0,0 +1,7 @@
+# THIS IS GENERATED CODE from `$ rake generate_self_sig`
+
+class Orthoses::AvoidRecursiveAncestorError
+  def initialize: (Orthoses::_Call loader) -> void
+  def call: () -> Orthoses::store
+  @loader: Orthoses::_Call
+end

data/sig/orthoses/builder/call_logable.rbs

@@ -1,4 +1,5 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 module Orthoses::Builder::CallLogable
+  def call: () -> untyped
 end

data/sig/orthoses/builder.rbs

@@ -1,5 +1,10 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Builder
+  def initialize: () { () -> untyped } -> void
+  def use: (untyped middleware, *untyped args, **untyped key) { () -> untyped } -> untyped
+  def run: (untyped loader) -> untyped
+  def to_loader: () -> untyped
+
   def call: () { () -> void } -> Orthoses::store
 end

data/sig/orthoses/call_tracer.rbs

@@ -2,5 +2,7 @@
 
 class Orthoses::CallTracer
   attr_accessor result: Array[[ Method, Hash[Symbol, untyped] ]]
+  def initialize: () -> void
+
   def trace: [T] (Method | UnboundMethod) ?{ () -> T } -> T
 end

data/sig/orthoses/const_load_error.rbs

@@ -5,4 +5,5 @@ class Orthoses::ConstLoadError < ::StandardError
   attr_reader const: Symbol
   attr_reader error: untyped
   def initialize: (root: Module, const: Symbol, error: untyped) -> void
+  def message: () -> ::String
 end

data/sig/orthoses/constant.rbs

@@ -1,7 +1,10 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Constant
+  def initialize: (Orthoses::_Call loader, strict: bool, ?if: ^(Module, Symbol, untyped) -> boolish, ?on_error: ^(Orthoses::ConstLoadError) -> void) -> void
+  def call: () -> Orthoses::store
   @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
-  def initialize: (Orthoses::_Call loader, ?if: ^(Module, Symbol, untyped) -> boolish, ?on_error: ^(Orthoses::ConstLoadError) -> void) -> void
+  @strict: bool
+  @if: ^(Module mod, Symbol const, untyped val, String rbs) -> boolish?
+  @on_error: ^(Orthoses::ConstLoadError) -> void | nil
 end

data/sig/orthoses/content/duplication_checker.rbs

@@ -1,4 +1,10 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Content::DuplicationChecker
+  def initialize: (untyped decl) -> void
+  def update_decl: () -> untyped
+  private
+  def drop_known_method_definition: (untyped uniq_map) -> untyped
+  def member_to_s: (untyped member) -> untyped
+  def member_key: (untyped member) -> untyped
 end

data/sig/orthoses/content/environment.rbs

@@ -1,13 +1,23 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Content::Environment
+  def self.load_from_paths: (untyped paths) -> untyped
+
+  def initialize: (?constant_filter: constant_filter?, ?mixin_filter: mixin_filter?, ?attribute_filter: attribute_filter?) -> void
+  def <<: (RBS::AST::Declarations::t decl) -> RBS::Environment
+  def write_to: (store: Orthoses::store) -> void
+  def each: () { (Orthoses::Content) -> void } -> void
+  private
+  # Avoid `RBS::GenericParameterMismatchError` from like rbs_prototype_rb
+  #     class Array # <= RBS::GenericParameterMismatchError
+  #     end
+  def avoid_generic_parameter_mismatch_error: () -> untyped
+  def decls_to_lines: (untyped decls) -> untyped
   type constant_filter = ^(RBS::AST::Declarations::Constant) -> boolish
   type mixin_filter = ^(RBS::AST::Members::Mixin) -> boolish
+  type attribute_filter = ^(RBS::AST::Members::Attribute) -> boolish
   @env: RBS::Environment
   @constant_filter: constant_filter?
   @mixin_filter: mixin_filter?
-  def initialize: (?constant_filter: constant_filter?, ?mixin_filter: mixin_filter?) -> void
-  def <<: (RBS::AST::Declarations::t decl) -> RBS::Environment
-  def write_to: (store: Orthoses::store) -> void
-  def each: () { (Orthoses::Content) -> void } -> void
+  @attribute_filter: attribute_filter?
 end

data/sig/orthoses/content/header_builder.rbs

@@ -0,0 +1,13 @@
+# THIS IS GENERATED CODE from `$ rake generate_self_sig`
+
+class Orthoses::Content::HeaderBuilder
+  def initialize: (env: untyped) -> void
+  def build: (entry: untyped, ?name_hint: untyped?) -> untyped
+  private
+  def build_module: (entry: untyped, ?name_hint: untyped?) -> ::String
+  def build_class: (entry: untyped, ?name_hint: untyped?) -> ::String
+  def build_super_class: (untyped primary) -> (nil | untyped)
+  def build_interface: (entry: untyped, ?name_hint: untyped?) -> ::String
+  def name_and_params: (untyped name, untyped params) -> ::String
+  def name_and_args: (untyped name, untyped args) -> (::String | nil)
+end

data/sig/orthoses/content.rbs

@@ -8,4 +8,14 @@ class Orthoses::Content
   def <<: (String) -> void
   def concat: (Array[String]) -> void
   def to_rbs: () -> String
+  def to_decl: () -> untyped
+
+  private
+  def original_rbs: () -> ::String
+
+  def uniqed_body_string: () -> String
+  def uniqed_body_decl: () -> RBS::AST::Declarations::t
+  def auto_header: () -> (nil | untyped)
+  def temporary_type_params: (untyped name) -> untyped
+  def type_params: (untyped name) -> untyped
 end

data/sig/orthoses/create_file_by_name.rbs

@@ -1,7 +1,7 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::CreateFileByName
-  @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
   def initialize: (Orthoses::_Call loader, base_dir: String, ?header: String?) -> void
+  def call: () -> Orthoses::store
+  @loader: Orthoses::_Call
 end

data/sig/orthoses/delegate_class.rbs

@@ -1,7 +1,8 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::DelegateClass
-  @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
   def initialize: (Orthoses::_Call loader) -> void
+  def call: () -> Orthoses::store
+  def temporary_type_params: (untyped name) -> untyped
+  @loader: Orthoses::_Call
 end

data/sig/orthoses/filter.rbs

@@ -1,7 +1,7 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Filter
-  @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
   def initialize: (Orthoses::_Call loader, if: ^(String, Orthoses::Content) -> boolish) -> void
+  def call: () -> Orthoses::store
+  @loader: Orthoses::_Call
 end

data/sig/orthoses/load_rbs.rbs

@@ -1,8 +1,8 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::LoadRBS
+  def initialize: (Orthoses::_Call loader, paths: Array[_ToS]) -> void
+  def call: () -> Orthoses::store
   @loader: Orthoses::_Call
   @paths: Array[_ToS]
-  include Orthoses::_MiddleWare
-  def initialize: (Orthoses::_Call loader, paths: Array[_ToS]) -> void
 end

data/sig/orthoses/mixin.rbs

@@ -1,4 +1,6 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Mixin
+  def initialize: (untyped loader, ?if: untyped?) -> void
+  def call: () -> untyped
 end

data/sig/orthoses/object_space_all.rbs

@@ -1,7 +1,7 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::ObjectSpaceAll
-  @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
   def initialize: (Orthoses::_Call loader, ?if: ^(Module) -> boolish) -> void
+  def call: () -> Orthoses::store
+  @loader: Orthoses::_Call
 end

data/sig/orthoses/pp.rbs

@@ -1,4 +1,6 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::PP
+  def initialize: (untyped loader) -> void
+  def call: () -> untyped
 end

data/sig/orthoses/rbs_prototype_rb.rbs

@@ -1,10 +1,11 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::RBSPrototypeRB
+  def initialize: (Orthoses::_Call loader, paths: Array[_ToS], ?constant_filter: Orthoses::Content::Environment::constant_filter?, ?mixin_filter: Orthoses::Content::Environment::mixin_filter?, ?attribute_filter: Orthoses::Content::Environment::attribute_filter?) -> void
+  def call: () -> Orthoses::store
   @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
-  def initialize: (Orthoses::_Call loader, paths: Array[_ToS], ?constant_filter: Orthoses::Content::Environment::constant_filter?, ?mixin_filter: Orthoses::Content::Environment::mixin_filter?) -> void
   @paths: Array[_ToS]
   @constant_filter: Orthoses::Content::Environment::constant_filter?
   @mixin_filter: Orthoses::Content::Environment::mixin_filter?
+  @attribute_filter: Orthoses::Content::Environment::mixin_filter?
 end

data/sig/orthoses/store.rbs

@@ -1,4 +1,6 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Store
+  def initialize: (untyped loader) -> void
+  def call: () -> untyped
 end

data/sig/orthoses/tap.rbs

@@ -1,7 +1,7 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Tap
-  @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
   def initialize: (Orthoses::_Call loader) -> void
+  def call: () -> Orthoses::store
+  @loader: Orthoses::_Call
 end

data/sig/orthoses/utils.rbs

@@ -1,11 +1,9 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 module Orthoses::Utils
-  UNBOUND_NAME_METHOD: UnboundMethod
-
   def self.unautoload!: () -> void
 
-  def self.each_const_recursive: (Module root, ?cache: Hash[untyped, true], ?on_error: ^(Orthoses::ConstLoadError) -> void?) ?{ (Module, Symbol, untyped) -> void } -> void
+  def self.each_const_recursive: (Module root, ?cache: Hash[untyped, true], ?on_error: ^(Orthoses::ConstLoadError) -> void) ?{ (Module, Symbol, untyped) -> void } -> void
 
   def self.rbs_defined_const?: (String name, ?library: (String | Array[String])?, ?collection: boolish) -> bool
 
@@ -13,10 +11,14 @@ module Orthoses::Utils
 
   def self.rbs_type_name: (String) -> RBS::TypeName
 
-  def self.rbs_environment: (?library: String | Array[String] | nil, ?collection: boolish) -> RBS::Environment
+  attr_accessor self.rbs_collection_pathname: Pathname
+
+  def self.rbs_environment: (?library: String | Array[String] | nil, ?collection: boolish, ?cache: boolish) -> RBS::Environment
 
   def self.object_to_rbs: (untyped object, strict: bool) -> String
 
+  UNBOUND_NAME_METHOD: UnboundMethod
+
   def self.module_name: (Module mod) -> String?
 
   def self.module_to_type_name: (Module) -> RBS::TypeName?

data/sig/orthoses/walk.rbs

@@ -1,8 +1,8 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Walk
+  def initialize: (Orthoses::_Call loader, root: Module | String) -> void
+  def call: () -> Orthoses::store
   @loader: Orthoses::_Call
   @root: Module | String
-  include Orthoses::_MiddleWare
-  def initialize: (Orthoses::_Call loader, root: Module | String) -> void
 end

data/sig/orthoses/writer.rbs

@@ -1,7 +1,7 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 class Orthoses::Writer
-  @loader: Orthoses::_Call
-  include Orthoses::_MiddleWare
   def initialize: (Orthoses::_Call loader, io: _Writer) -> void
+  def call: () -> Orthoses::store
+  @loader: Orthoses::_Call
 end

data/sig/orthoses.rbs

@@ -1,7 +1,7 @@
 # THIS IS GENERATED CODE from `$ rake generate_self_sig`
 
 module Orthoses
-  VERSION: String
+  VERSION: "0.6.0"
 
   attr_accessor self.logger: ::Logger
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: orthoses
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.6.0
 platform: ruby
 authors:
 - ksss
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-05-17 00:00:00.000000000 Z
+date: 2022-06-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rbs
@@ -36,13 +36,19 @@ files:
 - Gemfile.lock
 - LICENSE.txt
 - README.md
+- examples/minitest/Rakefile
+- examples/simple_middleware.rb
+- examples/simple_middleware.rbs
 - lib/orthoses.rb
+- lib/orthoses/attribute.rb
+- lib/orthoses/avoid_recursive_ancestor_error.rb
 - lib/orthoses/builder.rb
 - lib/orthoses/call_tracer.rb
 - lib/orthoses/constant.rb
 - lib/orthoses/content.rb
 - lib/orthoses/content/duplication_checker.rb
 - lib/orthoses/content/environment.rb
+- lib/orthoses/content/header_builder.rb
 - lib/orthoses/create_file_by_name.rb
 - lib/orthoses/delegate_class.rb
 - lib/orthoses/filter.rb
@@ -61,6 +67,9 @@ files:
 - sig/orthoses.rbs
 - sig/orthoses/_call.rbs
 - sig/orthoses/_middle_ware.rbs
+- sig/orthoses/attribute.rbs
+- sig/orthoses/attribute/hook.rbs
+- sig/orthoses/avoid_recursive_ancestor_error.rbs
 - sig/orthoses/builder.rbs
 - sig/orthoses/builder/call_logable.rbs
 - sig/orthoses/call_tracer.rbs
@@ -69,6 +78,7 @@ files:
 - sig/orthoses/content.rbs
 - sig/orthoses/content/duplication_checker.rbs
 - sig/orthoses/content/environment.rbs
+- sig/orthoses/content/header_builder.rbs
 - sig/orthoses/create_file_by_name.rbs
 - sig/orthoses/delegate_class.rbs
 - sig/orthoses/filter.rbs