rbs_rails 0.8.1 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 30c59f478f8fa0cd0c41751feb468a04208e057ff318edf80717b63d2f4664b8
-  data.tar.gz: f382afdc82aaa47b6fe14cba2cdcbb62f127dc04d95020b71ed1b6c1aa8dfd9d
+  metadata.gz: 4daa0714f5e127bc21e33e9737983999459a2dbb690202c6dc86408a68ef82b0
+  data.tar.gz: 54f9a8c04ca0f22c1a391a4d5a47c0294c54f50741843226655ae9e8175b0a51
 SHA512:
-  metadata.gz: 54ca5ed84a59c4e67892f8c4e72cdc190420d457345523eb11b3dd4efe7dde5d8662afc7ace9dbee3be1aa5577b257b028ccdba20297b45332919999ac648914
-  data.tar.gz: 61ba620c335f27ddb58ce96d4bf3d03ea61071c528b587b7fa419d93c462ba6d96e6ad4e508243a1e6aff903966bfa49e18a02ecf9eb69f053c2cdd215da20a7
+  metadata.gz: f85bdbf2f59f5accdd54a062f5d5412f432b9d91b62b6f4ced371fbcc0d565ad661a1015d9193e4ee9add64ef69cf7243dd5603dc22b05182b59f3a122ecad2e
+  data.tar.gz: f75cc735df8961b0714af8bf4f38771f01542ed4162e69fb8adc677dcbfec0885fe15296bc84006b955d9370004dfcd6070fbd761c0b220f02d16f062fadcbba

data/.dependabot/config.yml

@@ -0,0 +1,5 @@
+version: 1
+update_configs:
+  - package_manager: "ruby:bundler"
+    directory: "/"
+    update_schedule: "live"

data/.github/workflows/ci.yml

@@ -15,7 +15,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        ruby: [2.6, 2.7, '3.0', head]
+        ruby: [2.6, 2.7, '3.0', 3.1, head]
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v2
@@ -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

@@ -6,8 +6,7 @@
 /pkg/
 /spec/reports/
 /tmp/
-/Gemfile.lock
 /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,27 @@
 
 ## 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)
