typeprof 0.5.3 → 0.8.0

data/lib/typeprof/export.rb

@@ -91,9 +91,9 @@ module TypeProf
     def conv_class(namespace, class_def, inner_classes)
       @scratch.namespace = namespace
 
-      if class_def.superclass
-        omit = @class_defs[class_def.superclass].klass_obj == Type::Builtin[:obj] || class_def.klass_obj == Type::Builtin[:obj]
-        superclass = omit ? nil : @scratch.get_class_name(@class_defs[class_def.superclass].klass_obj)
+      if class_def.klass_obj.superclass != :__root__ && class_def.klass_obj.superclass
+        omit = class_def.klass_obj.superclass == Type::Builtin[:obj] || class_def.klass_obj == Type::Builtin[:obj]
+        superclass = omit ? nil : @scratch.get_class_name(class_def.klass_obj.superclass)
       end
 
       @scratch.namespace = class_def.name
@@ -105,12 +105,12 @@ module TypeProf
         consts[name] = ty.screen_name(@scratch)
       end
 
-      included_mods = class_def.modules[false].filter_map do |mod_def, absolute_paths|
+      included_mods = class_def.modules[false].filter_map do |mod_def, _type_args, absolute_paths|
         next if absolute_paths.all? {|path| !path || Config.check_dir_filter(path) == :exclude }
         Type::Instance.new(mod_def.klass_obj).screen_name(@scratch)
       end
 
-      extended_mods = class_def.modules[true].filter_map do |mod_def, absolute_paths|
+      extended_mods = class_def.modules[true].filter_map do |mod_def, _type_args, absolute_paths|
         next if absolute_paths.all? {|path| !path || Config.check_dir_filter(path) == :exclude }
         Type::Instance.new(mod_def.klass_obj).screen_name(@scratch)
       end
@@ -135,11 +135,12 @@ module TypeProf
               method_name = ctx.mid
               method_name = "self.#{ method_name }" if singleton
 
-              iseq_methods[method_name] ||= []
-              iseq_methods[method_name] << @scratch.show_method_signature(ctx)
+              iseq_methods[method_name] ||= [true, []]
+              iseq_methods[method_name][0] &&= mdef.pub_meth
+              iseq_methods[method_name][1] << @scratch.show_method_signature(ctx)
             end
           when AttrMethodDef
-            next if Config.check_dir_filter(mdef.absolute_path) == :exclude
+            next if !mdef.absolute_path || Config.check_dir_filter(mdef.absolute_path) == :exclude
             mid = mid.to_s[0..-2].to_sym if mid.to_s.end_with?("=")
             method_name = mid
             method_name = "self.#{ mid }" if singleton
@@ -168,12 +169,14 @@ module TypeProf
         next unless var.to_s.start_with?("@")
         var = "self.#{ var }" if singleton
         next if attr_methods[[singleton ? "self.#{ var.to_s[1..] }" : var.to_s[1..].to_sym, false]]
-        [var, ty.screen_name(@scratch), entry.rbs_declared]
+        next if entry.rbs_declared
+        [var, ty.screen_name(@scratch)]
       end.compact
 
       cvars = cvars.map do |var, entry|
         next if entry.absolute_paths.all? {|path| Config.check_dir_filter(path) == :exclude }
-        [var, entry.type.screen_name(@scratch), entry.rbs_declared]
+        next if entry.rbs_declared
+        [var, entry.type.screen_name(@scratch)]
       end.compact
 
       if !class_def.absolute_path || Config.check_dir_filter(class_def.absolute_path) == :exclude
@@ -277,14 +280,12 @@ module TypeProf
         output.puts indent + "  extend #{ mod }"
         first = false
       end
-      class_data.ivars.each do |var, ty, rbs_declared|
-        s = rbs_declared ? "# " : "  "
-        output.puts indent + s + "#{ var } : #{ ty }" unless var.start_with?("_")
+      class_data.ivars.each do |var, ty|
+        output.puts indent + "  #{ var } : #{ ty }" unless var.start_with?("_")
         first = false
       end
