rbs-siggen 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c368db792e08e8417cfc36a43cac9aa49e33eeeaf60c39cc7700db494832553b
-  data.tar.gz: 11ac12278d9d200e6258fb7230623213e3342006d138127b17f5fd24a454919d
+  metadata.gz: 06e23d1dc2eadd500a4871c55965497aa433e702c1a14702d3b6b92b62052d4d
+  data.tar.gz: 53f2519a3398677095889d18254eaf3dc4cdbc2e0619f6c30ec642ab897ebc30
 SHA512:
-  metadata.gz: d0ece64d8a739c938ad31197082f29fb7db72141a3932512117ea8e42073c56002b6db12fc4fdd130e25bc1d4c78178b1a39345299726344bebdab59a8059ed0
-  data.tar.gz: aec7870eb75d4fb9d568a546218d4807e1190ac9521b7e50392e0aac4089f79abc5bf6303c621cfddb6f54f44883ba6b0f0e4683a4855308b46147091173eb5c
+  metadata.gz: 830a11c06f15f3f2308370732726cb6d2b33f41102fca4c185b64b00f21aa6e06ebc5f4733522e47a817b496b03315d21d625fc701308def69001bf5c9838ffa
+  data.tar.gz: 983da6e48b25425bc98d9637c625d19ab1a21d6c5d9ad3a071d8bb926746f6c1e93fc243dcf6585ea3a1e75a99b057192dccb7f89df96308488573a6b7bb2686

data/lib/rbs/siggen/version.rb

@@ -2,6 +2,6 @@
 
 module RBS
   class Siggen
-    VERSION = "0.2.0"
+    VERSION = "0.2.1"
   end
 end

data/lib/rbs/siggen.rb

@@ -35,6 +35,8 @@ module RBS
     def analyze(path:)
       ruby_files = Dir.glob(path) + Dir.glob("#{path}/**/*.rb")
       ruby_files.each do |file|
+        next unless File.file?(file)
+
         analyze_ruby(File.read(file), name: file)
         yield self, file
       end
@@ -106,6 +108,10 @@ module RBS
 
       @typing = typing
       @node = source.node
+      @source = source
+      @context = context
+      @checker = checker
+      @annotations = annotations
       # steep:ignore:end
     end
 
@@ -138,7 +144,7 @@ module RBS
       io = ::StringIO.new
       ::RBS::Writer.new(out: io).write(merge_class_declarations(decls))
       io.rewind
-      io.readlines.reject { |line| line.strip.empty? }.join
+      io.read || ""
     end
 
     #: (Parser::AST::Node node, ?Array[untyped] stack) ?{ (String, untyped, Hash[untyped, untyped]) -> void } -> void
@@ -171,7 +177,7 @@ module RBS
           end
           arg_hash.merge!(create_arg_hash(node, method_decl))
 
-          annos.each { |anno| yield(create_class_name(method_decl), anno, arg_hash) }
+          annos.each { |anno| yield(create_class_name(self_type_of(node), method_decl), anno, arg_hash) }
         end
 
         if node.type == :block
@@ -188,10 +194,24 @@ module RBS
       end
     end
 
-    #: (untyped) -> String
-    def create_class_name(method_decl)
-      receiver_type = method_decl.method_name.type_name.relative!
-      "#{receiver_type.namespace}#{receiver_type.name}"
+    #: (Parser::AST::Node) -> untyped
+    def self_type_of(node)
+      # steep:ignore:start
+      typing = Steep::Typing.new(source: @source, root_context: @context, cursor: node.loc.expression.begin_pos)
+      construction = Steep::TypeConstruction.new(checker: @checker,
+                                                 source: @source,
+                                                 annotations: @annotations,
+                                                 context: @context,
+                                                 typing: typing)
+      construction.synthesize(@source.node) unless @source.node.nil?
+      # steep:ignore:end
+      typing.cursor_context.context&.self_type
+    end
+
+    #: (untyped, untyped) -> String
+    def create_class_name(reciver_type, method_decl)
+      type = reciver_type&.name&.relative! || method_decl.method_name.type_name.relative!
+      "#{type.namespace}#{type.name}"
     end
 
     #: (Parser::AST::Node, untyped) -> Hash[Symbol, untyped]

