tapioca 0.4.21 → 0.4.25

data/lib/tapioca/core_ext/string.rb

@@ -0,0 +1,18 @@
+# typed: strict
+# frozen_string_literal: true
+
+class String
+  extend T::Sig
+
+  sig { returns(String) }
+  def underscore
+    return self unless /[A-Z-]|::/.match?(self)
+
+    word = to_s.gsub("::", "/")
+    word.gsub!(/([A-Z\d]+)([A-Z][a-z])/, '\1_\2')
+    word.gsub!(/([a-z\d])([A-Z])/, '\1_\2')
+    word.tr!("-", "_")
+    word.downcase!
+    word
+  end
+end

data/lib/tapioca/gemfile.rb

@@ -53,21 +53,30 @@ module Tapioca
 
     sig { returns([T::Array[Gem], T::Array[String]]) }
     def load_dependencies
-      deps = definition.locked_gems.dependencies.values
-
-      missing_specs = T::Array[String].new
-
-      dependencies = definition
-        .resolve
-        .materialize(deps, missing_specs)
+      materialized_dependencies, missing_specs = materialize_deps
+      dependencies = materialized_dependencies
         .map { |spec| Gem.new(spec) }
         .reject { |gem| gem.ignore?(dir) }
         .uniq(&:rbi_file_name)
         .sort_by(&:rbi_file_name)
-
       [dependencies, missing_specs]
     end
 
+    sig { returns([T::Array[::Gem::Specification], T::Array[String]]) }
+    def materialize_deps
+      deps = definition.locked_gems.dependencies.values
+      missing_specs = T::Array[String].new
+      materialized_dependencies = if definition.resolve.method(:materialize).arity == 1 # Support bundler >= v2.2.25
+        md = definition.resolve.materialize(deps)
+        missing_spec_names = md.missing_specs.map(&:name)
+        missing_specs = T.cast(md.missing_specs.map { |spec| "#{spec.name} (#{spec.version})" }, T::Array[String])
+        md.to_a.reject { |spec| missing_spec_names.include?(spec.name) }
+      else
+        definition.resolve.materialize(deps, missing_specs)
+      end
+      [materialized_dependencies, missing_specs]
+    end
+
     sig { returns(Bundler::Runtime) }
     def runtime
       Bundler::Runtime.new(File.dirname(gemfile.path), definition)
@@ -108,8 +117,14 @@ module Tapioca
 
       sig { returns(T::Array[Pathname]) }
       def files
-        @spec.full_require_paths.flat_map do |path|
-          Pathname.glob((Pathname.new(path) / "**/*.rb").to_s)
+        if default_gem?
+          @spec.files.map do |file|
+            ruby_lib_dir.join(file)
+          end
+        else
+          @spec.full_require_paths.flat_map do |path|
+            Pathname.glob((Pathname.new(path) / "**/*.rb").to_s)
+          end
         end
       end
 
@@ -125,11 +140,25 @@ module Tapioca
 
       sig { params(path: String).returns(T::Boolean) }
       def contains_path?(path)
-        to_realpath(path).start_with?(full_gem_path) || has_parent_gemspec?(path)
+        if default_gem?
+          files.any? { |file| file.to_s == to_realpath(path) }
+        else
+          to_realpath(path).start_with?(full_gem_path) || has_parent_gemspec?(path)
+        end
       end
 
       private
 
+      sig { returns(T::Boolean) }
+      def default_gem?
+        @spec.respond_to?(:default_gem?) && @spec.default_gem?
+      end
+
+      sig { returns(Pathname) }
+      def ruby_lib_dir
+        Pathname.new(RbConfig::CONFIG["rubylibdir"])
+      end
+
       sig { returns(String) }
       def version_string
         version = @spec.version.to_s
@@ -176,7 +205,7 @@ module Tapioca
 
       sig { returns(T::Boolean) }
       def gem_in_bundle_path?
-        full_gem_path.start_with?(Bundler.bundle_path.to_s)
+        full_gem_path.start_with?(Bundler.bundle_path.to_s, Bundler.app_cache.to_s)
       end
     end
   end

data/lib/tapioca/generator.rb

@@ -3,6 +3,7 @@
 
 require 'pathname'
 require 'thor'
+require "tapioca/core_ext/string"
 
 module Tapioca
   class Generator < ::Thor::Shell::Color
@@ -285,7 +286,7 @@ module Tapioca
     def constantize(constant_names)
       constant_map = constant_names.map do |name|
         begin
-          [name, name.constantize]
+          [name, Object.const_get(name)]
         rescue NameError
           [name, nil]
         end
@@ -600,8 +601,9 @@ module Tapioca
       if diff.empty?
         say("Nothing to do, all RBIs are up-to-date.")
       else
