rbs_rails 0.1.0 → 0.2.0

data/assets/sig/libxml.rbs

@@ -0,0 +1,10 @@
+# is it correct?
+
+module LibXML
+end
+module LibXML::XML
+end
+module LibXML::XML::SaxParser
+end
+module LibXML::XML::SaxParser::Callbacks
+end

data/assets/sig/minitest.rbs

@@ -0,0 +1,2 @@
+class Minitest::Test
+end

data/assets/sig/nokogiri.rbs

@@ -0,0 +1,8 @@
+module Nokogiri
+end
+module Nokogiri::XML
+end
+module Nokogiri::XML::SAX
+end
+class Nokogiri::XML::SAX::Document
+end

data/assets/sig/rails.rbs

@@ -0,0 +1,11 @@
+module Rails
+  def self.root: () -> Pathname
+  def self.application: () -> Rails::Application
+end
+
+class Rails::Application
+  def routes: () -> untyped
+end
+
+class Rails::Railtie
+end

data/assets/sig/stdlib.rbs

@@ -0,0 +1,14 @@
+class Logger
+end
+
+class Logger::Formatter
+end
+
+module Mutex_m
+end
+
+module DRb
+end
+
+module DRb::DRbUndumped
+end

data/bin/add-type-params.rb

@@ -0,0 +1,59 @@
+require 'rbs'
+rbs = ARGF.read
+decls = RBS::Parser.parse_signature(rbs)
+
+def args(n)
+  (:T..).take(n)
+end
+
+def env
+  @env ||= RBS::Environment.new.tap do |env|
+    loader = RBS::EnvironmentLoader.new()
+    loader.load(env: env)
+  end
+end
+
+def apply_to_superclass(decl)
+  return unless decl.super_class
+
+  name = decl.super_class.name
+  type = env.find_class(name) || env.find_class(name.absolute!)
+  return unless type
+  return if type.type_params.empty?
+
+  args = args(type.type_params.size)
+  decl.super_class.instance_variable_set(:@args, args)
+  type_params = RBS::AST::Declarations::ModuleTypeParams.new.tap do |tp|
+    args.each do |a|
+      tp.add RBS::AST::Declarations::ModuleTypeParams::TypeParam.new(name: a)
+    end
+  end
+
+  decl.instance_variable_set(:@type_params, type_params)
+end
+
+def apply_to_includes(decl)
+  decl.members.each do |member|
+    next unless member.is_a?(RBS::AST::Members::Mixin)
+
+    name = member.name
+    type = env.find_class(name) || env.find_class(name.absolute!)
+    next unless type
+    next if type.type_params.empty?
+
+    args = type.type_params.size.times.map { :untyped }
+    member.instance_variable_set(:@args, args)
+  end
+end
+
+decls.each do |decl|
+  case decl
+  when RBS::AST::Declarations::Class
+    apply_to_superclass(decl)
+    apply_to_includes(decl)
+  when RBS::AST::Declarations::Module, RBS::AST::Declarations::Interface, RBS::AST::Declarations::Extension
+    apply_to_includes(decl)
+  end
+end
+
+puts RBS::Writer.new(out: $stdout).write(decls)

data/bin/merge-duplicate-decls.rb

@@ -0,0 +1,30 @@
+require 'rbs'
+rbs = ARGF.read
+
+decls = RBS::Parser.parse_signature(rbs)
+decl_map = {}
+decls.each do |decl|
+  if d = decl_map[decl.name]
+    # TODO: Is it right for decl is not a class / module?
+    c = RBS::AST::Declarations::Class
+    m = RBS::AST::Declarations::Module
+    e = RBS::AST::Declarations::Extension
+    raise 'class mismatch' if d.class != decl.class
+    raise 'class mismatch' unless [c,m,e].include?(d.class)
+
+    d.members.concat decl.members
+    d.members.uniq! { if _1.is_a?(RBS::AST::Members::MethodDefinition) then [_1.name, _1.singleton?] else _1 end}
+    case d
+    when c
+      decl_map[decl.name] = c.new(name: d.name, type_params: d.type_params, super_class: d.super_class || decl.super_class, members: d.members, annotations: d.annotations, location: d.location, comment: d.comment)
+    when m
+      decl_map[decl.name] = m.new(name: d.name, type_params: d.type_params, members: d.members, self_type: d.self_type, annotations: d.annotations, location: d.location, comment: d.comment)
+    when e
+      decl_map[decl.name] = e.new(name: d.name, type_params: d.type_params, extension_name: d.extension_name, members: d.members, annotations: d.annotations, location: d.location, comment: d.comment)
+    end
+  else
+    decl_map[decl.name] = decl
+  end
+end
+
+puts RBS::Writer.new(out: $stdout).write(decl_map.values)