data/sig/generated/rbs/siggen.rbs

@@ -26,8 +26,11 @@ module RBS
     # : (Parser::AST::Node node, ?Array[untyped] stack) ?{ (String, untyped, Hash[untyped, untyped]) -> void } -> void
     def traverse: (Parser::AST::Node node, ?Array[untyped] stack) ?{ (String, untyped, Hash[untyped, untyped]) -> void } -> void
 
-    # : (untyped) -> String
-    def create_class_name: (untyped) -> String
+    # : (Parser::AST::Node) -> untyped
+    def self_type_of: (Parser::AST::Node) -> untyped
+
+    # : (untyped, untyped) -> String
+    def create_class_name: (untyped, untyped) -> String
 
     # : (Parser::AST::Node, untyped) -> Hash[Symbol, untyped]
     def create_arg_hash: (Parser::AST::Node, untyped) -> Hash[Symbol, untyped]

data/siggen/activerecord_model.rbs

@@ -2,9 +2,11 @@ module ActiveRecord
   class Schema
     def self.[]: (untyped version) -> ActiveRecord::Migration::Current
   end
+
   class Migration
     class Current
       def define: (version: untyped) { () [self: instance] -> void } -> void
+
       %a{siggen:
         # In schema.rb, this model is declared as:
         #
@@ -18,6 +20,7 @@ module ActiveRecord
       def create_table: (untyped table_name, **untyped options) { (ActiveRecord::ConnectionAdapters::ColumnMethods t) -> void } -> void
     end
   end
+
   module ConnectionAdapters
     module ColumnMethods
         %a{siggen:
@@ -36,12 +39,27 @@ module ActiveRecord
                 # See the documentation comment for the delegated method below:
                 #
                 # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
-                def <%= name %>: () -> ::String<%= "?" unless options[:null] == false %>
+                def <%= name %>: () -> ::Integer<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::Integer<%= "?" unless options[:null] == false %> value) -> void
               <% end %>
             end
           end
         }
-        def text: (*Array[untyped] names, **untyped options) -> void
+        def bitint: (*Array[untyped] names, **untyped options) -> void
+
         %a{siggen:
           class <%= create_table.table_name.classify %>
             module GeneratedAttributeMethods
@@ -59,11 +77,100 @@ module ActiveRecord
                 #
                 # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
                 def <%= name %>: () -> ::String<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::String<%= "?" unless options[:null] == false %> value) -> void
               <% end %>
             end
           end
         }
-        def string: (*Array[untyped] names, **untyped options) -> void
+        def binary: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> bool<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (bool<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def boolean: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> ::Date<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::Date<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def date: (*Array[untyped] names, **untyped options) -> void
+
         %a{siggen:
           class <%= create_table.table_name.classify %>
             module GeneratedAttributeMethods
@@ -81,11 +188,100 @@ module ActiveRecord
                 #
                 # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
                 def <%= name %>: () -> ::ActiveSupport::TimeWithZone<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::ActiveSupport::TimeWithZone<%= "?" unless options[:null] == false %> value) -> void
               <% end %>
             end
           end
         }
         def datetime: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> ::BigDecimal<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::BigDecimal<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def decimal: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> ::Float<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::Float<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def float: (*Array[untyped] names, **untyped options) -> void