-      class_data.cvars.each do |var, ty, rbs_declared|
-        s = rbs_declared ? "# " : "  "
-        output.puts indent + s + "#{ var } : #{ ty }"
+      class_data.cvars.each do |var, ty|
+        output.puts indent + "  #{ var } : #{ ty }"
         first = false
       end
       class_data.attr_methods.each do |(method_name, hidden), (kind, ty)|
@@ -296,8 +297,13 @@ module TypeProf
         output.puts indent + "# def #{ method_name } : #{ sigs }"
         first = false
       end
-      class_data.iseq_methods.each do |method_name, sigs|
+      prev_pub_meth = true
+      class_data.iseq_methods.each do |method_name, (pub_meth, sigs)|
         sigs = sigs.sort.join("\n" + indent + " " * (method_name.size + 7) + "| ")
+        if prev_pub_meth != pub_meth
+          output.puts indent + "  #{ pub_meth ? "public" : "private" }"
+          prev_pub_meth = pub_meth
+        end
         output.puts indent + "  def #{ method_name } : #{ sigs }"
         first = false
       end

data/lib/typeprof/import.rb

@@ -100,14 +100,19 @@ module TypeProf
       json = {}
 
       each_class_decl do |name, decls|
-        super_class_name = get_super_class_name(name, decls)
         klass = conv_type_name(name)
-        superclass = super_class_name ? conv_type_name(super_class_name) : nil
+        super_class_name, super_class_args = get_super_class(name, decls)
+        if super_class_name
+          name = conv_type_name(super_class_name)
+          type_args = super_class_args.map {|type| conv_type(type) }
+          superclass = [name, type_args]
+        end
 
         type_params = nil
         included_modules = []
         extended_modules = []
         methods = {}
+        attr_methods = {}
         ivars = {}
         cvars = {}
         rbs_sources = {}
@@ -143,15 +148,15 @@ module TypeProf
               end
             when RBS::AST::Members::AttrReader
               ty = conv_type(member.type)
-              methods[[false, member.name]] = attr_reader_def(ty)
+              attr_methods[[false, member.name]] = attr_method_def(:reader, member.name, ty)
             when RBS::AST::Members::AttrWriter
               ty = conv_type(member.type)
-              methods[[false, :"#{ member.name }="]] = attr_writer_def(ty)
+              attr_methods[[false, member.name]] = attr_method_def(:writer, member.name, ty)
             when RBS::AST::Members::AttrAccessor
               ty = conv_type(member.type)
-              methods[[false, member.name]] = attr_reader_def(ty)
-              methods[[false, :"#{ member.name }="]] = attr_writer_def(ty)
+              attr_methods[[false, member.name]] = attr_method_def(:accessor, member.name, ty)
             when RBS::AST::Members::Alias
+              # XXX: an alias to attr methods?
               if member.instance?
                 method_def = methods[[false, member.old_name]]
                 methods[[false, member.new_name]] = method_def if method_def
@@ -165,7 +170,8 @@ module TypeProf
               name = member.name
               if name.kind == :class
                 mod = conv_type_name(name)
-                included_modules << mod
+                type_args = member.args.map {|type| conv_type(type) }
+                included_modules << [mod, type_args]
               else
                 # including an interface is not supported yet
               end
@@ -174,7 +180,8 @@ module TypeProf
               name = member.name
               if name.kind == :class
                 mod = conv_type_name(name)
-                extended_modules << mod
+                type_args = member.args.map {|type| conv_type(type) }
+                extended_modules << [mod, type_args]
               else
                 # extending a module with an interface is not supported yet
               end
@@ -205,6 +212,7 @@ module TypeProf
             included_modules: included_modules,
             extended_modules: extended_modules,
             methods: methods,
+            attr_methods: attr_methods,
             ivars: ivars,
             cvars: cvars,
             rbs_sources: rbs_sources,
