repl_type_completor 0.1.11 → 0.1.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 45c80e0c90081af7992d51765854f270adbaec795050775ae778560737db52b2
-  data.tar.gz: 9b4745ab99625ebfd3b6c1ecb526ff6e5186ef9320873b55cc4f6d88103f25ec
+  metadata.gz: e17428782dfde53e8b592d0fb16957c02ec537bee5122a624e3a99bbfc663e5a
+  data.tar.gz: ebebeec739da4ae0e46ac690f3e238cff6759a3cedebccdf9e180e28ee6c51a8
 SHA512:
-  metadata.gz: c1a11765f19a07a3cf1a387448e2a57f8d0a087991a9cdae946eebdb6000552e4b7ed7a172a1d7dc65ba273def75e345b76e7adc290cb99af153017cc84df8e8
-  data.tar.gz: 682c8671b614077226d873dba34543dfae6598fb555cefa69bdce155e09de2bb4cc5cfb107d7a5b4887e50d6381eb923b2fda1d4babb5d458b506484b118b2d5
+  metadata.gz: d2d803e3ae97d53a1b3287b564615f9f4f9790d58065ad224d9376ec4fd2407d6df66d10c20a54175ad10c57e2b9efafe10955e01306b21c0d92213b6cc5b9e9
+  data.tar.gz: 4ffb182ae4720b9e67d4abb575c09b1e07f7c00a33c1c2601bf66f763eaa4997f8a813980b9367f47ba9731f6ce318a13b277d8c80396c9e7cfe43178896198c

data/lib/repl_type_completor/methods.rb

@@ -5,6 +5,7 @@ module ReplTypeCompletor
     OBJECT_SINGLETON_METHODS_METHOD = Object.instance_method(:singleton_methods)
     OBJECT_PRIVATE_METHODS_METHOD = Object.instance_method(:private_methods)
     OBJECT_INSTANCE_VARIABLES_METHOD = Object.instance_method(:instance_variables)
+    OBJECT_INSTANCE_VARIABLE_DEFINED_METHOD = Object.instance_method(:instance_variable_defined?)
     OBJECT_INSTANCE_VARIABLE_GET_METHOD = Object.instance_method(:instance_variable_get)
     OBJECT_CLASS_METHOD = Object.instance_method(:class)
     MODULE_NAME_METHOD = Module.instance_method(:name)

data/lib/repl_type_completor/result.rb

@@ -72,7 +72,11 @@ module ReplTypeCompletor
       else
         []
       end
-      candidates.select { _1.start_with?(name) }.map { _1[name.size..] }
+
+      candidates.filter_map do
+        _1[name.size..] if _1.start_with?(name)
+      rescue EncodingError
+      end
     rescue Exception => e
       ReplTypeCompletor.handle_exception(e)
       []

data/lib/repl_type_completor/type_analyzer.rb

@@ -1179,6 +1179,12 @@ module ReplTypeCompletor
           end
         end
       end
+
+      if types.empty? && args.empty? && !kwargs && !block
+        t = Types.accessor_method_return_type(receiver, method_name)
+        types << t if t
+      end
+
       scope&.terminate if terminates && breaks.empty?
       Types::UnionType[*types, *breaks]
     end

data/lib/repl_type_completor/types.rb

@@ -105,6 +105,28 @@ module ReplTypeCompletor
       UnionType[*types]
     end
 
+    def self.accessor_method_return_type(type, method_name)
+      return unless method_name.match?(/\A[a-z_][a-z_0-9]*\z/)
+
+      ivar_name = :"@#{method_name}"
+      instances = type.types.filter_map do |t|
+        case t
+        in SingletonType
+          t.module_or_class
+        in InstanceType
+          t.instances
+        end
+      end.flatten
+      instances = instances.sample(OBJECT_TO_TYPE_SAMPLE_SIZE) if instances.size > OBJECT_TO_TYPE_SAMPLE_SIZE
+      objects = []
+      instances.each do |instance|
+        if Methods::OBJECT_INSTANCE_VARIABLE_DEFINED_METHOD.bind_call(instance, ivar_name)
+          objects << Methods::OBJECT_INSTANCE_VARIABLE_GET_METHOD.bind_call(instance, ivar_name)
+        end
+      end
+      union_type_from_objects(objects) unless objects.empty?
+    end
+
     def self.rbs_methods(type, method_name, args_types, kwargs_type, has_block)
       return [] unless rbs_builder
 
@@ -188,9 +210,10 @@ module ReplTypeCompletor
     end
 
     def self.union_type_from_objects(objects)
-      instanes = objects.size <= OBJECT_TO_TYPE_SAMPLE_SIZE ? objects : objects.sample(OBJECT_TO_TYPE_SAMPLE_SIZE)
-      class_instanes = instanes.group_by { Methods::OBJECT_CLASS_METHOD.bind_call(_1) }
-      UnionType[*class_instanes.map { InstanceType.new _1, nil, _2 }]
+      instances = objects.size <= OBJECT_TO_TYPE_SAMPLE_SIZE ? objects : objects.sample(OBJECT_TO_TYPE_SAMPLE_SIZE)
+      modules, instances = instances.partition { Module === _1 }
+      class_instances = instances.group_by { Methods::OBJECT_CLASS_METHOD.bind_call(_1) }
+      UnionType[*class_instances.map { InstanceType.new _1, nil, _2 }, *modules.uniq.map { SingletonType.new _1 }]
     end
 
     class SingletonType
@@ -273,7 +296,9 @@ module ReplTypeCompletor
       end
 
       def inspect
-        if params.empty?
+        if !@params && (@klass == Array || @klass == Hash) && @instances
+          "#{inspect_without_params}[unresolved]"
+        elsif params.empty?
           inspect_without_params
         else
           params_string = "[#{params.map { "#{_1}: #{_2.inspect}" }.join(', ')}]"

data/lib/repl_type_completor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ReplTypeCompletor
-  VERSION = "0.1.11"
+  VERSION = "0.1.12"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: repl_type_completor
 version: !ruby/object:Gem::Version
-  version: 0.1.11
+  version: 0.1.12
 platform: ruby
 authors:
 - tompng
 bindir: exe
 cert_chain: []
-date: 2025-04-01 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: prism
@@ -84,7 +84,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.3
+rubygems_version: 3.6.7
 specification_version: 4
 summary: Type based completion for REPL.
 test_files: []