-        say("RBI files are out-of-date, please run:")
-        say("  `#{Config::DEFAULT_COMMAND} dsl`")
+        say("RBI files are out-of-date. In your development environment, please run:", :green)
+        say("  `#{Config::DEFAULT_COMMAND} dsl`", [:green, :bold])
+        say("Once it is complete, be sure to commit and push any changes", :green)
 
         say("")
 

data/lib/tapioca/internal.rb

@@ -6,12 +6,20 @@ require "tapioca/loader"
 require "tapioca/constant_locator"
 require "tapioca/generic_type_registry"
 require "tapioca/sorbet_ext/generic_name_patch"
+require "tapioca/sorbet_ext/fixed_hash_patch"
 require "tapioca/config"
 require "tapioca/config_builder"
 require "tapioca/generator"
 require "tapioca/cli"
 require "tapioca/cli/main"
 require "tapioca/gemfile"
+require "tapioca/rbi/model"
+require "tapioca/rbi/visitor"
+require "tapioca/rbi/rewriters/nest_singleton_methods"
+require "tapioca/rbi/rewriters/nest_non_public_methods"
+require "tapioca/rbi/rewriters/group_nodes"
+require "tapioca/rbi/rewriters/sort_nodes"
+require "tapioca/rbi/printer"
 require "tapioca/compilers/sorbet"
 require "tapioca/compilers/requires_compiler"
 require "tapioca/compilers/symbol_table_compiler"

data/lib/tapioca/rbi/model.rb