@@ -231,7 +239,7 @@ module TypeProf
             @all_env.class_decls[name].decls.each do |decl|
               decl = decl.decl
               next if decl.is_a?(RBS::AST::Declarations::Module)
-              each_ancestor(decl) {|name| queue << [:visit, name] }
+              each_reference(decl) {|name| queue << [:visit, name] }
             end
             queue << [:visit, name.namespace.to_type_name] if !name.namespace.empty?
           end
@@ -246,24 +254,29 @@ module TypeProf
       end
     end
 
-    def each_ancestor(decl, &blk)
+    def each_reference(decl, &blk)
       yield decl.name
-      super_class = decl.super_class || RBS::BuiltinNames::Object
+      if decl.super_class
+        name = decl.super_class.name
+      else
+        name = RBS::BuiltinNames::Object.name
+      end
       return if decl.name == RBS::BuiltinNames::BasicObject.name
-      return if decl.name == super_class.name
-      @all_env.class_decls[super_class.name].decls.each do |decl|
-        each_ancestor(decl.decl, &blk)
+      return if decl.name == name
+      @all_env.class_decls[name].decls.each do |decl|
+        each_reference(decl.decl, &blk)
       end
     end
 
-    def get_super_class_name(name, decls)
+    def get_super_class(name, decls)
       return nil if name == RBS::BuiltinNames::BasicObject.name
 
       decls.each do |decl|
         decl = decl.decl
         case decl
         when RBS::AST::Declarations::Class
-          return decl.super_class.name if decl.super_class
+          super_class = decl.super_class
+          return super_class.name, super_class.args if super_class
         when RBS::AST::Declarations::Module, RBS::AST::Declarations::Interface
           return nil
         else
@@ -271,7 +284,7 @@ module TypeProf
         end
       end
 
-      return RBS::BuiltinNames::Object.name
+      return RBS::BuiltinNames::Object.name, []
     end
 
     def conv_method_def(rbs_method_types)
@@ -290,7 +303,7 @@ module TypeProf
       opt_kw_tys = func.optional_keywords.to_h {|key, type| [key, conv_type(type.type)] }
       req_kw_tys = func.required_keywords.to_h {|key, type| [key, conv_type(type.type)] }
       rest_kw_ty = func.rest_keywords
-      raise NotImplementedError if rest_kw_ty # XXX
+      rest_kw_ty = conv_type(rest_kw_ty.type) if rest_kw_ty
 
       ret_ty = conv_type(func.return_type)
 
@@ -307,32 +320,12 @@ module TypeProf
       }
     end
 
-    def attr_reader_def(ty)
-      [{
-        type_params: [],
-        lead_tys: [],
-        opt_tys: [],
-        rest_ty: nil,
-        req_kw_tys: {},
-        opt_kw_tys: {},
-        rest_kw_ty: nil,
-        blk: nil,
-        ret_ty: ty,
-      }]
-    end
-
-    def attr_writer_def(ty)
-      [{
-        type_params: [],
-        lead_tys: [ty],
-        opt_tys: [],
-        rest_ty: nil,
-        req_kw_tys: {},
-        opt_kw_tys: {},
-        rest_kw_ty: nil,
-        blk: nil,
-        ret_ty: ty,
-      }]
+    def attr_method_def(kind, name, ty)
+      {
+        kind: kind,
+        ivar: name,
+        ty: ty,
+      }
     end
 
     def conv_block(rbs_block)
@@ -467,8 +460,8 @@ module TypeProf
     def import(explicit = false)
       classes = @json[:classes].map do |classpath, cdef|
         type_params = cdef[:type_params]
-        superclass  = cdef[:superclass]
-        members     = cdef[:members]
+        superclass, superclass_type_args = cdef[:superclass]
+        members = cdef[:members]
 
         name = classpath.last
         superclass = path_to_klass(superclass) if superclass
@@ -492,23 +485,27 @@ module TypeProf
           end
         end
 
-        [klass, members]
+        [klass, superclass_type_args, members]
       end
 
