rbs_rails 0.8.2 → 0.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2d70009e239cba7c79ad16cf1861495051952a6930205678b05fcf5ad97f0809
-  data.tar.gz: 20e8d58362c2222651aa10430417ac90af1b8e38e5b3bbfcfdf43f186f77eff6
+  metadata.gz: ccb7076ac91a406f491cf3e6c4d8499525f860660061ff23809635d000d1aba3
+  data.tar.gz: a863d9e16999e7491cef930b8fb79515c538ad7d83b919abad762a5c1290e370
 SHA512:
-  metadata.gz: ef998ce7910f4169eb43716ed8ad0132338751b0c9346213cb02ddecae71fe1c0159d8eb2404877cd599dbd352fe42d3d52bd1316f6577fe2568c912ec0f402a
-  data.tar.gz: 846055b9290bfb250e933686a7dcd05bd52ff331cc1a83a3ddae4cd05ff480b2b43b7c164dbbbff164f25c568837b9d30353eb787464f430b08ffbed69737add
+  metadata.gz: 10dccf3356c898258d6c13aaed04d61a5f137787f9ea696fe0d7005e2063c2fd130d9fdda59c4c7b4fa58d1cde84a3ca0cf750dac28f1ac9ffbfe94ead90da57
+  data.tar.gz: b8f5c227bf58bdb16044867cd4abd883f21782817eda8fa6d85d1956418c763829748d5d0ee0e81e1005ffd18e022d6da6e83b64aa97cf4f77fe6942bacec48b

data/.dependabot/config.yml

@@ -3,6 +3,3 @@ update_configs:
   - package_manager: "ruby:bundler"
     directory: "/"
     update_schedule: "live"
-  - package_manager: "ruby:bundler"
-    directory: "/test/app/"
-    update_schedule: "live"

data/.github/workflows/ci.yml

@@ -23,7 +23,5 @@ jobs:
       with:
         ruby-version: ${{ matrix.ruby }}
         bundler-cache: true # runs 'bundle install' and caches installed gems automatically
-    - env:
-        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-      run: bin/gem_rbs
+    - run: bin/setup
     - run: bundle exec rake

data/.gitignore

@@ -9,4 +9,4 @@
 /test/app/sig/rbs_rails
 /test/app/sig/app
 /test/app/sig/path_helpers.rbs
-/gem_rbs
+/.gem_rbs_collection/

data/CHANGELOG.md

@@ -2,6 +2,22 @@
 
 ## master (unreleased)
 