data/lib/rbs_rails.rb

@@ -8,7 +8,7 @@ module RbsRails
   class Error < StandardError; end
 
   def self.copy_signatures(to:)
-    from = Pathname(__dir__) / '../assets/sig/'
+    from = Pathname(_ = __dir__) / '../assets/sig/'
     to = Pathname(to)
     FileUtils.cp_r(from, to)
   end

data/lib/rbs_rails/active_record.rb

@@ -56,7 +56,9 @@ module RbsRails
         when :extension
           "extension #{klass.name} (RbsRails)"
         when :class
-          "class #{klass.name} < #{klass.superclass.name}"
+          # @type var superclass: Class
+          superclass = _ = klass.superclass
+          "class #{klass.name} < #{superclass.name}"
         else
           raise "unexpected mode: #{mode}"
         end
@@ -91,6 +93,7 @@ module RbsRails
       end
 
       private def enum_instance_methods
+        # @type var methods: Array[String]
         methods = []
         enum_definitions.each do |hash|
           hash.each do |name, values|
@@ -108,6 +111,7 @@ module RbsRails
       end
 
       private def enum_scope_methods(singleton:)
+        # @type var methods: Array[String]
         methods = []
         enum_definitions.each do |hash|
           hash.each do |name, values|
@@ -191,6 +195,7 @@ module RbsRails
       end
 
       private def args_to_type(args_node)
+        # @type var res: Array[String]
         res = []
         args_node.children.each do |node|
           case node.type
@@ -213,7 +218,9 @@ module RbsRails
         return @parse_model_file if defined?(@parse_model_file)
 
 
-        path = Rails.root.join('app/models/', klass.name.underscore + '.rb')
+        # @type var class_name: String
+        class_name = _ = klass.name
+        path = Rails.root.join('app/models/', class_name.underscore + '.rb')
         return @parse_model_file = nil unless path.exist?
         return [] unless path.exist?
 
@@ -226,6 +233,7 @@ module RbsRails
       private def traverse(node, &block)
         return to_enum(__method__, node) unless block_given?
 
+        # @type var block: ^(Parser::AST::Node) -> untyped
         block.call node
         node.children.each do |child|
           traverse(child, &block) if child.is_a?(Parser::AST::Node)
@@ -246,7 +254,9 @@ module RbsRails
         case t
         when :integer
           Integer.name
-        when :string, :text, :uuid
+        when :float
+          Float.name
+        when :string, :text, :uuid, :binary
           String.name
         when :datetime
           # TODO
@@ -266,6 +276,7 @@ module RbsRails
       end
 
       private
+      # @dynamic klass, mode
       attr_reader :klass, :mode
     end
   end

data/lib/rbs_rails/path_helpers.rb

@@ -26,6 +26,7 @@ module RbsRails
     end
 
     private
+    # @dynamic routes
     attr_reader :routes
   end
 end

data/lib/rbs_rails/version.rb

@@ -1,3 +1,6 @@
 module RbsRails
-  VERSION = "0.1.0"
+  # Because of copy_signatures is defined by lib/rbs_rails.rb
+  # @dynamic self.copy_signatures
+
+  VERSION = "0.2.0"
 end

data/rbs_rails.gemspec

@@ -9,6 +9,7 @@ Gem::Specification.new do |spec|
   spec.summary       = %q{A RBS files generator for Rails application}
   spec.description   = %q{A RBS files generator for Rails application}
   spec.homepage      = "https://github.com/pocke/rbs_rails"
+  spec.license       = 'Apache-2.0'
   spec.required_ruby_version = Gem::Requirement.new(">= 2.3.0")
 
   spec.metadata["homepage_uri"] = spec.homepage

data/sig/fileutils.rbs

@@ -0,0 +1,3 @@
+module FileUtils
+  def self.cp_r: (Pathname, Pathname) -> untyped
+end

data/sig/parser.rbs

@@ -0,0 +1,14 @@
+module Parser
+end
+
+class Parser::CurrentRuby
+  def self.parse: (String) -> Parser::AST::Node
+end
+
+module Parser::AST
+end
+
+class Parser::AST::Node
+  def type: () -> Symbol
+  def children: () -> Array[untyped]
+end