-      classes.each do |klass, members|
+      classes.each do |klass, superclass_type_args, members|
+        @scratch.add_superclass_type_args!(klass, superclass_type_args&.map {|ty| conv_type(ty) })
         included_modules = members[:included_modules]
         extended_modules = members[:extended_modules]
         methods = members[:methods]
+        attr_methods = members[:attr_methods]
         ivars = members[:ivars]
         cvars = members[:cvars]
         rbs_sources = members[:rbs_sources]
 
-        included_modules.each do |mod|
-          @scratch.include_module(klass, path_to_klass(mod), false, nil)
+        included_modules.each do |mod, type_args|
+          type_args = type_args&.map {|ty| conv_type(ty) }
+          @scratch.include_module(klass, path_to_klass(mod), type_args, false, nil)
         end
 
-        extended_modules.each do |mod|
-          @scratch.include_module(klass, path_to_klass(mod), true, nil)
+        extended_modules.each do |mod, type_args|
+          type_args = type_args&.map {|ty| conv_type(ty) }
+          @scratch.include_module(klass, path_to_klass(mod), type_args, true, nil)
         end
 
         methods.each do |(singleton, method_name), mdef|
@@ -517,6 +514,14 @@ module TypeProf
           @scratch.add_method(klass, method_name, singleton, mdef)
         end
 
+        attr_methods.each do |(singleton, method_name), mdef|
+          kind = mdef[:kind]
+          ivar = mdef[:ivar]
+          ty = conv_type(mdef[:ty]).remove_type_vars
+          @scratch.add_attr_method(klass, nil, ivar, :"@#{ ivar }", kind)
+          @scratch.add_ivar_write!(Type::Instance.new(klass), :"@#{ ivar }", ty, nil)
+        end
+
         ivars.each do |ivar_name, ty|
           ty = conv_type(ty).remove_type_vars
           @scratch.add_ivar_write!(Type::Instance.new(klass), ivar_name, ty, nil)
@@ -583,7 +588,7 @@ module TypeProf
       req, lead_tys, opt_tys, ret_ty = blk
       lead_tys = lead_tys.map {|ty| conv_type(ty) }
       opt_tys = opt_tys.map {|ty| conv_type(ty) }
-      msig = MethodSignature.new(lead_tys, opt_tys, nil, nil, nil, nil, nil)
+      msig = MethodSignature.new(lead_tys, opt_tys, nil, [], {}, nil, Type.nil)
       ret_ty = conv_type(ret_ty)
       ret = [Type::Proc.new(TypedBlock.new(msig, ret_ty), Type::Builtin[:proc])]
       ret << Type.nil unless req

data/lib/typeprof/method.rb

@@ -4,12 +4,16 @@ module TypeProf
   end
 
   class ISeqMethodDef < MethodDef
-    def initialize(iseq, cref)
+    def initialize(iseq, cref, outer_ep, pub_meth)
       @iseq = iseq
       raise if iseq.nil?
       @cref = cref
+      @outer_ep = outer_ep
+      @pub_meth = pub_meth
     end
 
+    attr_accessor :pub_meth
+
     def do_send(recv, mid, aargs, caller_ep, caller_env, scratch, &ctn)
       recv = recv.base_type while recv.respond_to?(:base_type)
       recv = scratch.globalize_type(recv, caller_env, caller_ep)
@@ -25,8 +29,8 @@ module TypeProf
       end
 
       nctx = Context.new(@iseq, @cref, mid)
-      callee_ep = ExecutionPoint.new(nctx, 0, nil)
-      nenv = Env.new(StaticEnv.new(recv, blk_ty, false), locals, [], Utils::HashWrapper.new({}))
+      callee_ep = ExecutionPoint.new(nctx, 0, @outer_ep)
+      nenv = Env.new(StaticEnv.new(recv, blk_ty, false, true), locals, [], Utils::HashWrapper.new({}))
       alloc_site = AllocationSite.new(callee_ep)
       locals.each_with_index do |ty, i|
         alloc_site2 = alloc_site.add_id(i)