+*  Define find method based on primary key [#105](https://github.com/pocke/rbs_rails/pull/105)
+
 ## 0.8.1 (2021-01-09)
 
 * Skip generation RBS fro class that doesn't have table in DB. [#95](https://github.com/pocke/rbs_rails/pull/95)

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", "~> 12.0"
-gem 'rbs', '1.0.0'
-gem 'steep', '>= 0.34.0', git: 'https://github.com/soutaro/steep.git'
+gem "rake", "~> 13.0"
+gem 'rails', '< 6.1', '>= 6.0'
+gem 'rbs', '>= 2'
+gem 'steep'
 gem 'minitest'

data/Gemfile.lock

@@ -0,0 +1,174 @@
+PATH
+  remote: .
+  specs:
+    rbs_rails (0.10.0)
+      parser
+      rbs (>= 1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    actioncable (6.0.4.7)
+      actionpack (= 6.0.4.7)
+      nio4r (~> 2.0)
+      websocket-driver (>= 0.6.1)
+    actionmailbox (6.0.4.7)
+      actionpack (= 6.0.4.7)
+      activejob (= 6.0.4.7)
+      activerecord (= 6.0.4.7)
+      activestorage (= 6.0.4.7)
+      activesupport (= 6.0.4.7)
+      mail (>= 2.7.1)
+    actionmailer (6.0.4.7)
+      actionpack (= 6.0.4.7)
+      actionview (= 6.0.4.7)
+      activejob (= 6.0.4.7)
+      mail (~> 2.5, >= 2.5.4)
+      rails-dom-testing (~> 2.0)
+    actionpack (6.0.4.7)
+      actionview (= 6.0.4.7)
+      activesupport (= 6.0.4.7)
+      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.7)
+      actionpack (= 6.0.4.7)
+      activerecord (= 6.0.4.7)
+      activestorage (= 6.0.4.7)
+      activesupport (= 6.0.4.7)
+      nokogiri (>= 1.8.5)
+    actionview (6.0.4.7)
+      activesupport (= 6.0.4.7)
+      builder (~> 3.1)
+      erubi (~> 1.4)
+      rails-dom-testing (~> 2.0)
+      rails-html-sanitizer (~> 1.1, >= 1.2.0)
+    activejob (6.0.4.7)
+      activesupport (= 6.0.4.7)
+      globalid (>= 0.3.6)
+    activemodel (6.0.4.7)
+      activesupport (= 6.0.4.7)
+    activerecord (6.0.4.7)
+      activemodel (= 6.0.4.7)
+      activesupport (= 6.0.4.7)
+    activestorage (6.0.4.7)
+      actionpack (= 6.0.4.7)
+      activejob (= 6.0.4.7)
+      activerecord (= 6.0.4.7)
+      marcel (~> 1.0.0)
+    activesupport (6.0.4.7)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 0.7, < 2)
+      minitest (~> 5.1)
+      tzinfo (~> 1.1)
+      zeitwerk (~> 2.2, >= 2.2.2)
+    ast (2.4.2)
+    builder (3.2.4)
+    concurrent-ruby (1.1.9)
+    crass (1.0.6)
+    erubi (1.10.0)
+    ffi (1.15.5)
+    globalid (1.0.0)
+      activesupport (>= 5.0)
+    i18n (1.10.0)
+      concurrent-ruby (~> 1.0)
+    language_server-protocol (3.16.0.3)
+    listen (3.7.1)
+      rb-fsevent (~> 0.10, >= 0.10.3)
+      rb-inotify (~> 0.9, >= 0.9.10)
+    loofah (2.15.0)
+      crass (~> 1.0.2)
+      nokogiri (>= 1.5.9)
+    mail (2.7.1)
+      mini_mime (>= 0.1.1)
+    marcel (1.0.2)
+    method_source (1.0.0)
+    mini_mime (1.1.2)
+    mini_portile2 (2.8.0)
+    minitest (5.15.0)
+    nio4r (2.5.8)
+    nokogiri (1.13.3)
+      mini_portile2 (~> 2.8.0)
+      racc (~> 1.4)
+    parallel (1.21.0)
+    parser (3.1.1.0)
+      ast (~> 2.4.1)
+    racc (1.6.0)
+    rack (2.2.3)
+    rack-test (1.1.0)
+      rack (>= 1.0, < 3)
+    rails (6.0.4.7)
+      actioncable (= 6.0.4.7)
+      actionmailbox (= 6.0.4.7)
+      actionmailer (= 6.0.4.7)
+      actionpack (= 6.0.4.7)
+      actiontext (= 6.0.4.7)
+      actionview (= 6.0.4.7)
+      activejob (= 6.0.4.7)
+      activemodel (= 6.0.4.7)
+      activerecord (= 6.0.4.7)
+      activestorage (= 6.0.4.7)
+      activesupport (= 6.0.4.7)
+      bundler (>= 1.3.0)
+      railties (= 6.0.4.7)
+      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.7)
+      actionpack (= 6.0.4.7)
+      activesupport (= 6.0.4.7)
+      method_source
+      rake (>= 0.8.7)
+      thor (>= 0.20.3, < 2.0)
+    rainbow (3.1.1)
+    rake (13.0.6)
+    rb-fsevent (0.11.1)
+    rb-inotify (0.10.1)
+      ffi (~> 1.0)
+    rbs (2.2.2)
+    sprockets (4.0.3)
+      concurrent-ruby (~> 1.0)
+      rack (> 1, < 3)
+    sprockets-rails (3.4.2)
+      actionpack (>= 5.2)
+      activesupport (>= 5.2)
+      sprockets (>= 3.0.0)
+    steep (0.49.1)
+      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 (>= 2.2.0)
+      terminal-table (>= 2, < 4)
+    terminal-table (3.0.2)
+      unicode-display_width (>= 1.1.1, < 3)
+    thor (1.2.1)
+    thread_safe (0.3.6)
+    tzinfo (1.2.9)
+      thread_safe (~> 0.1)
+    unicode-display_width (2.1.0)
+    websocket-driver (0.7.5)
+      websocket-extensions (>= 0.1.0)
+    websocket-extensions (0.1.5)
+    zeitwerk (2.5.4)
+
+PLATFORMS
+  ruby
+  x86_64-linux
+
+DEPENDENCIES
+  minitest
+  rails (>= 6.0, < 6.1)
+  rake (~> 13.0)
+  rbs (>= 2)
+  rbs_rails!
+  steep
+
+BUNDLED WITH
+   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