+
         %a{siggen:
           class <%= create_table.table_name.classify %>
             module GeneratedAttributeMethods
@@ -103,12 +299,251 @@ module ActiveRecord
                 #
                 # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
                 def <%= name %>: () -> ::Integer<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::Integer<%= "?" unless options[:null] == false %> value) -> void
               <% end %>
             end
           end
         }
         def integer: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> untyped<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (untyped<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def json: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> ::String<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::String<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def string: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> ::String<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::String<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def text: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> ::ActiveSupport::TimeWithZone<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::ActiveSupport::TimeWithZone<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def time: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> ::ActiveSupport::TimeWithZone<%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (::ActiveSupport::TimeWithZone<%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def timestamp: (*Array[untyped] names, **untyped options) -> void
+
+        %a{siggen:
+          <% map = [[:bitint, "::Integer"], [:binary, "::String"], [:boolean, "bool"], [:date, "::Date"], [:datetime, "::ActiveSupport::TimeWithZone"], [:decimal, "::BigDecimal"], [:float, "::Float"], [:integer, "::Integer"], [:json, "untyped"], [:string, "::String"], [:text, "::String"], [:time, "::ActiveSupport::TimeWithZone"], [:timestamp, "::ActiveSupport::TimeWithZone"], [:virtual, "_"]].to_h %>
+          class <%= create_table.table_name.classify %>
+            module GeneratedAttributeMethods
+              <% names.each do |name| %>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Read` `#_read_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
+                def <%= name %>: () -> <%= map[options[:type]] %><%= "?" unless options[:null] == false %>
+
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%= name %>=: (<%= map[options[:type]] %><%= "?" unless options[:null] == false %> value) -> void
+              <% end %>
+            end
+          end
+        }
+        def virtual: (*Array[untyped] names, **untyped options) -> void
+
     end
+
     class TableDefinition
       include ColumnMethods
     end

data/siggen/activerecord_model.rbs.erb

@@ -3,9 +3,11 @@ module ActiveRecord
   class Schema
     def self.[]: (untyped version) -> ActiveRecord::Migration::Current
   end
+
   class Migration
     class Current
       def define: (version: untyped) { () [self: instance] -> void } -> void
+
       %a{siggen:
         # In schema.rb, this model is declared as:
         #
@@ -19,13 +21,36 @@ module ActiveRecord
       def create_table: (untyped table_name, **untyped options) { (ActiveRecord::ConnectionAdapters::ColumnMethods t) -> void } -> void
     end
   end
+
   module ConnectionAdapters
     module ColumnMethods
-      <%- { text: "::String", string: "::String", datetime: "::ActiveSupport::TimeWithZone", integer: "::Integer" }.each do |key, type| -%>
+      <%- map = {
+            # https://github.com/rails/rails/blob/v8.1.2/activerecord/lib/active_record/connection_adapters/abstract/schema_definitions.rb#L331-L332
+            bitint: "::Integer",
+            binary: "::String",
+            boolean: "bool",
+            date: "::Date",
+            datetime: "::ActiveSupport::TimeWithZone",
+            decimal: "::BigDecimal",
+            float: "::Float",
+            integer: "::Integer",
+            json: "untyped",
+            string: "::String",
+            text: "::String",
+            time: "::ActiveSupport::TimeWithZone",
+            timestamp: "::ActiveSupport::TimeWithZone",
+            virtual: "_",
+          }
+          map.each do |key, type| -%>
         %a{siggen:
+          <%- if key == :virtual -%>
+          <%% map = <%= map.to_a %>.to_h %>
+          <%- type = "<" + "%= map[options[:type]] %" + ">" -%>
+          <%- end -%>
           class <%%= create_table.table_name.classify %>
             module GeneratedAttributeMethods
               <%% names.each do |name| %>
+                <%-# https://github.com/rails/rails/blob/v8.1.2/activerecord/lib/active_record/attribute_methods/read.rb -%>
                 # In schema.rb, this column is declared as:
                 #
                 # ```ruby
@@ -39,13 +64,30 @@ module ActiveRecord
                 #
                 # <%%= ___comment_of["ActiveRecord::AttributeMethods::Read#read_attribute"] %>
                 def <%%= name %>: () -> <%= type %><%%= "?" unless options[:null] == false %>
+
+                <%-# https://github.com/rails/rails/blob/v8.1.2/activerecord/lib/active_record/attribute_methods/write.rb -%>
+                # In schema.rb, this column is declared as:
+                #
+                # ```ruby
+                # <%%= ___source %>
+                # ```
+                #
+                # This method delegates to `ActiveRecord::AttributeMethods::Write` `#_write_attribute`,
+                # passing `"<%%= name %>"` as the `attr_name` argument.
+                #
+                # See the documentation comment for the delegated method below:
+                #
+                # <%%= ___comment_of["ActiveRecord::AttributeMethods::Write#write_attribute"] %>
+                def <%%= name %>=: (<%= type %><%%= "?" unless options[:null] == false %> value) -> void
               <%% end %>
             end
           end
         }
         def <%= key %>: (*Array[untyped] names, **untyped options) -> void
+
       <%- end -%>
     end
+
     class TableDefinition
       include ColumnMethods
     end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rbs-siggen
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Koji NAKAMURA