@@ -44,6 +48,14 @@ module TypeProf
     end
 
     def do_check_send(msig, recv, mid, ep, scratch)
+      klass, singleton = recv.method_dispatch_info
+      cur_subst = {}
+      direct_method = true
+      scratch.adjust_substitution(klass, singleton, mid, self, recv.generate_substitution) do |subst, direct|
+        direct_method &&= direct
+        cur_subst = Type.merge_substitution(cur_subst, subst)
+      end
+
       lead_num = @iseq.fargs_format[:lead_num] || 0
       post_num = @iseq.fargs_format[:post_num] || 0
       rest_start = @iseq.fargs_format[:rest_start]
@@ -67,7 +79,8 @@ module TypeProf
       post_start = @iseq.fargs_format[:post_start]
       rest_start = @iseq.fargs_format[:rest_start]
       kw_start = @iseq.fargs_format[:kwbits]
-      kw_start -= @iseq.fargs_format[:keyword].size if kw_start
+      keyword = @iseq.fargs_format[:keyword]
+      kw_start -= keyword.size if kw_start
       kw_rest = @iseq.fargs_format[:kwrest]
       block_start = @iseq.fargs_format[:block_start]
 
@@ -77,18 +90,20 @@ module TypeProf
       callee_ep = ExecutionPoint.new(ctx, 0, nil)
 
       locals = [Type.nil] * @iseq.locals.size
-      nenv = Env.new(StaticEnv.new(recv, msig.blk_ty, false), locals, [], Utils::HashWrapper.new({}))
+      nenv = Env.new(StaticEnv.new(recv, msig.blk_ty, false, true), locals, [], Utils::HashWrapper.new({}))
       alloc_site = AllocationSite.new(callee_ep)
       idx = 0
       msig.lead_tys.each_with_index do |ty, i|
         alloc_site2 = alloc_site.add_id(idx += 1)
         # nenv is top-level, so it is okay to call Type#localize directly
+        ty = ty.substitute(cur_subst, Config.options[:type_depth_limit]).remove_type_vars
         nenv, ty = ty.localize(nenv, alloc_site2, Config.options[:type_depth_limit])
         nenv = nenv.local_update(i, ty)
       end
       if msig.opt_tys
         msig.opt_tys.each_with_index do |ty, i|
           alloc_site2 = alloc_site.add_id(idx += 1)
+          ty = ty.substitute(cur_subst, Config.options[:type_depth_limit]).remove_type_vars
           nenv, ty = ty.localize(nenv, alloc_site2, Config.options[:type_depth_limit])
           nenv = nenv.local_update(lead_num + i, ty)
         end
@@ -96,19 +111,27 @@ module TypeProf
       if msig.rest_ty
         alloc_site2 = alloc_site.add_id(idx += 1)
         ty = Type::Array.new(Type::Array::Elements.new([], msig.rest_ty), Type::Instance.new(Type::Builtin[:ary]))
+        ty = ty.substitute(cur_subst, Config.options[:type_depth_limit]).remove_type_vars
         nenv, rest_ty = ty.localize(nenv, alloc_site2, Config.options[:type_depth_limit])
         nenv = nenv.local_update(rest_start, rest_ty)
       end
       if msig.post_tys
         msig.post_tys.each_with_index do |ty, i|
           alloc_site2 = alloc_site.add_id(idx += 1)
+          ty = ty.substitute(cur_subst, Config.options[:type_depth_limit]).remove_type_vars
           nenv, ty = ty.localize(nenv, alloc_site2, Config.options[:type_depth_limit])
           nenv = nenv.local_update(post_start + i, ty)
         end
       end
       if msig.kw_tys
-        msig.kw_tys.each_with_index do |(_, key, ty), i|
+        msig.kw_tys.each do |_, key, ty|
+          i = keyword.index(key)
+          unless i
+            # warn
+            next
+          end
           alloc_site2 = alloc_site.add_id(key)