+## 0.9.0 (2021-09-18)
+
+### New Features
+
+* Support delegated_type association. [#181](https://github.com/pocke/rbs_rails/pull/181)
+
+### Bug Fixes
+
+* Fix error on a table that doesn't have the PK. [#121](https://github.com/pocke/rbs_rails/pull/121)
+* Quote variable names to avoid syntax errors. [#178](https://github.com/pocke/rbs_rails/pull/178)
+* Add scope methods to `ActiveRecord_Associations_CollectionProxy`. [#182](https://github.com/pocke/rbs_rails/pull/182)
+* Make `has_one` association optional. [#180](https://github.com/pocke/rbs_rails/pull/180)
+* Omit some methods for polymorphic associations. [#184](https://github.com/pocke/rbs_rails/pull/184)
+* Include enum methods to GeneratedRelationMethods. [#183](https://github.com/pocke/rbs_rails/pull/183)
+* Include `_ActiveRecord_Relation` to `CollectionProxy`. [#189](https://github.com/pocke/rbs_rails/pull/189)
+
 ## 0.8.2 (2021-02-20)
 
 *  Add ActiveRecord::AttributeMethods::Dirty methods [#104](https://github.com/pocke/rbs_rails/pull/104)

data/Gemfile

@@ -1,9 +1,12 @@
 source "https://rubygems.org"
 
+git_source(:github) { |repo_name| "https://github.com/#{repo_name}" }
+
 # Specify your gem's dependencies in rbs_rails.gemspec
 gemspec
 
 gem "rake", "~> 13.0"
-gem 'rbs', '>= 1.0.0'
-gem 'steep', '>= 0.34.0'
+gem 'rails', '< 6.1', '>= 6.0'
+gem 'rbs', '>= 1.6'
+gem 'steep', github: 'soutaro/steep'
 gem 'minitest'

data/Gemfile.lock

@@ -1,49 +1,166 @@
+GIT
+  remote: https://github.com/soutaro/steep
+  revision: 0673d88395fb0ea380463fef1aed568dfb077a60
+  specs:
+    steep (0.46.0)
+      activesupport (>= 5.1)
+      language_server-protocol (>= 3.15, < 4.0)
+      listen (~> 3.0)
+      parallel (>= 1.0.0)
+      parser (>= 3.0)
+      rainbow (>= 2.2.2, < 4.0)
+      rbs (>= 1.6.0)
+      terminal-table (>= 2, < 4)
+
 PATH
   remote: .
   specs:
-    rbs_rails (0.8.2)
+    rbs_rails (0.9.0)
       parser
       rbs (>= 1)
 
 GEM
   remote: https://rubygems.org/
   specs:
-    activesupport (6.1.2.1)
+    actioncable (6.0.4.1)
+      actionpack (= 6.0.4.1)
+      nio4r (~> 2.0)
+      websocket-driver (>= 0.6.1)
+    actionmailbox (6.0.4.1)
+      actionpack (= 6.0.4.1)
+      activejob (= 6.0.4.1)
+      activerecord (= 6.0.4.1)
+      activestorage (= 6.0.4.1)
+      activesupport (= 6.0.4.1)
+      mail (>= 2.7.1)
+    actionmailer (6.0.4.1)
+      actionpack (= 6.0.4.1)
+      actionview (= 6.0.4.1)
+      activejob (= 6.0.4.1)
+      mail (~> 2.5, >= 2.5.4)
+      rails-dom-testing (~> 2.0)
+    actionpack (6.0.4.1)
+      actionview (= 6.0.4.1)
+      activesupport (= 6.0.4.1)
+      rack (~> 2.0, >= 2.0.8)
+      rack-test (>= 0.6.3)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.0, >= 1.2.0)
+    actiontext (6.0.4.1)
+      actionpack (= 6.0.4.1)
+      activerecord (= 6.0.4.1)
+      activestorage (= 6.0.4.1)
+      activesupport (= 6.0.4.1)
+      nokogiri (>= 1.8.5)
+    actionview (6.0.4.1)
+      activesupport (= 6.0.4.1)
+      builder (~> 3.1)
+      erubi (~> 1.4)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.1, >= 1.2.0)
+    activejob (6.0.4.1)
+      activesupport (= 6.0.4.1)
+      globalid (>= 0.3.6)
+    activemodel (6.0.4.1)
+      activesupport (= 6.0.4.1)
+    activerecord (6.0.4.1)
+      activemodel (= 6.0.4.1)
+      activesupport (= 6.0.4.1)
+    activestorage (6.0.4.1)
+      actionpack (= 6.0.4.1)
+      activejob (= 6.0.4.1)
+      activerecord (= 6.0.4.1)
+      marcel (~> 1.0.0)
+    activesupport (6.0.4.1)
       concurrent-ruby (~> 1.0, >= 1.0.2)
-      i18n (>= 1.6, < 2)
-      minitest (>= 5.1)
-      tzinfo (~> 2.0)
-      zeitwerk (~> 2.3)
+      i18n (>= 0.7, < 2)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+      zeitwerk (~> 2.2, >= 2.2.2)
     ast (2.4.2)
-    ast_utils (0.4.0)
-      parser (>= 2.7.0)
-    concurrent-ruby (1.1.8)
-    ffi (1.14.2)
-    i18n (1.8.9)
+    builder (3.2.4)
+    concurrent-ruby (1.1.9)
+    crass (1.0.6)
+    erubi (1.10.0)
+    ffi (1.15.4)
+    globalid (0.5.2)
+      activesupport (>= 5.0)
+    i18n (1.8.10)
       concurrent-ruby (~> 1.0)
-    language_server-protocol (3.15.0.1)
-    listen (3.4.1)
+    language_server-protocol (3.16.0.3)
+    listen (3.7.0)
       rb-fsevent (~> 0.10, >= 0.10.3)
       rb-inotify (~> 0.9, >= 0.9.10)
-    minitest (5.14.3)
-    parser (3.0.0.0)
+    loofah (2.12.0)
+      crass (~> 1.0.2)
+      nokogiri (>= 1.5.9)
+    mail (2.7.1)
+      mini_mime (>= 0.1.1)
+    marcel (1.0.1)
+    method_source (1.0.0)
+    mini_mime (1.1.1)
+    mini_portile2 (2.6.1)
+    minitest (5.14.4)
+    nio4r (2.5.8)
+    nokogiri (1.12.4)
+      mini_portile2 (~> 2.6.1)
+      racc (~> 1.4)
+    parallel (1.20.1)
+    parser (3.0.2.0)
       ast (~> 2.4.1)
+    racc (1.5.2)
+    rack (2.2.3)
+    rack-test (1.1.0)
+      rack (>= 1.0, < 3)
+    rails (6.0.4.1)
+      actioncable (= 6.0.4.1)
+      actionmailbox (= 6.0.4.1)
+      actionmailer (= 6.0.4.1)
+      actionpack (= 6.0.4.1)
+      actiontext (= 6.0.4.1)
+      actionview (= 6.0.4.1)
+      activejob (= 6.0.4.1)
+      activemodel (= 6.0.4.1)
+      activerecord (= 6.0.4.1)
+      activestorage (= 6.0.4.1)
+      activesupport (= 6.0.4.1)
+      bundler (>= 1.3.0)
+      railties (= 6.0.4.1)
+      sprockets-rails (>= 2.0.0)
+    rails-dom-testing (2.0.3)
+      activesupport (>= 4.2.0)
+      nokogiri (>= 1.6)
+    rails-html-sanitizer (1.4.2)
+      loofah (~> 2.3)
+    railties (6.0.4.1)
+      actionpack (= 6.0.4.1)
+      activesupport (= 6.0.4.1)
+      method_source
+      rake (>= 0.8.7)
+      thor (>= 0.20.3, < 2.0)
     rainbow (3.0.0)
-    rake (13.0.3)
-    rb-fsevent (0.10.4)
+    rake (13.0.6)
+    rb-fsevent (0.11.0)
     rb-inotify (0.10.1)
       ffi (~> 1.0)
-    rbs (1.0.6)
-    steep (0.41.0)
-      activesupport (>= 5.1)
-      ast_utils (>= 0.4.0)
-      language_server-protocol (~> 3.15.0.1)
-      listen (~> 3.0)
-      parser (>= 2.7)
-      rainbow (>= 2.2.2, < 4.0)
-      rbs (~> 1.0.3)
-    tzinfo (2.0.4)
+    rbs (1.6.2)
+    sprockets (4.0.2)
       concurrent-ruby (~> 1.0)
+      rack (> 1, < 3)
+    sprockets-rails (3.2.2)
+      actionpack (>= 4.0)
+      activesupport (>= 4.0)
+      sprockets (>= 3.0.0)
+    terminal-table (3.0.1)
+      unicode-display_width (>= 1.1.1, < 3)
+    thor (1.1.0)
+    thread_safe (0.3.6)
+    tzinfo (1.2.9)
+      thread_safe (~> 0.1)
+    unicode-display_width (2.0.0)
+    websocket-driver (0.7.5)
+      websocket-extensions (>= 0.1.0)
+    websocket-extensions (0.1.5)
     zeitwerk (2.4.2)
 
 PLATFORMS
@@ -52,10 +169,11 @@ PLATFORMS
 
 DEPENDENCIES
   minitest
+  rails (>= 6.0, < 6.1)
   rake (~> 13.0)
-  rbs (>= 1.0.0)
+  rbs (>= 1.6)
   rbs_rails!
-  steep (>= 0.34.0)
+  steep!
 
 BUNDLED WITH
-   2.2.4
+   2.2.27

data/README.md

@@ -58,6 +58,9 @@ target :app do
   library 'monitor'
   library 'singleton'
   library 'tsort'
+  library 'time'
+
+  library 'rack'
 
   library 'activesupport'
   library 'actionpack'
@@ -69,12 +72,11 @@ target :app do
 end
 ```
 
-You need to put RBS repo to `path/to/rbs_repo`. See https://github.com/ruby/gem_rbs
+You need to put RBS repo to `path/to/rbs_repo`. See https://github.com/ruby/gem_rbs_collection
 
 ## Development
 
-After checking out the repo, run `GITHUB_TOKEN=xxx bin/setup` to install dependencies.
-`GITHUB_TOKEN` environment variable is required to fetch RBS from [ruby/gem_rbs](https://github.com/ruby/gem_rbs) repository.
+After checking out the repo, run `bin/setup` to install dependencies.
 
 You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 

data/Rakefile

@@ -9,8 +9,10 @@ task :steep do
 end
 
 task :rbs_validate do
-  rbs = Pathname(__dir__).join('bin/rbs').to_s
-  sh "ruby #{rbs} validate --silent"
+  repo = ENV['RBS_REPO_DIR']&.then do |env|
+    "--repo=#{env}"
+  end
+  sh "rbs #{repo} validate --silent"
 end
 
 Rake::TestTask.new do |test|

data/Steepfile

@@ -3,21 +3,5 @@ target :lib do
   signature 'assets/sig'
 
   check "lib"                       # Directory name
-  repo_path ENV['RBS_REPO_DIR'] || './gem_rbs/gems'
-
-  library "pathname"
-  library "logger"
-  library "mutex_m"
-  library "date"
-  library 'monitor'
-  library 'singleton'
-  library 'tsort'
-
-  library 'activesupport'
-  library 'actionpack'
-  library 'activejob'
-  library 'activemodel'
-  library 'actionview'
-  library 'activerecord'
-  library 'railties'
+  repo_path ENV['RBS_REPO_DIR'] if ENV['RBS_REPO_DIR']
 end

data/bin/setup

@@ -4,6 +4,6 @@ IFS=$'\n\t'
 set -vx
 
 bundle install
-bin/gem_rbs
+bundle exec rbs collection install --frozen
 
 # Do any other automated setup that you need to do here

data/lib/rbs_rails/active_record.rb

@@ -32,10 +32,14 @@ module RbsRails
 
           #{columns}
           #{associations}
+          #{delegated_type_instance}
+          #{delegated_type_scope(singleton: true)}
           #{enum_instance_methods}
           #{enum_scope_methods(singleton: true)}
           #{scopes(singleton: true)}
 
+          #{generated_relation_methods_decl}
+
           #{relation_decl}
 
           #{collection_proxy_decl}
@@ -46,18 +50,28 @@ module RbsRails
 
       private def pk_type
         pk = klass.primary_key
+        return 'top' unless pk
+
         col = klass.columns.find {|col| col.name == pk }
         sql_type_to_class(col.type)
       end
 
+      private def generated_relation_methods_decl
+        <<~RBS
+          module GeneratedRelationMethods
+            #{enum_scope_methods(singleton: false)}
+            #{scopes(singleton: false)}
+            #{delegated_type_scope(singleton: false)}
+          end
+        RBS
+      end
+
       private def relation_decl
         <<~RBS
           class #{relation_class_name} < ActiveRecord::Relation
+            include GeneratedRelationMethods
             include _ActiveRecord_Relation[#{klass_name}, #{pk_type}]
             include Enumerable[#{klass_name}]
-
-          #{enum_scope_methods(singleton: false)}
-          #{scopes(singleton: false)}
           end
         RBS
       end
@@ -65,6 +79,8 @@ module RbsRails
       private def collection_proxy_decl
         <<~RBS
           class ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
+            include GeneratedRelationMethods
+            include _ActiveRecord_Relation[#{klass_name}, #{pk_type}]
           end
         RBS
       end
@@ -121,7 +137,7 @@ module RbsRails
           type = a.polymorphic? ? 'untyped' : Util.module_name(a.klass)
           type_optional = optional(type)
           <<~RUBY.chomp
-            def #{a.name}: () -> #{type}
+            def #{a.name}: () -> #{type_optional}
             def #{a.name}=: (#{type_optional}) -> #{type_optional}
             def build_#{a.name}: (untyped) -> #{type}
             def create_#{a.name}: (untyped) -> #{type}
@@ -135,17 +151,91 @@ module RbsRails
         klass.reflect_on_all_associations(:belongs_to).map do |a|
           type = a.polymorphic? ? 'untyped' : Util.module_name(a.klass)
           type_optional = optional(type)
-          <<~RUBY.chomp
-            def #{a.name}: () -> #{type}
-            def #{a.name}=: (#{type_optional}) -> #{type_optional}
-            def build_#{a.name}: (untyped) -> #{type}
-            def create_#{a.name}: (untyped) -> #{type}
-            def create_#{a.name}!: (untyped) -> #{type}
-            def reload_#{a.name}: () -> #{type_optional}
-          RUBY
+          # @type var methods: Array[String]
+          methods = []
+          methods << "def #{a.name}: () -> #{type}"
+          methods << "def #{a.name}=: (#{type_optional}) -> #{type_optional}"
+          methods << "def reload_#{a.name}: () -> #{type_optional}"
+          if !a.polymorphic?
+            methods << "def build_#{a.name}: (untyped) -> #{type}"
+            methods << "def create_#{a.name}: (untyped) -> #{type}"
+            methods << "def create_#{a.name}!: (untyped) -> #{type}"
+          end
+          methods.join("\n")
         end.join("\n")
       end
 
+      private def delegated_type_scope(singleton:)
+        definitions = delegated_type_definitions
+        return "" unless definitions
+        definitions.map do |definition|
+          definition[:types].map do |type|
+            scope_name = type.tableize.gsub("/", "_")
+            "def #{singleton ? 'self.' : ''}#{scope_name}: () -> #{relation_class_name}"
+          end
+        end.flatten.join("\n")
+      end
+
+      private def delegated_type_instance
+        definitions = delegated_type_definitions
+        return "" unless definitions
+        # @type var methods: Array[String]
+        methods = []
+        definitions.each do |definition|
+          methods << "def #{definition[:role]}_class: () -> Class"
+          methods << "def #{definition[:role]}_name: () -> String"
+          methods << definition[:types].map do |type|
+            scope_name = type.tableize.gsub("/", "_")
+            singular = scope_name.singularize
+            <<~RUBY.chomp
+              def #{singular}?: () -> bool
+              def #{singular}: () -> #{type.classify}?
+              def #{singular}_id: () -> Integer?
+            RUBY
+          end.join("\n")
+        end
+        methods.join("\n")
+      end
+
+      private def delegated_type_definitions
+        ast = parse_model_file
+        return unless ast
+
+        traverse(ast).map do |node|
+          # @type block: { role: Symbol, types: Array[String] }?
+          next unless node.type == :send
+          next unless node.children[0].nil?
+          next unless node.children[1] == :delegated_type
+
+          role_node = node.children[2]
+          next unless role_node
+          next unless role_node.type == :sym
+          # @type var role: Symbol
+          role = role_node.children[0]
+
+          args_node = node.children[3]
+          next unless args_node
+          next unless args_node.type == :hash
+
+          types = traverse(args_node).map do |n|
+            # @type block: Array[String]?
+            next unless n.type == :pair
+            key_node = n.children[0]
+            next unless key_node
+            next unless key_node.type == :sym
+            next unless key_node.children[0] == :types
+
+            types_node = n.children[1]
+            next unless types_node
+            next unless types_node.type == :array
+            code = types_node.loc.expression.source
+            eval(code)
+          end.compact.flatten
+
+          { role: role, types: types }
+        end.compact
+      end
+
       private def enum_instance_methods
         # @type var methods: Array[String]
         methods = []
@@ -258,17 +348,17 @@ module RbsRails
         args_node.children.each do |node|
           case node.type
           when :arg
-            res << "untyped #{node.children[0]}"
+            res << "untyped `#{node.children[0]}`"
           when :optarg
-            res << "?untyped #{node.children[0]}"
+            res << "?untyped `#{node.children[0]}`"
           when :kwarg
             res << "#{node.children[0]}: untyped"
           when :kwoptarg
             res << "?#{node.children[0]}: untyped"
           when :restarg
-            res << "*untyped #{node.children[0]}"
+            res << "*untyped `#{node.children[0]}`"
           when :kwrestarg
-            res << "**untyped #{node.children[0]}"
+            res << "**untyped `#{node.children[0]}`"
           when :blockarg
             block = " { (*untyped) -> untyped }"
           else
@@ -292,7 +382,7 @@ module RbsRails
       end
 
       private def traverse(node, &block)
-        return to_enum(__method__, node) unless block_given?
+        return to_enum(__method__ || raise, node) unless block_given?
 
         # @type var block: ^(Parser::AST::Node) -> untyped
         block.call node

data/lib/rbs_rails/version.rb

@@ -2,5 +2,5 @@ module RbsRails
   # Because of copy_signatures is defined by lib/rbs_rails.rb
   # @dynamic self.copy_signatures
 
-  VERSION = "0.8.2"
+  VERSION = "0.9.0"
 end

data/rbs_collection.lock.yaml

@@ -0,0 +1,132 @@
+---
+sources:
+- name: ruby/gem_rbs_collection
+  remote: https://github.com/ruby/gem_rbs_collection.git
+  revision: main
+  repo_dir: gems
+path: ".gem_rbs_collection"
+gems:
+- name: pathname
+  version: '0'
+  source:
+    type: stdlib
+- name: logger
+  version: '0'
+  source:
+    type: stdlib
+- name: mutex_m
+  version: '0'
+  source:
+    type: stdlib
+- name: date
+  version: '0'
+  source:
+    type: stdlib
+- name: monitor
+  version: '0'
+  source:
+    type: stdlib
+- name: singleton
+  version: '0'
+  source:
+    type: stdlib
+- name: tsort
+  version: '0'
+  source:
+    type: stdlib
+- name: time
+  version: '0'
+  source:
+    type: stdlib
+- name: set
+  version: '0'
+  source:
+    type: stdlib
+- name: rack
+  version: 2.2.2
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: json
+  version: '0'
+  source:
+    type: stdlib
+- name: strscan
+  version: '0'
+  source:
+    type: stdlib
+- name: optparse
+  version: '0'
+  source:
+    type: stdlib
+- name: actionpack
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: actionview
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: activejob
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: activemodel
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: activerecord
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: activesupport
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: parallel
+  version: '1.20'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: railties
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 51880bed87fbc3dc8076fedb5cf798be05148220
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rbs
+  version: 1.6.2
+  source:
+    type: rubygems

data/rbs_collection.yaml

@@ -0,0 +1,45 @@
+# Download sources
+sources:
+  - name: ruby/gem_rbs_collection
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    revision: main
+    repo_dir: gems
+
+# A directory to install the downloaded RBSs
+path: .gem_rbs_collection
+
+gems:
+  # stdlibs
+  - name: 'pathname'
+  - name: 'logger'
+  - name: 'mutex_m'
+  - name: 'date'
+  - name: 'monitor'
+  - name: 'singleton'
+  - name: 'tsort'
+  - name: 'time'
+  - name: 'set'
+
+  # gems
+  - name: 'rack'
+
+  # RBS gem dependencies
+  - name: 'json'
+  - name: 'strscan'
+  - name: 'optparse'
+
+  # Ignores - Gemfile.lock contains them but their RBSs are unnecessary
+  - name: steep
+    ignore: true
+  - name: nokogiri
+    ignore: true
+  - name: ast
+    ignore: true
+  - name: rainbow
+    ignore: true
+  - name: listen
+    ignore: true
+
+  # ignore RBS Rails itself
+  - name: rbs_rails
+    ignore: true

data/sig/rbs_rails/active_record.rbs

@@ -14,6 +14,8 @@ class RbsRails::ActiveRecord::Generator
 
   def pk_type: () -> String
 
+  def generated_relation_methods_decl: () -> String
+
   def relation_decl: () -> String
 
   def collection_proxy_decl: () -> String
@@ -30,6 +32,12 @@ class RbsRails::ActiveRecord::Generator
 
   def belongs_to: () -> String
 
+  def delegated_type_scope: (singleton: bool) -> String
+
+  def delegated_type_instance: () -> String
+
+  def delegated_type_definitions: () -> Array[{ role: Symbol, types: Array[String] }]?
+
   def enum_instance_methods: () -> String
 
   def enum_scope_methods: (singleton: untyped `singleton`) -> String

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbs_rails
 version: !ruby/object:Gem::Version
-  version: 0.8.2
+  version: 0.9.0
 platform: ruby
 authors:
 - Masataka Pocke Kuwabara
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-02-20 00:00:00.000000000 Z
+date: 2021-09-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: parser
@@ -58,9 +58,7 @@ files:
 - Steepfile
 - bin/add-type-params.rb
 - bin/console
-- bin/gem_rbs
 - bin/postprocess.rb
-- bin/rbs
 - bin/rbs-prototype-rb.rb
 - bin/setup
 - bin/to-ascii.rb
@@ -71,6 +69,8 @@ files:
 - lib/rbs_rails/rake_task.rb
 - lib/rbs_rails/util.rb
 - lib/rbs_rails/version.rb
+- rbs_collection.lock.yaml
+- rbs_collection.yaml
 - rbs_rails.gemspec
 - sig/activerecord.rbs
 - sig/fileutils.rbs
@@ -105,7 +105,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.2.3
+rubygems_version: 3.3.0.dev
 signing_key:
 specification_version: 4
 summary: A RBS files generator for Rails application

data/bin/gem_rbs

@@ -1,94 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'net/http'
-require 'json'
-require 'securerandom'
-require 'pathname'
-
-TOKEN = ENV.fetch('GITHUB_TOKEN')
-VERSION = "6.0.3.2"
-
-def req(query)
-  http = Net::HTTP.new("api.github.com", 443)
-  http.use_ssl = true
-  header = {
-    "Authorization" => "Bearer #{TOKEN}",
-    'Content-Type' => 'application/json',
-    'User-Agent' => 'gem_rbs client',
-  }
-  resp = http.request_post('/graphql', JSON.generate(query), header)
-  JSON.parse(resp.body, symbolize_names: true).tap do |content|
-    raise content[:errors].inspect if content[:errors]
-  end
-end
-
-class QueryBuilder
-  attr_reader :variables
-
-  def initialize
-    @queries = []
-    @variables = {}
-  end
-
-  def add(query, variables)
-    query = query.dup
-    variables = variables.transform_keys do |key|
-      next key unless @variables.key?(key)
-
-      new_key = key + '_' + SecureRandom.hex(8)
-      query.gsub!(key, new_key)
-      new_key
-    end
-
-    @queries << query
-    @variables.merge!(variables)
-  end
-
-  def query
-    # TODO: Allow non-String type for variables
-    "query(#{variables.keys.map { |v| "$#{v}: String!" }.join(',')}) { #{@queries.join("\n")} }"
-  end
-end
-
-gems = %w[activesupport actionpack activejob activemodel actionview activerecord railties]
-
-builder = QueryBuilder.new
-gems.each do |gem|
-  path = "main:gems/#{gem}/#{VERSION}"
-  builder.add(<<~GRAPHQL, { 'path' => path })
-    #{gem}:repository(owner: "ruby", name: "gem_rbs_collection") {
-      object(expression: $path) {
-        ... on Tree {
-          entries {
-            name
-            object {
-              ... on Blob {
-                isTruncated
-                text
-              }
-            }
-          }
-        }
-      }
-    }
-  GRAPHQL
-end
-
-resp = req(query: builder.query, variables: builder.variables)
-
-resp[:data].each do |gem_name, gem_value|
-  gem_value.dig(:object, :entries).each do |entry|
-    fname = entry[:name]
-    if fname.end_with?('.rbs')
-      content =
-        if entry.dig(:object, :isTruncated)
-          `curl -H 'Accept: application/vnd.github.v3.raw' -H Authorization: token #{TOKEN} https://api.github.com/repos/ruby/gem_rbs_collection/contents/gems/#{gem_name}/#{VERSION}/#{fname}`
-        else
-          entry.dig(:object, :text)
-        end
-      dir = Pathname("gem_rbs/gems/#{gem_name}/#{VERSION}")
-      dir.mkpath
-      dir.join(fname).write(content)
-    end
-  end
-end

data/bin/rbs

@@ -1,30 +0,0 @@
-#!ruby
-
-require 'pathname'
-root = Pathname(__dir__) / '../'
-
-def v(require)
-  if v = ENV['RAILS_VERSION']
-    "#{require}:#{v}"
-  else
-    require
-  end
-end
-
-def repo
-  ENV['RBS_REPO_DIR'] || Pathname(__dir__).join('../gem_rbs/gems').to_s
-end
-
-exec(
-  'rbs',
-  # Require stdlibs
-  '-rlogger', '-rpathname', '-rmutex_m', '-rdate', '-rmonitor', '-rsingleton', '-rtsort',
-  "--repo=#{repo}",
-  # Require Rails libraries
-  v('-ractivesupport'), v('-ractionpack'), v('-ractivejob'), v('-ractivemodel'), v('-ractionview'), v('-ractiverecord'), v('-rrailties'),
-  # Load signatures that are bundled in rbs_rails
-  '-I' + root.join('sig').to_s, '-I' + root.join('assets/sig').to_s,
-  # Expand arguments
-  *ARGV,
-)
-