data/sig/rbs_rails.rbs

@@ -0,0 +1,6 @@
+module RbsRails
+  def self.copy_signatures: (to: untyped to) -> untyped
+end
+
+class RbsRails::Error < StandardError
+end

data/sig/rbs_rails/active_record.rbs

@@ -0,0 +1,59 @@
+module RbsRails::ActiveRecord
+  def self.class_to_rbs: (untyped klass, mode: untyped mode) -> untyped
+end
+
+class RbsRails::ActiveRecord::Generator
+  @parse_model_file: nil | Parser::AST::Node
+
+  def initialize: (singleton(ActiveRecord::Base) klass, mode: (:extension | :class) mode) -> untyped
+
+  def generate: () -> String
+
+  def klass_decl: () -> String
+
+  def relation_decl: () -> String
+
+  def collection_proxy_decl: () -> String
+
+  def header: () -> String
+
+  def associations: () -> String
+
+  def has_many: () -> String
+
+  def has_one: () -> String
+
+  def belongs_to: () -> String
+
+  def enum_instance_methods: () -> String
+
+  def enum_scope_methods: (singleton: untyped `singleton`) -> String
+
+  def enum_definitions: () -> Array[Hash[Symbol, untyped]]
+
+  # We need static analysis to detect enum.
+  # ActiveRecord has `defined_enums` method,
+  # but it does not contain _prefix and _suffix information.
+  def build_enum_definitions: () -> Array[Hash[Symbol, untyped]]
+
+  def enum_method_name: (Hash[Symbol, untyped] hash, Symbol name, Symbol label) -> String
+
+  def scopes: (singleton: untyped `singleton`) -> untyped
+
+  def args_to_type: (untyped args_node) -> untyped
+
+  def parse_model_file: () -> untyped
+
+  def traverse: (Parser::AST::Node node) { (Parser::AST::Node) -> untyped } -> untyped
+              | (Parser::AST::Node node) -> Enumerator[Parser::AST::Node, untyped]
+
+  def relation_class_name: () -> untyped
+
+  def columns: () -> untyped
+
+  def sql_type_to_class: (untyped t) -> untyped
+
+  attr_reader klass: singleton(ActiveRecord::Base)
+
+  attr_reader mode: (:extension | :class)
+end

data/sig/rbs_rails/path_helpers.rbs

@@ -0,0 +1,11 @@
+class RbsRails::PathHelpers
+  def self.generate: (?routes: untyped routes) -> untyped
+
+  def initialize: (routes: untyped routes) -> untyped
+
+  def generate: () -> untyped
+
+  def helpers: () -> untyped
+
+  attr_reader routes: untyped
+end

data/sig/rbs_rails/version.rbs

@@ -0,0 +1 @@
+RbsRails::VERSION: ::String

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbs_rails
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Masataka Pocke Kuwabara
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-02-20 00:00:00.000000000 Z
+date: 2020-06-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: parser
@@ -32,19 +32,42 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".travis.yml"
 - Gemfile
+- LICENSE
 - README.md
 - Rakefile
+- Steepfile
+- assets/sig/action_controller.rbs
+- assets/sig/action_mailer.rbs
+- assets/sig/action_view.rbs
 - assets/sig/active_record.rbs
+- assets/sig/builtin.rbs
+- assets/sig/generated/activemodel.rbs
+- assets/sig/generated/activesupport.rbs
+- assets/sig/libxml.rbs
+- assets/sig/minitest.rbs
+- assets/sig/nokogiri.rbs
+- assets/sig/rails.rbs
+- assets/sig/stdlib.rbs
+- bin/add-type-params.rb
 - bin/console
+- bin/merge-duplicate-decls.rb
 - bin/setup
 - lib/rbs_rails.rb
 - lib/rbs_rails/active_record.rb
 - lib/rbs_rails/path_helpers.rb
 - lib/rbs_rails/version.rb
 - rbs_rails.gemspec
+- sig/fileutils.rbs
+- sig/parser.rbs
+- sig/rbs_rails.rbs
+- sig/rbs_rails/active_record.rbs
+- sig/rbs_rails/path_helpers.rbs
+- sig/rbs_rails/version.rbs
 homepage: https://github.com/pocke/rbs_rails
-licenses: []
+licenses:
+- Apache-2.0
 metadata:
   homepage_uri: https://github.com/pocke/rbs_rails
   source_code_uri: https://github.com/pocke/rbs_rails