+          ty = ty.substitute(cur_subst, Config.options[:type_depth_limit]).remove_type_vars
           nenv, ty = ty.localize(nenv, alloc_site2, Config.options[:type_depth_limit])
           nenv = nenv.local_update(kw_start + i, ty)
         end
@@ -116,6 +139,7 @@ module TypeProf
       if msig.kw_rest_ty
         ty = msig.kw_rest_ty
         alloc_site2 = alloc_site.add_id(:**)
+        ty = ty.substitute(cur_subst, Config.options[:type_depth_limit]).remove_type_vars
         nenv, ty = ty.localize(nenv, alloc_site2, Config.options[:type_depth_limit])
         nenv = nenv.local_update(kw_rest, ty)
       end
@@ -170,6 +194,15 @@ module TypeProf
 
     def do_send(recv_orig, mid, aargs, caller_ep, caller_env, scratch, &ctn)
       recv = scratch.globalize_type(recv_orig, caller_env, caller_ep)
+
+      klass, singleton = recv_orig.method_dispatch_info
+      cur_subst = {}
+      direct_method = true
+      scratch.adjust_substitution(klass, singleton, mid, self, recv.generate_substitution) do |subst, direct|
+        direct_method &&= direct
+        cur_subst = Type.merge_substitution(cur_subst, subst)
+      end
+
       found = false
       aargs = scratch.globalize_type(aargs, caller_env, caller_ep)
       @sig_rets.each do |msig, ret_ty|
@@ -178,56 +211,15 @@ module TypeProf
         # XXX: support self type in msig
         subst = aargs.consistent_with_method_signature?(msig)
         next unless subst
+
+        if direct_method && recv_orig.is_a?(Type::Local)
+          ncaller_env = recv_orig.update_container_elem_type(subst, ncaller_env, caller_ep, scratch)
+        end
+
+        subst = Type.merge_substitution(subst, cur_subst)
         # need to check self tyvar?
         subst[Type::Var.new(:self)] = recv
-        case
-        when recv.is_a?(Type::Cell) && recv_orig.is_a?(Type::LocalCell)
-          tyvars = recv.base_type.klass.type_params.map {|name,| Type::Var.new(name) }
-          tyvars.each_with_index do |tyvar, idx|
-            ty = subst[tyvar]
-            if ty
-              ncaller_env, ty = scratch.localize_type(ty, ncaller_env, caller_ep)
-              ncaller_env = scratch.update_container_elem_types(ncaller_env, caller_ep, recv_orig.id, recv_orig.base_type) do |elems|
-                elems.update(idx, ty)
-              end
-            end
-          end
-          tyvars.zip(recv.elems.elems) do |tyvar, elem|
-            if subst[tyvar]
-              subst[tyvar] = subst[tyvar].union(elem)
-            else
-              subst[tyvar] = elem
-            end
-          end
-        when recv.is_a?(Type::Array) && recv_orig.is_a?(Type::LocalArray)
-          tyvar_elem = Type::Var.new(:Elem)
-          if subst[tyvar_elem]
-            ty = subst[tyvar_elem]
-            ncaller_env, ty = scratch.localize_type(ty, ncaller_env, caller_ep)
-            ncaller_env = scratch.update_container_elem_types(ncaller_env, caller_ep, recv_orig.id, recv_orig.base_type) do |elems|
-              elems.update(nil, ty)
-            end
-          end
-          subst.merge!({ tyvar_elem => recv.elems.squash })
-        when recv.is_a?(Type::Hash) && recv_orig.is_a?(Type::LocalHash)
-          tyvar_k = Type::Var.new(:K)
-          tyvar_v = Type::Var.new(:V)
-          k_ty0, v_ty0 = recv.elems.squash
-          if subst[tyvar_k] && subst[tyvar_v]
-            k_ty = subst[tyvar_k]
-            v_ty = subst[tyvar_v]
-            k_ty0 = k_ty0.union(k_ty)
-            v_ty0 = v_ty0.union(v_ty)
-            alloc_site = AllocationSite.new(caller_ep)
-            ncaller_env, k_ty = scratch.localize_type(k_ty, ncaller_env, caller_ep, alloc_site.add_id(:k))
-            ncaller_env, v_ty = scratch.localize_type(v_ty, ncaller_env, caller_ep, alloc_site.add_id(:v))
-            ncaller_env = scratch.update_container_elem_types(ncaller_env, caller_ep, recv_orig.id, recv_orig.base_type) do |elems|
-              elems.update(k_ty, v_ty)
-            end
-          end
-          # XXX: need to heuristically replace ret type Hash[K, V] with self, instead of conversative type?
-          subst.merge!({ tyvar_k => k_ty0, tyvar_v => v_ty0 })
-        end
+
         found = true
         if aargs.blk_ty.is_a?(Type::Proc)
           #raise NotImplementedError unless aargs.blk_ty.block_body.is_a?(ISeqBlock) # XXX