@@ -28,14 +28,20 @@ module RbsRails
       private def klass_decl
         <<~RBS
           #{header}
-            extend _ActiveRecord_Relation_ClassMethods[#{klass_name}, #{relation_class_name}]
+            extend _ActiveRecord_Relation_ClassMethods[#{klass_name}, #{relation_class_name}, #{pk_type}]
 
           #{columns}
           #{associations}
+          #{generated_association_methods}
+          #{has_secure_password}
+          #{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}
@@ -44,20 +50,39 @@ module RbsRails
         RBS
       end
 
+      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 _ActiveRecord_Relation[#{klass_name}]
+          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
 
       private def collection_proxy_decl
         <<~RBS
-          class ActiveRecord_Associations_CollectionProxy < ActiveRecord::Associations::CollectionProxy
+          class ActiveRecord_Associations_CollectionProxy < ::ActiveRecord::Associations::CollectionProxy
+            include GeneratedRelationMethods
+            include _ActiveRecord_Relation[#{klass_name}, #{pk_type}]
           end
         RBS
       end
@@ -74,7 +99,7 @@ module RbsRails
             superclass_name = Util.module_name(superclass)
             @dependencies << superclass_name
 
-            "class #{mod_name} < #{superclass_name}"
+            "class #{mod_name} < ::#{superclass_name}"
           when Module
             "module #{mod_name}"
           else
@@ -114,7 +139,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}
@@ -128,23 +153,162 @@ 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 generated_association_methods
+        # @type var sigs: Array[String]
+        sigs = []
+
+        # Needs to require "active_storage/engine"
+        if klass.respond_to?(:attachment_reflections)
+          sigs << "module GeneratedAssociationMethods"
+          sigs << klass.attachment_reflections.map do |name, reflection|
+            case reflection.macro
+            when :has_one_attached
+              <<~EOS
+                def #{name}: () -> ActiveStorage::Attached::One
+                def #{name}=: (ActionDispatch::Http::UploadedFile) -> ActionDispatch::Http::UploadedFile
+                            | (Rack::Test::UploadedFile) -> Rack::Test::UploadedFile
+                            | (ActiveStorage::Blob) -> ActiveStorage::Blob
+                            | (String) -> String
+                            | ({ io: IO, filename: String, content_type: String? }) -> { io: IO, filename: String, content_type: String? }
+                            | (nil) -> nil
+              EOS
+            when :has_many_attached
+              <<~EOS
+                def #{name}: () -> ActiveStorage::Attached::Many
+                def #{name}=: (untyped) -> untyped
+              EOS
+            else
+              raise
+            end
+          end.join("\n")
+          sigs << "end"
+          sigs << "include GeneratedAssociationMethods"
+        end
+
+        sigs.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 has_secure_password
+        ast = parse_model_file
+        return unless ast
+
+        traverse(ast).map do |node|
+          # @type block: String?
+          next unless node.type == :send
+          next unless node.children[0].nil?
+          next unless node.children[1] == :has_secure_password
+
+          attribute_node = node.children[2]
+          attribute = if attribute_node && attribute_node.type == :sym
+                        attribute_node.children[0]
+                      else
+                        :password
+                      end
+
+          <<~EOS
+            module ActiveModel_SecurePassword_InstanceMethodsOnActivation_#{attribute}
+              attr_reader #{attribute}: String?
+              def #{attribute}=: (String) -> String
+              def #{attribute}_confirmation=: (String) -> String
+              def authenticate_#{attribute}: (String) -> (#{klass_name} | false)
+              #{attribute == :password ? "alias authenticate authenticate_password" : ""}
+            end
+            include ActiveModel_SecurePassword_InstanceMethodsOnActivation_#{attribute}
+          EOS
+        end.compact.join("\n")
+      end
+
       private def enum_instance_methods
         # @type var methods: Array[String]
         methods = []
         enum_definitions.each do |hash|
           hash.each do |name, values|
-            next if name == :_prefix || name == :_suffix
+            next if name == :_prefix || name == :_suffix || name == :_default
 
             values.each do |label, value|
               value_method_name = enum_method_name(hash, name, label)
@@ -162,7 +326,7 @@ module RbsRails
         methods = []
         enum_definitions.each do |hash|
           hash.each do |name, values|
-            next if name == :_prefix || name == :_suffix
+            next if name == :_prefix || name == :_suffix || name == :_default
 
             values.each do |label, value|
               value_method_name = enum_method_name(hash, name, label)
@@ -223,7 +387,9 @@ module RbsRails
         ast = parse_model_file
         return '' unless ast
 
-        traverse(ast).map do |node|
+        prefix = singleton ? 'self.' : ''
+
+        sigs = traverse(ast).map do |node|
           # @type block: nil | String
           next unless node.type == :send
           next unless node.children[0].nil?
@@ -239,8 +405,16 @@ module RbsRails
           next unless body_node.type == :block
 
           args = args_to_type(body_node.children[1])
-          "def #{singleton ? 'self.' : ''}#{name}: #{args} -> #{relation_class_name}"
-        end.compact.join("\n")
+          "def #{prefix}#{name}: #{args} -> #{relation_class_name}"
+        end.compact
+
+        if klass.respond_to?(:attachment_reflections)
+          klass.attachment_reflections.each do |name, _reflection|
+            sigs << "def #{prefix}with_attached_#{name}: () -> #{relation_class_name}"
+          end
+        end
+
+        sigs.join("\n")
       end
 
       private def args_to_type(args_node)
@@ -251,17 +425,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
@@ -285,7 +459,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
@@ -299,7 +473,8 @@ module RbsRails
       end
 
       private def columns
-        klass.columns.map do |col|
+        mod_sig = +"module GeneratedAttributeMethods\n"
+        mod_sig << klass.columns.map do |col|
           class_name = if enum_definitions.any? { |hash| hash.key?(col.name) || hash.key?(col.name.to_sym) }
                          'String'
                        else
@@ -308,7 +483,9 @@ module RbsRails
           class_name_opt = optional(class_name)
           column_type = col.null ? class_name_opt : class_name
           sig = <<~EOS
-            attr_accessor #{col.name} (): #{column_type}
+            def #{col.name}: () -> #{column_type}
+            def #{col.name}=: (#{column_type}) -> #{column_type}
+            def #{col.name}?: () -> bool
             def #{col.name}_changed?: () -> bool
             def #{col.name}_change: () -> [#{class_name_opt}, #{class_name_opt}]
             def #{col.name}_will_change!: () -> void
@@ -316,12 +493,21 @@ module RbsRails
             def #{col.name}_previously_changed?: () -> bool
             def #{col.name}_previous_change: () -> Array[#{class_name_opt}]?
             def #{col.name}_previously_was: () -> #{class_name_opt}
+            def #{col.name}_before_last_save: () -> #{class_name_opt}
+            def #{col.name}_change_to_be_saved: () -> Array[#{class_name_opt}]?
+            def #{col.name}_in_database: () -> #{class_name_opt}
+            def saved_change_to_#{col.name}: () -> Array[#{class_name_opt}]?
+            def saved_change_to_#{col.name}?: () -> bool
+            def will_save_change_to_#{col.name}?: () -> bool
             def restore_#{col.name}!: () -> void
             def clear_#{col.name}_change: () -> void
           EOS
-          sig << "attr_accessor #{col.name}? (): #{class_name}\n" if col.type == :boolean
+          sig << "\n"
           sig
         end.join("\n")
+        mod_sig << "\nend\n"
+        mod_sig << "include GeneratedAttributeMethods"
+        mod_sig
       end
 
       private def optional(class_name)

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.1"
+  VERSION = "0.10.0"
 end

data/rbs_collection.lock.yaml

@@ -0,0 +1,144 @@
+---
+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: logger
+  version: '0'
+  source:
+    type: stdlib
+- name: set
+  version: '0'
+  source:
+    type: stdlib
+- name: pathname
+  version: '0'
+  source:
+    type: stdlib
+- name: json
+  version: '0'
+  source:
+    type: stdlib
+- name: optparse
+  version: '0'
+  source:
+    type: stdlib
+- name: rubygems
+  version: '0'
+  source:
+    type: stdlib
+- name: tsort
+  version: '0'
+  source:
+    type: stdlib
+- name: actionpack
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    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: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    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: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    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: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    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: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: activestorage
+  version: '6.0'
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    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: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    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: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rack
+  version: 2.2.2
+  source:
+    type: git
+    name: ruby/gem_rbs_collection
+    revision: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    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: 6320a194c85afbf4ff332c3eb75fcae2a6518c92
+    remote: https://github.com/ruby/gem_rbs_collection.git
+    repo_dir: gems
+- name: rbs
+  version: 2.2.2
+  source:
+    type: rubygems
+- name: monitor
+  version: '0'
+  source:
+    type: stdlib
+- name: tempfile
+  version: '0'
+  source:
+    type: stdlib
+- name: date
+  version: '0'
+  source:
+    type: stdlib
+- name: singleton
+  version: '0'
+  source:
+    type: stdlib
+- name: mutex_m
+  version: '0'
+  source:
+    type: stdlib
+- name: time
+  version: '0'
+  source:
+    type: stdlib

data/rbs_collection.yaml

@@ -0,0 +1,36 @@
+# 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:
+  # 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
+
+  # They'are necessary to load RBS gem.
+  # See https://github.com/ruby/rbs/pull/921
+  - name: logger
+  - name: set
+  - name: pathname
+  - name: json
+  - name: optparse
+  - name: rubygems
+  - name: tsort

data/sig/activerecord.rbs

@@ -0,0 +1,4 @@
+# TODO: Define it in gem_rbs_collection repository
+class ActiveRecord::Base
+  def self.primary_key: () -> String
+end

data/sig/rbs_rails/active_record.rbs

@@ -12,6 +12,10 @@ class RbsRails::ActiveRecord::Generator
 
   def klass_decl: () -> String
 
+  def pk_type: () -> String
+
+  def generated_relation_methods_decl: () -> String
+
   def relation_decl: () -> String
 
   def collection_proxy_decl: () -> String
@@ -28,6 +32,16 @@ class RbsRails::ActiveRecord::Generator
 
   def belongs_to: () -> String
 
+  def generated_association_methods: () -> String
+
+  def delegated_type_scope: (singleton: bool) -> String
+
+  def delegated_type_instance: () -> String
+
+  def delegated_type_definitions: () -> Array[{ role: Symbol, types: Array[String] }]?
+
+  def has_secure_password: () -> 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.1
+  version: 0.10.0
 platform: ruby
 authors:
 - Masataka Pocke Kuwabara
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-01-08 00:00:00.000000000 Z
+date: 2022-03-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: parser
@@ -45,20 +45,20 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".dependabot/config.yml"
 - ".github/workflows/ci.yml"
 - ".gitignore"
 - ".gitmodules"
 - CHANGELOG.md
 - Gemfile
+- Gemfile.lock
 - LICENSE
 - README.md
 - Rakefile
 - 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
@@ -69,7 +69,10 @@ 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
 - sig/parser.rbs
 - sig/rake.rbs
@@ -102,7 +105,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.3.0.dev
+rubygems_version: 3.4.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") {
-      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/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,
-)
-