@@ -0,0 +1,405 @@
+# typed: strict
+# frozen_string_literal: true
+
+module Tapioca
+  module RBI
+    class Node
+      extend T::Sig
+      extend T::Helpers
+
+      abstract!
+
+      sig { returns(T.nilable(Tree)) }
+      attr_accessor :parent_tree
+
+      sig { void }
+      def initialize
+        @parent_tree = nil
+      end
+
+      sig { void }
+      def detach
+        tree = parent_tree
+        return unless tree
+        tree.nodes.delete(self)
+        self.parent_tree = nil
+      end
+    end
+
+    class Tree < Node
+      extend T::Sig
+
+      sig { returns(T::Array[Node]) }
+      attr_reader :nodes
+
+      sig { void }
+      def initialize
+        super()
+        @nodes = T.let([], T::Array[Node])
+      end
+
+      sig { params(node: Node).void }
+      def <<(node)
+        node.parent_tree = self
+        @nodes << node
+      end
+
+      sig { returns(T::Boolean) }
+      def empty?
+        nodes.empty?
+      end
+    end
+
+    # Scopes
+
+    class Scope < Tree
+      extend T::Helpers
+
+      abstract!
+    end
+
+    class Module < Scope
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_accessor :name
+
+      sig { params(name: String).void }
+      def initialize(name)
+        super()
+        @name = name
+      end
+    end
+
+    class Class < Scope
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_accessor :name
+
+      sig { returns(T.nilable(String)) }
+      attr_accessor :superclass_name
+
+      sig { params(name: String, superclass_name: T.nilable(String)).void }
+      def initialize(name, superclass_name: nil)
+        super()
+        @name = name
+        @superclass_name = superclass_name
+      end
+    end
+
+    class SingletonClass < Scope
+      extend T::Sig
+
+      sig { void }
+      def initialize
+        super()
+      end
+    end
+
+    # Consts
+
+    class Const < Node
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_reader :name, :value
+
+      sig { params(name: String, value: String).void }
+      def initialize(name, value)
+        super()
+        @name = name
+        @value = value
+      end
+    end
+
+    # Methods and args
+
+    class Method < Node
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_accessor :name
+
+      sig { returns(T::Array[Param]) }
+      attr_reader :params
+
+      sig { returns(T::Boolean) }
+      attr_accessor :is_singleton
+
+      sig { returns(Visibility) }
+      attr_accessor :visibility
+
+      sig { returns(T::Array[Sig]) }
+      attr_accessor :sigs
+
+      sig do
+        params(
+          name: String,
+          params: T::Array[Param],
+          is_singleton: T::Boolean,
+          visibility: Visibility,
+          sigs: T::Array[Sig]
+        ).void
+      end
+      def initialize(name, params: [], is_singleton: false, visibility: Visibility::Public, sigs: [])
+        super()
+        @name = name
+        @params = params
+        @is_singleton = is_singleton
+        @visibility = visibility
+        @sigs = sigs
+      end
+
+      sig { params(param: Param).void }
+      def <<(param)
+        @params << param
+      end
+    end
+
+    class Param < Node
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_reader :name
+
+      sig { params(name: String).void }
+      def initialize(name)
+        super()
+        @name = name
+      end
+    end
+
+    class OptParam < Param
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_reader :value
+
+      sig { params(name: String, value: String).void }
+      def initialize(name, value)
+        super(name)
+        @value = value
+      end
+    end
+
+    class RestParam < Param; end
+    class KwParam < Param; end
+    class KwOptParam < OptParam; end
+    class KwRestParam < Param; end
+    class BlockParam < Param; end
+
+    # Mixins
+
+    class Mixin < Node
+      extend T::Sig
+      extend T::Helpers
+
+      abstract!
+
+      sig { returns(String) }
+      attr_reader :name
+
+      sig { params(name: String).void }
+      def initialize(name)
+        super()
+        @name = name
+      end
+    end
+
+    class Include < Mixin; end
+    class Extend < Mixin; end
+
+    # Visibility
+
+    class Visibility < Node
+      extend T::Sig
+      extend T::Helpers
+
+      abstract!
+
+      sig { returns(Symbol) }
+      attr_reader :visibility
+
+      sig { params(visibility: Symbol).void }
+      def initialize(visibility)
+        super()
+        @visibility = visibility
+      end
+
+      sig { returns(T::Boolean) }
+      def public?
+        visibility == :public
+      end
+
+      Public = T.let(Visibility.new(:public), Visibility)
+      Protected = T.let(Visibility.new(:protected), Visibility)
+      Private = T.let(Visibility.new(:private), Visibility)
+    end
+
+    # Sorbet's sigs
+
+    class Sig < Node
+      extend T::Sig
+
+      sig { returns(T::Array[SigParam]) }
+      attr_reader :params
+
+      sig { returns(T.nilable(String)) }
+      attr_accessor :return_type
+
+      sig { returns(T::Boolean) }
+      attr_accessor :is_abstract, :is_override, :is_overridable
+
+      sig { returns(T::Array[String]) }
+      attr_reader :type_params
+
+      sig do
+        params(
+          params: T::Array[SigParam],
+          return_type: T.nilable(String),
+          is_abstract: T::Boolean,
+          is_override: T::Boolean,
+          is_overridable: T::Boolean,
+          type_params: T::Array[String]
+        ).void
+      end
+      def initialize(
+        params: [],
+        return_type: nil,
+        is_abstract: false,
+        is_override: false,
+        is_overridable: false,
+        type_params: []
+      )
+        super()
+        @params = params
+        @return_type = return_type
+        @is_abstract = is_abstract
+        @is_override = is_override
+        @is_overridable = is_overridable
+        @type_params = type_params
+      end
+
+      sig { params(param: SigParam).void }
+      def <<(param)
+        @params << param
+      end
+    end
+
+    class SigParam < Node
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_reader :name, :type
+
+      sig { params(name: String, type: String).void }
+      def initialize(name, type)
+        super()
+        @name = name
+        @type = type
+      end
+    end
+
+    # Sorbet's T::Struct
+
+    class TStruct < Class
+      extend T::Sig
+
+      sig { params(name: String).void }
+      def initialize(name)
+        super(name, superclass_name: "::T::Struct")
+      end
+    end
+
+    class TStructField < Node
+      extend T::Sig
+      extend T::Helpers
+
+      abstract!
+
+      sig { returns(String) }
+      attr_accessor :name, :type
+
+      sig { returns(T.nilable(String)) }
+      attr_accessor :default
+
+      sig do
+        params(
+          name: String,
+          type: String,
+          default: T.nilable(String)
+        ).void
+      end
+      def initialize(name, type, default: nil)
+        super()
+        @name = name
+        @type = type
+        @default = default
+      end
+    end
+
+    class TStructProp < TStructField; end
+    class TStructConst < TStructField; end
+
+    # Sorbet's T::Enum
+
+    class TEnum < Class
+      extend T::Sig
+
+      sig { params(name: String).void }
+      def initialize(name)
+        super(name, superclass_name: "::T::Enum")
+      end
+    end
+
+    class TEnumBlock < Node
+      extend T::Sig
+
+      sig { returns(T::Array[String]) }
+      attr_reader :names
+
+      sig { params(names: T::Array[String]).void }
+      def initialize(names = [])
+        super()
+        @names = names
+      end
+
+      sig { returns(T::Boolean) }
+      def empty?
+        names.empty?
+      end
+    end
+
+    # Sorbet's misc.
+
+    class Helper < Node
+      extend T::Helpers
+
+      sig { returns(String) }
+      attr_reader :name
+
+      sig { params(name: String).void }
+      def initialize(name)
+        super()
+        @name = name
+      end
+    end
+
+    class TypeMember < Node
+      extend T::Sig
+
+      sig { returns(String) }
+      attr_reader :name, :value
+
+      sig { params(name: String, value: String).void }
+      def initialize(name, value)
+        super()
+        @name = name
+        @value = value
+      end
+    end
+
+    class MixesInClassMethods < Mixin; end
+  end
+end