@@ -235,91 +227,33 @@ module TypeProf
           dummy_ep = ExecutionPoint.new(dummy_ctx, -1, caller_ep)
           s_recv = recv
           s_recv = s_recv.base_type while s_recv.respond_to?(:base_type)
-          dummy_env = Env.new(StaticEnv.new(s_recv, msig.blk_ty, false), [], [], Utils::HashWrapper.new({}))
+          dummy_env = Env.new(StaticEnv.new(s_recv, msig.blk_ty, false, true), [], [], Utils::HashWrapper.new({}))
           if msig.blk_ty.is_a?(Type::Proc)
             scratch.add_callsite!(dummy_ctx, caller_ep, ncaller_env, &ctn)
-            nfargs = msig.blk_ty.block_body.msig
+            bsig = msig.blk_ty.block_body.msig
             alloc_site = AllocationSite.new(caller_ep).add_id(self)
-            nlead_tys = (nfargs.lead_tys + nfargs.opt_tys).map.with_index do |ty, i|
-              if recv.is_a?(Type::Array)
-                tyvar_elem = Type::Var.new(:Elem)
-                ty = ty.substitute(subst.merge({ tyvar_elem => recv.elems.squash_or_any }), Config.options[:type_depth_limit])
-              else
-                ty = ty.substitute(subst, Config.options[:type_depth_limit])
-              end
-              ty = ty.remove_type_vars
-              alloc_site2 = alloc_site.add_id(i)
-              dummy_env, ty = scratch.localize_type(ty, dummy_env, dummy_ep, alloc_site2)
+            nlead_tys = (bsig.lead_tys + bsig.opt_tys).map.with_index do |ty, i|
+              ty = ty.substitute(subst, Config.options[:type_depth_limit]).remove_type_vars
+              dummy_env, ty = scratch.localize_type(ty, dummy_env, dummy_ep, alloc_site.add_id(i))
               ty
             end
-            0.upto(nfargs.opt_tys.size) do |n|
-              naargs = ActualArguments.new(nlead_tys[0, nfargs.lead_tys.size + n], nil, {}, Type.nil) # XXX: support block to block?
+            0.upto(bsig.opt_tys.size) do |n|
+              naargs = ActualArguments.new(nlead_tys[0, bsig.lead_tys.size + n], nil, {}, Type.nil) # XXX: support block to block?
               scratch.do_invoke_block(aargs.blk_ty, naargs, dummy_ep, dummy_env) do |blk_ret_ty, _ep, _env|
                 subst2 = Type.match?(blk_ret_ty, msig.blk_ty.block_body.ret_ty)
                 if subst2
                   subst2 = Type.merge_substitution(subst, subst2)
-                  case
-                  when recv.is_a?(Type::Cell) && recv_orig.is_a?(Type::LocalCell)
-                    tyvars = recv.base_type.klass.type_params.map {|name,| Type::Var.new(name) }
-                    tyvars.each_with_index do |tyvar, idx|
-                      ty = subst2[tyvar]
-                      if ty
-                        ncaller_env, ty = scratch.localize_type(ty, ncaller_env, caller_ep)
-                        ncaller_env = scratch.update_container_elem_types(ncaller_env, caller_ep, recv_orig.id, recv_orig.base_type) do |elems|
-                          elems.update(idx, ty)
-                        end
-                        scratch.merge_return_env(caller_ep) {|env| env ? env.merge(ncaller_env) : ncaller_env }
-                      end
-                    end
-                    tyvars.zip(recv.elems.elems) do |tyvar, elem|
-                      if subst2[tyvar]
-                        subst2[tyvar] = subst2[tyvar].union(elem)
-                      else
-                        subst2[tyvar] = elem
-                      end
-                    end
-                    ret_ty = ret_ty.substitute(subst2, Config.options[:type_depth_limit])
-                  when recv.is_a?(Type::Array) && recv_orig.is_a?(Type::LocalArray)
-                    tyvar_elem = Type::Var.new(:Elem)
-                    if subst2[tyvar_elem]
-                      ty = subst2[tyvar_elem]
-                      ncaller_env, ty = scratch.localize_type(ty, ncaller_env, caller_ep)
-                      ncaller_env = scratch.update_container_elem_types(ncaller_env, caller_ep, recv_orig.id, recv_orig.base_type) do |elems|
-                        elems.update(nil, ty)
-                      end
-                      scratch.merge_return_env(caller_ep) {|env| env ? env.merge(ncaller_env) : ncaller_env }
-                    end
-                    ret_ty = ret_ty.substitute(subst2, Config.options[:type_depth_limit])
-                  when recv.is_a?(Type::Hash) && recv_orig.is_a?(Type::LocalHash)
-                    tyvar_k = Type::Var.new(:K)
-                    tyvar_v = Type::Var.new(:V)
-                    k_ty0, v_ty0 = recv.elems.squash
-                    if subst2[tyvar_k] && subst2[tyvar_v]
-                      k_ty = subst2[tyvar_k]
-                      v_ty = subst2[tyvar_v]
-                      k_ty0 = k_ty0.union(k_ty)
-                      v_ty0 = v_ty0.union(v_ty)
-                      alloc_site = AllocationSite.new(caller_ep)
-                      ncaller_env, k_ty = scratch.localize_type(k_ty, ncaller_env, caller_ep, alloc_site.add_id(:k))
-                      ncaller_env, v_ty = scratch.localize_type(v_ty, ncaller_env, caller_ep, alloc_site.add_id(:v))
-                      ncaller_env = scratch.update_container_elem_types(ncaller_env, caller_ep, recv_orig.id, recv_orig.base_type) do |elems|
-                        elems.update(k_ty, v_ty)
-                      end
-                      scratch.merge_return_env(caller_ep) {|env| env ? env.merge(ncaller_env) : ncaller_env }
-                    end
-                    ret_ty = ret_ty.substitute(subst2, Config.options[:type_depth_limit])
-                  else
-                    ret_ty = ret_ty.substitute(subst2, Config.options[:type_depth_limit])
+                  if direct_method && recv_orig.is_a?(Type::Local)
+                    ncaller_env = recv_orig.update_container_elem_type(subst2, ncaller_env, caller_ep, scratch)
+                    scratch.merge_return_env(caller_ep) {|env| env ? env.merge(ncaller_env) : ncaller_env }
                   end
+                  ret_ty2 = ret_ty.substitute(subst2, Config.options[:type_depth_limit]).remove_type_vars
                 else
-                  # raise "???"
-                  # XXX: need warning
-                  ret_ty = Type.any
+                  ret_ty2 = Type.any
                 end
-                ret_ty = ret_ty.remove_type_vars
                 # XXX: check the return type from the block
                 # sig.blk_ty.block_body.ret_ty.eql?(_ret_ty) ???
-                scratch.add_return_value!(dummy_ctx, ret_ty)
+                scratch.add_return_value!(dummy_ctx, ret_ty2)
               end
               # scratch.add_return_value!(dummy_ctx, ret_ty) ?
               # This makes `def foo; 1.times { return "str" }; end` return Integer|String